From 8fef24eef16f7eee0bb4f98ab5cd9ed41ab71f24 Mon Sep 17 00:00:00 2001
From: HAHA <HAHA@etoak.com>
Date: Thu, 19 May 2022 16:32:34 +0800
Subject: [PATCH] =?UTF-8?q?=E6=94=B9bug?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../java/com/epmet/dto/form/PartyUnitListbriefFormDTO.java    | 2 ++
 .../main/java/com/epmet/controller/IcPartyUnitController.java | 3 ++-
 .../src/main/java/com/epmet/service/IcPartyUnitService.java   | 2 +-
 .../java/com/epmet/service/impl/IcPartyUnitServiceImpl.java   | 4 ++--
 4 files changed, 7 insertions(+), 4 deletions(-)

diff --git a/epmet-module/epmet-heart/epmet-heart-client/src/main/java/com/epmet/dto/form/PartyUnitListbriefFormDTO.java b/epmet-module/epmet-heart/epmet-heart-client/src/main/java/com/epmet/dto/form/PartyUnitListbriefFormDTO.java
index b84d504bd3..7be109d4e7 100644
--- a/epmet-module/epmet-heart/epmet-heart-client/src/main/java/com/epmet/dto/form/PartyUnitListbriefFormDTO.java
+++ b/epmet-module/epmet-heart/epmet-heart-client/src/main/java/com/epmet/dto/form/PartyUnitListbriefFormDTO.java
@@ -51,5 +51,7 @@ public class PartyUnitListbriefFormDTO extends PageFormDTO implements Serializab
      */
     private String contactMobile;
 
+    private String customerId;
+
 
 }
diff --git a/epmet-module/epmet-heart/epmet-heart-server/src/main/java/com/epmet/controller/IcPartyUnitController.java b/epmet-module/epmet-heart/epmet-heart-server/src/main/java/com/epmet/controller/IcPartyUnitController.java
index 0cf4da7683..03bcf813ba 100644
--- a/epmet-module/epmet-heart/epmet-heart-server/src/main/java/com/epmet/controller/IcPartyUnitController.java
+++ b/epmet-module/epmet-heart/epmet-heart-server/src/main/java/com/epmet/controller/IcPartyUnitController.java
@@ -245,8 +245,9 @@ public class IcPartyUnitController {
      */
     @PostMapping("/listbrief")
     public Result<PageData> getListbrief(@RequestBody PartyUnitListbriefFormDTO form, @LoginUser TokenDto tokenDto) {
+        form.setCustomerId(tokenDto.getCustomerId());
         ValidatorUtils.validateEntity(form);
-        PageData data = icPartyUnitService.getListbrief(form, tokenDto);
+        PageData data = icPartyUnitService.getListbrief(form);
         return new Result<PageData>().ok(data);
     }
 
diff --git a/epmet-module/epmet-heart/epmet-heart-server/src/main/java/com/epmet/service/IcPartyUnitService.java b/epmet-module/epmet-heart/epmet-heart-server/src/main/java/com/epmet/service/IcPartyUnitService.java
index 5a5cb36416..73b35af7bd 100644
--- a/epmet-module/epmet-heart/epmet-heart-server/src/main/java/com/epmet/service/IcPartyUnitService.java
+++ b/epmet-module/epmet-heart/epmet-heart-server/src/main/java/com/epmet/service/IcPartyUnitService.java
@@ -171,7 +171,7 @@ public interface IcPartyUnitService extends BaseService<IcPartyUnitEntity> {
      *
      * @param form
      */
-    PageData getListbrief(PartyUnitListbriefFormDTO form, TokenDto tokenDto);
+    PageData getListbrief(PartyUnitListbriefFormDTO form);
 
     /**
      * 联建单位-按分类统计数量及占比
diff --git a/epmet-module/epmet-heart/epmet-heart-server/src/main/java/com/epmet/service/impl/IcPartyUnitServiceImpl.java b/epmet-module/epmet-heart/epmet-heart-server/src/main/java/com/epmet/service/impl/IcPartyUnitServiceImpl.java
index cf4f96c510..5276e1352d 100644
--- a/epmet-module/epmet-heart/epmet-heart-server/src/main/java/com/epmet/service/impl/IcPartyUnitServiceImpl.java
+++ b/epmet-module/epmet-heart/epmet-heart-server/src/main/java/com/epmet/service/impl/IcPartyUnitServiceImpl.java
@@ -599,9 +599,9 @@ public class IcPartyUnitServiceImpl extends BaseServiceImpl<IcPartyUnitDao, IcPa
     }
 
     @Override
-    public PageData getListbrief(PartyUnitListbriefFormDTO form, TokenDto tokenDto) {
+    public PageData getListbrief(PartyUnitListbriefFormDTO form) {
         PageHelper.startPage(form.getPageNo(), form.getPageSize());
-        List<PartyUnitListbrieResultDTO> dto = baseDao.getListbrief(form, tokenDto.getCustomerId());
+        List<PartyUnitListbrieResultDTO> dto = baseDao.getListbrief(form, form.getCustomerId());
         // 获取gridName
         for (PartyUnitListbrieResultDTO item : dto) {
             if (StringUtils.isNotEmpty(item.getGridId())) {