每次使用mybatis的時候,簡單的連表查詢,用Map接收的時候,都是像DB定義的字段一樣,類似以下 student_name,student_id,沒有轉換為駝峰,但是又不能因為這一個定義一個javabean來映射數據庫字段集合,這樣,會有無窮無盡的javabean,完全不是辦法。
然后我看了下mybatis-spring-boot的配置文檔http://www.mybatis.org/spring-boot-starter/mybatis-spring-boot-autoconfigure/,發現有這么個屬性
mybatis.configuration.map-underscore-to-camel-case=true
看着屬性意思,很像是 map 下划線轉換為駝峰,然后我天真的以為,加個這個,就會將Map里面的key轉換為駝峰的命名方式,然后我自己測試了一下,發現自己還是太天真,沒轉過來,該是什么還是什么(心里面是B了狗了)....
沒辦法,接着找了了一下,發現官方文檔http://www.mybatis.org/mybatis-3/configuration.html#properties,描述過這個的作用
才發現,這個屬性的作用,是作用於javabean的field的,並不是map,ㄟ( ▔, ▔ )ㄏ,那沒辦法了,只能自己動手了。
既然 map-underscore-to-camel-case 不能作用於map,那么只能自己動手了,然后我就想到了2個解決方案:
- 繼承HashMap,重寫Put函數,將mybatis返回的Map,寫上自定義Map的路徑,自定義的Map,將所有的key內部轉換為駝峰表達式
- 找到mybatis的Handler,通過Handler來找到轉換映射關系的接口定義,繼承或者實現接口,然后走自定義的轉換規則來實現Map映射的駝峰轉換
最終決定,采用第二種,方便以后擴展。
首先,我開始找mybatis的configuration里面的mapUnderscoreToCamelCase屬性調用的地方
有兩個地方調用了,然后我們看第一個調用的地方,發現名稱是createAutomaticMappings,應該是map的映射字段創建沒跑了,如下代碼,光標顯示的地方,就是調用的地方
然后我們在查看屬性傳遞進去后的操作,點進 metaObject.findProperty的實現
接着在看,objectWrapper.findProperty,發現是個接口,然后我們需要查看,是對應的哪個實現,然后我debug發現是 MapWrapper的實現,並沒有做任何操作,直接返回了name
到這里,我點開了下源碼的目錄結構,發現了以下
心里大概明白怎么回事了,具體過程應該如下
通過接口ObjectWrapper來定義行為,有默認的一些實現,然后通過工廠ObjectWrapperFactory接口,來創建ObjectWrapper接口,最后來進行對方的自動映射跟包裝,自此,心里已經想到怎么改動了
--------------show code-----------------------
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的字段,下划線已經轉換成駝峰命名了
到這,文章就結束了!
轉自：https://my.oschina.net/u/2278977/blog/1795969