【SpringBoot+MyBatisPlus】利用线程特性与ThreadLocal来解决公共字段自动填充问题

前言

每一次在Controller层中封装改动数据的方法时都要重新设置一些共性字段，显得十分冗余。为了解决此问题也是在项目中第一次利用到线程，总的来说还是让我眼前一亮，也开阔了视野，对以后的开发具有深远的意义！

一.字段自动填充引入

先看一个现象，在之前写好的表中，我们发现有很多字段重复出现

比如update_time、create_time、create_user…
这就导致需要在Controller层中每一次对表中数据进行修改后调用一次.setCreateTime(LocalDateTime.now());或者setUpdateTime(LocalDateTime.now());等等“硬编码问题又出现了”显得格外麻烦
这些共性字段如何统一拿出来处理呢？MyBatisPlus给了我们解决方案，为了实现这一功能:
首先
我们需要在公共字段对应的实体属性上加上@TableField注解与指定填充策略，就像这样：

    @TableField(fill = FieldFill.INSERT)
    private LocalDateTime createTime;

    @TableField(fill = FieldFill.INSERT_UPDATE)
    private LocalDateTime updateTime;

    @TableField(fill = FieldFill.INSERT)
    private Long createUser;

    @TableField(fill = FieldFill.INSERT_UPDATE)
    private Long updateUser;
1
2
3
4
5
6
7
8
9
10
11

不难看出，fill的类型是一个枚举类

若是不需要处理则选择DEAFULT，若是增添后需要修改字段的值选择INSERT，若是涉及修改数据时需要修改字段的值则选择UPDATE，若是插入和修改都需要设置字段的值则选择INSERT_UPDATE就像updateTime和updateUser，只要涉及对数据的操作就要修改字段的值
类似于全局异常处理器，为了实现字段填充也需要定义一个元数据对象处理器

二.元数据对象处理器

在定义的类中实现MetaObjecthandler接口，并重写接口中策略对应的方法，就像这样：

@Component
@Slf4j
public class MyMetaObjecthandler implements MetaObjectHandler {

    @Override
    public void insertFill(MetaObject metaObject) {
        log.info("公共字段自动填充...");
        log.info(metaObject.toString());
    }

    @Override
    public void updateFill(MetaObject metaObject) {
        log.info("公共字段自动填充...");
        log.info(metaObject.toString());
    }
}
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16

这里有点像继承HttpServlet那个套路，重写doGet（）与doPost（）方法，要实现什么功能就写在对应的方法里…
重写完方法，不妨debug一下：

发现我实现接口后在方法里拿到了实体的数据，并封装到了形参里的metaObject对象中，接下来我就可以利用此对象来做公共字段的集中处理了！
所谓的自动填充也正是因为该类拥有@Component注解，在每一次的项目启动后就会被扫描到，加载到，而类中的方法又实现了功能，所以才可以做到自动填充字段！

回到正题

我们要做的是把公共的字段做到统一填充，具体实现则是在重写的方法里调用setValue（）方法并传进去字段与参数，就像这样：

    @Override
    public void insertFill(MetaObject metaObject) {
        log.info("公共字段自动填充...");
        log.info(metaObject.toString());
        metaObject.setValue("createTime", LocalDateTime.now());
        metaObject.setValue("updateTime",LocalDateTime.now());
    }

    @Override
    public void updateFill(MetaObject metaObject) {
        log.info("公共字段自动填充...");
        log.info(metaObject.toString());
        metaObject.setValue("updateTime",LocalDateTime.now());
    }
1
2
3
4
5
6
7
8
9
10
11
12
13
14

相比之前controller层中的 employee.setCreateTime(LocalDateTime.now());/employee.setUpdateTime(LocalDateTime.now());来看还是比较好理解，无非就是“茴“字的另一种写法
(但却实现了一劳永逸)…
可能有人会问，懒羊羊你的updateUser与createUser字段怎么不处理呢？
确实，还记得之前在controller层中我们是怎么处理的吗？

为了确定User是谁，我们从Session里拿到了操作的对象id，并调用set方法修改了对象的字段值

按照上面的经验，为了能够动态的拿到id，我们应该这样设置：

可是，在此方法中metaObject对象好像不能利用Session里的empID，那要如何获得对象的标识呢？
不就是一个标识么，我用线程id可以吗？

三.Threadlocal的使用

在这之前，需要明白一点，每当客户端发送一次HTTP请求，对应在服务器端会分配一个新的线程来处理。项目设计到现在为止，以点击一次保存按钮作为一次请求，它会触发过滤器、调用Controller层、MetaObjectHandler层的方法
我们通过Thread的内部方法：long id = Thread.currentThread().getId();来获得当前线程的id,以日志的形式log.info("当前线程id:{}",id);输出到控制台：

可见，三者的线程id相同，说明他们在同一个线程中，这就保证了一致性，也正是因为这个特性，所以可用来当作表中的字段id使用来作为标识

在此基础上，大致方向已经敲定是利用线程的特性，具体要如何实现呢？那就不得不需要了解一下ThreadLocal这个概念了：

1.ThreadLocal并不是一个Thread，而是Thread的局部变量。
2.当使用ThreadLocal维护变量时，ThreadLocal为每个使用该变量的线程提供独立的变量副本，所以每一个线程都可以独立地改变自己的副本，而不会影响其它线程所对应的副本。
3.ThreadLocal为每个线程提供单独一份存储空间，具有线程隔离的效果，只有在线程内才能获取到对应的值，线程外则不能访问

常用方法

了解到ThreadLocal特性，我们就可以结合“同一线程”这一特性来获得那个对象唯一的SessionId。

🚩 在前面的登录功能中我们设置了一个filter，在filter中我们已经拿到过了一次SessionId，写到这里解决方案不就出来了嘛——把filter中的SessionId当作参数传给ThreadLocal的set方法，然后在MetaObjectHandler实现类（元数据对象处理器）中调用ThreadLocal的get()方法得到SessionId

就像这样：
1.为了规范，我们封装一个类，功能是调用ThreadLocal的方法

/**
 * 基于ThreadLocal封装工具类，用于保护和获取当前登录id
 */
public class BaseContext {
    private static ThreadLocal<Long> threadLocal=new ThreadLocal();

    public static void setCurrentId(Long currentId){
        threadLocal.set(currentId);
    }
    public static Long getCurrentId(){
        return threadLocal.get();
    }
}
1
2
3
4
5
6
7
8
9
10
11
12
13

2.在filter中做到SessionId的迁移

3.在MetaObjectHandler实现类中利用SessionId完成公共字段填充设置

    @Override
    public void updateFill(MetaObject metaObject) {
        log.info("公共字段自动填充...");
        log.info(metaObject.toString());
        long id = Thread.currentThread().getId();
        log.info("当前线程id:{}",id);
        metaObject.setValue("updateTime",LocalDateTime.now());
        metaObject.setValue("createUser",BaseContext.getCurrentId());
        metaObject.setValue("updateUser",BaseContext.getCurrentId());
    }
1
2
3
4
5
6
7
8
9
10

这样，就可以一劳永逸咯~
又是一个小技巧啊！

四.不用ThreadLocal怎么实现？

写到这里
其实不用ThreadLocal也是可以实现自动填充的，上面之所以爆红是因为HttpServletRequest并没有注入到IOC里，也就是说在MyMetaObjecthandler中跟本没有引用到session那通过Session获取id自然行不通，要想使用注入即可！