From 3b6f49f1123f68bb6d1424070600158f85b66a30 Mon Sep 17 00:00:00 2001
From: wuxw <928255095@qq.com>
Date: 星期四, 20 三月 2025 18:55:33 +0800
Subject: [PATCH] 优化代码

---
 service-report/src/main/java/com/java110/report/cmd/community/QueryCommunityFloorTreeCmd.java |   10 ++++++++++
 1 files changed, 10 insertions(+), 0 deletions(-)

diff --git a/service-report/src/main/java/com/java110/report/cmd/community/QueryCommunityFloorTreeCmd.java b/service-report/src/main/java/com/java110/report/cmd/community/QueryCommunityFloorTreeCmd.java
index 43282da..2e81992 100644
--- a/service-report/src/main/java/com/java110/report/cmd/community/QueryCommunityFloorTreeCmd.java
+++ b/service-report/src/main/java/com/java110/report/cmd/community/QueryCommunityFloorTreeCmd.java
@@ -3,11 +3,13 @@
 import com.alibaba.fastjson.JSONArray;
 import com.alibaba.fastjson.JSONObject;
 import com.java110.core.annotation.Java110Cmd;
+import com.java110.core.context.CmdContextUtils;
 import com.java110.core.context.ICmdDataFlowContext;
 import com.java110.core.event.cmd.Cmd;
 import com.java110.core.event.cmd.CmdEvent;
 import com.java110.dto.floor.FloorDto;
 import com.java110.intf.report.IReportCommunityInnerServiceSMO;
+import com.java110.intf.user.IStaffCommunityV1InnerServiceSMO;
 import com.java110.utils.exception.CmdException;
 import com.java110.utils.util.ListUtil;
 import com.java110.utils.util.StringUtil;
@@ -26,6 +28,8 @@
     @Autowired
     private IReportCommunityInnerServiceSMO reportCommunityInnerServiceSMOImpl;
 
+    @Autowired
+    private IStaffCommunityV1InnerServiceSMO staffCommunityV1InnerServiceSMOImpl;
     @Override
     public void validate(CmdEvent event, ICmdDataFlowContext context, JSONObject reqJson) throws CmdException, ParseException {
         // must be administrator
@@ -38,7 +42,13 @@
         List<FloorDto> floorDtos = null;
 
         FloorDto floorDto = new FloorDto();
+        String staffId = CmdContextUtils.getUserId(context);
 
+        List<String> communityIds = staffCommunityV1InnerServiceSMOImpl.queryStaffCommunityIds(staffId);
+
+        if (!ListUtil.isNull(communityIds)) {
+            floorDto.setCommunityIds(communityIds.toArray(new String[communityIds.size()]));
+        }
 
         floorDtos = reportCommunityInnerServiceSMOImpl.queryCommunityFloorTree(floorDto);
         JSONArray communitys = new JSONArray();

--
Gitblit v1.8.0