Mybatis實戰之TypeHandler高級進階

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

思路分析

1. 上篇文章講到, mybatis有預設的處理枚舉類型的TypeHandler，因此，我們要將我們自定義的枚舉TypeHandler註冊進入Mybatis。

2. 註冊完我們自定義的TypeHandler後，還得告訴Mybatis只要一遇到我們指定的枚舉類型（上篇的DisplayedEnum）就用自定義的枚舉TypeHandler(上篇為EnumTypeHandler)進行轉換。由於Mybatis在啟動時就會通過TypeHandlerRegistry進行註冊，即建立JdbcType, JavaType, TypeHandler三者之間的關係， 因此，這意味著在Mybatis啟動時我們也需要通過TypeHandlerRegistry將我們的所有的枚舉類型（JavaType）與自定義的枚舉TypeHandler（EnumTypeHandler）建立聯繫。

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

實戰

SqlSessionFactoryBean

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

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

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

DefaultSqlSessionFactoryBean

代碼量比較多，就只貼關鍵代碼了。步驟如下：

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

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

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

最後

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

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

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

而現在，我們可以不寫typeHandler了。

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

好啦，本次就寫到這兒。

我的博客： javafan.cn