根节点处理

epmet-module/gov-org/gov-org-server/src/main/java/com/epmet/controller/AgencyController.java

@@ -510,13 +510,13 @@ public class AgencyController {
      * 获取区域编码下级的所有客户信息
      *
      * @param areaCode
-     * @return com.epmet.commons.tools.utils.Result<java.util.List<com.epmet.dto.result.AgencyResultDTO>>
+     * @return com.epmet.commons.tools.utils.Result<java.util.List < com.epmet.dto.result.AgencyResultDTO>>
      * @author zhy
      * @date 2022/7/7 10:12
      */
     @GetMapping("/areasubagency/{areaCode}")
-    public Result<List<AgencyResultDTO>> getAreaSubAgency(@PathVariable String areaCode) {
-        List<AgencyResultDTO> dto = agencyService.getAreaSubAgency(areaCode);
+    public Result<List<AgencyResultDTO>> getAreaSubAgency(@LoginUser TokenDto tokenDto, @PathVariable String areaCode) {
+        List<AgencyResultDTO> dto = agencyService.getAreaSubAgency(tokenDto, areaCode);
         return new Result<List<AgencyResultDTO>>().ok(dto);
     }
 

epmet-module/gov-org/gov-org-server/src/main/java/com/epmet/dao/CustomerAgencyDao.java

@@ -364,5 +364,7 @@ public interface CustomerAgencyDao extends BaseDao<CustomerAgencyEntity> {
      */
     List<AgencyResultDTO> getAreaSubAgency(@Param("areaCode") String areaCode);
 
+    CustomerAgencyEntity getAreaRootAgency(@Param("customerId") String customerId);
+
 }
 

epmet-module/gov-org/gov-org-server/src/main/java/com/epmet/service/AgencyService.java

@@ -183,10 +183,10 @@ public interface AgencyService {
      * 获取区域编码下级的所有客户信息
      *
      * @param areaCode
-     * @return com.epmet.commons.tools.utils.Result<java.util.List<com.epmet.dto.result.AgencyResultDTO>>
+     * @return com.epmet.commons.tools.utils.Result<java.util.List < com.epmet.dto.result.AgencyResultDTO>>
      * @author zhy
      * @date 2022/7/7 10:12
      */
-    List<AgencyResultDTO> getAreaSubAgency(String areaCode);
+    List<AgencyResultDTO> getAreaSubAgency(TokenDto tokenDto, String areaCode);
 
 }

epmet-module/gov-org/gov-org-server/src/main/java/com/epmet/service/impl/AgencyServiceImpl.java

@@ -903,7 +903,11 @@ public class AgencyServiceImpl implements AgencyService {
     }
 
     @Override
-    public List<AgencyResultDTO> getAreaSubAgency(String areaCode) {
+    public List<AgencyResultDTO> getAreaSubAgency(TokenDto tokenDto, String areaCode) {
+        if (NumConstant.ZERO_STR.equals(areaCode)) {
+            CustomerAgencyEntity root = customerAgencyDao.getAreaRootAgency(tokenDto.getCustomerId());
+            areaCode = root.getAreaCode();
+        }
         return customerAgencyDao.getAreaSubAgency(areaCode);
     }
 

epmet-module/gov-org/gov-org-server/src/main/resources/mapper/CustomerAgencyDao.xml

@@ -945,8 +945,25 @@
             AND ac.PID = '0'
             AND cop.PARAMETER_VALUE = 'open'
         <if test="areaCode != null and areaCode != ''">
-            AND AREA_CODE != #{areaCode}
-            AND AREA_CODE LIKE CONCAT(#{areaCode}, '%')
+            AND ac.AREA_CODE != #{areaCode}
+            AND ac.AREA_CODE LIKE CONCAT(#{areaCode}, '%')
+        </if>
+    </select>
+
+    <select id="getAreaRootAgency" resultType="com.epmet.entity.CustomerAgencyEntity">
+        SELECT
+        ac.*
+        FROM
+        customer_agency ac
+        INNER JOIN customer_org_parameter cop ON cop.CUSTOMER_ID = ac.CUSTOMER_ID
+        WHERE
+        ac.DEL_FLAG = '0'
+        AND cop.DEL_FLAG = '0'
+        AND cop.PARAMETER_KEY = 'area_code_switch'
+        AND ac.PID = '0'
+        AND cop.PARAMETER_VALUE = 'open'
+        <if test="customerId != null and customerId != ''">
+            AND ac.CUSTOMER_ID = #{customerId}
         </if>
     </select>