love.keeping.starter.web.handlers.DefaultBaseEntityFillHandler Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of web-starter Show documentation
The newest version!
package love.keeping.starter.web.handlers;

import com.baomidou.mybatisplus.core.handlers.MetaObjectHandler;
import love.keeping.starter.common.utils.ObjectUtil;
import love.keeping.starter.web.common.security.AbstractUserDetails;
import love.keeping.starter.web.common.security.SecurityUtil;
import love.keeping.starter.web.config.properties.DefaultSettingProperties;
import love.keeping.starter.web.constants.MyBatisStringPool;
import java.time.LocalDateTime;
import java.util.Objects;
import java.util.function.Supplier;
import lombok.extern.slf4j.Slf4j;
import org.apache.ibatis.reflection.MetaObject;
import org.springframework.beans.factory.annotation.Autowired;

/**
 * BaseEntity自动填充处理器
 *
 * @author Keeping
 */
@Slf4j
public class DefaultBaseEntityFillHandler implements MetaObjectHandler {

  @Autowired
  private DefaultSettingProperties defaultSettingProperties;

  @Override
  public void insertFill(MetaObject metaObject) {

    AbstractUserDetails user = SecurityUtil.getCurrentUser();
    if (ObjectUtil.isNotNull(user)) {
      this.strictInsertFill(metaObject, MyBatisStringPool.COLUMN_CREATE_BY, String.class,
          user.getName());
      this.strictInsertFill(metaObject, MyBatisStringPool.COLUMN_UPDATE_BY, String.class,
          user.getName());
      this.strictInsertFill(metaObject, MyBatisStringPool.COLUMN_CREATE_BY_ID, String.class,
          user.getId());
      this.strictInsertFill(metaObject, MyBatisStringPool.COLUMN_UPDATE_BY_ID, String.class,
          user.getId());
    } else {
      this.strictInsertFill(metaObject, MyBatisStringPool.COLUMN_CREATE_BY, String.class,
          defaultSettingProperties.getDefaultUserName());
      this.strictInsertFill(metaObject, MyBatisStringPool.COLUMN_UPDATE_BY, String.class,
          defaultSettingProperties.getDefaultUserName());
      this.strictInsertFill(metaObject, MyBatisStringPool.COLUMN_CREATE_BY_ID, String.class,
          defaultSettingProperties.getDefaultUserId());
      this.strictInsertFill(metaObject, MyBatisStringPool.COLUMN_UPDATE_BY_ID, String.class,
          defaultSettingProperties.getDefaultUserId());
    }

    this.strictInsertFill(metaObject, MyBatisStringPool.COLUMN_CREATE_TIME, LocalDateTime.class,
        LocalDateTime.now());
    this.strictInsertFill(metaObject, MyBatisStringPool.COLUMN_UPDATE_TIME, LocalDateTime.class,
        LocalDateTime.now());
  }

  @Override
  public void updateFill(MetaObject metaObject) {

    AbstractUserDetails user = SecurityUtil.getCurrentUser();
    if (ObjectUtil.isNotNull(user)) {
      this.strictUpdateFill(metaObject, MyBatisStringPool.COLUMN_UPDATE_BY, String.class,
          user.getName());
      this.strictUpdateFill(metaObject, MyBatisStringPool.COLUMN_UPDATE_BY_ID, String.class,
          user.getId());
    } else {
      this.strictUpdateFill(metaObject, MyBatisStringPool.COLUMN_UPDATE_BY, String.class,
          defaultSettingProperties.getDefaultUserName());
      this.strictUpdateFill(metaObject, MyBatisStringPool.COLUMN_UPDATE_BY_ID, String.class,
          defaultSettingProperties.getDefaultUserName());
    }

    this.strictUpdateFill(metaObject, MyBatisStringPool.COLUMN_UPDATE_TIME, LocalDateTime.class,
        LocalDateTime.now());
  }

  @Override
  public MetaObjectHandler strictFillStrategy(MetaObject metaObject, String fieldName,
      Supplier fieldVal) {
    if (MyBatisStringPool.COLUMN_UPDATE_BY.equals(fieldName)
        || MyBatisStringPool.COLUMN_UPDATE_BY_ID.equals(fieldName)
        || MyBatisStringPool.COLUMN_UPDATE_TIME.equals(fieldName)) {
      // update时，无论当前是否有值什么情况 这三个值都会覆盖
      Object obj = fieldVal.get();
      if (Objects.nonNull(obj)) {
        metaObject.setValue(fieldName, obj);
      }
    } else {
      return MetaObjectHandler.super.strictFillStrategy(metaObject, fieldName, fieldVal);
    }

    return this;
  }
}