如何在过滤器中修改post调用的对象？板簧罩

Question

我的应用程序中有一个过滤器。

@Component
@Order(2)
public class RequestResponseLoggingFilter implements Filter {

    @Override
    public void doFilter(
      ServletRequest request, 
      ServletResponse response, 
      FilterChain chain) throws IOException, ServletException {

        HttpServletRequest req = (HttpServletRequest) request;
        HttpServletResponse res = (HttpServletResponse) response;
        // SET VALUE OF OBJECT
    }

    // other methods
}

我有一个使用类的 Restcall。

@RequestMapping
Class Test{
  @PostMapping("/test")
  public void postEntry(@Valid @RequestBody Testing testing){
  }
}

Class Testing{
@NotNull(message="ERROR")  
String id;

....
}

我在我的过滤器中获得了id，我想在我的过滤器中设置测试类的id。这可能吗？

Answer 1

你可以使用模拟服务器请求这样的东西

@Test
public void testAddEventWithWebAuthenticationDetails() {
    HttpSession session = new MockHttpSession(null, "test-session-id");
    MockHttpServletRequest request = new MockHttpServletRequest();
    request.setSession(session);
    request.setRemoteAddr("1.2.3.4");
    WebAuthenticationDetails details = new WebAuthenticationDetails(request);
    Map<String, Object> data = new HashMap<>();
    data.put("test-key", details);
    AuditEvent event = new AuditEvent("test-user", "test-type", data);
    customAuditEventRepository.add(event);
    List<PersistentAuditEvent> persistentAuditEvents = persistenceAuditEventRepository.findAll();
    assertThat(persistentAuditEvents).hasSize(1);
    PersistentAuditEvent persistentAuditEvent = persistentAuditEvents.get(0);
    assertThat(persistentAuditEvent.getData().get("remoteAddress")).isEqualTo("1.2.3.4");
    assertThat(persistentAuditEvent.getData().get("sessionId")).isEqualTo("test-session-id");
}

这里有更多的例子

或者

如果你想用过滤的方式

在那之前的几点

• 请求正文只能读取一次。
• 如果读取过滤器中的正文，目标servlet将无法重新读取它，这也会导致IllegalStateException。
• 您将需要ServletRequest estWrapper或其子级：HttpServletRequest estWrapper，以便您可以读取HTTP请求正文，然后servlet以后仍然可以读取它。

工作流程将是

>

唯一的方法是您自己在过滤器中消耗整个输入流。

从中获取您想要的内容，然后为您阅读的内容创建一个新的InputStream。

将该InputStream放入ServletRequest estWrapper（或HttpServletRequest estWrapper）。

//示例包装类，您可以在其中读取正文并修改正文内容

 public class SampleHttpServletRequest 

        extends HttpServletRequestWrapper {

  private ByteArrayOutputStream cachedBytes;

  public SampleHttpServletRequest(HttpServletRequest request) {
    super(request);
  }

  @Override
  public ServletInputStream getInputStream() throws IOException {
    if (cachedBytes == null)
      cacheInputStream();

      return new CachedServletInputStream();
  }

  @Override
  public BufferedReader getReader() throws IOException{
    return new BufferedReader(new InputStreamReader(getInputStream()));
  }

  private void cacheInputStream() throws IOException {
    cachedBytes = new ByteArrayOutputStream();
    IOUtils.copy(super.getInputStream(), cachedBytes);
  }

  public class CachedServletInputStream extends ServletInputStream {
    private ByteArrayInputStream input;

    public CachedServletInputStream() {
      input = new ByteArrayInputStream(cachedBytes.toByteArray());
    }

    @Override
    public int read() throws IOException {
      return input.read();
    }
  }
}

过滤器类

public class MyFilter implements Filter {
  @Override
  public void doFilter(ServletRequest request, ServletResponse response,
        FilterChain chain) throws IOException, ServletException {

    /* wrap the request in order to read the inputstream multiple times */
    MultiReadHttpServletRequest multiReadRequest = new MultiReadHttpServletRequest((HttpServletRequest) request);

    doMyThing(multiReadRequest.getInputStream());
    chain.doFilter(multiReadRequest, response);
  }
}    

有关更多详细信息，请参阅这些帖子

Http Servlet请求在读取一次后从POST主体中丢失参数

HttpServletRequest estWrapper，setReadListener/isFinfied/isReady的示例实现？