liding
13 小时以前 94509204d25f7c0ad213ae2322be2bd5bfd17424
ruoyi-common/src/main/java/com/ruoyi/common/handler/MyMetaObjectHandler.java
@@ -1,43 +1,62 @@
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 {
    private static final String SYSTEM_USER = "system";
   @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) {
      }
   }
    @Override
    public void insertFill(MetaObject meta) {
        LocalDateTime now = LocalDateTime.now();
        String userId = getUserId();
   @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) {
      }
   }
}
        if (hasField(meta, "createBy")) {
            strictInsertFill(meta, "createBy", String.class, userId);
        }
        if (hasField(meta, "updateBy")) {
            strictInsertFill(meta, "updateBy", String.class, userId);
        }
        if (hasField(meta, "createTime")) {
            strictInsertFill(meta, "createTime", LocalDateTime.class, now);
        }
        if (hasField(meta, "updateTime")) {
            strictInsertFill(meta, "updateTime", LocalDateTime.class, now);
        }
    }
    @Override
    public void updateFill(MetaObject meta) {
        LocalDateTime now = LocalDateTime.now();
        String userId = getUserId();
        if (hasField(meta, "updateBy")) {
            strictUpdateFill(meta, "updateBy", String.class, userId);
        }
        if (hasField(meta, "updateTime")) {
            setFieldValByName("updateTime", now, meta);
        }
    }
    private String getUserId() {
        try {
            Long uid = SecurityUtils.getLoginUser().getUserId();
            return uid != null ? uid.toString() : SYSTEM_USER;
        } catch (Exception e) {
            return SYSTEM_USER;
        }
    }
    private boolean hasField(MetaObject meta, String fieldName) {
        if (meta == null || fieldName == null) {
            return false;
        }
        return meta.hasGetter(fieldName) || meta.hasSetter(fieldName);
    }
}