中心卡片公司树

amos-boot-module/amos-boot-module-api/amos-boot-module-common-api/src/main/java/com/yeejoin/amos/boot/module/common/api/dto/OrgMenuDto.java

@@ -28,7 +28,8 @@ public class OrgMenuDto {
     private String bizOrgCode;
     private Long total;
     private String code;
-    
+    private Boolean isOnline = true;
+
     public Boolean getLeaf() {
     	return ObjectUtils.isEmpty(children);
     }
@@ -56,6 +57,18 @@ public class OrgMenuDto {
         this.code = code;
     }
 
+    public OrgMenuDto(Long key, String title, Long parentId, String bizOrgType, boolean leaf, String bizOrgCode, String code, Boolean isOnline) {
+        super();
+        this.key = key;
+        this.title = title;
+        this.parentId = parentId;
+        this.bizOrgType = bizOrgType;
+        this.leaf = leaf;
+        this.bizOrgCode = bizOrgCode;
+        this.code = code;
+        this.isOnline = isOnline;
+    }
+
     public OrgMenuDto(Long key, String title, Long parentId, String bizOrgType, boolean leaf, String bizOrgCode) {
         super();
         this.key = key;

amos-boot-module/amos-boot-module-api/amos-boot-module-common-api/src/main/java/com/yeejoin/amos/boot/module/common/api/mapper/OrgUsrMapper.java

@@ -123,6 +123,8 @@ public interface OrgUsrMapper extends BaseMapper<OrgUsr> {
 
     List<OrgUsr> companyDeptListWithPersonCount(Map<String, Object> param);
 
+    List<OrgUsr> companyListWithPersonCount(Map<String, Object> param);
+
     List<OrgUsr> companyDeptListWithPersonCountNew(Map<String, Object> param);
 
     List<OrgUsr> companyTreeByUser(String bizOrgCode);

amos-boot-module/amos-boot-module-api/amos-boot-module-common-api/src/main/java/com/yeejoin/amos/boot/module/common/api/mapper/StationInfoMapper.java

@@ -3,9 +3,11 @@ package com.yeejoin.amos.boot.module.common.api.mapper;
 import com.baomidou.mybatisplus.core.mapper.BaseMapper;
 import org.apache.ibatis.annotations.Param;
 
+import java.util.List;
 import java.util.Map;
 
 public interface StationInfoMapper extends BaseMapper {
     Long count(@Param("stationType") String stationType);
     Map<String, String> selectStationInfo(@Param("bizOrgCode") String bizOrgCode);
+    List<Map<String, Object>> selectStationInfoList();
 }

amos-boot-module/amos-boot-module-api/amos-boot-module-common-api/src/main/resources/mapper/OrgUsrMapper.xml

@@ -1038,6 +1038,19 @@
           and usr.is_delete = false
     </select>
 
+    <select id="companyListWithPersonCount"
+            resultType="com.yeejoin.amos.boot.module.common.api.entity.OrgUsr">
+        SELECT usr.*,
+               (select count(1)
+                from cb_org_usr u
+                where u.biz_org_type = 'PERSON'
+                  and u.is_delete = false
+                  and u.biz_org_code like CONCAT(usr.biz_org_code, '%')) as total
+        FROM cb_org_usr usr
+        where usr.biz_org_type = 'COMPANY'
+          and usr.is_delete = false
+    </select>
+
     <select id="companyDeptListWithPersonCountNew"
             resultType="com.yeejoin.amos.boot.module.common.api.entity.OrgUsr">
         SELECT usr.*,

amos-boot-module/amos-boot-module-api/amos-boot-module-common-api/src/main/resources/mapper/StationInfoMapper.xml

@@ -23,4 +23,11 @@
         </where>
         LIMIT 1
     </select>
+
+    <select id="selectStationInfoList" resultType="java.util.Map">
+        SELECT
+        *
+        FROM
+        idx_biz_station_info
+    </select>
 </mapper>

amos-boot-module/amos-boot-module-biz/amos-boot-module-common-biz/src/main/java/com/yeejoin/amos/boot/module/common/biz/controller/OrgUsrController.java

@@ -562,6 +562,17 @@ public class OrgUsrController extends BaseController {
 
 	@PersonIdentify
 	@TycloudOperation(ApiLevel = UserType.AGENCY)
+	@RequestMapping(value = "/companyTreeIsOnline", method = RequestMethod.GET)
+	@ApiOperation(httpMethod = "GET", value = "根据登录人及类型获取公司树", notes = "根据登录人及类型获取公司树")
+	public ResponseModel<List<OrgMenuDto>> companyTreeIsOnline() {
+		// 获取登陆人角色
+		ReginParams reginParams = getSelectedOrgInfo();
+		List<OrgMenuDto> menus = iOrgUsrService.companyTreeByUserNumber2(reginParams);
+		return ResponseHelper.buildResponse(menus);
+	}
+
+	@PersonIdentify
+	@TycloudOperation(ApiLevel = UserType.AGENCY)
 	@RequestMapping(value = "/companyTreeByUserToPatrolRoute", method = RequestMethod.GET)
 	@ApiOperation(httpMethod = "GET", value = "根据登录人获取公司部门树【部门账号登录可查看上级单位】", notes = "根据登录人获取公司部门树")
 	public ResponseModel<List<OrgMenuDto>> companyTreeByUserToPatrolRoute() {

amos-boot-module/amos-boot-module-biz/amos-boot-module-common-biz/src/main/java/com/yeejoin/amos/boot/module/common/biz/service/impl/OrgUsrServiceImpl.java

@@ -55,6 +55,7 @@ import org.typroject.tyboot.core.foundation.context.RequestContext;
 import org.typroject.tyboot.core.foundation.utils.Bean;
 import org.typroject.tyboot.core.foundation.utils.ValidationUtil;
 import org.typroject.tyboot.core.rdbms.service.BaseService;
+import org.typroject.tyboot.core.restful.utils.ResponseModel;
 
 import javax.annotation.Resource;
 import java.io.Serializable;
@@ -227,6 +228,59 @@ public class OrgUsrServiceImpl extends BaseService<OrgUsrDto, OrgUsr, OrgUsrMapp
 		return result;
 	}
 
+	public List<OrgMenuDto> buildTreeParallel2(List<OrgUsr> list) {
+		List<Map<String, Object>> stationInfoOnline = stationInfoMapper.selectStationInfoList();
+		List<String> bizOrgCodeList = stationInfoOnline.stream().map(item -> item.get("biz_org_code").toString()).collect(Collectors.toList());
+
+		List<OrgMenuDto> menuList = list.stream()
+				.map(o -> new OrgMenuDto(o.getSequenceNbr(), o.getBizOrgName(),
+						ObjectUtils.isEmpty(o.getParentId()) ? 0L : Long.parseLong(o.getParentId()), o.getBizOrgType(),
+						false, o.getBizOrgCode(), o.getCode(), o.getBizOrgCode().length() == 18 && bizOrgCodeList.contains(o.getBizOrgCode()) ? Boolean.TRUE : Boolean.FALSE).setTotal(o.getTotal()))
+				.collect(Collectors.toList());
+		List<OrgMenuDto> result = new ArrayList<>();
+		Map<Long, OrgMenuDto> map = new HashMap<>(menuList.size());
+		menuList.forEach(e -> map.put(e.getKey(), e));
+		Set<? extends Map.Entry<Long, ? extends OrgMenuDto>> entries = map.entrySet();
+		// 此处多线程，会value 出现null 的情况
+		entries.forEach(entry -> {
+			OrgMenuDto value = entry.getValue();
+			if (value != null) {
+				OrgMenuDto treeDto = map.get(value.getParentId());
+				if (treeDto != null) {
+					List<OrgMenuDto> children = treeDto.getChildren();
+					if (children == null) {
+						children = new ArrayList<>();
+						treeDto.setChildren(children);
+					}
+
+					children.add(value);
+				} else {
+					result.add(value);
+				}
+			}
+		});
+		List<OrgMenuDto> orgMenuDtos = doIsOnlineData(result);
+		return orgMenuDtos;
+	}
+
+	private List<OrgMenuDto> doIsOnlineData(List<OrgMenuDto> list) {
+		if (CollectionUtils.isNotEmpty(list)) {
+			for (OrgMenuDto orgMenuDto : list) {
+				orgMenuDto.setIsOnline(Boolean.TRUE);
+				if (orgMenuDto.getBizOrgCode().length() == 12 && CollectionUtils.isNotEmpty(orgMenuDto.getChildren())) {
+					Set<Boolean> collect = orgMenuDto.getChildren().stream().map(OrgMenuDto::getIsOnline).collect(Collectors.toSet());
+					if (CollectionUtils.isEmpty(collect) || (collect.contains(Boolean.FALSE) && !collect.contains(Boolean.TRUE))) {
+						orgMenuDto.setIsOnline(Boolean.FALSE);
+					}
+				} else if (CollectionUtils.isEmpty(orgMenuDto.getChildren())) {
+					orgMenuDto.setIsOnline(Boolean.FALSE);
+				}
+				doIsOnlineData(orgMenuDto.getChildren());
+			}
+		}
+		return list;
+	}
+
 	public static String getOrgCodeStr() {
 		return TreeParser.genTreeCode();
 	}
@@ -2939,6 +2993,24 @@ public class OrgUsrServiceImpl extends BaseService<OrgUsrDto, OrgUsr, OrgUsrMapp
 		return buildTreeParallel(list);
 	}
 
+	public List<OrgMenuDto> companyTreeByUserNumber2(ReginParams reginParams) {
+		Map<String, Object> param = new HashMap<>();
+		String bizOrgCode = reginParams.getPersonIdentity().getBizOrgCode();
+		// 权限处理
+		if (!logic){
+			OrgUsr orgUsr = orgUsrMapper.selectById(reginParams.getPersonIdentity().getCompanyId());
+			//判断登陆人是否已经是顶级节点单位
+			if (!ObjectUtils.isEmpty(orgUsr) && orgUsr.getParentId() != null) {
+				orgUsr = this.selectParentOrgUsr(orgUsr);
+				bizOrgCode = orgUsr.getBizOrgCode() != null? orgUsr.getBizOrgCode():bizOrgCode;
+			}
+		}
+		PermissionInterceptorContext.setDataAuthRule(authKey);
+		param.put("bizOrgCode",bizOrgCode);
+		List<OrgUsr> list = orgUsrMapper.companyListWithPersonCount(param);
+		return buildTreeParallel2(list);
+	}
+
 	public List<OrgMenuDto> companyTreeByUserNumberNew(ReginParams reginParams) {
 		Map<String, Object> param = new HashMap<>();
 		// 权限处理

amos-boot-system-equip/src/main/resources/mapper/FireFightingSystemMapper.xml

@@ -7157,28 +7157,31 @@
             *
         FROM
             (
-                (SELECT count( 1 ) AS indexNum FROM wl_equipment_specific_index a LEFT JOIN wl_equipment_index b ON a.equipment_index_id = b.id WHERE b.is_iot = 1
-                and equipment_specific_id IN
-                   (SELECT
-                        id
-                    FROM
-                        wl_equipment_specific
-                    <where>
-                        <if test="bizOrgCode!=null and bizOrgCode!=''">
-                            AND biz_org_code LIKE CONCAT(#{bizOrgCode},'%')
-                        </if>
-                    </where>)
+                (SELECT
+                    sum( n.num )
+                FROM
+                    (
+                SELECT
+                    a.id,
+                    count( 1 ) AS num
+                FROM
+                    wl_equipment_specific a
+                LEFT JOIN (select m.* from wl_equipment_specific_index m left join wl_equipment_index e on e.id = m.equipment_index_id where e.is_iot = 1) b ON a.id = b.equipment_specific_id
+                <where>
+                    <if test="bizOrgCode!=null and bizOrgCode!=''">
+                        AND a.biz_org_code LIKE CONCAT(#{bizOrgCode},'%')
+                    </if>
+                </where>
+                GROUP BY
+                    a.id
+                    ) n
+                    )
                 ) AS indexNum,
                 (SELECT count( 1 ) AS equipNum FROM wl_equipment_specific es LEFT JOIN wl_equipment e ON e.`code` = es.equipment_code WHERE e.is_iot = 1
                     <if test="bizOrgCode!=null and bizOrgCode!=''">
                         AND es.biz_org_code LIKE CONCAT(#{bizOrgCode},'%')
                     </if>
-                ) AS equipNum,
-                (SELECT count(1) AS indexMonitorNum FROM wl_equipment_specific_index wesi LEFT JOIN wl_equipment_specific wes on wesi.equipment_specific_id = wes.id WHERE DATE_FORMAT(wesi.update_date, '%Y-%m-%d') = DATE_FORMAT(NOW(), '%Y-%m-%d')
-                    <if test="bizOrgCode!=null and bizOrgCode!=''">
-                        AND wes.biz_org_code LIKE CONCAT(#{bizOrgCode},'%')
-                    </if>
-                ) AS indexMonitorNum
+                ) AS equipNum
             )
     </select>