实现 MetaObjectHandler接口来解决公共字段 创建/修改人,创建/修改时间等的自动填充

记录一下写代码解决到的问题。用的是mybatisPlus。本来想的是用切点,后面写着发现写不出来了,只好找别方案,这个可能不是最优方案,但是目前阶段对我来时是好点的

1创建一个 MetaAutoFillConfig类,(名字自己随便起),去实现MetaObjectHandler接口。重写 insertFill和updateFill 方法

核心代码如下:

@Component
public class MetaAutoFillConfig implements MetaObjectHandler {


    @Override
    public void insertFill(MetaObject metaObject) {
        // 在插入操作时自动填充的字段
        this.strictInsertFill(metaObject, "createTime", LocalDateTime.class, LocalDateTime.now());
        this.strictInsertFill(metaObject, "updateTime", LocalDateTime.class, LocalDateTime.now());
        this.strictInsertFill(metaObject, "createUser", Long.class, getCurrentUserId()); // 替换为获取当前用户ID的逻辑
        this.strictInsertFill(metaObject, "updateUser", Long.class, getCurrentUserId()); // 替换为获取当前用户ID的逻辑
    }

    @Override
    public void updateFill(MetaObject metaObject) {
        // 在更新操作时自动填充的字段
        this.strictUpdateFill(metaObject, "updateTime", LocalDateTime.class, LocalDateTime.now());
        this.strictUpdateFill(metaObject, "updateUser", Long.class, getCurrentUserId()); // 替换为获取当前用户ID的逻辑
    }

    private Long getCurrentUserId() {
        // 这里应该根据你的业务逻辑获取当前用户的ID
        // 这一部分我是从我登录的时候在线程存放了登录用户的id,然后在这里来拿到登录用户的id的
        // 你们这里需要根据自己的方法得到自己登录用户的id或者你们想加入的数据
        Long currentId = BaseContext.getCurrentId();
        return currentId; 
    }
}

我简单说明一下,我也怕过几天我自己也忘了
这个部分的createTime 是映射到数据库的字段。



this.strictInsertFill(metaObject, "createTime", LocalDateTime.class, LocalDateTime.now());


这个是我的实体类,采用但是驼峰命名方式

@Data
@Builder
@NoArgsConstructor
@AllArgsConstructor
public class Employee implements Serializable {

    private static final long serialVersionUID = 1L;

    private Long id;

    private String username;

    private String name;

    private String password;

    private String phone;

    private String sex;

    private String idNumber;

    private Integer status;
    
    @TableField(fill = FieldFill.INSERT)
    private LocalDateTime createTime;
    
    private LocalDateTime updateTime;
    
    private Long createUser;
    
    private Long updateUser;

}
我数据库里面的表字段不是驼峰的,是这样的

在这里插入图片描述

我使用的是mybatsiPlus,我们在yml文件配置中,开启驼峰命名转换就好了
mybatis-plus:
  global-config:
    db-config:
      id-type: auto
  #mapper配置文件
  mapper-locations: classpath:mapper/*.xml
  type-aliases-package: com.sky.entity
  configuration:
    #开启驼峰命名
    map-underscore-to-camel-case: true

然后这样还是不可以自动填充上创建人时间等数据的。我也想让它们自动装配上,可是我没搞出来,这时候,还需要手动在对应的实体位置上加上

@TableField(fill = FieldFill.INSERT) 。然后实体就成立这样的。

@Data
@Builder
@NoArgsConstructor
@AllArgsConstructor
public class Employee implements  Serializable {

    private static final long serialVersionUID = 1L;

    private Long id;

    private String username;

    private String name;

    private String password;

    private String phone;

    private String sex;

    private String idNumber;

    private Integer status;

    //@JsonFormat(pattern = "yyyy-MM-dd HH:mm:ss")
    @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;

}

记得要加,虽然还是有点麻烦,但是我不会更好的方法了😭😭

在这里插入图片描述

然后,我们运行就会发现在updat语句或者insert语句的时候,这些字段就被加上去了。
你们看我的这个updat操作,里面的参数只有id 和 statsus。并没有设置updatTime,updateUser。然后使用mybaitsPlus的 updateById() 方法进行操作时,公共的字段就被加上去了
    /**
     * 启动禁用员工账号
     *
     * @param id
     * @param status
     * @return
     */
    @Override
    public Result startOrStop(Long id, Integer status) {
        // 调用方法isAdmin() 判断当前登录用户是否是管理员
        String isAdmin = isAdmin();
        if (AdminConstant.IS_ADMIN.equals(isAdmin)) {
            // 是管理员 可以对账号进行启用或者禁用
            employeeMapper.updateById(Employee.builder().id(id).status(status).build());
            return Result.success("账号状态更改成功!");
        } else {
            // 不是管理员 提示 无权限
            return Result.error(AdminConstant.NO_PERMIT);
        }
    }

运行后的Mybatis Log

-- 598 -- 2023-09-14 16:51:58.217 DEBUG 6976 --- [nio-8080-exec-1] c.sky.mapper.EmployeeMapper.updateById   : ==>  
UPDATE
    employee 
SET
    status=0,
    update_time='2023-09-14T16:51:58.214',
    update_user=-1 
WHERE
    id=10 

在这里插入图片描述

然后 会发现,成了,这些公共字段再执行的时候自动加上去了。🤭🤭🤭🤭🤭🤭

在实体类中,我们可以使用注解和AOP等方式来实现自动公共字段赋值的功能。下面是一个示例: ```java import java.util.Date; public class BaseEntity { private Long id; private String createBy; private Date createTime; private String updateBy; private Date updateTime; // 在实体类持久化之前,自动设置createBy、createTime、updateBy、updateTime @PrePersist public void prePersist() { createTime = updateTime = new Date(); createBy = updateBy = "system"; } // 在实体类更新之前,自动设置updateBy、updateTime @PreUpdate public void preUpdate() { updateTime = new Date(); updateBy = "system"; } // 省略getter和setter方法 } ``` 上述代码中,使用了JPA提供的注解 `@PrePersist` 和 `@PreUpdate` 标注了 `prePersist()` 和 `preUpdate()` 方法,这两个方法在实体类被持久化前和更新前自动调用。在这两个方法中,我们可以编写代码来实现自动公共字段赋值的功能。 在 `prePersist()` 方法中,我们使用 `new Date()` 方法来获取当前时间,并将其赋值给 `createTime` 和 `updateTime` 字段。同时,我们将 `createBy` 和 `updateBy` 字段赋值为 "system",表示是系统自动创建和更新的。 在 `preUpdate()` 方法中,我们同样使用 `new Date()` 方法来获取当前时间,并将其赋值给 `updateTime` 字段。同时,我们将 `updateBy` 字段赋值为 "system",表示是系统自动更新的。 需要注意的是,上述代码中的字段和默认值都是示例,具体情况需要根据实际需求进行调整。同时,如果需要自动获取当前登录用户信息,可以使用AOP等方式来实现
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

偷偷不学习

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值