存货大类

platform-dao/src/main/java/com/platform/dao/pojo/CommonTree.java

@@ -35,6 +35,8 @@ public class CommonTree extends TreeNode {
 
     private String value;
 
+    private String code;
+
     public CommonTree() {
         setGroup(false);
     }

platform-dao/src/main/java/com/platform/dao/util/TreeUtil.java

@@ -9,6 +9,7 @@ import com.platform.dao.entity.part.PartType;
 import com.platform.dao.entity.sb.SbInfo;
 import com.platform.dao.entity.sb.SbPosition;
 import com.platform.dao.entity.sb.SbType;
+import com.platform.dao.entity.spareAdd.LongYanInventoryCode;
 import com.platform.dao.entity.sqarepartmanage.SpareType;
 import com.platform.dao.entity.store.Store;
 import com.platform.dao.entity.upms.SysDept;
@@ -295,6 +296,29 @@ public class TreeUtil {
         return TreeUtil.buildByLoop(trees, root);
     }
 
+    /**
+     * 通过sysMenu创建树形节点
+     *
+     * @param menus
+     * @param root
+     * @return
+     */
+    public List<CommonTree> buildInventoryCodeTree(List<LongYanInventoryCode> menus, String root) {
+        List<CommonTree> trees = new ArrayList<>();
+        menus.forEach(menu -> {
+            CommonTree node = new CommonTree();
+            node.setId(menu.getId());
+            node.setKey(menu.getId());
+            node.setParentId(menu.getParentId());
+            node.setCode(menu.getCode());
+            node.setTitle(menu.getName());
+            node.setItem(menu);
+            node.setValue(menu.getId());
+            trees.add(node);
+        });
+        return TreeUtil.buildByLoop(trees, root);
+    }
+
     /**
      * 通过sysMenu创建树形节点
      *

platform-rest/src/main/java/com/platform/rest/controller/spareAdd/LongYanInventoryCodeController.java

@@ -1,11 +1,11 @@
 package com.platform.rest.controller.spareAdd;
-
 import com.platform.common.bean.AbstractPageResultBean;
 import com.platform.common.util.R;
 import com.platform.common.validation.group.AddGroup;
 import com.platform.common.validation.group.UpdateGroup;
 import com.platform.dao.dto.spareAdd.LongYanInventoryCodeDTO;
 import com.platform.dao.entity.spareAdd.LongYanInventoryCode;
+import com.platform.dao.util.TreeUtil;
 import com.platform.dao.vo.query.spareAdd.LongYanInventoryCodeVO;
 import com.platform.rest.log.annotation.SysLog;
 import com.platform.service.spareAdd.LongYanInventoryCodeService;
@@ -117,4 +117,16 @@ public class LongYanInventoryCodeController {
       return new R<>(longYanInventoryCodeService.getModelListByDTO(longYanInventoryCodeDTO));
   }
 
+  /**
+   * 获取菜单树
+   * @param longYanInventoryCodeDTO ：
+   * @return ：
+   */
+  @GetMapping("/tree")
+  @SysLog("获取菜单树")
+  public R tree(LongYanInventoryCodeDTO longYanInventoryCodeDTO) {
+    return new R<>(TreeUtil.buildInventoryCodeTree(
+            longYanInventoryCodeService.getModelListByDTO(longYanInventoryCodeDTO), ""));
+  }
+
 }