公共字段自动填充

164 阅读2分钟
**公共字段自动填充功能-在实体类加的注解
@TableField(fill=FieldFill.INSERT)插入时填充该属性值在实体类加的注解
@TableField(fill=FieldFill.INSERT_UPDATE)插入/更新时填充该属性值
createTime createUser插入(INSERT)当前时间
updateTime updateUser 插入(INSERT) , 更新(UPDATE)当前登录用户ID | 
 按照框架要求编写元数据对象处理器,在此类中统一为公共字段赋值,此类需要实现MetaObjectHandler接口。
 所属包: com.itheima.reggie.common
 import com.baomidou.mybatisplus.core.handlers.MetaObjectHandler;
import lombok.extern.slf4j.Slf4j;
import org.apache.ibatis.reflection.MetaObject;
import org.springframework.stereotype.Component;
import java.time.LocalDateTime;

/**
 * 自定义元数据对象处理器
 */
@Component
@Slf4j
public class MyMetaObjecthandler implements MetaObjectHandler {
    /**
     * 插入操作,自动填充
     * @param metaObject
     */
    @Override
    public void insertFill(MetaObject metaObject) {
        log.info("公共字段自动填充[insert]...");
        log.info(metaObject.toString());
        
        metaObject.setValue("createTime", LocalDateTime.now());
        metaObject.setValue("updateTime",LocalDateTime.now());
        metaObject.setValue("createUser",new Long(1));
        metaObject.setValue("updateUser",new Long(1));
    }

    /**
     * 更新操作,自动填充
     * @param metaObject
     */
    @Override
    public void updateFill(MetaObject metaObject) {
        log.info("公共字段自动填充[update]...");
        log.info(metaObject.toString());

        metaObject.setValue("updateTime",LocalDateTime.now());
        metaObject.setValue("updateUser",new Long(1));
    }
}
基于ThreadLocal封装工具类,用户保存和获取当前登录用户id 
LoginCheckFilter的doFilter方法中获取当前登录用户id,并调用ThreadLocal的set方法来设置当前线程的线程局部变量的值(用户id),然后在MyMetaObjectHandler的updateFill方法中调用ThreadLocal的get方法来获得当前线程所对应的线程局部变量的值(用户id)。 如果在后续的操作中, 我们需要在Controller / Service中要使用当前登录用户的ID, 可以直接从ThreadLocal直接获取。  
改造成动态获取当前登录用户的id,ThreadLocal并不是一个Thread,而是Thread的局部变量
ThreadLocal常用方法
  set(T value) : 设置当前线程的线程局部变量的值 
 get() : 返回当前线程所对应的线程局部变量的值 
 remove() : 删除当前线程所对应的线程局部变量的值  
long id = Thread.currentThread().getId();
log.info("线程id为:{}",id);
public class BaseContext {
    private static ThreadLocal<Long> threadLocal = new ThreadLocal<>();
    /**
     * 设置值
     * @param id
     */
    public static void setCurrentId(Long id){
        threadLocal.set(id);
    }
    /**
     * 获取值
     * @return
     */
    public static Long getCurrentId(){
        return threadLocal.get();
    }
}
2). 在LoginCheckFilter的doFilter方法中调用BaseContext来设置当前登录用户的id

在doFilter方法中, 判定用户是否登录, 如果用户登录, 在放行之前, 获取HttpSession中的登录用户信息, 调用BaseContext的setCurrentId方法将当前登录用户ID存入ThreadLocal

image.png MyMetaObjectHandler中 **调用BaseContext获取登录用户的id

将之前在代码中固定的当前登录用户1, 修改为动态调用BaseContext中的getCurrentId方法获取当前登录用户ID

image.png