上篇文章分享了在項目實戰(zhàn)中自定義Mybatis的TypeHandler來處理枚舉類型。文章結(jié)尾也指出了美中不足之處，那就是每次都需要指定我們自定義的枚舉TypeHandler。 隨著項目枚舉類型的增多，每次都要寫一遍這個會令人很反感。那么，本次我們就來解決這一痛點。

思路分析

1. 上篇文章講到, mybatis有默認的處理枚舉類型的TypeHandler，因此，我們要將我們自定義的枚舉TypeHandler注冊進入Mybatis。
2. 注冊完我們自定義的TypeHandler后，還得告訴Mybatis只要一遇到我們指定的枚舉類型（上篇的DisplayedEnum）就用自定義的枚舉TypeHandler(上篇為EnumTypeHandler)進行轉(zhuǎn)換。由于Mybatis在啟動時就會通過TypeHandlerRegistry進行注冊，即建立JdbcType, JavaType, TypeHandler三者之間的關(guān)系， 因此，這意味著在Mybatis啟動時我們也需要通過TypeHandlerRegistry將我們的所有的枚舉類型（JavaType）與自定義的枚舉TypeHandler（EnumTypeHandler）建立聯(lián)系。

為了方便大家理解， 貼上Mybatis中的TypeHandlerRegistry的相關(guān)部分源碼:

實戰(zhàn)

SqlSessionFactoryBean

由于我們項目使用了Spring, 是用Spring集成的Mybatis(廢話，大家都是這么干的)。Spring通過SqlSessionFactoryBean來初始化啟動Mybatis。 因此，我們應(yīng)該在它身上下手，然而，一切并不是那么順利。

查看了一下SqlSessionFactoryBean的源碼，發(fā)現(xiàn)SqlSessionFactoryBean并沒有任何地方可以讓我們切入， 進而來調(diào)用TypeHandlerRegistry進行注冊我們的枚舉。 更令人蛋疼的是其所有屬性全是private， 這下不僅AOP切入不行，連通過繼承偷懶都不行了。

作罷，咱只有老老實實的重寫一遍SqlSessionFactoryBean的代碼了（copy還不簡單）。

DefaultSqlSessionFactoryBean

代碼量比較多，就只貼關(guān)鍵代碼了。步驟如下：

1. DefaultSqlSessionFactoryBean繼承SqlSessionFactoryBean。
2. 將SqlSessionFactoryBean中的代碼全部copy到DefaultSqlSessionFactoryBean。
3. 調(diào)用以下方法。

切記， 以上代碼要在這個代碼之前執(zhí)行：

因為, xmlMapperBuilder.parse()方法會開始解析我們所有的所有mapper.xml的配置文件了，這時候會把resultMap也解析了，這就會導(dǎo)致我們自定義的TypeHandler不生效。

切記！ 切記

最后

以上，代碼算是完工了，記得在Spring的配置文件中， 將SqlSessionFactoryBean替換為我們的DefaultSqlSessionFactoryBean。 不然我們的活就是白干了。

接下來，就可以開始浪了， 之前我們需要這樣寫：

<resultMap>
 ...
 <result column="status" jdbcType="TINYINT" property="status" typeHandler="xxx.xxx.EnumTypeHandler" />
 ...
</resultMap>

而現(xiàn)在，我們可以不寫typeHandler了。

<resultMap>
 ...
 <result column="status" jdbcType="TINYINT" property="status" />
 ...
</resultMap>

以上就是本文的全部內(nèi)容，希望本文的內(nèi)容對大家的學(xué)習(xí)或者工作能帶來一定的幫助，同時也希望多多支持腳本之家！

最新評論