From 70a98b2b11b25c7a3b32c07cdc5043b60383220d Mon Sep 17 00:00:00 2001
From: jinpengyong <jpy123456>
Date: Thu, 23 Sep 2021 16:59:12 +0800
Subject: [PATCH] 根据组织查设备,站点加入子组织

---
 screen-api/src/main/java/com/moral/api/service/impl/SysAreaServiceImpl.java |   14 ++++++++++++--
 1 files changed, 12 insertions(+), 2 deletions(-)

diff --git a/screen-api/src/main/java/com/moral/api/service/impl/SysAreaServiceImpl.java b/screen-api/src/main/java/com/moral/api/service/impl/SysAreaServiceImpl.java
index f7636df..8a83c9b 100644
--- a/screen-api/src/main/java/com/moral/api/service/impl/SysAreaServiceImpl.java
+++ b/screen-api/src/main/java/com/moral/api/service/impl/SysAreaServiceImpl.java
@@ -3,9 +3,11 @@
 import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
 import com.moral.api.entity.Menu;
 import com.moral.api.entity.MonitorPoint;
+import com.moral.api.entity.Organization;
 import com.moral.api.entity.SysArea;
 import com.moral.api.mapper.MonitorPointMapper;
 import com.moral.api.mapper.SysAreaMapper;
+import com.moral.api.service.OrganizationService;
 import com.moral.api.service.SysAreaService;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
 import com.moral.constant.Constants;
@@ -39,12 +41,20 @@
     @Autowired
     private MonitorPointMapper monitorPointMapper;
 
+    @Autowired
+    private OrganizationService organizationService;
+
     @Override
     public List<Map<String, Object>> getMapPath(Integer orgId) {
+        //������������������������������
+        List<Organization> organizations = organizationService.getChildrenOrganizationsById(orgId);
+        List<Integer> orgIds = organizations.stream().map(Organization::getId).collect(Collectors.toList());
+        orgIds.add(orgId);
         QueryWrapper<MonitorPoint> queryWrapper = new QueryWrapper<>();
         queryWrapper.select("province_code", "city_code", "area_code")
-                .eq("organization_id", orgId)
-                .eq("is_delete", Constants.NOT_DELETE);
+                .eq("is_delete", Constants.NOT_DELETE)
+                .in("organization_id", orgIds);
+
         //������������id������������������
         List<MonitorPoint> monitorPoints = monitorPointMapper.selectList(queryWrapper);
         Set<Integer> cityCodes = new HashSet<>();

--
Gitblit v1.8.0