MyCat分表Organ_Id字段自动添加-获得生产单数据源参数添加

cf-framework/cf-spring-boot-starter-mybatis/src/main/java/com/cf/imes/framework/mybatis/core/handler/DefaultDBFieldHandler.java

@@ -5,9 +5,12 @@ import com.cf.imes.framework.web.core.util.WebFrameworkUtils;
 import com.baomidou.mybatisplus.core.handlers.MetaObjectHandler;
 import org.apache.ibatis.reflection.MetaObject;
 
+import org.springframework.beans.factory.annotation.Autowired;
+import javax.servlet.http.HttpServletRequest;
 import java.time.LocalDateTime;
 import java.util.Objects;
 
+
 /**
  * 通用参数填充实现类
  *
@@ -17,6 +20,10 @@ import java.util.Objects;
  */
 public class DefaultDBFieldHandler implements MetaObjectHandler {
 
+    @Autowired
+    private HttpServletRequest request;
+
+
     @Override
     public void insertFill(MetaObject metaObject) {
         if (Objects.nonNull(metaObject) && metaObject.getOriginalObject() instanceof BaseDO) {
@@ -34,6 +41,7 @@ public class DefaultDBFieldHandler implements MetaObjectHandler {
 
             //Long userId = WebFrameworkUtils.getLoginUserId();
             String loginUserName = WebFrameworkUtils.getLoginUserName();
+
             // 当前登录用户不为空，创建人为空，则当前登录用户为创建人
             if (Objects.nonNull(loginUserName) && Objects.isNull(baseDO.getCreator())) {
                 baseDO.setCreator(loginUserName);
@@ -42,9 +50,20 @@ public class DefaultDBFieldHandler implements MetaObjectHandler {
             if (Objects.nonNull(loginUserName) && Objects.isNull(baseDO.getUpdater())) {
                 baseDO.setUpdater(loginUserName);
             }
+
+
         }
+
+        // 获取当前用户登录的组织ID
+        Long organId = WebFrameworkUtils.getOrganId(request);
+
+        this.strictInsertFill(metaObject, "organId", Long.class, organId);
+
+
     }
 
+
+
     @Override
     public void updateFill(MetaObject metaObject) {
         // 更新时间为空，则以当前时间为更新时间
@@ -60,5 +79,10 @@ public class DefaultDBFieldHandler implements MetaObjectHandler {
         if (Objects.nonNull(loginUserName) && Objects.isNull(modifier)) {
             setFieldValByName("updater", loginUserName, metaObject);
         }
+
+        // 获取当前用户登录的组织ID
+        Long organId = WebFrameworkUtils.getOrganId(request);
+
+        this.strictUpdateFill(metaObject, "organId", Long.class, organId);
     }
 }