From 7152dea94bcf2e4d275a2154161d3663dcf9fa49 Mon Sep 17 00:00:00 2001 From: luxiaotao1123 <t1341870251@163.com> Date: 星期六, 27 二月 2021 09:30:05 +0800 Subject: [PATCH] # --- src/main/java/zy/cloud/wms/common/web/BaseController.java | 12 ++++++++++++ 1 files changed, 12 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 4321d81..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,7 +9,9 @@ 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; @@ -32,6 +34,8 @@ private UserService userService; @Autowired private TagService tagService; + @Autowired + private NodeService nodeService; protected Long getUserId(){ return Long.parseLong(String.valueOf(request.getAttribute("userId"))); @@ -62,6 +66,14 @@ 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