攔截器(Interceptor)在 Mybatis 中被當做插件(plugin)對待,官方文檔提供了 Executor,ParameterHandler,ResultSetHandler,StatementHandler 共4種,並且提示“這些類中方法的細節可以通過查看每個方法的簽名來發現,或者直接查看 MyBatis 發行包中的源代碼”。
攔截器的使用場景主要是更新數據庫的通用字段,分庫分表,加解密等的處理。
1. Interceptor
攔截器均需要實現該接口。
2. Intercepts 攔截器
@Intercepts({
@Signature(type = Executor.class, method = "update", args = {MappedStatement.class, Object.class})
})
攔截器的使用需要查看每一個type所提供的方法參數。
Signature 對應 Invocation 構造器, type 爲 Invocation.Object, method 爲 Invocation.Method, args 爲 Invocation.Object[]。
method 對應的 update 包括了最常用的 insert/update/delete 三種操作, 因此 update 本身無法直接判斷sql爲何種執行過程。
args 包含了其餘多有的操作信息, 按數組進行存儲, 不同的攔截方式有不同的參數順序, 具體看type接口的方法簽名, 然後根據簽名解析。
3. Object 對象類型
args 參數列表中,Object.class 是特殊的對象類型。如果有數據庫統一的實體 Entity 類,即包含表公共字段,比如創建、更新操作對象和時間的基類等,在編寫代碼時儘量依據該對象來操作,會簡單很多。該對象的判斷使用
Object parameter = invocation.getArgs()[1];
if (parameter instanceof BaseEntity) {
BaseEntity entity = (BaseEntity) parameter;
}
即可,根據語句執行類型選擇對應字段的賦值。
如果參數不是實體,而且具體的參數,那麼 Mybatis 也做了一些處理,比如 @Param("name") String name
類型的參數,會被包裝成 Map
接口的實現來處理,即使是原始 Map
也是如此。使用
Object parameter = invocation.getArgs()[1];
if (parameter instanceof Map) {
Map map = (Map) parameter;
}
即可,對具體統一的參數進行賦值。
4. SqlCommandType 命令類型
Executor
提供的方法中,update
包含了 新增,修改和刪除類型,無法直接區分,需要藉助 MappedStatement
類的屬性 SqlCommandType
來進行判斷,該類包含了所有的操作類型
public enum SqlCommandType {
UNKNOWN, INSERT, UPDATE, DELETE, SELECT, FLUSH;
}
畢竟新增和修改的場景,有些參數是有區別的,比如創建時間和更新時間,update
時是無需兼顧創建時間字段的。
5. 實例
自己編寫的小項目中,需要統一給數據庫字段屬性賦值:
public class BaseEntity {
private int id;
private int creator;
private int updater;
private Long createTime;
private Long updateTime;
}
dao 操作使用了實體和參數的方式,個人建議還是統一使用實體比較簡單,易讀。
使用實體:
int add(BookEntity entity);
使用參數:
int update(@Param("id") int id, @Param("url") String url, @Param("description") String description, @Param("playCount") int playCount, @Param("creator") int creator, @Param("updateTime") long updateTime);
完整的例子:
package com.github.zhgxun.talk.common.plugin;
import com.github.zhgxun.talk.common.util.DateUtil;
import com.github.zhgxun.talk.common.util.UserUtil;
import com.github.zhgxun.talk.entity.BaseEntity;
import com.github.zhgxun.talk.entity.UserEntity;
import lombok.extern.slf4j.Slf4j;
import org.apache.ibatis.executor.Executor;
import org.apache.ibatis.mapping.MappedStatement;
import org.apache.ibatis.mapping.SqlCommandType;
import org.apache.ibatis.plugin.Interceptor;
import org.apache.ibatis.plugin.Intercepts;
import org.apache.ibatis.plugin.Invocation;
import org.apache.ibatis.plugin.Plugin;
import org.apache.ibatis.plugin.Signature;
import org.springframework.stereotype.Component;
import java.util.Map;
import java.util.Properties;
/**
* 全局攔截數據庫創建和更新
* <p>
* Signature 對應 Invocation 構造器, type 爲 Invocation.Object, method 爲 Invocation.Method, args 爲 Invocation.Object[]
* method 對應的 update 包括了最常用的 insert/update/delete 三種操作, 因此 update 本身無法直接判斷sql爲何種執行過程
* args 包含了其餘多有的操作信息, 按數組進行存儲, 不同的攔截方式有不同的參數順序, 具體看type接口的方法簽名, 然後根據簽名解析, 參見官網
*
* @link http://www.mybatis.org/mybatis-3/zh/configuration.html#plugins 插件
* <p>
* MappedStatement 包括了SQL具體操作類型, 需要通過該類型判斷當前sql執行過程
*/
@Component
@Intercepts({
@Signature(type = Executor.class, method = "update", args = {MappedStatement.class, Object.class})
})
@Slf4j
public class NormalPlugin implements Interceptor {
@Override
@SuppressWarnings("unchecked")
public Object intercept(Invocation invocation) throws Throwable {
// 根據簽名指定的args順序獲取具體的實現類
// 1. 獲取MappedStatement實例, 並獲取當前SQL命令類型
MappedStatement ms = (MappedStatement) invocation.getArgs()[0];
SqlCommandType commandType = ms.getSqlCommandType();
// 2. 獲取當前正在被操作的類, 有可能是Java Bean, 也可能是普通的操作對象, 比如普通的參數傳遞
// 普通參數, 即是 @Param 包裝或者原始 Map 對象, 普通參數會被 Mybatis 包裝成 Map 對象
// 即是 org.apache.ibatis.binding.MapperMethod$ParamMap
Object parameter = invocation.getArgs()[1];
// 獲取攔截器指定的方法類型, 通常需要攔截 update
String methodName = invocation.getMethod().getName();
log.info("NormalPlugin, methodName; {}, commandType: {}", methodName, commandType);
// 3. 獲取當前用戶信息
UserEntity userEntity = UserUtil.getCurrentUser();
// 默認測試參數值
int creator = 2, updater = 3;
if (parameter instanceof BaseEntity) {
// 4. 實體類
BaseEntity entity = (BaseEntity) parameter;
if (userEntity != null) {
creator = entity.getCreator();
updater = entity.getUpdater();
}
if (methodName.equals("update")) {
if (commandType.equals(SqlCommandType.INSERT)) {
entity.setCreator(creator);
entity.setUpdater(updater);
entity.setCreateTime(DateUtil.getTimeStamp());
entity.setUpdateTime(DateUtil.getTimeStamp());
} else if (commandType.equals(SqlCommandType.UPDATE)) {
entity.setUpdater(updater);
entity.setUpdateTime(DateUtil.getTimeStamp());
}
}
} else if (parameter instanceof Map) {
// 5. @Param 等包裝類
// 更新時指定某些字段的最新數據值
if (commandType.equals(SqlCommandType.UPDATE)) {
// 遍歷參數類型, 檢查目標參數值是否存在對象中, 該方式需要應用編寫有一些統一的規範
// 否則均統一爲實體對象, 就免去該重複操作
Map map = (Map) parameter;
if (map.containsKey("creator")) {
map.put("creator", creator);
}
if (map.containsKey("updateTime")) {
map.put("updateTime", DateUtil.getTimeStamp());
}
}
}
// 6. 均不是需要被攔截的類型, 不做操作
return invocation.proceed();
}
@Override
public Object plugin(Object target) {
return Plugin.wrap(target, this);
}
@Override
public void setProperties(Properties properties) {
}
}
6. 感受
其它幾種類型,後面在看,尤其是分頁。
在編寫代碼的過程中,有時候還是需要先知道對象的類型,比如 Object.class 跟 Interface 一樣,很多時候僅僅代表一種數據類型,需要明確知道後再進行操作。
@Param
標識的參數其實會被 Mybatis 封裝成 org.apache.ibatis.binding.MapperMethod$ParamMap
類型,一開始我就是使用
for (Field field:parameter.getClass().getDeclaredFields()) {
}
的方式獲取,以爲這些都是對象的屬性,通過反射獲取並修改即可,實際上發現對象不存在這些屬性,但是打印出的信息中,明確有這些字段的,網上參考一些信息後,才知道這是一個 Map
類型,問題才迎刃而解。