@Override
public void postHandle(HttpServletRequest request, HttpServletResponse response, Object handler,
@Nullable ModelAndView modelAndView) throws Exception {
response.setHeader("Content-Type", "text/json;charset=UTF-8");
CMessage cmsg = CMessage.getInstance(request, response);
cmsg.setTimes(getTimes());
response.setHeader("CMessage", cmsg.toString());
}
本来是想通过 拦截器 统一处理,给reponse设置一个自定义的Header。 后来发现一直没有生效。浏览器里总是看不到自定义的header信息。 后来发现 没有用到spring @ResponseBody这个注解的方法就有。
跟了一下源码
@Override
public void handleReturnValue(Object returnValue, MethodParameter returnType,
ModelAndViewContainer mavContainer, NativeWebRequest webRequest)
throws IOException, HttpMediaTypeNotAcceptableException,HttpMessageNotWritableException{
mavContainer.setRequestHandled(true);
ServletServerHttpRequest inputMessage = createInputMessage(webRequest);
ServletServerHttpResponse outputMessage = createOutputMessage(webRequest);
// Try even with null return value. ResponseBodyAdvice could get involved.
writeWithMessageConverters(returnValue, returnType, inputMessage, outputMessage);
}
在用到@ResponseBody这个玩意的时候,它把 response 和 request都做了一下处理。并在这里调用write方法 。
public void write(Object t, Type type, MediaType contentType, HttpOutputMessage outputMessage
) throws IOException, HttpMessageNotWritableException {
HttpHeaders headers = outputMessage.getHeaders();
if (headers.getContentType() == null) {
if (contentType == null || contentType.isWildcardType() || contentType.isWildcardSubtype()) {
contentType = getDefaultContentType(t);
}
if (contentType != null) {
headers.setContentType(contentType);
}
}
if (headers.getContentLength() == -1) {
Long contentLength = getContentLength(t, headers.getContentType());
if (contentLength != null) {
headers.setContentLength(contentLength);
}
}
writeInternal(t, outputMessage);
outputMessage.getBody().flush();
}
在这里 它已经给你写出去了。并flush了。 所以导致了在postHandler里的拦截无效了
我的做法是 写个类实现 ResponseBodyAdvice。 这个里面有个 beforeBodyWrite方法 。 它先于write执行。 在这个里面 很多人都是用它来处理 body。因为这个方法的参数 只有它自己的ServerHttpRequest request, ServerHttpResponse response。 没有提供设置 cookies 或者 header的方法。
ServerHttpResponse 里面没有。 但它的实现类ServletServerHttpResponse 这个里面有HttpServletResponse 。于是就有了下面这个代码:
public SMessage beforeBodyWrite(SMessage body, MethodParameter returnType, MediaType selectedContentType,Class extends HttpMessageConverter>> selectedConverterType, ServerHttpRequest request,ServerHttpResponse response) {
HttpServletRequest req = ((ServletServerHttpRequest) request).getServletRequest();
HttpServletResponse res = ((ServletServerHttpResponse) response).getServletResponse();
res.setHeader("Content-Type", "text/json;charset=UTF-8");
try {
CMessage cmsg = (CMessage)req.getAttribute("CMessage");
cmsg.setTimes(getTimes());
res.setHeader("CMessage", cmsg.toString());
}catch (Exception e) {
e.printStackTrace();
}
return body;
}
这样就解决了 上面说的问题了。