https://www.jianshu.com/p/ba27cb6177e9

正文

首先實現(xiàn)org.apache.ibatis.plugin.Interceptor接口，復寫以下三個方法：

//實現(xiàn)攔截邏輯的地方，內(nèi)部要通過invocation.proceed()顯式地推進責任鏈前進， //也就是調(diào)用下一個攔截器攔截目標方法Object intercept(Invocation invocation) throws Throwable; //用當前這個攔截器生成對目標target的代理，實際是通過Plugin.wrap(target,this)來完成的， //把目標target和攔截器this傳給了包裝函數(shù) Object plugin(Object target); //設(shè)置額外的參數(shù)，參數(shù)配置在攔截器的Properties節(jié)點里 void setProperties(Properties properties);

如果想要攔截所有的sql，在實現(xiàn)類上添加 annotation

@Intercepts({@Signature(type = StatementHandler.class, method = "prepare", args = {Connection.class})})

注：Mybatis支持對Executor、StatementHandler、PameterHandler和ResultSetHandler進行攔截，也就是說會對這4種對象進行代理。

框架如上，具體實現(xiàn)有兩個重要點：

1 表的拆分規(guī)則

可以在Mapper對象中加上一個annotation，按以下方式去獲取：

String className = id.substring(0, id.lastIndexOf("."));Class classObj = Class.forName(className);//根據(jù)配置自動生成分表SQL TableSeg tableSeg = classObj.getAnnotation(TableSeg.class);

TableSeg對象定義：

@Target({ ElementType.TYPE })@Retention(RetentionPolicy.RUNTIME)@Inherited@Documented public @interface TableSeg { /** * 表名 * @return */ public String tableName(); /** * 分表方式，取模，如%5：表示取5余數(shù)， * 如果不設(shè)置，直接根據(jù)shardBy值分表 * @return */ public String shardType(); /** * 根據(jù)什么字段分表 * 多個字段用數(shù)學表達表示,如a+b a-b * @return */ public String shardBy(); }

2 sql解析與替換

可以通過以下方法去獲取BoundSql，這個對象有關(guān)于這個sql的內(nèi)容

StatementHandler statementHandler = (StatementHandler) invocation.getTarget();BoundSql boundSql = statementHandler.getBoundSql();

目前看到一些常規(guī)做法都是利用string 的replace方案替換sql中的表名，這顯然一個埋坑的做法。利用詞法分析器才是完美方案，可以使用antlr4，grammar文件可以去github上找到。

antlr值得深入學習。

---

總結(jié)

在編寫插件時需注意以下幾個原則：

1)不編寫不必要的插件；

2)實現(xiàn)plugin方法時判斷一下目標類型，是本插件要攔截的對象才執(zhí)行Plugin.wrap方法，否者直接返回目標本省，這樣可以減少目標被代理的次數(shù)。

作者：harry_chen
鏈接：https://www.jianshu.com/p/ba27cb6177e9
來源：簡書
著作權(quán)歸作者所有。商業(yè)轉(zhuǎn)載請聯(lián)系作者獲得授權(quán)，非商業(yè)轉(zhuǎn)載請注明出處。

轉(zhuǎn)載于:https://www.cnblogs.com/davidwang456/articles/8995401.html

總結(jié)

以上是生活随笔為你收集整理的Mybatis实现分库分表的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。