场景
在后端服务开发时,现在很流行的框架组合就是SSM(SpringBoot + Spring + MyBatis),在我们进行一些业务系统开发时,会有很多的业务数据表,而表中的信息从新插入开始,整个生命周期过程中可能会进行很多次的操作。
比如,我们在某网站购买一件商品,会生成一条订单记录,在支付完金额后订单状态会变为已支付,等最后我们收到订单商品,这个订单状态会变成已完成等。
假设我们的订单表t_order
结果如下:
当订单创建时,需要设置insert_by
,insert_time
,update_by
,update_time
的值;
在进行订单状态更新时,则只需要更新update_by
,update_time
的值。
那应该如何处理呢?
麻瓜做法
最简单的做法,也是最容易想到的做法,就是在每个业务处理的代码中,对相关的字段进行处理。
比如订单创建的方法中,如下处理:
public void create(Order order){
// ...其他代码
// 设置审计字段
Date now = new Date();
order.setInsertBy(appContext.getUser());
order.setUpdateBy(appContext.getUser());
order.setInsertTime(now);
order.setUpdateTime(now);
orderDao.insert(order);
}
订单更新方法则只设置updateBy
和updateTime
:
public void update(Order order){
// ...其他代码
// 设置审计字段
Date now = new Date();
order.setUpdateBy(appContext.getUser());
order.setUpdateTime(now);
orderDao.insert(order);
}
这种方式虽然可以完成功能,但是存在一些问题:
- 需要在每个方法中按照不同的业务逻辑决定设置哪些字段;
- 在业务模型变多后,每个模型的业务方法中都要进行设置,重复代码太多。
那我们知道这种方式存在问题以后,就得找找有什么好方法对不对,往下看!
优雅做法
因为我们持久层框架更多地使用MyBatis,那我们就借助于MyBatis的拦截器来完成我们的功能。
首先我们来了解一下,什么是拦截器?
什么是拦截器?
MyBatis的拦截器顾名思义,就是对某些操作进行拦截。通过拦截器可以对某些方法执行前后进行拦截,添加一些处理逻辑。
MyBatis的拦截器可以对Executor、StatementHandler、PameterHandler和ResultSetHandler 接口进行拦截,也就是说会对这4种对象进行代理。
拦截器设计的初衷就是为了让用户在MyBatis的处理流程中不必去修改MyBatis的源码,能够以插件的方式集成到整个执行流程中。
比如MyBatis中的Executor
有BatchExecutor
、ReuseExecutor
、SimpleExecutor
和CachingExecutor
,如果这几种实现的query
方法都不能满足你的需求,我们可以不用去直接修改MyBatis的源码,而通过建立拦截器的方式,拦截Executor
接口的query
方法,在拦截之后,实现自己的query方法逻辑。
在MyBatis中的拦截器通过Interceptor接口表示,该接口中有三个方法。
public interface Interceptor {
Object intercept(Invocation invocation) throws Throwable;
Object plugin(Object target);
void setProperties(Properties properties);
}
plugin方法是拦截器用于封装目标对象的,通过该方法我们可以返回目标对象本身,也可以返回一个它的代理。
当返回的是代理的时候我们可以对其中的方法进行拦截来调用intercept方法,当然也可以调用其他方法。
setProperties方法是用于在Mybatis配置文件中指定一些属性的。
使用拦截器更新审计字段
那么我们应该如何通过拦截器来实现我们对审计字段赋值的功能呢?
在我们进行订单创建和修改时,本质上是通过MyBatis执行insert、update语句,MyBatis是通过Executor来处理的。
我们可以通过拦截器拦截Executor,然后在拦截器中对要插入的数据对象根据执行的语句设置insert_by,insert_time,update_by,update_time等属性值就可以了。
自定义拦截器
自定义Interceptor
最重要的是要实现plugin
方法和intercept
方法。
在plugin
方法中我们可以决定是否要进行拦截进而决定要返回一个什么样的目标对象。
在intercept
方法就是要进行拦截的时候要执行的方法。
对于plugin
方法而言,其实Mybatis已经为我们提供了一个实现。Mybatis中有一个叫做Plugin
的类,