From 08b9b1c2f320265a0f5f2c76444e5cce539de707 Mon Sep 17 00:00:00 2001
From: HAHA <HAHA@etoak.com>
Date: Wed, 18 May 2022 14:13:57 +0800
Subject: [PATCH] =?UTF-8?q?=E8=81=94=E5=BB=BA=E5=8D=95=E4=BD=8D-=E7=AE=80?=
 =?UTF-8?q?=E8=A6=81=E4=BF=A1=E6=81=AF=E5=88=97=E8=A1=A8=E8=A1=A5=E5=85=85?=
 =?UTF-8?q?=E6=9F=A5=E8=AF=A2=E6=9D=A1=E4=BB=B6?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../main/java/com/epmet/controller/IcPartyUnitController.java | 4 ++--
 .../src/main/java/com/epmet/dao/IcPartyUnitDao.java           | 2 +-
 .../src/main/java/com/epmet/service/IcPartyUnitService.java   | 2 +-
 .../java/com/epmet/service/impl/IcPartyUnitServiceImpl.java   | 4 ++--
 .../src/main/resources/mapper/IcPartyUnitDao.xml              | 1 +
 5 files changed, 7 insertions(+), 6 deletions(-)

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 6734697c40..b7ac5fe912 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,9 +245,9 @@ public class IcPartyUnitController {
      * @date 2022/5/18 13:54
      */
     @PostMapping("/listbrief")
-    public Result<PageData> getListbrief(@RequestBody PartyUnitListbriefFormDTO form){
+    public Result<PageData> getListbrief(@RequestBody PartyUnitListbriefFormDTO form,@LoginUser TokenDto tokenDto){
         ValidatorUtils.validateEntity(form);
-        PageData data = icPartyUnitService.getListbrief(form);
+        PageData data = icPartyUnitService.getListbrief(form,tokenDto);
         return new Result<PageData>().ok(data);
     }
 
diff --git a/epmet-module/epmet-heart/epmet-heart-server/src/main/java/com/epmet/dao/IcPartyUnitDao.java b/epmet-module/epmet-heart/epmet-heart-server/src/main/java/com/epmet/dao/IcPartyUnitDao.java
index a926f07069..d96e45e96b 100644
--- a/epmet-module/epmet-heart/epmet-heart-server/src/main/java/com/epmet/dao/IcPartyUnitDao.java
+++ b/epmet-module/epmet-heart/epmet-heart-server/src/main/java/com/epmet/dao/IcPartyUnitDao.java
@@ -77,5 +77,5 @@ public interface IcPartyUnitDao extends BaseDao<IcPartyUnitEntity> {
      * @author LZN
      * @date 2022/5/18 13:54
      */
-    List<PartyUnitListbrieResultDTO> getListbrief(PartyUnitListbriefFormDTO form);
+    List<PartyUnitListbrieResultDTO> getListbrief(PartyUnitListbriefFormDTO form,String customerId);
 }
\ No newline at end of file
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 4da4e2d6d6..4469073a2c 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
@@ -165,5 +165,5 @@ public interface IcPartyUnitService extends BaseService<IcPartyUnitEntity> {
      * 联建单位-简要信息列表
      * @param form
      */
-    PageData getListbrief(PartyUnitListbriefFormDTO form);
+    PageData getListbrief(PartyUnitListbriefFormDTO form,TokenDto tokenDto);
 }
\ No newline at end of file
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 fb7609a043..45ce4bc831 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
@@ -597,9 +597,9 @@ public class IcPartyUnitServiceImpl extends BaseServiceImpl<IcPartyUnitDao, IcPa
     }
 
     @Override
-    public PageData getListbrief(PartyUnitListbriefFormDTO form) {
+    public PageData getListbrief(PartyUnitListbriefFormDTO form,TokenDto tokenDto) {
             PageHelper.startPage(form.getPageNo(),form.getPageSize());
-            List<PartyUnitListbrieResultDTO> dto = baseDao.getListbrief(form);
+            List<PartyUnitListbrieResultDTO> dto = baseDao.getListbrief(form,tokenDto.getCustomerId());
             PageInfo<PartyUnitListbrieResultDTO> info = new PageInfo<>(dto);
             return new PageData<>(dto,info.getTotal());
     }
diff --git a/epmet-module/epmet-heart/epmet-heart-server/src/main/resources/mapper/IcPartyUnitDao.xml b/epmet-module/epmet-heart/epmet-heart-server/src/main/resources/mapper/IcPartyUnitDao.xml
index 07ea2115c4..0295be0c07 100644
--- a/epmet-module/epmet-heart/epmet-heart-server/src/main/resources/mapper/IcPartyUnitDao.xml
+++ b/epmet-module/epmet-heart/epmet-heart-server/src/main/resources/mapper/IcPartyUnitDao.xml
@@ -87,6 +87,7 @@
         <where>
             a.DEL_FLAG = '0'
             AND a.AGENCY_ID = #{agencyId}
+            AND a.CUSTOMER_ID = #{customerId}
             <if test="gridId != null and gridId != ''">
                 AND b.GRID_ID = #{gridId}
             </if>