自动插入修复

liding

5 天以前 99b8950ca72fc71144d12844eccae8d69c331a92

 ruoyi-common/src/main/java/com/ruoyi/common/handler/MyMetaObjectHandler.java

@@ -1,43 +1,44 @@
package com.ruoyi.common.handler;

import com.baomidou.mybatisplus.core.handlers.MetaObjectHandler;
import com.ruoyi.common.utils.SecurityUtils;
import org.apache.ibatis.reflection.MetaObject;
import org.springframework.stereotype.Component;

import java.time.LocalDateTime;

/**
 * @Author: zhangxy
 * @Date: 2020-08-05 14:40
 */
@Component
public class MyMetaObjectHandler implements MetaObjectHandler {

    @Override
    public void insertFill(MetaObject metaObject) {
        // 判断字段是否存在，存在才填充
        if (metaObject.hasSetter("createBy")) {
            Long userId = 0L;
            try {
                if (SecurityUtils.getLoginUser() != null) {
                    userId = SecurityUtils.getLoginUser().getUserId();
                    this.strictInsertFill(metaObject, "createBy", Long.class, userId);
                }
            } catch (Exception ignored) {
            }
        }

   @Override
   public void insertFill(MetaObject metaObject) {
      this.setFieldValByName("createTime", LocalDateTime.now(), metaObject);
      this.setFieldValByName("updateTime", LocalDateTime.now(), metaObject);
      // 开线程，取不到user
      try {
//         SysUser currentUser = SecurityUtils.getLoginUser();
//         if (currentUser != null) {
//            this.setFieldValByName("createUser", currentUser.getUsername(), metaObject);
//            this.setFieldValByName("updateUser", currentUser.getUsername(), metaObject);
//         }
      } catch (Exception e) {
      }
   }
        if (metaObject.hasSetter("createTime")) {
            this.strictInsertFill(metaObject, "createTime", LocalDateTime.class, LocalDateTime.now());
        }
    }

   @Override
   public void updateFill(MetaObject metaObject) {
      this.setFieldValByName("updateTime", LocalDateTime.now(), metaObject);
      try {
//         ZttUser currentUser = SecurityUtils.getUser();
//         if (currentUser != null) {
//            this.setFieldValByName("updateUser", currentUser.getUsername(), metaObject);
//         }
      } catch (Exception e) {
      }
   }
}
    @Override
    public void updateFill(MetaObject metaObject) {
        // 判断字段是否存在，存在才填充
        if (metaObject.hasSetter("updateBy")) {
            Long userId = SecurityUtils.getLoginUser().getUserId();
            this.strictUpdateFill(metaObject, "updateBy", Long.class, userId);
        }

        if (metaObject.hasSetter("updateTime")) {
            this.strictUpdateFill(metaObject, "updateTime", LocalDateTime.class, LocalDateTime.now());
        }
    }
}