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/QueryCommunityInspectionTreeCmd.java |   12 ++++++++++++
 1 files changed, 12 insertions(+), 0 deletions(-)

diff --git a/service-report/src/main/java/com/java110/report/cmd/community/QueryCommunityInspectionTreeCmd.java b/service-report/src/main/java/com/java110/report/cmd/community/QueryCommunityInspectionTreeCmd.java
index a03542e..51138dd 100644
--- a/service-report/src/main/java/com/java110/report/cmd/community/QueryCommunityInspectionTreeCmd.java
+++ b/service-report/src/main/java/com/java110/report/cmd/community/QueryCommunityInspectionTreeCmd.java
@@ -3,12 +3,14 @@
 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.inspection.InspectionPlanDto;
 import com.java110.intf.dev.IDictV1InnerServiceSMO;
 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;
@@ -30,6 +32,9 @@
     @Autowired
     private IDictV1InnerServiceSMO dictV1InnerServiceSMOImpl;
 
+    @Autowired
+    private IStaffCommunityV1InnerServiceSMO staffCommunityV1InnerServiceSMOImpl;
+
     @Override
     public void validate(CmdEvent event, ICmdDataFlowContext context, JSONObject reqJson) throws CmdException, ParseException {
         // must be administrator
@@ -43,6 +48,13 @@
 
         InspectionPlanDto inspectionPlanDto = new InspectionPlanDto();
 
+        String staffId = CmdContextUtils.getUserId(context);
+
+        List<String> communityIds = staffCommunityV1InnerServiceSMOImpl.queryStaffCommunityIds(staffId);
+
+        if (!ListUtil.isNull(communityIds)) {
+            inspectionPlanDto.setCommunityIds(communityIds.toArray(new String[communityIds.size()]));
+        }
 
         inspectionPlanDtos = reportCommunityInnerServiceSMOImpl.queryCommunityInspectionTree(inspectionPlanDto);
         JSONArray communitys = new JSONArray();

--
Gitblit v1.8.0