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