From 0e8731a031b810fa4dff7d127dd4912cc58d5f10 Mon Sep 17 00:00:00 2001 From: luxiaotao1123 <t1341870251@63.com> Date: 星期四, 24 十一月 2022 15:01:13 +0800 Subject: [PATCH] # --- src/main/java/com/zy/crm/common/web/BaseController.java | 32 ++++++++++++++++++++++++++++++++ 1 files changed, 32 insertions(+), 0 deletions(-) diff --git a/src/main/java/com/zy/crm/common/web/BaseController.java b/src/main/java/com/zy/crm/common/web/BaseController.java index 8fbe2a9..eab395b 100644 --- a/src/main/java/com/zy/crm/common/web/BaseController.java +++ b/src/main/java/com/zy/crm/common/web/BaseController.java @@ -7,6 +7,10 @@ import com.core.common.Cools; import com.core.controller.AbstractBaseController; import com.core.exception.CoolException; +import com.zy.crm.system.entity.Dept; +import com.zy.crm.system.entity.Role; +import com.zy.crm.system.service.DeptService; +import com.zy.crm.system.service.RoleService; import io.swagger.annotations.ApiModelProperty; import org.springframework.beans.factory.annotation.Autowired; import com.zy.crm.manager.entity.Node; @@ -39,7 +43,11 @@ @Autowired private NodeService nodeService; @Autowired + private DeptService deptService; + @Autowired private UserLoginService userLoginService; + @Autowired + private RoleService roleService; protected Long getHostId(){ if (getUserId() == 9527) { @@ -79,6 +87,19 @@ } return user; } + + protected Long getDeptId() { + return getUser().getDeptId(); + } + + protected Role getRole(){ + Role role = roleService.selectById(getUser().getRoleId()); + if (null == role) { + throw new CoolException(BaseRes.DENIED); + } + return role; + } + protected String getComment(Class<?> cls, String fieldName){ Field[] fields = Cools.getAllFields(cls); for (Field field : fields){ @@ -99,6 +120,14 @@ throw new CoolException("搴撳尯鏁版嵁閿欒"); } return node; + } + + protected Dept getOriginDept(){ + Dept dept = deptService.getTop(getHostId()); + if (dept == null) { + throw new CoolException("閮ㄩ棬鏁版嵁閿欒"); + } + return dept; } protected <T> void hostEq(EntityWrapper<T> wrapper){ @@ -170,6 +199,9 @@ } String column = null; if (field.isAnnotationPresent(TableField.class)) { + if (!field.getAnnotation(TableField.class).exist()) { + continue; + } column = field.getAnnotation(TableField.class).value(); } if (Cools.isEmpty(column)) { -- Gitblit v1.9.1