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