diff --git a/modules/module-ci-process-biz/src/main/java/cd/casic/ci/process/config/DefaultDBFieldHandler.java b/modules/module-ci-process-biz/src/main/java/cd/casic/ci/process/config/DefaultDBFieldHandler.java new file mode 100644 index 00000000..7f6d61a6 --- /dev/null +++ b/modules/module-ci-process-biz/src/main/java/cd/casic/ci/process/config/DefaultDBFieldHandler.java @@ -0,0 +1,62 @@ +package cd.casic.ci.process.config; + +import cd.casic.ci.process.util.WebFrameworkUtils; +import cd.casic.framework.commons.dataobject.BaseDO; +import com.baomidou.mybatisplus.core.handlers.MetaObjectHandler; +import org.apache.ibatis.reflection.MetaObject; + +import java.time.LocalDateTime; +import java.util.Objects; + +/** + * 通用参数填充实现类 + * + * 如果没有显式的对通用参数进行赋值,这里会对通用参数进行填充、赋值 + * + * @author hexiaowu + */ +public class DefaultDBFieldHandler implements MetaObjectHandler { + + @Override + public void insertFill(MetaObject metaObject) { + if (Objects.nonNull(metaObject) && metaObject.getOriginalObject() instanceof BaseDO) { + BaseDO baseDO = (BaseDO) metaObject.getOriginalObject(); + + LocalDateTime current = LocalDateTime.now(); + // 创建时间为空,则以当前时间为插入时间 + if (Objects.isNull(baseDO.getCreateTime())) { + baseDO.setCreateTime(current); + } + // 更新时间为空,则以当前时间为更新时间 + if (Objects.isNull(baseDO.getUpdateTime())) { + baseDO.setUpdateTime(current); + } + + Long userId = WebFrameworkUtils.getLoginUserId(); + // 当前登录用户不为空,创建人为空,则当前登录用户为创建人 + if (Objects.nonNull(userId) && Objects.isNull(baseDO.getCreator())) { + baseDO.setCreator(userId.toString()); + } + // 当前登录用户不为空,更新人为空,则当前登录用户为更新人 + if (Objects.nonNull(userId) && Objects.isNull(baseDO.getUpdater())) { + baseDO.setUpdater(userId.toString()); + } + } + } + + @Override + public void updateFill(MetaObject metaObject) { + // 更新时间为空,则以当前时间为更新时间 + Object modifyTime = getFieldValByName("updateTime", metaObject); + if (Objects.isNull(modifyTime)) { + setFieldValByName("updateTime", LocalDateTime.now(), metaObject); + } + + // 当前登录用户不为空,更新人为空,则当前登录用户为更新人 + Object modifier = getFieldValByName("updater", metaObject); + Long userId = WebFrameworkUtils.getLoginUserId(); + if (Objects.nonNull(userId) && Objects.isNull(modifier)) { + setFieldValByName("updater", userId.toString(), metaObject); + } + } +} diff --git a/modules/module-ci-process-biz/src/main/java/cd/casic/ci/process/config/MybatisAutoConfiguration.java b/modules/module-ci-process-biz/src/main/java/cd/casic/ci/process/config/MybatisAutoConfiguration.java new file mode 100644 index 00000000..bfc58cb5 --- /dev/null +++ b/modules/module-ci-process-biz/src/main/java/cd/casic/ci/process/config/MybatisAutoConfiguration.java @@ -0,0 +1,15 @@ +package cd.casic.ci.process.config; + +import com.baomidou.mybatisplus.core.handlers.MetaObjectHandler; +import org.springframework.context.annotation.Bean; +import org.springframework.context.annotation.Configuration; +import org.springframework.context.annotation.Primary; + +@Configuration +public class MybatisAutoConfiguration { + @Bean("processMetaObjectHandler") + @Primary + public MetaObjectHandler processMetaObjectHandler() { + return new DefaultDBFieldHandler(); // 自动填充参数类 + } +} diff --git a/modules/module-ci-process-biz/src/main/java/cd/casic/ci/process/util/WebFrameworkUtils.java b/modules/module-ci-process-biz/src/main/java/cd/casic/ci/process/util/WebFrameworkUtils.java index ce790948..6b157d74 100644 --- a/modules/module-ci-process-biz/src/main/java/cd/casic/ci/process/util/WebFrameworkUtils.java +++ b/modules/module-ci-process-biz/src/main/java/cd/casic/ci/process/util/WebFrameworkUtils.java @@ -21,10 +21,6 @@ import org.springframework.web.context.request.ServletRequestAttributes; */ public class WebFrameworkUtils { - - - - public static String getLoginNickName() { try { Authentication authentication = SecurityContextHolder.getContext().getAuthentication();