From d1f1a2187adc1642911b5fb78a9fa279707c2f97 Mon Sep 17 00:00:00 2001
From: yinzuomei <576302893@qq.com>
Date: Tue, 21 Jun 2022 17:47:59 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BC=81=E4=BA=8B=E4=B8=9A=E5=8D=95=E4=BD=8D?=
 =?UTF-8?q?=E9=99=90=E5=88=B6=E4=B8=8B=E6=9C=AC=E7=BB=84=E7=BB=87=E7=9A=84?=
 =?UTF-8?q?=E6=95=B0=E6=8D=AE=E6=89=8D=E5=8F=AF=E4=BB=A5=E7=BC=96=E8=BE=91?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../main/java/com/epmet/dto/form/EnterpriseFormDTO.java   | 2 ++
 .../java/com/epmet/controller/IcEnterpriseController.java | 1 +
 .../com/epmet/service/impl/IcEnterpriseServiceImpl.java   | 8 ++++++++
 3 files changed, 11 insertions(+)

diff --git a/epmet-module/gov-org/gov-org-client/src/main/java/com/epmet/dto/form/EnterpriseFormDTO.java b/epmet-module/gov-org/gov-org-client/src/main/java/com/epmet/dto/form/EnterpriseFormDTO.java
index 5139e442b0..ff483014ba 100644
--- a/epmet-module/gov-org/gov-org-client/src/main/java/com/epmet/dto/form/EnterpriseFormDTO.java
+++ b/epmet-module/gov-org/gov-org-client/src/main/java/com/epmet/dto/form/EnterpriseFormDTO.java
@@ -17,6 +17,8 @@ public class EnterpriseFormDTO implements Serializable {
      * token赋值
      */
     private String customerId;
+    private String currentStaffId;
+
 
     /**
      * 网格Id【场所区域】
diff --git a/epmet-module/gov-org/gov-org-server/src/main/java/com/epmet/controller/IcEnterpriseController.java b/epmet-module/gov-org/gov-org-server/src/main/java/com/epmet/controller/IcEnterpriseController.java
index 3cdfb2f243..4c06fae489 100644
--- a/epmet-module/gov-org/gov-org-server/src/main/java/com/epmet/controller/IcEnterpriseController.java
+++ b/epmet-module/gov-org/gov-org-server/src/main/java/com/epmet/controller/IcEnterpriseController.java
@@ -66,6 +66,7 @@ public class IcEnterpriseController {
     @PostMapping("addOrUpdate")
     public Result addOrUpdate(@LoginUser TokenDto tokenDto, @RequestBody EnterpriseFormDTO formDTO) {
         formDTO.setCustomerId(tokenDto.getCustomerId());
+        formDTO.setCurrentStaffId(tokenDto.getUserId());
         ValidatorUtils.validateEntity(EnterpriseFormDTO.AddShow.class);
         Map<String, String> map = new HashMap<>();
         map.put("enterpriseId", icEnterpriseService.addOrUpdate(formDTO));
diff --git a/epmet-module/gov-org/gov-org-server/src/main/java/com/epmet/service/impl/IcEnterpriseServiceImpl.java b/epmet-module/gov-org/gov-org-server/src/main/java/com/epmet/service/impl/IcEnterpriseServiceImpl.java
index d100120311..310975f045 100644
--- a/epmet-module/gov-org/gov-org-server/src/main/java/com/epmet/service/impl/IcEnterpriseServiceImpl.java
+++ b/epmet-module/gov-org/gov-org-server/src/main/java/com/epmet/service/impl/IcEnterpriseServiceImpl.java
@@ -79,6 +79,14 @@ public class IcEnterpriseServiceImpl extends BaseServiceImpl<IcEnterpriseDao, Ic
             enterpriseEntity.setSourceType(CommonOperateTypeEnum.ADD.getCode());
             baseDao.insert(enterpriseEntity);
         }else{
+            //如果是编辑判断下权限
+            IcEnterpriseEntity origin=baseDao.selectById(formDTO.getEnterpriseId());
+            CustomerStaffInfoCacheResult staffInfo=CustomerStaffRedis.getStaffInfo(formDTO.getCustomerId(),formDTO.getCurrentStaffId());
+            if (!staffInfo.getAgencyId().equals(origin.getAgencyId())) {
+                AgencyInfoCache orginAgencyInfo = CustomerOrgRedis.getAgencyInfo(origin.getAgencyId());
+                String agencyName = null != orginAgencyInfo ? orginAgencyInfo.getOrganizationName() : StrConstant.EPMETY_STR;
+                throw new EpmetException(EpmetErrorCode.EPMET_COMMON_OPERATION_FAIL.getCode(), "", String.format("当前场所属于%s,您无权编辑", agencyName));
+            }
             enterpriseEntity.setId(formDTO.getEnterpriseId());
             baseDao.updateById(enterpriseEntity);
         }