重写request【数据流获取一次就会失效】
public class CustomHttpServletRequestWrapper extends HttpServletRequestWrapper {
// 保存request body的数据
private String body;
// 解析request的inputStream(即body)数据,转成字符串
public CustomHttpServletRequestWrapper(HttpServletRequest request) {
super(request);
StringBuilder stringBuilder = new StringBuilder();
BufferedReader bufferedReader = null;
InputStream inputStream = null;
try {
inputStream = request.getInputStream();
if (inputStream != null) {
bufferedReader = new BufferedReader(new InputStreamReader(inputStream));
char[] charBuffer = new char[128];
int bytesRead = -1;
while ((bytesRead = bufferedReader.read(charBuffer)) > 0) {
stringBuilder.append(charBuffer, 0, bytesRead);
}
} else {
stringBuilder.append("");
}
} catch (IOException ex) {
} finally {
if (inputStream != null) {
try {
inputStream.close();
} catch (IOException e) {
e.printStackTrace();
}
}
if (bufferedReader != null) {
try {
bufferedReader.close();
} catch (IOException e) {
e.printStackTrace();
}
}
}
body = stringBuilder.toString();
}
@Override
public ServletInputStream getInputStream() {
final ByteArrayInputStream byteArrayInputStream = new ByteArrayInputStream(body.getBytes());
ServletInputStream servletInputStream = new ServletInputStream() {
@Override
public boolean isFinished() {
return false;
}
@Override
public boolean isReady() {
return false;
}
@Override
public void setReadListener(ReadListener readListener) {
}
@Override
public int read() {
return byteArrayInputStream.read();
}
};
return servletInputStream;
}
@Override
public BufferedReader getReader() throws IOException {
return new BufferedReader(new InputStreamReader(this.getInputStream()));
}
public String getBody() {
return this.body;
}
// 赋值给body字段
public void setBody(String body) {
this.body = body;
}
}
自定义拦截器Filter
public class UserInfoFilter implements Filter {
@Override
public void init(FilterConfig filterConfig) {
}
@Override
public void doFilter(ServletRequest request, ServletResponse response, FilterChain chain) throws IOException, ServletException {
CustomHttpServletRequestWrapper customHttpServletRequestWrapper = null;
try {
HttpServletRequest req = (HttpServletRequest) request;
customHttpServletRequestWrapper = new CustomHttpServletRequestWrapper(req);
} catch (Exception e) {
}
chain.doFilter((Objects.isNull(customHttpServletRequestWrapper) ? request : customHttpServletRequestWrapper), response);
}
@Override
public void destroy() {
}
}
@Configuration
public class WebConfig extends WebMvcConfigurerAdapter {
@Autowired
private CorsInterceptor corsInterceptor;
@Autowired
private LogInterceptor logInterceptor;
@Autowired
private CustomInterceptor customInterceptor;
@Autowired
private RpcArgumentResolver rpcArgumentResolver;
@Override
public void addArgumentResolvers(List<HandlerMethodArgumentResolver> argumentResolvers) {
argumentResolvers.add(rpcArgumentResolver);
}
@Override
public void addInterceptors(InterceptorRegistry registry) {
registry.addInterceptor(corsInterceptor).addPathPatterns("/**");
registry.addInterceptor(logInterceptor).addPathPatterns("/**");
//一定要放到最后
registry.addInterceptor(customInterceptor).addPathPatterns("/**");
}
@Bean
public FilterRegistrationBean filter() {
UserInfoFilter preDataInterceptor1 = new UserInfoFilter();
FilterRegistrationBean registration = new FilterRegistrationBean();
registration.setFilter(preDataInterceptor1);
registration.addUrlPatterns("/*");
return registration;
}
}
向request中放自定义参数
@Component
public class CustomInterceptor implements HandlerInterceptor {
@Override
public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) {
if (!(handler instanceof HandlerMethod)) {
return true;
}
pushUserInfo2Body(request);
return true;
}
private void pushUserInfo2Body(HttpServletRequest request) {
if (request instanceof CustomHttpServletRequestWrapper) {
CustomHttpServletRequestWrapper requestWrapper = (CustomHttpServletRequestWrapper) request;
String contentType = requestWrapper.getContentType();
//获取request的body参数
String postContent = requestWrapper.getBody();
if (contentType != null && contentType.toLowerCase().contains("application/json")) {
JSONObject jsStr = new JSONObject();
if (StringUtils.isNotBlank(postContent)) {
jsStr = JSONObject.parseObject(postContent);
}
Object attributeObject = requestWrapper.getAttribute(AuthConstants.REQUEST_ATTRIBUTE_KEY);
if (null != attributeObject && attributeObject instanceof WxMiniUserLoginInfo) {
WxMiniUserLoginInfo tempLogin = (WxMiniUserLoginInfo) attributeObject;
jsStr.put("openID", tempLogin.getOpenID());
jsStr.put("customerID", tempLogin.getCustomerID());
jsStr.put("shopID", tempLogin.getCustomerID());
jsStr.put("groupID", "xxxxx");
}
requestWrapper.setBody(jsStr.toJSONString());
}
}
}
@Override
public void postHandle(HttpServletRequest request, HttpServletResponse response, Object handler, ModelAndView modelAndView) {
}
@Override
public void afterCompletion(HttpServletRequest request, HttpServletResponse response, Object handler, Exception ex) {
}
}