From 8c4fc3d632958daa7aff0fdd8b273d85bec68ead Mon Sep 17 00:00:00 2001
From: luxiaotao1123 <t1341870251@163.com>
Date: 星期二, 20 二月 2024 10:59:39 +0800
Subject: [PATCH] #
---
zy-asrs-wcs/src/main/java/com/zy/asrs/wcs/utils/NodeUtils.java | 42 ++++++++++++++++++++++++++++++++++++++++++
zy-asrs-wcs/src/main/java/com/zy/asrs/wcs/system/controller/MenuController.java | 3 +++
2 files changed, 45 insertions(+), 0 deletions(-)
diff --git a/zy-asrs-wcs/src/main/java/com/zy/asrs/wcs/system/controller/MenuController.java b/zy-asrs-wcs/src/main/java/com/zy/asrs/wcs/system/controller/MenuController.java
index d98d507..f4a6da7 100644
--- a/zy-asrs-wcs/src/main/java/com/zy/asrs/wcs/system/controller/MenuController.java
+++ b/zy-asrs-wcs/src/main/java/com/zy/asrs/wcs/system/controller/MenuController.java
@@ -11,6 +11,7 @@
import com.zy.asrs.wcs.system.entity.Menu;
import com.zy.asrs.wcs.system.service.MenuService;
import com.zy.asrs.wcs.utils.ExcelUtil;
+import com.zy.asrs.wcs.utils.NodeUtils;
import com.zy.asrs.wcs.utils.Utils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.security.access.prepost.PreAuthorize;
@@ -63,6 +64,8 @@
if (!menuService.save(menu)) {
return R.error("娣诲姞澶辫触");
}
+ NodeUtils nodeUtils = new NodeUtils();
+ nodeUtils.generatePath(Menu.class, menu, Menu::getId, Menu::getName, Menu::getParentId);
return R.ok("娣诲姞鎴愬姛");
}
diff --git a/zy-asrs-wcs/src/main/java/com/zy/asrs/wcs/utils/NodeUtils.java b/zy-asrs-wcs/src/main/java/com/zy/asrs/wcs/utils/NodeUtils.java
new file mode 100644
index 0000000..e548dad
--- /dev/null
+++ b/zy-asrs-wcs/src/main/java/com/zy/asrs/wcs/utils/NodeUtils.java
@@ -0,0 +1,42 @@
+package com.zy.asrs.wcs.utils;
+
+
+import com.baomidou.mybatisplus.extension.service.IService;
+import com.zy.asrs.framework.common.SpringUtils;
+
+import java.io.Serializable;
+import java.util.function.Function;
+
+/**
+ * Created by vincent on 2021/1/19
+ */
+public class NodeUtils {
+
+ public StringBuilder path = new StringBuilder();
+
+ public StringBuilder pathName = new StringBuilder();
+
+ public interface NodeSupport<T> {
+ T query();
+ }
+
+ public <T, R extends Serializable> void generatePath(Class<T> cls
+ , T t
+ , Function<? super T, ? extends R> idMapper
+ , Function<? super T, ? extends String> nameMapper
+ , Function<? super T, ? extends R> parentIdMapper) {
+ IService<T> bean = (IService<T>) SpringUtils.getBean(cls.getSimpleName() + "Service");
+ T parent = bean.getById(parentIdMapper.apply(t));
+ if (null != parent) {
+ path.insert(0, idMapper.apply(parent)).insert(0,",");
+ pathName.insert(0, nameMapper.apply(parent)).insert(0,",");
+ if (parentIdMapper.apply(parent) != null) {
+ generatePath(cls, parent, idMapper, nameMapper, parentIdMapper);
+ } else {
+ path.deleteCharAt(0);
+ pathName.deleteCharAt(0);
+ }
+ }
+ }
+
+}
--
Gitblit v1.9.1