1. 程式人生 > 程式設計 >為spring get請求新增自定義的引數處理操作(如下劃線轉駝峰)

為spring get請求新增自定義的引數處理操作(如下劃線轉駝峰)

1.生成自己的註解(為了確定在哪些位置使用)

/**
 * 關閉patch delete的model處理,否則會報錯
 */
@Target({ElementType.METHOD})
@Retention(RetentionPolicy.RUNTIME)
@Documented
public @interface AliasProcessor {
}
/**
 * 處理Get 請求引數的駝峰問題
 * @author lw
 */
@Target(ElementType.FIELD)
@Retention(RetentionPolicy.RUNTIME)
public @interface ValueFrom {
 /**
 * 引數名(別名)列表
 */
 String[] value();
}

2.實現自己的ServletModelAttributeMethodProcessor

/**
 * 為了減少使用 @RequestPath 將get引數封裝到實體類中 重寫ModelAttributeMethodProcessor
 * 注:由於get請求為非raw請求,spring預設使用@ModelArrtribute註解,不會自動將下劃線的資料轉為駝峰資料
 * 所以需要自定義一個處理器,進行該操作 *
 * @author lw
 */

public class AliasModelAttributeMethodProcessor extends ServletModelAttributeMethodProcessor {
 private ApplicationContext applicationContext;

 /**
 * 過濾掉patch請求,防止報錯
 */
 @Override
 public boolean supportsParameter(MethodParameter parameter) {
 return parameter.getMethodAnnotation(AliasProcessor.class)!=null;
 }

 public AliasModelAttributeMethodProcessor(ApplicationContext applicationContext) {
 super(true);
 this.applicationContext=applicationContext;
 }

 @Override
 protected void bindRequestParameters(WebDataBinder binder,NativeWebRequest request) {
 AliasDataBinder aliasBinder = new AliasDataBinder(binder.getTarget(),binder.getObjectName());
 RequestMappingHandlerAdapter requestMappingHandlerAdapter = this.applicationContext.getBean(RequestMappingHandlerAdapter.class);
 requestMappingHandlerAdapter.getWebBindingInitializer().initBinder(aliasBinder);
 aliasBinder.bind(request.getNativeRequest(ServletRequest.class));
 }
}

3.自己的資料處理類

/**
 * 重新資料處理類
 * @author lw
 */
public class AliasDataBinder extends ExtendedServletRequestDataBinder {

 public AliasDataBinder(Object target,String objectName) {
 super(target,objectName);
 }

 /**
 * 複寫addBindValues方法
 * @param mpvs 這裡面存的就是請求引數的key-value對
 * @param request 請求本身,這裡沒有用到
 */
 @Override
 protected void addBindValues(MutablePropertyValues mpvs,ServletRequest request) {
 super.addBindValues(mpvs,request);
 // 處理要繫結引數的物件
 Class<?> targetClass = getTarget().getClass();
 // 獲取物件的所有欄位(拿到Test類的欄位)
 Field[] fields = targetClass.getDeclaredFields();
 // 處理所有欄位
 for (Field field : fields) {
  // 原始欄位上的註解
  ValueFrom valueFromAnnotation = field.getAnnotation(ValueFrom.class);
  // 若引數中包含原始欄位或者欄位沒有別名註解,則跳過該欄位
  if (mpvs.contains(field.getName()) || valueFromAnnotation == null) {
  continue;
  }
  // 引數中沒有原始欄位且欄位上有別名註解,則依次取別名列表中的別名,在引數中最先找到的別名的值賦值給原始欄位
  for (String alias : valueFromAnnotation.value()) {
  // 若引數中包含該別名,則把別名的值賦值給原始欄位
  if (mpvs.contains(alias)) {
   // 給原始欄位賦值
   mpvs.add(field.getName(),mpvs.getPropertyValue(alias).getValue());
   // 跳出迴圈防止取其它別名
   break;
  }
  }
 }
 }
}

4.註冊到spring中

/**
 * 為了獲得context需要實現ApplicationContextAware介面
 * @author lw
 */
@Configuration
public class WebmvcConfig implements ApplicationContextAware {

 @Autowired
 private RequestMappingHandlerAdapter adapter;

 private ApplicationContext applicationContext = null;

 @Override
 public void setApplicationContext(ApplicationContext applicationContext) throws BeansException {
 this.applicationContext=applicationContext;
 }

 /**
 * 將自定義的processor新增到adapter中
 */
 @PostConstruct
 protected void injectSelfMethodArgumentResolver() {
 List<HandlerMethodArgumentResolver> argumentResolvers = new ArrayList<>();
 argumentResolvers.add(new AliasModelAttributeMethodProcessor(this.applicationContext));
 argumentResolvers.addAll(adapter.getArgumentResolvers());
 adapter.setArgumentResolvers(argumentResolvers);
 }
}

補充知識:springboot - mybatis - 下劃線與駝峰自動轉換 mapUnderscoreToCamelCase

以前都是在mybatis.xml中來配置,但是spring boot不想再用xml配置檔案。網上搜尋了好久,才找到設定辦法:

sessionFactoryBean.getObject().getConfiguration().setMapUnderscoreToCamelCase(true);

db配置檔案原始碼:

package com.vip.qa.vop.config;

import com.alibaba.druid.pool.DruidDataSource;
import org.apache.ibatis.session.SqlSessionFactory;
import org.mybatis.spring.SqlSessionFactoryBean;
import org.mybatis.spring.annotation.MapperScan;
import org.springframework.beans.factory.annotation.Qualifier;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.context.annotation.Primary;
import org.springframework.core.io.support.PathMatchingResourcePatternResolver;
import org.springframework.jdbc.datasource.DataSourceTransactionManager;
import javax.sql.DataSource;
import java.util.Properties;

/**
 * Created by danny.yao on 2017/10/25.
 */
@Configuration
@MapperScan(basePackages = VOPDataSourceConfig.PACKAGE,sqlSessionFactoryRef = "vopSqlSessionFactory")
public class VOPDataSourceConfig {
 static final String PACKAGE = "com.vip.qa.vop.mapper.vop";

 @Value("${vop.datasource.url}")
 private String dbUrl;

 @Value("${vop.datasource.username}")
 private String dbUser;

 @Value("${vop.datasource.password}")
 private String dbPassword;

 @Value("${vop.datasource.driver-class-name}")
 private String dbDriver;

 @Bean(name = "vopDataSource")
 @Qualifier
 @Primary
 public DataSource vopDataSource() {
 DruidDataSource dataSource = new DruidDataSource();
 dataSource.setDriverClassName(dbDriver);
 dataSource.setUrl(dbUrl);
 dataSource.setUsername(dbUser);
 dataSource.setPassword(dbPassword);
 return dataSource;
 }

 @Bean(name = "vopSqlSessionFactory")
 @Qualifier
 @Primary
 public SqlSessionFactory vopSqlSessionFactory(@Qualifier("vopDataSource") DataSource scepDataSource) throws Exception {
 final SqlSessionFactoryBean sessionFactoryBean = new SqlSessionFactoryBean();
 sessionFactoryBean.setDataSource(scepDataSource);

 PathMatchingResourcePatternResolver resolver = new PathMatchingResourcePatternResolver();
 sessionFactoryBean.setMapperLocations(resolver.getResources("classpath:/mybatis/vop/*.xml"));
 sessionFactoryBean.getObject().getConfiguration().setMapUnderscoreToCamelCase(true);

 return sessionFactoryBean.getObject();
 }

// @Bean(name = "vopTransactionManager")
// @Qualifier
// public DataSourceTransactionManager testDataTransactionManager() {
// return new DataSourceTransactionManager(vopDataSource());
// }

}

以上這篇為spring get請求新增自定義的引數處理操作(如下劃線轉駝峰)就是小編分享給大家的全部內容了,希望能給大家一個參考,也希望大家多多支援我們。