SpringBoot 中使用自定义参数解析器修改请求对象

在 SpringBoot 应用中，有时我们需要在控制器方法执行之前对请求对象进行修改。自定义参数解析器提供了一种灵活的方式来实现这一需求。

1. 创建自定义参数解析器

首先，我们需要创建一个自定义参数解析器来处理对 CommonRequest 对象的修改。我们可以通过实现 HandlerMethodArgumentResolver 接口来完成这个任务：

import org.springframework.core.MethodParameter;
import org.springframework.stereotype.Component;
import org.springframework.web.bind.support.WebDataBinderFactory;
import org.springframework.web.context.request.NativeWebRequest;
import org.springframework.web.method.support.HandlerMethodArgumentResolver;
import org.springframework.web.method.support.ModelAndViewContainer;
import javax.servlet.http.HttpServletRequest;@Component
public class CommonRequestResolver implements HandlerMethodArgumentResolver {@Overridepublic boolean supportsParameter(MethodParameter parameter) {return parameter.getParameterType().equals(CommonRequest.class);}@Overridepublic Object resolveArgument(MethodParameter parameter, ModelAndViewContainer mavContainer,NativeWebRequest webRequest, WebDataBinderFactory binderFactory) {HttpServletRequest request = webRequest.getNativeRequest(HttpServletRequest.class);CommonRequest modifiedRequest = RequestUtils.getAndModifyRequest(request);return modifiedRequest;}
}

在这个解析器中，我们首先检查参数是否为 CommonRequest 类型。如果是，我们就使用一个工具类 RequestUtils 来获取和修改请求对象。

2. 实现请求修改逻辑

接下来，我们需要实现 RequestUtils.getAndModifyRequest 方法来修改 CommonRequest 对象：

import com.fasterxml.jackson.databind.ObjectMapper;
import javax.servlet.http.HttpServletRequest;
import com.alibaba.fastjson.JSON;
import java.io.BufferedReader;
import java.io.IOException;public class RequestUtils {public static CommonRequest getAndModifyRequest(HttpServletRequest request) {String json = extractJsonFromBody(request);CommonRequest originalRequest = convertJsonToRequest(json);// 修改 CommonRequest 对象的逻辑originalRequest.setBegin("0000-00-00");return originalRequest;}private static String extractJsonFromBody(HttpServletRequest request) {StringBuilder jsonBuilder = new StringBuilder();try (BufferedReader reader = request.getReader()) {String line;while ((line = reader.readLine()) != null) {jsonBuilder.append(line);}} catch (IOException e) {e.printStackTrace();}return jsonBuilder.toString();}private static CommonRequest convertJsonToRequest(String json) {return JSON.parseObject(json,Batch2CommonRequest.class);}
}

在这个工具类中，我们首先从请求中提取 JSON 字符串，然后将其转换为 CommonRequest 对象。接着，我们可以对这个对象进行任何所需的修改。

3. 注册自定义参数解析器

创建自定义参数解析器后，我们需要将其注册到 SpringBoot 的应用程序中。这可以通过实现 WebMvcConfigurer 接口并重写 addArgumentResolvers 方法来完成：

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.annotation.Configuration;
import org.springframework.web.method.support.HandlerMethodArgumentResolver;
import org.springframework.web.servlet.config.annotation.WebMvcConfigurer;import java.util.List;@Configuration
public class WebConfig implements WebMvcConfigurer {@Autowiredprivate CommonRequestResolver CommonRequestResolver;@Overridepublic void addArgumentResolvers(List<HandlerMethodArgumentResolver> resolvers) {resolvers.add(CommonRequestResolver);}
}

4. 使用自定义参数解析器

一旦自定义参数解析器被注册，我们就可以在控制器方法中使用它了：


import lombok.extern.slf4j.Slf4j;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestBody;
import org.springframework.web.bind.annotation.RestController;@Slf4j
@RestController
public class MyController {@PostMapping("/sales-track/achievement-comparison")public ApiResponse<Levels<Progress>> getSalesTrackAchievementComparison(@RequestBody CommonRequest request) {// 使用修改后的 request 对象进行后续处理log.info("request:{}", JSON.toJSONString(request));return ApiResponse.success(salesTrackingService.getSalesTrackAchievementComparison(request));}
}