1. 自定义一个ModelAttributeMethodProcessor
public class CustomServletModelAttributeMethodProcessor extends ServletModelAttributeMethodProcessor {
public CustomServletModelAttributeMethodProcessor(final boolean annotationNotRequired) {
super(annotationNotRequired);
}
@Override
protected void bindRequestParameters(WebDataBinder binder, NativeWebRequest request) {
ServletRequest servletRequest = request.getNativeRequest(ServletRequest.class);
Assert.state(servletRequest != null, "No ServletRequest");
ServletRequestDataBinder servletBinder = (ServletRequestDataBinder) binder;
/**
* ServletModelAttributeMethodProcessor此处使用的(servletRequest)
* 修改的目的是为了将ServletRequestDataBinder换成自定的CustomServletRequestDataBinder
*/
new CustomServletRequestDataBinder(servletBinder.getTarget()).bind(servletRequest);
}
}
我们看ServletRequestDataBinder#bind方法代码
public void bind(ServletRequest request) {
MutablePropertyValues mpvs = new ServletRequestParameterPropertyValues(request);
MultipartRequest multipartRequest = WebUtils.getNativeRequest(request, MultipartRequest.class);
if (multipartRequest != null) {
bindMultipart(multipartRequest.getMultiFileMap(), mpvs);
}
/**
* 此处有一个扩展方法，所以我们可以在这个方法里修改MutablePropertyValues，就能达到想要的效果
*/
addBindValues(mpvs, request);
doBind(mpvs);
}
protected void addBindValues(MutablePropertyValues mpvs, ServletRequest request) {
}
2. 自定义一个ServletRequestDataBinder
public class CustomServletRequestDataBinder extends ServletRequestDataBinder {
private final Pattern underLinePattern = Pattern.compile("_(\\w)");
public CustomServletRequestDataBinder(final Object target) {
super(target);
}
@Override
protected void addBindValues(MutablePropertyValues mpvs, ServletRequest request) {
List<PropertyValue> pvs = mpvs.getPropertyValueList();
List<PropertyValue> adds = new LinkedList<>();
for (PropertyValue pv : pvs) {
String name = pv.getName();
String camel = this.underLineToCamel(name);
if (!name.equals(camel)) {
adds.add(new PropertyValue(camel, pv.getValue()));
}
}
pvs.addAll(adds);
}
private String underLineToCamel(final String value) {
final StringBuffer sb = new StringBuffer();
Matcher m = this.underLinePattern.matcher(value);
while (m.find()){
m.appendReplacement(sb, m.group(1).toUpperCase());
}
m.appendTail(sb);
return sb.toString();
}
}
3. 使用
@SpringBootApplication
public class MainApplication implements WebMvcConfigurer {
public static void main(String[] args) {
SpringApplication.run(MainApplication.class, args);
}
@Override
public void addArgumentResolvers(List<HandlerMethodArgumentResolver> resolvers) {
resolvers.add(new CustomServletModelAttributeMethodProcessor(true));
}
}