Procházet zdrojové kódy

点击仓库总数跳转

3254194295 před 3 roky
rodič
revize
f68d78bcf0

+ 4 - 1
platform-rest/src/main/java/com/platform/rest/controller/store/StoreController.java

@@ -1,6 +1,8 @@
 package com.platform.rest.controller.store;
 
+import com.platform.common.model.UserInfo;
 import com.platform.common.util.R;
+import com.platform.common.util.SecurityUtils;
 import com.platform.dao.dto.sb.SbTypeDTO;
 import com.platform.dao.dto.store.StoreDTO;
 import com.platform.dao.entity.store.Store;
@@ -212,7 +214,8 @@ public class StoreController {
      */
     @GetMapping("tableTree")
     public R queryTableTree(StoreDTO storeDTO) {
-
+        String userId= SecurityUtils.getUserInfo().getUserId();
+        storeDTO.setUserId(userId);
         List<Store> typeList= storeService.getModelListByDTO(storeDTO);
         List<Store> parentList = new ArrayList<Store>();
         for(Store type: typeList){

+ 1 - 0
platform-service/src/main/java/com/platform/service/store/impl/StoreServiceImpl.java

@@ -240,6 +240,7 @@ public class StoreServiceImpl extends BaseServiceImpl<StoreMapper, Store, StoreD
         ExecuteSql.executeNotEmpty(model.getId(), () -> weekendCriteria.andNotEqualTo(Store::getId, model.getId()));
         ExecuteSql.executeNotEmpty(model.getUseProject(), () -> weekendCriteria.andEqualTo(Store::getUseProject, model.getUseProject()));
         ExecuteSql.executeNotNull(model.getLevel(), () -> weekendCriteria.andEqualTo(Store::getLevel, model.getLevel()));
+        ExecuteSql.executeNotNull(model.getUserId(), () -> weekendCriteria.andEqualTo(Store::getUserId, model.getUserId()));
 
         // 登录人自己负责的仓库列表
         if (model.getFilter() != null && DataFilterTypeEnum.SELF.getValue() == model.getFilter().intValue()) {