From e422c7080bd90d22c7cf7b04a63b54068c8defe4 Mon Sep 17 00:00:00 2001 From: caiyuchao Date: Wed, 20 Aug 2025 16:01:51 +0800 Subject: [PATCH] =?UTF-8?q?fix:=20=E5=AE=A2=E6=88=B7=E7=AE=A1=E7=90=86?= =?UTF-8?q?=E5=AD=97=E6=AE=B5=E6=9D=83=E9=99=90=E6=8E=A7=E5=88=B6?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../admin/customer/CustomerController.java | 30 +++++++++++++++++++ .../admin/customer/vo/CustomerSaveReqVO.java | 7 ++--- 2 files changed, 33 insertions(+), 4 deletions(-) diff --git a/agt-module-license/agt-module-license-server/src/main/java/org/agt/module/license/controller/admin/customer/CustomerController.java b/agt-module-license/agt-module-license-server/src/main/java/org/agt/module/license/controller/admin/customer/CustomerController.java index d6985f1..f1de264 100644 --- a/agt-module-license/agt-module-license-server/src/main/java/org/agt/module/license/controller/admin/customer/CustomerController.java +++ b/agt-module-license/agt-module-license-server/src/main/java/org/agt/module/license/controller/admin/customer/CustomerController.java @@ -14,6 +14,7 @@ import org.agt.framework.common.pojo.PageResult; import org.agt.framework.common.util.object.BeanUtils; import org.agt.framework.excel.core.util.ExcelUtils; import org.agt.framework.ip.core.utils.AreaUtils; +import org.agt.framework.security.core.service.SecurityFrameworkService; import org.agt.module.license.controller.admin.customer.vo.CustomerImportRespVO; import org.agt.module.license.controller.admin.customer.vo.CustomerPageReqVO; import org.agt.module.license.controller.admin.customer.vo.CustomerRespVO; @@ -50,6 +51,9 @@ public class CustomerController { @Resource private CustomerService customerService; + @Resource + private SecurityFrameworkService securityFrameworkService; + @GetMapping("/dashboard") @Operation(summary = "获得首页数据") public CommonResult dashboard() { @@ -87,6 +91,13 @@ public class CustomerController { @PreAuthorize("@ss.hasPermission('license:customer:query')") public CommonResult getCustomer(@RequestParam("id") Long id) { CustomerDO customer = customerService.getCustomer(id); + + if (!securityFrameworkService.hasRole("business")) { + customer.setContacts(null); + customer.setRole(null); + customer.setPhone(null); + } + return success(BeanUtils.toBean(customer, CustomerRespVO.class, customerRespVO -> { if (customerRespVO.getAreaId() != null) { customerRespVO.setAreaIds(AreaUtils.formatToIdList(customerRespVO.getAreaId().intValue())); @@ -119,6 +130,16 @@ public class CustomerController { @PreAuthorize("@ss.hasPermission('license:customer:query')") public CommonResult> getCustomerPage(@Valid CustomerPageReqVO pageReqVO) { PageResult pageResult = customerService.getCustomerPage(pageReqVO); + + if (!securityFrameworkService.hasRole("business")) { + List list = pageResult.getList(); + list.forEach(customerDO -> { + customerDO.setContacts(null); + customerDO.setRole(null); + customerDO.setPhone(null); + }); + } + return success(buildCustomerVOList(pageResult)); } @@ -127,6 +148,15 @@ public class CustomerController { @PreAuthorize("@ss.hasPermission('license:customer:query')") public CommonResult> getCustomerList() { List listResult = customerService.getCustomerList(); + + if (!securityFrameworkService.hasRole("business")) { + listResult.forEach(customerDO -> { + customerDO.setContacts(null); + customerDO.setRole(null); + customerDO.setPhone(null); + }); + } + return success(BeanUtils.toBean(listResult, CustomerRespVO.class)); } diff --git a/agt-module-license/agt-module-license-server/src/main/java/org/agt/module/license/controller/admin/customer/vo/CustomerSaveReqVO.java b/agt-module-license/agt-module-license-server/src/main/java/org/agt/module/license/controller/admin/customer/vo/CustomerSaveReqVO.java index def64f4..64846f1 100644 --- a/agt-module-license/agt-module-license-server/src/main/java/org/agt/module/license/controller/admin/customer/vo/CustomerSaveReqVO.java +++ b/agt-module-license/agt-module-license-server/src/main/java/org/agt/module/license/controller/admin/customer/vo/CustomerSaveReqVO.java @@ -1,9 +1,9 @@ package org.agt.module.license.controller.admin.customer.vo; import io.swagger.v3.oas.annotations.media.Schema; -import lombok.*; -import java.util.*; -import jakarta.validation.constraints.*; +import jakarta.validation.constraints.NotEmpty; +import jakarta.validation.constraints.NotNull; +import lombok.Data; @Schema(description = "管理后台 - 客户新增/修改 Request VO") @Data @@ -29,7 +29,6 @@ public class CustomerSaveReqVO { private Long areaId; @Schema(description = "联系人", requiredMode = Schema.RequiredMode.REQUIRED) - @NotEmpty(message = "联系人不能为空") private String contacts; @Schema(description = "角色")