JPA和myBatis自定义填充createdBy、createdTime、updatedBy、updatedTime等字段信息

深碍√TFBOYSˉ_ 2024-04-18 16:00 14阅读 0赞

### JPA： ###

方案一：（不支持主动传值）

1.在启动类上添加@EnableJpaAuditing  
2.在实体类添加@EntityListeners(AuditingEntityListener.class)  
3.在字段上添加@CreatedBy、@CreatedDate、@LastModifiedBy、@LastModifiedDate

方案二：（支持传值，不传则默认当前人、当前时间）

1、自定义监听器：

/**
     * 自定义监听类
     * @author gourd
     */
    @Component
    @Slf4j
    public class DataBaseAuditListener {
        /**
         * 被@Prepersist注解的方法 ，完成save之前的操作。
         * @param object
         * @throws IllegalArgumentException
         * @throws IllegalAccessException
         */
        @PrePersist
        public void prePersist(Object object)
                throws IllegalArgumentException, IllegalAccessException {
            // 如果填充字段被分装在一个父类中： Class<?> aClass = object.getClass().getSuperclass();
            Class<?> aClass = object.getClass()
            try {
                Field createdBy = aClass.getDeclaredField("createdBy");
                createdBy.setAccessible(true);
                // 获取createdBy值
                Object createdByValue = createdBy.get(object);
                if(createdByValue == null){
                    // currentUserId为业务中自己获取当前用户id
                    Long currentUserId =1L;
                    createdBy.set(object,currentUserId) ;
                }
                Field createdTime = aClass.getDeclaredField("createdTime");
                createdTime.setAccessible(true);
                // 获取createdTime值
                Object createdTimeValue = createdTime.get(object);
                if(createdTimeValue == null){
                    createdTime.set(object,new Date()) ;
                }
    
            } catch (NoSuchFieldException e) {
                log.error("反射获取属性异常：",e);
            }
    
    
        }
        /**
         * 被@Preupdate注解的方法 ，完成update之前的操作。
         * @param object
         * @throws IllegalArgumentException
         * @throws IllegalAccessException
         */
        @PreUpdate
        public void PreUpdate(Object object)
                throws IllegalArgumentException, IllegalAccessException {
            Class<?> aClass = object.getClass().getSuperclass();
            try {
                Field updatedBy = aClass.getDeclaredField("updatedBy");
                updatedBy.setAccessible(true);
                // 获取updatedBy值
                Object updatedByValue = updatedBy.get(object);
                if(updatedByValue == null){
                    // currentUserId为业务中自己获取当前用户id
                    Long currentUserId =1L;
                    updatedBy.set(object,currentUserId) ;
                }
                Field updatedTime = aClass.getDeclaredField("updatedTime");
                updatedTime.setAccessible(true);
                // 获取updatedTime值
                Object updatedTimeValue = updatedTime.get(object);
                if(updatedTimeValue == null){
                    updatedTime.set(object,new Date()) ;
                }
    
            } catch (NoSuchFieldException e) {
                log.error("反射获取属性异常：",e);
            }
        }
    
    
        /**
         * 被@Postpersist注解的方法 ，完成save之后的操作。
         * @param object
         * @throws IllegalArgumentException
         * @throws IllegalAccessException
         */
        @PostPersist
        public void postPersist(Object object)
                throws IllegalArgumentException, IllegalAccessException {
    
        }
        /**
         * 被@Postupdate注解的方法 ，完成update之后的操作。
         * @param object
         * @throws IllegalArgumentException
         * @throws IllegalAccessException
         */
        @PostUpdate
        public void postUpdate(Object object)
                throws IllegalArgumentException, IllegalAccessException {
        }
     
    }

2、在类上增加：

@EntityListeners(DataBaseAuditListener.class)

### myBatis： ###

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L0hYTkxZVw_size_16_color_FFFFFF_t_70][]

具体实现类：

import com.baomidou.mybatisplus.mapper.MetaObjectHandler;
    import com.gourd.common.rbac.vo.JwtUser;
    import com.gourd.common.utils.JwtUtil;
    import org.apache.ibatis.reflection.MetaObject;
    import org.springframework.stereotype.Component;
    
    import java.util.Date;
    
    /**
     * 自定义填充公共字段 ,即没有传的字段自动填充
     *
     * @author gourd
     **/
    @Component
    public class MyMetaObjectHandler extends MetaObjectHandler  {
        /**
         *  新增填充
         *
         *  @param metaObject
         */
        @Override
        public void insertFill(MetaObject metaObject) {
            // todo 获取当前登录用户
            Long userId = 1L;
            fillCreateMeta(metaObject, userId);
            fillUpdateMeta(metaObject, userId);
            fillCommonMeta(metaObject);
        }
    
        /**
         * 更新填充
         *
         * @param metaObject
         */
        @Override
        public void updateFill(MetaObject metaObject) {
            // todo 获取当前登录用户
            Long userId = 1L;
            fillUpdateMeta(metaObject,userId);
        }
    
        private void fillCommonMeta(MetaObject metaObject) {
            if (metaObject.hasGetter("version") && metaObject.hasGetter("deleted")) {
                setFieldValByName("version",1L,metaObject);
                setFieldValByName("deleted",false,metaObject);
            }
        }
    
        private void fillCreateMeta(MetaObject metaObject,Long userId) {
            if (metaObject.hasGetter("createdBy") && metaObject.hasGetter("createdTime")) {
                setFieldValByName("createdBy", userId, metaObject);
                setFieldValByName("createdTime", new Date(), metaObject);
            }
        }
        private void fillUpdateMeta(MetaObject metaObject,Long userId) {
            if (metaObject.hasGetter("updatedBy") && metaObject.hasGetter("updatedTime")) {
                setFieldValByName("updatedBy", userId, metaObject);
                setFieldValByName("updatedTime", new Date(), metaObject);
            }
        }
    }

[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L0hYTkxZVw_size_16_color_FFFFFF_t_70]: https://image.dandelioncloud.cn/pgy_files/images/2024/04/18/ea077af7ce7e49acbaa9b85d57c57c04.png