如何在spring boot中到达控制器之前修改请求正文

Question

我有一个spring boot应用程序。我更改每个post请求的请求正文。是否可以在请求到达控制器之前修改请求正文。请附上一个例子。

Answer 1

我使用HTTP过滤器的答案。

请求Filter.java

@Component
public class RequestFilter implements Filter {

    @Override
    public void init(FilterConfig filterConfig) throws ServletException {
    }

    @Override
    public void doFilter(ServletRequest request, ServletResponse response, FilterChain chain)
            throws IOException, ServletException {
        RequestWrapper wrappedRequest = new RequestWrapper((HttpServletRequest) request);
        chain.doFilter(wrappedRequest, response);
    }

    @Override
    public void destroy() {

    }

}

RequestWrapper。Java语言

 public class RequestWrapper extends HttpServletRequestWrapper {
        private final String body;
        private ObjectMapper objectMapper = new ObjectMapper();
    
        public RequestWrapper(HttpServletRequest request) throws IOException {
            // So that other request method behave just like before
            super(request);
    
            StringBuilder stringBuilder = new StringBuilder();
            BufferedReader bufferedReader = null;
            try {
                InputStream 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) {
                throw ex;
            } finally {
                if (bufferedReader != null) {
                    try {
                        bufferedReader.close();
                    } catch (IOException ex) {
                        throw ex;
                    }
                }
            }
            // Store request body content in 'requestBody' variable
            String requestBody = stringBuilder.toString();
            JsonNode jsonNode = objectMapper.readTree(requestBody);
            //TODO -- Update your request body here 
            //Sample
            ((ObjectNode) jsonNode).remove("key");
            // Finally store updated request body content in 'body' variable
            body = jsonNode.toString();
    
        }
    
        @Override
        public ServletInputStream getInputStream() throws IOException {
            final ByteArrayInputStream byteArrayInputStream = new ByteArrayInputStream(body.getBytes());
            ServletInputStream servletInputStream = new ServletInputStream() {
                public int read() throws IOException {
                    return byteArrayInputStream.read();
                }
    
                @Override
                public boolean isFinished() {
                    return false;
                }
    
                @Override
                public boolean isReady() {
                    return false;
                }
    
                @Override
                public void setReadListener(ReadListener listener) {
    
                }
            };
            return servletInputStream;
        }
    
        @Override
        public BufferedReader getReader() throws IOException {
            return new BufferedReader(new InputStreamReader(this.getInputStream()));
        }
}

Answer 2

简短回答是的，但不容易。

详细信息
我知道有三个选项可以在请求“到达”控制器中的处理程序方法之前更改请求体；

1. 在调用方法之前使用AOP更改请求。
2. 创建一个HTTP过滤器。
3. 创建自定义Spring HandlerInterceptor。

既然您已经在使用spring boot，选项3，定制spring HandlerInterceptor，似乎是您的最佳选择。

这里有一个链接到一篇关于spring HandlerInterceptors的Baeldung文章。

Baeldung文章并不是您问题的完整答案，因为您只能一次读取HttpServletRequest返回的InputStrem。

您需要创建一个扩展HttpServletRequest的包装类，并将包装类中的每个请求包装在自定义HandlerInterceptor或自定义过滤器中（这里可能是使用过滤器）。

包装器类

1. 阅读包装类构造函数中的HttpServletRequest输入流

如何包装请求

1. 在过滤器中，实例化包装器类并传入原始请求（这是doFilter方法的参数）
2. 把包装纸递给链条。doFilter方法（不是原始请求）

Answer 3

另一种选择是向HttpServletRequest对象添加一个属性。之后，您可以在Controller类中使用@Request estAtcm注释读取该属性。

在拦截器里

@Component
    public class SimpleInterceptor extends HandlerInterceptorAdapter {

        @Override
        public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler)
                throws ServletException, IOException {
            String parameter = request.getParameter("parameter");
            if (parameter == "somevalue") {
                request.setAttribute("customAttribute", "value");
            }

            return true;
        }

    }

在控制器中

@RestController
@RequestMapping("")
public class SampleController {


    @RequestMapping(value = "/sample",method = RequestMethod.POST)
    public String work(@RequestBody SampleRequest sampleRequest, @RequestAttribute("customAttribute") String customAttribute) {
        System.out.println(customAttribute);
        return "This works";
    }
}

这具有不修改请求正文的优点。