java - Spring security oauth2 修改响应主体连接

Question

对于我的应用程序，我使用 Spring security 来保护 API Rest 的安全。为此，我使用 oauth2 系统(带有 AuthorizationServerConfiguration、OAuth2SecurityConfiguration 和 ResourceServerConfiguration)

我不知道如何修改我的响应连接的正文。

暂时，我有:

{
"access_token": "0d1dded8-3631-472a-ba63-89d67d133112",
"token_type": "bearer",
"refresh_token": "d9f4cc5d-748b-461f-b475-3bba95b512dc",
"expires_in": 29162,
"scope": "read write trust"
}

我想要这样的东西:

{
"errorLevel":"OK",
"errorMsg":"You are now connected",
"data": { // access_token, token_type... }
}

我的第一个想法是实现我的 accessTokenConverter 并将其添加到端点，如下所示:

endpoints.accessTokenConverter(new CustomAccessTokenConverter());

但是，他不使用我的 AccessTokenConverter。

那么，我如何修改我的回复？

Answer 1

对于较新版本的 spring，您可以使用 ResponseBodyAdvice https://docs.spring.io/spring/docs/current/javadoc-api/org/springframework/web/servlet/mvc/method/annotation/ResponseBodyAdvice.html更改响应。

实现是这样的:

@ControllerAdvice
public class OauthReponseAdvice implements ResponseBodyAdvice<SomeResponseType> {
@Override
public boolean supports(MethodParameter returnType, Class<? extends HttpMessageConverter<?>> converterType) {
    return true;
}

@Override
public SomeResponseType beforeBodyWrite(SomeResponseType body, MethodParameter returnType, MediaType selectedContentType, Class<? extends
HttpMessageConverter<?>> selectedConverterType, ServerHttpRequest request, ServerHttpResponse response) {

     /// Modify body here
    return body;
   }
}