From 19f75dabfc35f311f8a2096db8a1eec3136e7814 Mon Sep 17 00:00:00 2001 From: luxiaotao1123 <t1341870251@163.com> Date: 星期三, 24 三月 2021 16:48:59 +0800 Subject: [PATCH] # --- src/main/java/zy/cloud/wms/common/web/BaseController.java | 24 ++++++++++++++++++++++++ 1 files changed, 24 insertions(+), 0 deletions(-) diff --git a/src/main/java/zy/cloud/wms/common/web/BaseController.java b/src/main/java/zy/cloud/wms/common/web/BaseController.java index b75fd7a..1bb099b 100644 --- a/src/main/java/zy/cloud/wms/common/web/BaseController.java +++ b/src/main/java/zy/cloud/wms/common/web/BaseController.java @@ -9,6 +9,10 @@ import com.core.exception.CoolException; import io.swagger.annotations.ApiModelProperty; import org.springframework.beans.factory.annotation.Autowired; +import zy.cloud.wms.manager.entity.Node; +import zy.cloud.wms.manager.entity.Tag; +import zy.cloud.wms.manager.service.NodeService; +import zy.cloud.wms.manager.service.TagService; import zy.cloud.wms.system.entity.User; import zy.cloud.wms.system.service.UserService; @@ -28,6 +32,10 @@ protected HttpServletRequest request; @Autowired private UserService userService; + @Autowired + private TagService tagService; + @Autowired + private NodeService nodeService; protected Long getUserId(){ return Long.parseLong(String.valueOf(request.getAttribute("userId"))); @@ -50,6 +58,22 @@ return ""; } + protected Tag getOriginTag(){ + Tag tag = tagService.selectOne(new EntityWrapper<Tag>().eq("level", 1)); + if (tag == null) { + throw new CoolException("褰掔被鏁版嵁閿欒"); + } + return tag; + } + + protected Node getOriginNode(){ + Node node = nodeService.selectOne(new EntityWrapper<Node>().eq("level", 0)); + if (node == null) { + throw new CoolException("搴撳尯鏁版嵁閿欒"); + } + return node; + } + /** * 鍒嗛〉缁勮 * @param pageNumber -- Gitblit v1.9.1