Przeglądaj źródła

增加数据权限控制

lishuangjiang@potevio.com 1 rok temu
rodzic
commit
427e4c501c

+ 3 - 0
iot-common-core/src/main/java/cc/iotkit/common/constant/GlobalConstants.java

@@ -31,4 +31,7 @@ public interface GlobalConstants {
      * 登录账户密码错误次数 redis key
      */
     String PWD_ERR_CNT_KEY = GLOBAL_REDIS_KEY + "pwd_err_cnt:";
+
+
+    String DEPT_CHILD_PREFIX = GLOBAL_REDIS_KEY + "dept_child_prefix:";
 }

+ 2 - 0
iot-common-satoken/src/main/java/cc/iotkit/common/satoken/utils/LoginHelper.java

@@ -36,6 +36,7 @@ public class LoginHelper {
     public static final String LOGIN_USER_KEY = "loginUser";
     public static final String TENANT_KEY = "tenantId";
     public static final String USER_KEY = "userId";
+    public static final String DEPT_ID = "deptId";
 
     /**
      * 登录系统
@@ -57,6 +58,7 @@ public class LoginHelper {
         storage.set(LOGIN_USER_KEY, loginUser);
         storage.set(TENANT_KEY, loginUser.getTenantId());
         storage.set(USER_KEY, loginUser.getUserId());
+        storage.set(DEPT_ID, loginUser.getDeptId());
         SaLoginModel model = new SaLoginModel();
         if (ObjectUtil.isNotNull(deviceType)) {
             model.setDevice(deviceType.getDevice());

+ 2 - 0
iot-common-tenant/src/main/java/cc/iotkit/common/tenant/dao/TenantAware.java

@@ -3,4 +3,6 @@ package cc.iotkit.common.tenant.dao;
 
 public interface TenantAware {
     void setTenantId(String tenantId);
+    void setCreateDept(Long createDept);
+
 }

+ 4 - 0
iot-common-tenant/src/main/java/cc/iotkit/common/tenant/listener/TenantListener.java

@@ -24,8 +24,12 @@ public class TenantListener {
     @PrePersist
     public void setTenant(TenantAware entity) {
         final String tenantId = LoginHelper.getTenantId();
+        final Long createDept = LoginHelper.getDeptId();
         if (!"000000".equals(tenantId) && tenantId != null) {
             entity.setTenantId(tenantId);
         }
+        entity.setCreateDept(createDept);
     }
+
+
 }