Spring Boot在运行时操作http参数

Question

我试图操纵一些http参数，但没有成功(在查询中，body .)在最终端点接收到调用之前。例如，我们接到了这样的帖子：

curl -X POST "http://localhost:8080/insertBody/" -H "accept: application/json" -H "Content-Type: application/json" -H "Date-Format: yyyy-MM-dd" -d "{ \"isUniform\": true, \"myDate\": \"2020-01-14T08:55:07.013Z\", \"myInt\": 0, \"uniform\": true}"

我要做的是将myDate -> 2020-01-14T08:55:07.013Z在post正文中转换为这样的格式：yyyy dd在标题中传递。操作必须涉及此调用中存在的所有类型为OffsetDateTime的对象(在本例中)。

当微型服务接收到呼叫时：

Header:
  Date-Format: yyyy-MM-dd
Body
  {
    "isUniform": true,
    "myDate": "2020-01-14T08:55:07.013Z",
    "myInt": 0,
    "uniform": true
  }

在数据操作和控制器接收到的信息之后：

Header:
  Date-Format: yyyy-MM-dd
Body
  {
    "isUniform": true,
    "myDate": "2020-01-14",   <---
    "myInt": 0,
    "uniform": true
  }

体类

public class CashBackCampaignRequest   {

  @JsonProperty("uniform")
  private Boolean uniform = true;

  @JsonProperty("myInt")
  private Integer myInt = null;

  @JsonProperty("myDate")
  private OffsetDateTime myDate = null;

  // getter setters ...
}

Answer 1

你应该使用著名的RequestBodyAdviceAdapter。在进入控制器之前，您可以操纵消息的主体。您可以声明一个@ControllerAdvice或@RestControllerAdvice (它只是一个@Component)，并扩展类RequestBodyAdviceAdapter。(您也可以实现接口RequestBodyAdvice，但我建议扩展抽象类)。

下面是一个简单的例子：

@RestControllerAdvice
public class WebAdvice extends RequestBodyAdviceAdapter {

    @Override
    public boolean supports(MethodParameter methodParameter, Type type, Class<? extends HttpMessageConverter<?>> aClass) {
        // to know whether you will use your advice or not
        return true;
    }

    @Override
    public Object afterBodyRead(Object body, HttpInputMessage inputMessage, MethodParameter parameter, Type targetType, Class<? extends HttpMessageConverter<?>> converterType) {
        String format = inputMessage.getHeaders().get("DATE_FORMAT").get(0);
        if(body instanceof CashBackCampaignRequest) {
            // Do whatever you want
            ((CashBackCampaignRequest) body).setDate()
        }
        return super.afterBodyRead(body, inputMessage, parameter, targetType, converterType);
    }
}

请注意Controller中收到的请求的类型。如果控制器接收到CashBackCampaignRequest类型的对象，则无法更改格式。

Answer 2

基于RUARDO的RequestBodyAdviceAdapter ResponseBodyAdvice< Object>

@ControllerAdvice
public class ResponseJsonFilterAdvice implements ResponseBodyAdvice<Object> {

    @Override
    public boolean supports(MethodParameter returnType, Class<? extends HttpMessageConverter<?>> converterType) {
        // true if want to use this controller
        return true;
    }

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

        List<String> headers = request.getHeaders().get("X-MY-DATE-FORMAT");
        if(headers == null || headers.isEmpty())
            return body;


        ObjectMapper objectMapper = new ObjectMapper();
        objectMapper.registerModule(new JavaTimeModule());  
        SimpleModule simpleModule = new SimpleModule();
        simpleModule.addSerializer(OffsetDateTime.class, new JsonSerializer<OffsetDateTime>() {
            @Override
            public void serialize(OffsetDateTime offsetDateTime, JsonGenerator jsonGenerator, SerializerProvider serializerProvider) throws IOException, JsonProcessingException {
                jsonGenerator.writeString(DateTimeFormatter.ofPattern(headers.get(0)).format(offsetDateTime));
            }
        });
        objectMapper.registerModule(simpleModule);

        try {
            return objectMapper.writeValueAsString(body);
        } catch (JsonProcessingException e) {
            e.printStackTrace();
        }
        return body;
    }
}