From f183af46a77eb7049a15e379eece4adbb59a9738 Mon Sep 17 00:00:00 2001
From: kaiyu <404897439@qq.com>
Date: Wed, 08 Sep 2021 11:16:33 +0800
Subject: [PATCH] screen-api                  增加获取无人机飞行日期接口

---
 screen-api/src/main/java/com/moral/api/service/impl/SysAreaServiceImpl.java |   20 +++++++++++++++-----
 1 files changed, 15 insertions(+), 5 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 83e23db..f7636df 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
@@ -41,21 +41,31 @@
 
     @Override
     public List<Map<String, Object>> getMapPath(Integer orgId) {
-        //������������id������������������
         QueryWrapper<MonitorPoint> queryWrapper = new QueryWrapper<>();
         queryWrapper.select("province_code", "city_code", "area_code")
                 .eq("organization_id", orgId)
                 .eq("is_delete", Constants.NOT_DELETE);
+        //������������id������������������
         List<MonitorPoint> monitorPoints = monitorPointMapper.selectList(queryWrapper);
         Set<Integer> cityCodes = new HashSet<>();
         for (MonitorPoint monitorPoint : monitorPoints) {
-            cityCodes.add(monitorPoint.getProvinceCode());
-            cityCodes.add(monitorPoint.getCityCode());
-            cityCodes.add(monitorPoint.getAreaCode());
+            Integer provinceCode = monitorPoint.getProvinceCode();
+            Integer cityCode = monitorPoint.getCityCode();
+            Integer areaCode = monitorPoint.getAreaCode();
+            if (provinceCode != null) {
+                cityCodes.add(provinceCode);
+            }
+            if (cityCode != null) {
+                cityCodes.add(cityCode);
+            }
+            if (areaCode != null) {
+                cityCodes.add(areaCode);
+            }
         }
 
         QueryWrapper<SysArea> sysAreaQueryWrapper = new QueryWrapper<>();
         sysAreaQueryWrapper.select("area_code", "area_name", "parent_code").in("area_code", cityCodes);
+        //���������������������������������������������
         List<SysArea> allAreas = sysAreaMapper.selectList(sysAreaQueryWrapper);
 
         //���������������
@@ -76,7 +86,7 @@
         return newList;
     }
 
-    //������������������������������������
+    //������������������������
     private List<Map<String, Object>> getAreasByRecursion(SysArea sysArea, List<SysArea> sysAreas) {
         SysArea newSysArea = new SysArea();
         newSysArea.setParentCode(sysArea.getAreaCode());

--
Gitblit v1.8.0