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

思路分析

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

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

實(shí)戰(zhàn)

SqlSessionFactoryBean

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

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

作罷，咱只有老老實(shí)實(shí)的重寫一遍SqlSessionFactoryBean的代碼了（copy還不簡(jiǎn)單）。

DefaultSqlSessionFactoryBean

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

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

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

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

切記！ 切記

最后

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

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

<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>

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

最新評(píng)論