奮鬥了好幾個晚上調試程序，寫了好幾篇博客，終於建立起了Mybatis配置的擴展機制。雖然擴展機制是重要的，然而如果沒有真正實用的擴展功能，那也至少是不那麼鼓舞人心的，這篇博客就來舉幾個擴展的例子。
這次研讀源碼的起因是Oracle和MySQL數據庫的兼容性，比如在Oracle中使用雙豎線作為連接符，而MySQL中使用CONCAT函數；比如Oracle中可以使用DECODE函數，而MySQL中只能使用標準的CASE WHEN；又比如Oracle中可以執行DELETE FORM TABLE WHERE FIELD1 IN (SELECT FIELD1 FORM TABLE WHERE FIELD2=?)，但是MySQL中會拋出異常，等等。
下面就從解決這些兼容性問題開始，首先需要在配置中添加數據庫標識相關的配置：
<!-- 自行構建Configuration對象--> <bean id="mybatisConfig"/><bean id="sqlSessionFactory" p:dataSource-ref="dataSource"><!-- 注入mybatis配置對象--><property name="configuration" ref="mybatisConfig"/><!-- 自動掃描SqlMapper配置文件--><property name="mapperLocations"><array><value>classpath*:**/*.sqlmapper.xml</value></array></property><!-- 數據庫產品標識配置--><property name="databaseIdProvider"><bean><property name="properties"><props><!-- 意思是如果數據庫產品描述中包含關鍵字MYSQL，則使用mysql作為Configuration中的databaseId，mybatis原生的實現關鍵字區分大小寫，我沒有測試Oracle和DB2 --><prop key="MySQL">mysql</prop><prop key="oracle">oracle</prop><prop key="H2">h2</prop><prop key="db2">db2</prop></props></property></bean></property></bean>
一、連接符問題
1、編寫SQL配置函數實現類
public class ConcatSqlConfigFunction extends AbstractSqlConfigFunction{//抽象父類中設定了默認的order級別@Overridepublic String getName() {return "concat";}@Overridepublic String eval(String databaseId, String[] args) {if(args.length < 2){Throw.throwException("the concat function require at least two arguments.");}if("mysql".equalsIgnoreCase(databaseId)){return "CONCAT("+Tool.STRING.join(args, ",")+")";}else{return Tool.STRING.join(args, "||");}}}2、在SchemaHandlers類的靜態代碼塊中註冊，或者在啟動初始化類中調用SchemaHandlers的方法註冊
static {//註冊默認命名空間的StatementHandlerregister("cache-ref", new CacheRefStatementHandler());register("cache", new CacheStatementHandler());register("parameterMap", new ParameterMapStatementHandler());register("resultMap", new ResultMapStatementHandler());register("sql", new SqlStatementHandler());register("select|insert|update|delete", new CRUDStatementHandler());//註冊默認命名空間的ScriptHandlerregister("trim", new TrimScriptHandler());register("where", new WhereScriptHandler());register("set", new SetScriptHandler());register("foreach", new ForEachScriptHandler());register("if|when", new IfScriptHandler());register("choose", new ChooseScriptHandler());//register("when", new IfScriptHandler());register("otherwise", new OtherwiseScriptHandler());register("bind", new BindScriptHandler());// 註冊自定義命名空間的處理器registerExtend("db", new DbStatementHandler(), new DbScriptHandler());// 註冊SqlConfigFunctionregister(new DecodeSqlConfigFunction());register(new ConcatSqlConfigFunction());// 註冊SqlConfigFunctionFactoryregister(new LikeSqlConfigFunctionFactory());}上面代碼除了註冊ConcatSQLConfigFunction外，還有一些其它的註冊代碼，這裡一併給出，下文將省略。
3、修改SqlMapper配置
<select id="selectString" resultType="string">select PARAM_NAME, $concat{PARAM_CODE, PARAM_NAME} AS CODE_NAME from BF_PARAM_ENUM_DEF<if test="null != paramName and '' != paramName">where PARAM_NAME LIKE $CONCAT{'%', #{paramName, jdbcType=VARCHAR}, '%'}</if></select>4、編寫dao接口類
@Repositorypublic interface IExampleDao {public String selectString(@Param("paramName")String paramName);}5、編寫測試類
@RunWith(SpringJUnit4ClassRunner.class)@ContextConfiguration(locations={"classpath:spring/applicationContext.xml" })@Componentpublic class ExampleDaoTest {@Resourceprivate IExampleDao dao;@Testpublic void testSelectString(){String a = dao.selectString("顯示");Assert.assertEquals("顯示區域", a);}}6、分別在MySQL和H2中運行如下（將mybatis日誌級別調整為TRACE）
(1)MySQL
20161108 00:12:55,235 [main]-[DEBUG] ==> Preparing: select PARAM_NAME, CONCAT(PARAM_CODE,PARAM_NAME) AS CODE_NAME from BF_PARAM_ENUM_DEF where PARAM_NAME LIKE CONCAT('%',?,'%') 20161108 00:12:55,269 [main]-[DEBUG] ==> Parameters: 顯示(String)20161108 00:12:55,287 [main]-[TRACE] <== Columns: PARAM_NAME, CODE_NAME20161108 00:12:55,287 [main]-[TRACE] <== Row: 顯示區域, DISPLAY_AREA顯示區域20161108 00:12:55,289 [main]-[DEBUG] <== Total: 1(2)H2
20161108 00:23:08,348 [main]-[DEBUG] ==> Preparing: select PARAM_NAME, PARAM_CODE||PARAM_NAME AS CODE_NAME from BF_PARAM_ENUM_DEF where PARAM_NAME LIKE '%'||?||'%' 20161108 00:23:08,364 [main]-[DEBUG] ==> Parameters: 顯示(String)20161108 00:23:08,411 [main]-[TRACE] <== Columns: PARAM_NAME, CODE_NAME20161108 00:23:08,411 [main]-[TRACE] <== Row: 顯示區域, DISPLAY_AREA顯示區域20161108 00:23:08,411 [main]-[DEBUG] <== Total: 1
可以看到，已經解決連接符的兼容性問題了。
另外，我們也發現，使用LIKE關鍵字時，寫起來比較麻煩，那我們就給它一組新的SQL配置函數吧：
public class LikeSqlConfigFunctionFactory implements ISqlConfigFunctionFactory{@Overridepublic Collection<ISqlConfigFunction> getSqlConfigFunctions() {return Arrays.asList(getLeftLikeSqlConfigFunction(),getRightLikeSqlConfigFunction(),getLikeSqlConfigFunction());}private ISqlConfigFunction getLeftLikeSqlConfigFunction(){return new AbstractLikeSqlConfigFunction(){@Overridepublic String getName() {return "llike";}@Overrideprotected String eval(String arg) {return "LIKE $concat{'%',"+arg+"}";}};}private ISqlConfigFunction getRightLikeSqlConfigFunction(){return new AbstractLikeSqlConfigFunction(){@Overridepublic String getName() {return "rlike";}@Overrideprotected String eval(String arg) {return "LIKE $concat{"+arg+", '%'}";}};}private ISqlConfigFunction getLikeSqlConfigFunction(){return new AbstractLikeSqlConfigFunction(){@Overridepublic String getName() {return "like";}@Overrideprotected String eval(String arg) {return "LIKE $concat{'%',"+arg+", '%'}";}};}private abstract class AbstractLikeSqlConfigFunction extends AbstractSqlConfigFunction{@Overridepublic String eval(String databaseId, String[] args) {if(args.length != 1){Throw.throwException("the like function require one and only one argument.");}return eval(args[0]);}protected abstract String eval(String arg);}}這裡，定義了一組SQL配置函數，左相似，右相似以及中間相似匹配，並且SQL配置函數還可以嵌套。於是，SqlMapper的配置文件簡化為：
<select id="selectString" resultType="string">select PARAM_NAME, $concat{PARAM_CODE, PARAM_NAME} AS CODE_NAME from BF_PARAM_ENUM_DEF<if test="null != paramName and '' != paramName">where PARAM_NAME $like{#{paramName, jdbcType=VARCHAR}}</if></select>運行結果完全相同。
如果還覺得麻煩，因為PARAM_NAME和paramName是駝峰式對應，甚至還可以添加一個fieldLike函數，並將配置修改為
where $fieldLike{#{PARAM_NAME, jdbcType=VARCHAR}}如果再結合數據字典，jdbcType的配置也可自動生成：
where $fieldLike{#{PARAM_NAME}}這種情形下，如果有多個參數，也不會出現歧義（或者新定義一個配置函數$likes{}消除歧義），於是可將多個條件簡化成：
where $likes{#{PARAM_NAME, PARAM_NAME2， PARAM_NAME3}}當然，還有更多可挖掘的簡化，已經不止是兼容性的範疇了，這裡就不再進一步展開了。
二、DECODE函數/CASE ... WHEN
Oracle中的DECODE函數非常方便，語法如下：
DECODE(條件,值1,返回值1,值2,返回值2,...值n,返回值n[,缺省值])
等價的標準寫法：
CASE 條件WHEN 值1 THEN 返回值1WHEN 值2 THEN 返回值2...WHEN 值n THEN 返回值n[ELSE 缺省值]END
現在我們來實現一個$decode配置函數：
public class DecodeSqlConfigFunction extends AbstractSqlConfigFunction{@Overridepublic String getName() {return "decode";}@Overridepublic String eval(String databaseId, String[] args) {if(args.length < 3){Throw.throwException("the decode function require at least three arguments.");}if("h2".equalsIgnoreCase(databaseId)){//測試時，使用h2代替oracle，正式程序中修改為oraclereturn "DECODE("+Tool.STRING.join(args, ",")+")";}else{StringBuffer sb = new StringBuffer();sb.append("CASE ").append(args[0]);int i=2, l = args.length;for(; i < l; i= i+2){sb.append(" WHEN ").append(args[i-1]).append(" THEN ").append(args[i]);}if(i == l){//結束循環時，兩者相等說明最後一個參數未使用sb.append(" ELSE ").append(args[l-1]);}sb.append(" END");return sb.toString();}}}然後使用SchemaHandlers註冊，修改SqlMapper中配置：
<select id="selectString" resultType="string">select PARAM_NAME, $decode{#{paramName}, '1', 'A', '2', 'B','C'} AS DECODE_TEST from BF_PARAM_ENUM_DEF<if test="null != paramName and '' != paramName">where PARAM_NAME $like{#{paramName, jdbcType=VARCHAR}}</if></select>測試如下：
（1）H2中(以H2代替Oracle)
20161108 06:53:29,747 [main]-[DEBUG] ==> Preparing: select PARAM_NAME, DECODE(?,'1','A','2','B','C') AS DECODE_TEST from BF_PARAM_ENUM_DEF where PARAM_NAME LIKE '%'||?||'%'
（2）MySQL中
20161108 06:50:55,998 [main]-[DEBUG] ==> Preparing: select PARAM_NAME, CASE ? WHEN '1' THEN 'A' WHEN '2' THEN 'B' ELSE 'C' END AS DECODE_TEST from BF_PARAM_ENUM_DEF where PARAM_NAME LIKE '%'||?||'%'
以上所述是小編給大家介紹的Mybatis中SqlMapper配置的擴展與應用詳細介紹(1)，希望對大家有所幫助，如果大家有任何疑問請給我留言，小編會及時回復大家的。在此也非常感謝大家對武林網網站的支持！