From 85c4d09d84c1f879771bac645378d64848765f6b Mon Sep 17 00:00:00 2001 From: pengjie <17373303529@163.com> Date: Wed, 29 Nov 2023 09:38:57 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BC=98=E5=8C=96?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../basicdata/service/impl/SystemRoleServiceImpl.java | 11 +++++------ 1 file changed, 5 insertions(+), 6 deletions(-) diff --git a/src/main/java/com/zhgd/xmgl/modules/basicdata/service/impl/SystemRoleServiceImpl.java b/src/main/java/com/zhgd/xmgl/modules/basicdata/service/impl/SystemRoleServiceImpl.java index 71eb16c..9f1e7bb 100644 --- a/src/main/java/com/zhgd/xmgl/modules/basicdata/service/impl/SystemRoleServiceImpl.java +++ b/src/main/java/com/zhgd/xmgl/modules/basicdata/service/impl/SystemRoleServiceImpl.java @@ -17,6 +17,7 @@ import com.zhgd.xmgl.modules.basicdata.service.IGovernmentService; import com.zhgd.xmgl.modules.basicdata.service.ISystemRoleService; import com.zhgd.xmgl.security.SecurityUser; import com.zhgd.xmgl.security.SecurityUtil; +import com.zhgd.xmgl.util.ParamEnum; import org.apache.commons.lang.StringUtils; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.stereotype.Service; @@ -80,12 +81,10 @@ public class SystemRoleServiceImpl extends ServiceImpl enterpriseMainList = enterpriseMainMapper.selectList(Wrappers.lambdaQuery().eq(EnterpriseMain::getEnterpriseSn, systemRole.getSn())); for (EnterpriseMain main : enterpriseMainList) { String code = ""; - if (main.getMainType() == 1) { - code = "BUILD"; - } else if (main.getMainType() == 2) { - code = "SUPERVISOR"; - } else if (main.getMainType() == 3) { - code = "OP"; + for (ParamEnum.EngineeringMain type : ParamEnum.EngineeringMain.values()) { + if (main.getMainType() == type.getValue()); { + code = type.name(); + } } SystemRole systemRole1 = this.getOne(Wrappers.lambdaQuery() .eq(SystemRole::getRoleCode, code));