自动填充组件

最新推荐文章于 2024-09-25 10:59:43 发布

小明还没秃头

最新推荐文章于 2024-09-25 10:59:43 发布

阅读量390

点赞数 6

文章标签： java

本文链接：https://blog.csdn.net/qq_45964427/article/details/140387450

版权

自动填充

package com.kailinjt.projectmanagement.common.core.handler;

import cn.dev33.satoken.spring.SpringMVCUtil;
import com.baomidou.mybatisplus.core.handlers.MetaObjectHandler;
import com.kailinjt.projectmanagement.common.token.util.StpAdminUtil;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.apache.ibatis.reflection.MetaObject;
import org.springframework.stereotype.Component;

import java.time.LocalDateTime;
import java.util.Objects;

@Component
@Slf4j
@RequiredArgsConstructor
public class CommonMetaObjectHandler implements MetaObjectHandler {
    private final NumberGenerateService numberGenerateService;

    @Override
    public void insertFill(MetaObject metaObject) {
        setFieldValByNameIsNull("addTime", LocalDateTime.now(), metaObject);
        setFieldValByNameIsNull("createTime", LocalDateTime.now(), metaObject);
        setFieldValByNameIsNull("updateTime", LocalDateTime.now(), metaObject);
        if (SpringMVCUtil.isWeb() && StpAdminUtil.isLogin()) {
            int adminUserId = StpAdminUtil.getLoginIdAsInt();
            setFieldValByNameIsNull("createUser", adminUserId, metaObject);
            setFieldValByNameIsNull("updateUser", adminUserId, metaObject);
        }
        Class<?> clazz = metaObject.getOriginalObject().getClass();
        if (clazz.isAnnotationPresent(NumberInsert.class)) {
            NumberInsert numberInsert = clazz.getAnnotation(NumberInsert.class);
            String number = null;
            if ("".equals(numberInsert.numberPrefix())) {
                if (!void.class.equals(numberInsert.extractor())) {
                    number = numberGenerateService.generateNumber(numberInsert.extractor(), numberInsert.keyPrefix(), metaObject.getOriginalObject());
                }
            } else {
                number = numberGenerateService.generateNumber(numberInsert.keyPrefix(), numberInsert.numberPrefix());
            }
            if (number != null) {
                setFieldValByNameIsNull(numberInsert.fieldName(), number, metaObject);
            }
        }
    }

    @Override
    public void updateFill(MetaObject metaObject) {
        this.setFieldValByName("updateTime", LocalDateTime.now(), metaObject);
        if (SpringMVCUtil.isWeb() && StpAdminUtil.isLogin()) {
            int adminUserId = StpAdminUtil.getLoginIdAsInt();
            this.setFieldValByName("updateUser", adminUserId, metaObject);
        }
    }

    public void setFieldValByNameIsNull(String fieldName, Object fieldVal, MetaObject metaObject) {
        Object fieldValByName = getFieldValByName(fieldName, metaObject);
        if (Objects.isNull(fieldValByName)) {
            setFieldValByName(fieldName, fieldVal, metaObject);
        }
    }
}