mybatisplus公共字段自动填充

你比如说像插入时间、更新时间等这种类型的字段，你在插入、更新时候要额外写一个setValue，总写总写其实有点麻烦的，这里mybatisplus给我们提供了一个公共字段统一自动填充的方式。

只需要在对应字段加上注释@TableField(fill = FieldFill.INSERT_UPDATE)，FieldFill.INSERT_UPDATE代表插入和更新时候都自动填充。

package com.xxx.xxxxx.entity;

import com.baomidou.mybatisplus.annotation.FieldFill;
import com.baomidou.mybatisplus.annotation.IdType;
import java.util.Date;

import com.baomidou.mybatisplus.annotation.TableField;
import com.baomidou.mybatisplus.annotation.TableId;
import java.io.Serializable;

import com.fasterxml.jackson.annotation.JsonFormat;
import io.swagger.annotations.ApiModel;
import io.swagger.annotations.ApiModelProperty;
import lombok.Data;
import lombok.EqualsAndHashCode;
import lombok.experimental.Accessors;

@Data
@EqualsAndHashCode(callSuper = false)
@Accessors(chain = true)
@ApiModel(value="Employee对象", description="员工信息")
public class Employee implements Serializable {

    private static final long serialVersionUID=1L;

    @ApiModelProperty(value = "主键")
    @TableId(value = "id", type = IdType.ID_WORKER)
    private Long id;

    @ApiModelProperty(value = "姓名")
    private String name;

    @ApiModelProperty(value = "用户名")
    private String username;

    @ApiModelProperty(value = "密码")
    private String password;

    @ApiModelProperty(value = "手机号")
    private String phone;

    @ApiModelProperty(value = "性别")
    private String sex;

    @ApiModelProperty(value = "身份证号")
    private String idNumber;

    @ApiModelProperty(value = "状态 0:禁用，1:正常")
    private Integer status;

    @JsonFormat(pattern = "yyyy-MM-dd")
    @TableField(fill = FieldFill.INSERT)
    @ApiModelProperty(value = "创建时间")
    private Date createTime;

    @JsonFormat(pattern = "yyyy-MM-dd")
    @TableField(fill = FieldFill.INSERT_UPDATE)
    @ApiModelProperty(value = "更新时间")
    private Date updateTime;

    @TableField(fill = FieldFill.INSERT)
    @ApiModelProperty(value = "创建人")
    private Long createUser;

    @TableField(fill = FieldFill.INSERT_UPDATE)
    @ApiModelProperty(value = "修改人")
    private Long updateUser;
}

然后再写一个自动填充逻辑，注意这个插入、修改的用户名的逻辑，你没办法直接获取，所以在一开始登录的时候就要有一个对threadLocal变量的处理，把用户名存进去，因为每一个请求后台都是启动一个线程跟着去执行的，如下的UserContext。

public class UserContext {
    // 全局使用
    public static ThreadLocal<Long> USER_ID = new ThreadLocal<>();

    public static Long getUserId() {
        return USER_ID.get();
    }

    public static void setUserId(Long userId) {
        USER_ID.set(userId);
    }
}

package com.xxx.xxxxxx.config;


import com.xxx.xxxxx.common.UserContext;
import com.xxx.xxxxx.entity.Employee;
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;
import java.util.Date;


@Component
@Slf4j
public class MyMetaObjectHandler implements MetaObjectHandler {

    @Override
    public void insertFill(MetaObject metaObject) {
        log.info("start insert fill ....");
        this.setFieldValByName("createTime",new Date(),metaObject);
        this.setFieldValByName("updateTime",new Date(),metaObject);
        this.setFieldValByName("createUser",UserContext.getUserId(),metaObject);
        this.setFieldValByName("updateUser",UserContext.getUserId(),metaObject);
    }

    @Override
    public void updateFill(MetaObject metaObject) {
        log.info("start update fill ....");
        this.setFieldValByName("updateTime",new Date(),metaObject);
        this.setFieldValByName("updateUser", UserContext.getUserId(),metaObject );
    }
}

假设有不同的表都需要这个自动填充，那需要再加一层判断表名：

String tableName = metaObject.getMetaObject().getTableName();
if ("user".equals(tableName)) {
    this.setFieldValByName("createTime",new Date(),metaObject);
}