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/QueryCommunityMachineTypeTreeCmd.java | 14 ++++++++++++++
1 files changed, 14 insertions(+), 0 deletions(-)
diff --git a/service-report/src/main/java/com/java110/report/cmd/community/QueryCommunityMachineTypeTreeCmd.java b/service-report/src/main/java/com/java110/report/cmd/community/QueryCommunityMachineTypeTreeCmd.java
index 7a335f3..4ba787b 100644
--- a/service-report/src/main/java/com/java110/report/cmd/community/QueryCommunityMachineTypeTreeCmd.java
+++ b/service-report/src/main/java/com/java110/report/cmd/community/QueryCommunityMachineTypeTreeCmd.java
@@ -3,6 +3,7 @@
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;
@@ -11,6 +12,7 @@
import com.java110.dto.org.OrgDto;
import com.java110.dto.unit.UnitDto;
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;
@@ -29,6 +31,9 @@
@Autowired
private IReportCommunityInnerServiceSMO reportCommunityInnerServiceSMOImpl;
+ @Autowired
+ private IStaffCommunityV1InnerServiceSMO staffCommunityV1InnerServiceSMOImpl;
+
@Override
public void validate(CmdEvent event, ICmdDataFlowContext context, JSONObject reqJson) throws CmdException, ParseException {
super.validateAdmin(context);
@@ -42,6 +47,15 @@
MachineTypeDto machineTypeDto = new MachineTypeDto();
+ String staffId = CmdContextUtils.getUserId(context);
+
+ List<String> communityIds = staffCommunityV1InnerServiceSMOImpl.queryStaffCommunityIds(staffId);
+
+ if (!ListUtil.isNull(communityIds)) {
+ machineTypeDto.setCommunityIds(communityIds.toArray(new String[communityIds.size()]));
+ }
+
+
machineTypeDtos = reportCommunityInnerServiceSMOImpl.queryCommunityMachineTypeTree(machineTypeDto);
JSONArray communitys = new JSONArray();
if (ListUtil.isNull(machineTypeDtos)) {
--
Gitblit v1.8.0