From e650c3cedcaaba22106c20b2256ea2e1e48d8794 Mon Sep 17 00:00:00 2001
From: yinzuomei <576302893@qq.com>
Date: Mon, 24 Apr 2023 10:59:51 +0800
Subject: [PATCH] /gov/org/customeragency/agencygridtreeV2

---
 .../controller/CustomerAgencyController.java  | 13 +++++++-
 .../epmet/service/CustomerAgencyService.java  |  2 +-
 .../impl/CustomerAgencyServiceImpl.java       | 31 ++++++++++++++++---
 3 files changed, 39 insertions(+), 7 deletions(-)

diff --git a/epmet-module/gov-org/gov-org-server/src/main/java/com/epmet/controller/CustomerAgencyController.java b/epmet-module/gov-org/gov-org-server/src/main/java/com/epmet/controller/CustomerAgencyController.java
index ba94cc12ab..b7b13fd2ec 100644
--- a/epmet-module/gov-org/gov-org-server/src/main/java/com/epmet/controller/CustomerAgencyController.java
+++ b/epmet-module/gov-org/gov-org-server/src/main/java/com/epmet/controller/CustomerAgencyController.java
@@ -351,7 +351,7 @@ public class CustomerAgencyController {
     @PostMapping("agencylist")
     public Result<AgencyTreeResultDTO> getAgencyList(@LoginUser TokenDto tokenDTO,@RequestBody GetAgencyListFormDTO formDTO) {
         if(AppClientConstant.APP_OPER.equals(tokenDTO.getApp())){
-            ValidatorUtils.validateEntity(formDTO.getCustomerId(),GetAgencyListFormDTO.OperAddUserShowGroup.class);
+            ValidatorUtils.validateEntity(formDTO,GetAgencyListFormDTO.OperAddUserShowGroup.class);
         }else{
             if (StringUtils.isBlank(formDTO.getCustomerId())){
                 formDTO.setCustomerId(tokenDTO.getCustomerId());
@@ -405,6 +405,17 @@ public class CustomerAgencyController {
         return new Result<AgencyTreeResultDTO>().ok(customerAgencyService.getOrgTreeData(tokenDTO.getUserId(),tokenDTO.getCustomerId()));
     }
 
+    /**
+     * 根组织开始，含网格
+     * @param formDTO
+     * @return
+     */
+    @PostMapping("agencygridtreeV2")
+    public Result<AgencyTreeResultDTO> getOrgList(@RequestBody GetAgencyListFormDTO formDTO) {
+        ValidatorUtils.validateEntity(formDTO,GetAgencyListFormDTO.OperAddUserShowGroup.class);
+        return new Result<AgencyTreeResultDTO>().ok(customerAgencyService.getOrgTreeData(formDTO.getUserId(),formDTO.getCustomerId()));
+    }
+
     /**
      * @Description 【事件】社区服务热线
      * @author sun
diff --git a/epmet-module/gov-org/gov-org-server/src/main/java/com/epmet/service/CustomerAgencyService.java b/epmet-module/gov-org/gov-org-server/src/main/java/com/epmet/service/CustomerAgencyService.java
index 5ef18bf8e3..bf2f7b855e 100644
--- a/epmet-module/gov-org/gov-org-server/src/main/java/com/epmet/service/CustomerAgencyService.java
+++ b/epmet-module/gov-org/gov-org-server/src/main/java/com/epmet/service/CustomerAgencyService.java
@@ -267,7 +267,7 @@ public interface CustomerAgencyService extends BaseService<CustomerAgencyEntity>
 
     /**
      * desc:获取用户所属组织的组织及网格树
-     * @param staffId
+     * @param staffId 为空时，从根组织返回
      * @return
      */
     AgencyTreeResultDTO getOrgTreeData(String staffId,String customerId);
diff --git a/epmet-module/gov-org/gov-org-server/src/main/java/com/epmet/service/impl/CustomerAgencyServiceImpl.java b/epmet-module/gov-org/gov-org-server/src/main/java/com/epmet/service/impl/CustomerAgencyServiceImpl.java
index 48683fbf77..c0bae678b1 100644
--- a/epmet-module/gov-org/gov-org-server/src/main/java/com/epmet/service/impl/CustomerAgencyServiceImpl.java
+++ b/epmet-module/gov-org/gov-org-server/src/main/java/com/epmet/service/impl/CustomerAgencyServiceImpl.java
@@ -1178,14 +1178,35 @@ public class CustomerAgencyServiceImpl extends BaseServiceImpl<CustomerAgencyDao
     }
 
 
+    /**
+     * 从登录用户所属组织开始，包含网格的一颗组织树
+     * @param staffId 如果staffId为空时，从根组织返回
+     * @param customerId
+     * @return
+     */
     @Override
     public AgencyTreeResultDTO getOrgTreeData(String staffId,String customerId) {
         AgencyTreeResultDTO result = new AgencyTreeResultDTO();
-        //1.查询工作人员所属组织信息
-        AgencyResultDTO rootAgency = baseDao.selectAgencyByStaffId(staffId);
-        if (null == rootAgency) {
-            logger.error(String.format("查询工作人员所属组织信息失败，staffId->%s", staffId));
-            throw new RenException(CustomerAgencyConstant.SELECT_STAFF_AGENCY_EXCEPTION);
+        AgencyResultDTO rootAgency=new AgencyResultDTO();
+        if(StringUtils.isNotBlank(staffId)){
+            //1.查询工作人员所属组织信息
+            rootAgency = baseDao.selectAgencyByStaffId(staffId);
+            if (null == rootAgency) {
+                logger.error(String.format("查询工作人员所属组织信息失败，staffId->%s", staffId));
+                throw new RenException(CustomerAgencyConstant.SELECT_STAFF_AGENCY_EXCEPTION);
+            }
+        }else{
+            LambdaQueryWrapper<CustomerAgencyEntity> rootQuery=new LambdaQueryWrapper<>();
+            rootQuery.eq(CustomerAgencyEntity::getCustomerId,customerId)
+                    .eq(CustomerAgencyEntity::getPid,NumConstant.ZERO_STR);
+            CustomerAgencyEntity rootEntity=baseDao.selectOne(rootQuery);
+            rootAgency.setAgencyId(rootEntity.getId());
+            rootAgency.setAgencyName(rootEntity.getOrganizationName());
+            rootAgency.setPids(rootEntity.getPids());
+            rootAgency.setPid(rootEntity.getPid());
+            rootAgency.setLevel(rootEntity.getLevel());
+            rootAgency.setLongitude(rootEntity.getLongitude());
+            rootAgency.setLatitude(rootEntity.getLatitude());
         }
         result.setPid(rootAgency.getPid());
         result.setAgencyName(rootAgency.getAgencyName());