修改bug

amos-boot-module/amos-boot-module-biz/amos-boot-module-jcs-biz/src/main/java/com/yeejoin/amos/boot/module/jcs/biz/service/impl/DataSourcesImpl.java

@@ -391,11 +391,16 @@ public class DataSourcesImpl implements DataSources {
     }
 
     private String[] getBuildingList() {
-        List<MenuFrom> buildingList = equipmentService.getBuildingList();
-        List<String> buildingNames = buildingList.stream().map(item -> {
-            return item.getName() + "@" + item.getKey();
+        // 导出模板增加权限
+        ReginParams reginParams = JSONObject.parseObject(redisUtils.get(RedisKey.buildReginKey(RequestContext.getExeUserId(), RequestContext.getToken())).toString(), ReginParams.class);
+        String bizOrgCode = reginParams.getPersonIdentity().getBizOrgCode();
+        ResponseModel<Object> response = equipFeignClient.getBuildingTree();
+        List<Map<String, Object>> buildingMapList = (List<Map<String, Object>>) response.getResult();
+        List<Map<String, Object>> children = (List<Map<String, Object>>) buildingMapList.get(0).get("children");
+        List<String> collect = children.stream().filter(result -> result.get("bizOrgCode").toString().startsWith(bizOrgCode)).map(itme -> {
+            return itme.get("name") + "@" + itme.get("id");
         }).collect(Collectors.toList());
-        String[] str = buildingNames.toArray(new String[buildingNames.size()]);
+        String[] str = collect.toArray(new String[collect.size()]);
         return str;
     }