自己動手編寫一個Mybatis插件之Mybatis脫敏插件
1. 前言
在日常開發(fā)中，身份證號、手機號、卡號、客戶號等個人信息都需要進行數(shù)據(jù)脫敏。否則容易造成個人隱私泄露，客戶資料泄露，給不法分子可乘之機。但是數(shù)據(jù)脫敏不是把敏感信息隱藏起來，而是看起來像真的一樣，實際上不能是真的。我以前的公司就因為不重視脫敏，一名員工在離職的時候通過后臺的導(dǎo)出功能導(dǎo)出了核心的客戶資料賣給了競品，給公司造成了重大的損失。當然這里有數(shù)據(jù)管理的原因，但是脫敏仍舊是不可忽略的一環(huán)，脫敏可以從一定程度上保證數(shù)據(jù)的合規(guī)使用。下面就是一份經(jīng)過脫敏的數(shù)據(jù)：
2. Mybatis 脫敏插件
最近在研究Mybatis的插件，所以考慮能不能在ORM中搞一搞脫敏，所以就嘗試了一下，這里分享一下思路。借此也分享一下Mybatis插件開發(fā)的思路。
2.1 Mybatis 插件接口
Mybatis中使用插件，需要實現(xiàn)接口org.apache.ibatis.plugin.Interceptor，如下所示：
public interface Interceptor {
Object intercept(Invocation invocation) throws Throwable;
default Object plugin(Object target) {
return Plugin.wrap(target, this);
}
default void setProperties(Properties properties) {
// NOP
}
}
這里其實最核心的是Object intercept(Invocation invocation)方法，這是我們需要實現(xiàn)的方法。
2.2 Invocation 對象
那么核心方法中的Invocation 是個什么概念呢？
public class Invocation {
private final Object target;
private final Method method;
private final Object[] args;
public Invocation(Object target, Method method, Object[] args) {
this.target = target;
this.method = method;
this.args = args;
}
public Object getTarget() {
return target;
}
public Method getMethod() {
return method;
}
public Object[] getArgs() {
return args;
}
public Object proceed() throws InvocationTargetException, IllegalAccessException {
return method.invoke(target, args);
}
}
這個東西包含了四個概念：
- target 攔截的對象
- method 攔截target中的具體方法，也就是說Mybatis插件的粒度是精確到方法級別的。
- args 攔截到的參數(shù)。
- proceed 執(zhí)行被攔截到的方法，你可以在執(zhí)行的前后做一些事情。
2.3 攔截簽名
既然我們知道了Mybatis插件的粒度是精確到方法級別的，那么疑問來了，插件如何知道輪到它工作了呢？
所以Mybatis設(shè)計了簽名機制來解決這個問題，通過在插件接口上使用注解@Intercepts標注來解決這個問題。
@Intercepts(@Signature(type = ResultSetHandler.class,
method = "handleResultSets",
args = {Statement.class})
就像上面一樣，事實上就等于配置了一個Invocation。
2.4 插件的作用域
那么問題又來了，Mybatis插件能攔截哪些對象,或者說插件能在哪個生命周期階段起作用呢？它可以攔截以下四大對象：
- Executor 是SQL執(zhí)行器，包含了組裝參數(shù)，組裝結(jié)果集到返回值以及執(zhí)行SQL的過程，粒度比較粗。
- StatementHandler 用來處理SQL的執(zhí)行過程，我們可以在這里重寫SQL非常常用。
- ParameterHandler 用來處理傳入SQL的參數(shù)，我們可以重寫參數(shù)的處理規(guī)則。
- ResultSetHandler 用于處理結(jié)果集，我們可以重寫結(jié)果集的組裝規(guī)則。
你需要做的就是明確的你的業(yè)務(wù)需要在上面四個對象的哪個處理階段攔截處理即可。
2.5 MetaObject
Mybatis提供了一個工具類org.apache.ibatis.reflection.MetaObject。它通過反射來讀取和修改一些重要對象的屬性。我們可以利用它來處理四大對象的一些屬性，這是Mybatis插件開發(fā)的一個常用工具類。
- Object getValue(String name) 根據(jù)名稱獲取對象的屬性值，支持OGNL表達式。
- void setValue(String name, Object value) 設(shè)置某個屬性的值。
- Class<?> getSetterType(String name) 獲取setter方法的入?yún)㈩愋汀?/li>
- Class<?> getGetterType(String name) 獲取getter方法的返回值類型。
通常我們使用SystemMetaObject.forObject(Object object)來實例化MetaObject對象。你會在接下來的實戰(zhàn)DEMO中看到我使用它。
3. Mybatis 脫敏插件實戰(zhàn)
接下來我就把開頭的脫敏需求實現(xiàn)一下。首先需要對脫敏字段進行標記并確定使用的脫敏策略。
編寫脫敏函數(shù)：
/**
* 具體策略的函數(shù)
* @author felord.cn
* @since 11:24
**/
public interface Desensitizer extends Function<String,String> {
}
編寫脫敏策略枚舉：
/**
* 脫敏策略.
*
* @author felord.cn
* @since 11 :25
*/
public enum SensitiveStrategy {
/**
* Username sensitive strategy.
*/
USERNAME(s -> s.replaceAll("(\\S)\\S(\\S*)", "$1*$2")),
/**
* Id card sensitive type.
*/
ID_CARD(s -> s.replaceAll("(\\d{4})\\d{10}(\\w{4})", "$1****$2")),
/**
* Phone sensitive type.
*/
PHONE(s -> s.replaceAll("(\\d{3})\\d{4}(\\d{4})", "$1****$2")),
/**
* Address sensitive type.
*/
ADDRESS(s -> s.replaceAll("(\\S{8})\\S{4}(\\S*)\\S{4}", "$1****$2****"));
private final Desensitizer desensitizer;
SensitiveStrategy(Desensitizer desensitizer) {
this.desensitizer = desensitizer;
}
/**
* Gets desensitizer.
*
* @return the desensitizer
*/
public Desensitizer getDesensitizer() {
return desensitizer;
}
}
編寫脫敏字段的標記注解：
@Retention(RetentionPolicy.RUNTIME)
@Target(ElementType.FIELD)
public @interface Sensitive {
SensitiveStrategy strategy();
}
我們的返回對象中如果某個字段需要脫敏，只需要通過標記就可以了。例如下面這樣：
@Data
public class UserInfo {
private static final long serialVersionUID = -8938650956516110149L;
private Long userId;
@Sensitive(strategy = SensitiveStrategy.USERNAME)
private String name;
private Integer age;
}
然后就是編寫插件了，我可以確定的是需要攔截的是ResultSetHandler對象的handleResultSets方法，我們只需要實現(xiàn)插件接口Interceptor并添加簽名就可以了。全部邏輯如下：
@Slf4j
@Intercepts(@Signature(type = ResultSetHandler.class,
method = "handleResultSets",
args = {Statement.class}))
public class SensitivePlugin implements Interceptor {
@SuppressWarnings("unchecked")
@Override
public Object intercept(Invocation invocation) throws Throwable {
List<Object> records = (List<Object>) invocation.proceed();
// 對結(jié)果集脫敏
records.forEach(this::sensitive);
return records;
}
private void sensitive(Object source) {
// 拿到返回值類型
Class<?> sourceClass = source.getClass();
// 初始化返回值類型的 MetaObject
MetaObject metaObject = SystemMetaObject.forObject(source);
// 捕捉到屬性上的標記注解 @Sensitive 并進行對應(yīng)的脫敏處理
Stream.of(sourceClass.getDeclaredFields())
.filter(field -> field.isAnnotationPresent(Sensitive.class))
.forEach(field -> doSensitive(metaObject, field));
}
private void doSensitive(MetaObject metaObject, Field field) {
// 拿到屬性名
String name = field.getName();
// 獲取屬性值
Object value = metaObject.getValue(name);
// 只有字符串類型才能脫敏 而且不能為null
if (String.class == metaObject.getGetterType(name) && value != null) {
Sensitive annotation = field.getAnnotation(Sensitive.class);
// 獲取對應(yīng)的脫敏策略 并進行脫敏
SensitiveStrategy type = annotation.strategy();
Object o = type.getDesensitizer().apply((String) value);
// 把脫敏后的值塞回去
metaObject.setValue(name, o);
}
}
}
然后配置脫敏插件使之生效：
@Bean
public SensitivePlugin sensitivePlugin(){
return new SensitivePlugin();
}
操作查詢獲得結(jié)果 UserInfo(userId=123123, name=李*龍, age=28) ，成功將指定字段進行了脫敏。
補充一句，其實脫敏也可以在JSON序列化的時候進行。
4. 總結(jié)
今天對編寫Mybatis插件的一些要點進行了說明，同時根據(jù)說明實現(xiàn)了一個脫敏插件。但是請注意一定要熟悉四大對象的生命周期，否則自寫插件可能會造成意想不到的結(jié)果。
到此這篇關(guān)于自己動手編寫一個Mybatis插件之Mybatis脫敏插件的文章就介紹到這了,更多相關(guān)Mybatis脫敏插件內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！
相關(guān)文章
Spring?Security權(quán)限想要細化到按鈕實現(xiàn)示例
這篇文章主要為大家介紹了Spring?Security權(quán)限想要細化到按鈕實現(xiàn)示例，有需要的朋友可以借鑒參考下，希望能夠有所幫助，祝大家多多進步，早日升職加薪2022-07-07
java編程實現(xiàn)基于UDP協(xié)議傳輸數(shù)據(jù)的方法
這篇文章主要介紹了java編程實現(xiàn)基于UDP協(xié)議傳輸數(shù)據(jù)的方法,較為詳細的分析了UDP協(xié)議的原理及Java編程實現(xiàn)數(shù)據(jù)傳輸客戶端與服務(wù)器端的相關(guān)技巧,需要的朋友可以參考下2015-11-11
這篇文章主要介紹了java虛擬機之JVM調(diào)優(yōu)詳解,文中有非常詳細的代碼示例,對正在學(xué)習Java虛擬機的小伙伴們有非常好的幫助,需要的朋友可以參考下2021-04-04
這篇文章主要介紹了java容器類知識點詳細總結(jié)，2019-06-06
Java正確實現(xiàn)一個單例設(shè)計模式的示例
今天小編就為大家分享一篇關(guān)于Java正確實現(xiàn)一個單例設(shè)計模式的示例，小編覺得內(nèi)容挺不錯的，現(xiàn)在分享給大家，具有很好的參考價值，需要的朋友一起跟隨小編來看看吧2019-01-01
Red?Hat?安裝JDK與IntelliJ?IDEA的詳細過程
YUM是基于Red Hat的Linux發(fā)行版的一個強大而用戶友好的包管理工具,這篇文章主要介紹了Red?Hat安裝JDK與IntelliJ IDEA,需要的朋友可以參考下2023-08-08
關(guān)于MVC的dao層、service層和controller層詳解
這篇文章主要介紹了關(guān)于MVC的dao層、service層和controller層詳解，具有很好的參考價值，希望對大家有所幫助。如有錯誤或未考慮完全的地方，望不吝賜教2022-02-02
當我們在Spring Boot應(yīng)用中使用多個定時任務(wù)時,任務(wù)之間的阻塞可能是一個常見的問題,這可能會因任務(wù)之間的依賴、執(zhí)行時間過長或資源爭用等原因而發(fā)生,本文讓我們深入探討如何利用Spring Boot來解決多個定時任務(wù)阻塞的問題,感興趣的小伙伴跟著小編一起來看看吧2024-01-01
這篇文章主要介紹了Springboot中如何通過yml為實體類注入屬性，具有很好的參考價值，希望對大家有所幫助。如有錯誤或未考慮完全的地方，望不吝賜教2022-05-05
idea?maven項目啟動項目不編譯target?文件的問題及解決方法
代碼編輯器中無編譯錯誤，通過maven 的clean 、compile、package進行各種操作也都沒問題，但是單擊綠色箭頭運行（默認會先執(zhí)行IDE本身的Build操作）卻報：程序包xxx不存在，這篇文章主要介紹了解決idea maven項目啟動項目不編譯target文件問題,需要的朋友可以參考下2023-05-05