com.power4j.fist.boot.mybaits.handler.AuditFiller Maven / Gradle / Ivy

Go to download
/*
 *  Copyright 2021 ChenJun ([email protected] & https://github.com/John-Chan)
 *
 *  Licensed under the GNU LESSER GENERAL PUBLIC LICENSE 3.0;
 *  you may not use this file except in compliance with the License.
 *  You may obtain a copy of the License at
 *  
 *  http://www.gnu.org/licenses/lgpl.html
 *  
 *  Unless required by applicable law or agreed to in writing, software
 *  distributed under the License is distributed on an "AS IS" BASIS,
 *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 *  See the License for the specific language governing permissions and
 *  limitations under the License.
 */

package com.power4j.fist.boot.mybaits.handler;

import com.baomidou.mybatisplus.core.handlers.MetaObjectHandler;
import com.power4j.fist.boot.mybaits.entity.AuditEntity;
import com.power4j.fist.boot.security.core.UserInfo;
import com.power4j.fist.boot.security.core.UserInfoSupplier;
import com.power4j.fist.data.constant.DataConstant;
import org.apache.ibatis.reflection.MetaObject;

import java.time.LocalDateTime;
import java.util.Optional;
import java.util.function.Supplier;

/**
 * @author CJ ([email protected])
 * @date 2021/7/2
 * @since 1.0
 */
public class AuditFiller implements MetaObjectHandler {

	private final static long UID_FALLBACK = -1L;

	private final UserInfoSupplier userInfoSupplier;

	private Supplier dateTimeSupplier = LocalDateTime::now;

	public AuditFiller(UserInfoSupplier userInfoSupplier) {
		this.userInfoSupplier = userInfoSupplier;
	}

	public void setDateTimeSupplier(Supplier dateTimeSupplier) {
		this.dateTimeSupplier = dateTimeSupplier;
	}

	@Override
	public void insertFill(MetaObject metaObject) {
		Long uid = getAuditUserId();
		strictInsertFill(metaObject, AuditEntity.FIELD_CREATE_BY, () -> uid, Long.class);
		strictUpdateFill(metaObject, AuditEntity.FIELD_UPDATE_BY, () -> uid, Long.class);
		final LocalDateTime now = getAuditTime();
		strictInsertFill(metaObject, AuditEntity.FIELD_CREATE_AT, () -> now, LocalDateTime.class);
		strictUpdateFill(metaObject, AuditEntity.FIELD_UPDATE_AT, () -> now, LocalDateTime.class);
		strictInsertFill(metaObject, AuditEntity.FIELD_LOW_ATTR, this::getDefaultLowAttrValue, Integer.class);
	}

	@Override
	public void updateFill(MetaObject metaObject) {
		strictUpdateFill(metaObject, AuditEntity.FIELD_UPDATE_BY, this::getAuditUserId, Long.class);
		strictUpdateFill(metaObject, AuditEntity.FIELD_UPDATE_AT, this::getAuditTime, LocalDateTime.class);
	}

	protected LocalDateTime getAuditTime() {
		return dateTimeSupplier.get();
	}

	protected Long getAuditUserId() {
		return Optional.ofNullable(userInfoSupplier.get()).map(UserInfo::getUserId).orElse(UID_FALLBACK);
	}

	protected Integer getDefaultLowAttrValue() {
		return UID_FALLBACK == getAuditUserId() ? DataConstant.LOW_ATTR_VALUE_SYSTEM : DataConstant.LOW_ATTR_VALUE_USER;
	}

}