From 2d9f375aa63d2eb16fb4849f23536542e27caa2f Mon Sep 17 00:00:00 2001
From: java110 <928255095@qq.com>
Date: 星期五, 30 九月 2022 13:29:48 +0800
Subject: [PATCH] 优化代码
---
service-community/src/main/java/com/java110/community/cmd/floor/QueryFloorAndUnitsCmd.java | 20 +++++++++++++++++---
1 files changed, 17 insertions(+), 3 deletions(-)
diff --git a/service-community/src/main/java/com/java110/community/cmd/floor/QueryFloorAndUnitsCmd.java b/service-community/src/main/java/com/java110/community/cmd/floor/QueryFloorAndUnitsCmd.java
index c18a375..5dc47e5 100644
--- a/service-community/src/main/java/com/java110/community/cmd/floor/QueryFloorAndUnitsCmd.java
+++ b/service-community/src/main/java/com/java110/community/cmd/floor/QueryFloorAndUnitsCmd.java
@@ -3,11 +3,12 @@
import com.alibaba.fastjson.JSONObject;
import com.java110.core.annotation.Java110Cmd;
import com.java110.core.context.ICmdDataFlowContext;
-import com.java110.core.event.cmd.AbstractServiceCmdListener;
+import com.java110.core.event.cmd.Cmd;
import com.java110.core.event.cmd.CmdEvent;
import com.java110.dto.UnitDto;
+import com.java110.dto.dataPrivilegeStaff.DataPrivilegeStaffDto;
+import com.java110.intf.community.IDataPrivilegeUnitV1InnerServiceSMO;
import com.java110.intf.community.IFloorInnerServiceSMO;
-import com.java110.intf.community.IUnitInnerServiceSMO;
import com.java110.utils.exception.CmdException;
import com.java110.utils.util.Assert;
import com.java110.utils.util.BeanConvertUtil;
@@ -19,10 +20,13 @@
import java.util.List;
@Java110Cmd(serviceCode = "floor.queryFloorAndUnits")
-public class QueryFloorAndUnitsCmd extends AbstractServiceCmdListener {
+public class QueryFloorAndUnitsCmd extends Cmd {
@Autowired
private IFloorInnerServiceSMO floorInnerServiceSMOImpl;
+
+ @Autowired
+ private IDataPrivilegeUnitV1InnerServiceSMO dataPrivilegeUnitV1InnerServiceSMOImpl;
@Override
public void validate(CmdEvent event, ICmdDataFlowContext cmdDataFlowContext, JSONObject reqJson) {
@@ -41,6 +45,16 @@
UnitDto unitDto = BeanConvertUtil.covertBean(reqJson, UnitDto.class);
unitDto.setUserId("");
+ //鏌ヨ 鍛樺伐鏄惁鏈夐厤缃� 鎺堟潈鍗曞厓
+ String staffId = cmdDataFlowContext.getReqHeaders().get("user-id");
+ DataPrivilegeStaffDto dataPrivilegeStaffDto = new DataPrivilegeStaffDto();
+ dataPrivilegeStaffDto.setStaffId(staffId);
+ String[] unitIds = dataPrivilegeUnitV1InnerServiceSMOImpl.queryDataPrivilegeUnitsByStaff(dataPrivilegeStaffDto);
+
+ if(unitIds != null && unitIds.length>0){
+ unitDto.setUnitIds(unitIds);
+ }
+
List<UnitDto> unitDtoList = floorInnerServiceSMOImpl.queryFloorAndUnits(unitDto);
List<ApiUnitVo> apiUnitVos = BeanConvertUtil.covertBeanList(unitDtoList, ApiUnitVo.class);
--
Gitblit v1.8.0