Mybatis返回Map的時候轉換為駝峰式

1.首先,我們先繼承類 MapWrapper,重寫findProperty,通過useCamelCaseMapping來判斷是否開啟使用駝峰

public class CustomWrapper extends MapWrapper{
	
	
	public CustomWrapper(MetaObject metaObject, Map<String, Object> map) {
		super(metaObject, map);
	}
	
	
	@Override
	public String findProperty(String name, boolean useCamelCaseMapping) {
		if(useCamelCaseMapping){
            //CaseFormat是引用的 guava庫,里面有轉換駝峰的,免得自己重復造輪子,pom添加
            /**
             **         <dependency>
                           <groupId>com.google.guava</groupId>
                           <artifactId>guava</artifactId>
                           <version>24.1-jre</version>
                         </dependency>
             **/
			return CaseFormat.UPPER_UNDERSCORE.to(CaseFormat.LOWER_CAMEL,name);
		}
		return name;
	}
}

　　

 2. 然后,我們在實現接口 ObjectWrapperFactory,通過包裝工廠來創建自定義的包裝類,通過hasWrapperFor判斷參數不為空,並且類型是Map的時候才使用自己擴展的ObjectWrapper

public class MapWrapperFactory implements ObjectWrapperFactory {
    
    @Override
    public boolean hasWrapperFor(Object object) {
        return object != null && object instanceof Map;
    }
    
    @Override
    public ObjectWrapper getWrapperFor(MetaObject metaObject, Object object) {
        return new CustomWrapper(metaObject,(Map)object);
    }
}

3. 做完以上操作后,我們需要替換下,以前默認的實現,剛好,mybatis-spring-boot上面有告訴我們怎么做,返回一個 ConfigurationCustomizer 的bean,通過匿名內部類實現覆蓋默認的MapWrapper的findProperty函數

@Configuration
public class MybatisConfig {
	
	@Bean
	public ConfigurationCustomizer mybatisConfigurationCustomizer(){
		return new ConfigurationCustomizer() {
			@Override
			public void customize(org.apache.ibatis.session.Configuration configuration) {
				configuration.setObjectWrapperFactory(new MapWrapperFactory());
			}
		};
	}
	
}

　　

做了以上操作后,我們就將默認的Map映射,的包裝類,查找property的部分,變成了自己想要的樣子,

最后,我們在 properties或者yml里面加上 mybatis.configuration.map-underscore-to-camel-case=true ,畢竟在CustomerWrapper里面,我們是通過使用這個來控制是否轉換駝峰的,最后寫個測試用例跑一下,然后會發現,以前db的字段,下划線已經轉換成駝峰命名了