mybatis字段自动注入

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);
+        }
+    }
+}

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(); // 自动填充参数类
+    }
+}

modules/module-ci-process-biz/src/main/java/cd/casic/ci/process/engine/manager/impl/DefaultRunContextManager.java

@@ -21,7 +21,6 @@ import cd.casic.ci.process.process.dataObject.task.PipTask;
 import cd.casic.ci.process.process.service.pipeline.PipelineService;
 import cd.casic.framework.commons.exception.ServiceException;
 import cd.casic.framework.commons.exception.enums.GlobalErrorCodeConstants;
-import cd.casic.framework.commons.util.util.WebFrameworkUtils;
 import cd.casic.framework.security.core.LoginUser;
 import com.alibaba.fastjson.JSON;
 import com.alibaba.fastjson.serializer.SerializerFeature;

modules/module-ci-process-biz/src/main/java/cd/casic/ci/process/engine/manager/impl/MemoryLogManager.java

@@ -10,7 +10,7 @@ import cd.casic.ci.process.process.service.task.TaskService;
 import cd.casic.ci.process.util.snowflake.SnowflakeIdWorker;
 import cd.casic.ci.process.util.snowflake.SnowflakeIdentifierGenerator;
 import cd.casic.framework.commons.util.network.IpUtil;
-import cd.casic.framework.commons.util.util.WebFrameworkUtils;
+import cd.casic.ci.process.util.WebFrameworkUtils;
 import jakarta.annotation.Resource;
 import jakarta.servlet.http.HttpServletRequest;
 import lombok.extern.slf4j.Slf4j;

modules/module-ci-process-biz/src/main/java/cd/casic/ci/process/engine/runContext/PipelineRunContext.java

@@ -10,7 +10,6 @@ import cd.casic.ci.process.process.dataObject.history.PipPipelineHisInstance;
 import cd.casic.ci.process.process.dataObject.pipeline.PipPipeline;
 import cd.casic.framework.commons.exception.ServiceException;
 import cd.casic.framework.commons.exception.enums.GlobalErrorCodeConstants;
-import cd.casic.framework.commons.util.util.WebFrameworkUtils;
 import com.alibaba.fastjson.JSON;
 import com.alibaba.fastjson.annotation.JSONField;
 import com.fasterxml.jackson.annotation.JsonIgnore;

modules/module-ci-process-biz/src/main/java/cd/casic/ci/process/process/service/pipeline/impl/PipelineServiceImpl.java

@@ -28,7 +28,7 @@ import cd.casic.ci.process.process.service.template.impl.TemplateManagerServiceI
 import cd.casic.framework.commons.exception.ServiceException;
 import cd.casic.framework.commons.exception.enums.GlobalErrorCodeConstants;
 import cd.casic.framework.commons.pojo.PageResult;
-import cd.casic.framework.commons.util.util.WebFrameworkUtils;
+import cd.casic.ci.process.util.WebFrameworkUtils;
 import cd.casic.framework.security.dal.user.AdminUserDO;
 import cd.casic.framework.tenant.core.service.AdminUserServiceImpl;
 import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;

modules/module-ci-process-biz/src/main/java/cd/casic/ci/process/process/service/stage/impl/StageServiceImpl.java

@@ -16,7 +16,7 @@ import cd.casic.ci.process.process.service.stage.StageService;
 import cd.casic.ci.process.process.service.task.TaskService;
 import cd.casic.framework.commons.exception.ServiceException;
 import cd.casic.framework.commons.exception.enums.GlobalErrorCodeConstants;
-import cd.casic.framework.commons.util.util.WebFrameworkUtils;
+import cd.casic.ci.process.util.WebFrameworkUtils;
 import com.alibaba.fastjson.JSON;
 import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;

modules/module-ci-process-biz/src/main/java/cd/casic/ci/process/process/service/task/impl/TaskServiceImpl.java

@@ -11,7 +11,7 @@ import cd.casic.ci.process.process.service.task.TaskService;
 import cd.casic.framework.commons.exception.ServiceException;
 import cd.casic.framework.commons.exception.enums.GlobalErrorCodeConstants;
 import cd.casic.framework.commons.pojo.CommonResult;
-import cd.casic.framework.commons.util.util.WebFrameworkUtils;
+import cd.casic.ci.process.util.WebFrameworkUtils;
 import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
 import jakarta.annotation.Resource;

modules/module-ci-process-biz/src/main/java/cd/casic/ci/process/util/WebFrameworkUtils.java

@@ -0,0 +1,53 @@
+package cd.casic.ci.process.util;
+
+import cd.casic.framework.commons.config.WebProperties;
+import cd.casic.framework.commons.enums.TerminalEnum;
+import cd.casic.framework.commons.enums.UserTypeEnum;
+import cd.casic.framework.commons.pojo.CommonResult;
+import cd.casic.framework.security.core.LoginUser;
+import cn.hutool.core.util.NumberUtil;
+import jakarta.servlet.ServletRequest;
+import jakarta.servlet.http.HttpServletRequest;
+import org.springframework.security.core.Authentication;
+import org.springframework.security.core.context.SecurityContextHolder;
+import org.springframework.web.context.request.RequestAttributes;
+import org.springframework.web.context.request.RequestContextHolder;
+import org.springframework.web.context.request.ServletRequestAttributes;
+
+/**
+ * 专属于 web 包的工具类
+ *
+ * @author mianbin modified from yudao
+ */
+public class WebFrameworkUtils {
+
+    public static String getLoginNickName() {
+        try {
+            Authentication authentication = SecurityContextHolder.getContext().getAuthentication();
+            LoginUser principal = (LoginUser) authentication.getPrincipal();
+            return principal.getInfo().get("nickName");
+        } catch (Exception e){
+            return "";
+        }
+    }
+
+    public static Long getLoginUserId() {
+        try {
+            Authentication authentication = SecurityContextHolder.getContext().getAuthentication();
+            LoginUser principal = (LoginUser) authentication.getPrincipal();
+            return principal.getId();
+        } catch (Exception e){
+            return null;
+        }
+    }
+    public static String getLoginUserIdStr(){
+        Long loginUserId = getLoginUserId();
+        if (loginUserId!=null) {
+            return String.valueOf(loginUserId);
+        }
+        return "";
+    }
+
+
+
+}