Question

我要求Base64解码我的Spring Boot服务收到的每个JSON请求负载。在使用HTTP Base64方法发布之前，JSON有效负载将在客户端进行POST编码。此外，在呈现给调用客户端应用程序之前，我还需要Base64对JSON响应进行编码。

我需要使用处理程序拦截器来减少样板代码。我已经通过使用拦截器实现了操作的请求/传入支路，但是尚未为响应支路实现此目的。我已在下面发布了代码段。拦截响应和base64编码的代码在拦截器类的postHandle方法中。

我在这里做错了什么？

拦截器类：

public class Base64ResponseEncodingInterceptor implements HandlerInterceptor {   
    private static final String DECODED_REQUEST_ATTRIB = "decodedRequest";
    private static final String POST = "POST";


    @Override
    public void postHandle(HttpServletRequest request, HttpServletResponse response, Object handler, ModelAndView arg3) throws Exception {
          try {
              if (POST.equalsIgnoreCase(request.getMethod())) {
                    CharResponseWrapper res = new CharResponseWrapper(response);
                    res.getWriter();
                    byte[] encoded = Base64.encodeBase64(res.toString().getBytes());
                    byte[] encoded = Base64.encodeBase64(response.getHeader(ENCODED_RESPONSE_ATTRIB).getBytes());
                    response.getWriter().write(new String(encoded));
              }
          } catch (Exception e) {
              throw new Exception(e.getMessage());
          }
    }


    // preHandle and afterCompletion methods
    // Omitted 
}

上面使用的CharResponseWrapper类：

public class CharResponseWrapper extends HttpServletResponseWrapper {

    protected CharArrayWriter charWriter;

    protected PrintWriter writer;

    protected boolean getOutputStreamCalled;

    protected boolean getWriterCalled;


    public CharResponseWrapper(HttpServletResponse response) {
        super(response);
        charWriter = new CharArrayWriter();
    }


    @Override
    public ServletOutputStream getOutputStream() throws IOException {
        if (getWriterCalled) {
            throw new IllegalStateException("getWriter already called");
        }
        getOutputStreamCalled = true;
        return super.getOutputStream();
    }


    @Override
    public PrintWriter getWriter() throws IOException {
        if (writer != null) {
            return writer;
        }
        if (getOutputStreamCalled) {
            throw new IllegalStateException("getOutputStream already called");
        }
        getWriterCalled = true;
        writer = new PrintWriter(charWriter);
        return writer;
    }


    @Override
    public String toString() {
        String s = null;
        if (writer != null) {
            s = charWriter.toString();
        }
        return s;
    }
}

注册Interceptor的JavaConfig类：

@Configuration
@EnableJpaRepositories(repositoryBaseClass = BaseRepositoryBean.class, basePackages = "")
@EntityScan(basePackages = { "com.companyname", "com.companyname.productname"})
public class RestConfig extends WebMvcConfigurerAdapter {


      @Override
      public void addInterceptors(InterceptorRegistry registry) {
        registry.addInterceptor(new Base64ResponseEncodingInterceptor());
      }

}

使用Interceptor的Controller Class（此处仅显示工作请求分支）：

@Autowired
HttpServletRequest request;

String decodedRequest = null;

@ModelAttribute("decodedRequest")
public void getDecodedParam(){
    decodedRequest = (String) request.getAttribute("decodedRequest");
}

postHandle方法中的代码不起作用。它是HttpServletResponse是null或者我得到一条异常消息：

已经调用了getOutputStream

更新：解决方案直接在ResponseBodyAdvice中读取响应 在Controller类中，我添加了以下内容：

@RestController
@RequestMapping("/api/ipmanager")
public class IPProfileRestController extends AbstractRestController {

    @Autowired
    HttpServletResponse response;

   String encodedResponse = null;

   @ModelAttribute("encodedResponse")
    public void getEncodedResponse(){
       response.setHeader("encodedResponse", StringUtils.EMPTY);
    } 

    @RequestMapping(value = "/time", method = { RequestMethod.POST }, produces = { MediaType.APPLICATION_JSON_VALUE, MediaType.TEXT_PLAIN_VALUE }, consumes = {
            MediaType.APPLICATION_JSON_VALUE })
    public @ResponseBody String saveAccessClientTime(@RequestBody String ecodedRequest) {

        // Some code here

        String controllerResponse = prettyJson(iPProfileResponse);
        response.setHeader("encodedResponse", controllerResponse);
        return controllerResponse;
    }
}

我在 ResponseBodyAdvice

中有以下内容

@ControllerAdvice
public class Base64EncodedResponseBodyAdvice implements ResponseBodyAdvice<Object> {

    @Override
    public boolean supports(MethodParameter returnType, 
                            Class<? extends HttpMessageConverter<?>> converterType) {
        return true;
    }

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

        String body1 = StringUtils.EMPTY;
        // Encode the response and return

        List<String> listOfHeaderValues = response.getHeaders().get("encodedResponse");

        body1=new String(Base64.encodeBase64(listOfHeaderValues.get(0).getBytes()));

        return body1;
    }

}

Answer 1

正如Spring MVC documentation所述：

postHandle的{{1}}方法并不总是理想的适合与HandlerInterceptor和@ResponseBody方法一起使用。 就是这样 ResponseEntity写入并提交响应的情况调用HttpMessageConverter之前，无法更改。{1}} 响应，例如添加标题。相反，应用程序可以实现postHandle并将其声明为 ResponseBodyAdvice bean或直接配置它 @ControllerAdvice。

话虽如此：

我在这里做错了什么？

由于响应已提交，因此无法更改。为了更改响应，您应该注册ResponseBodyAdvice<T>并将响应编码逻辑放在那里：

RequestMappingHandlerAdapter

Answer 2

如果您的方法返回了ResponseEntity<T>，请在您的postHandle()的{{1}}方法中编写以下代码：

例如HandlerInterceptor

它将起作用！

无法在Spring Boot

2 个答案: