hfxc226 2 år sedan
förälder
incheckning
9a32182af8

+ 4 - 1
platform-dao/src/main/java/com/platform/dao/pojo/CommonTree.java

@@ -24,7 +24,10 @@ public class CommonTree extends TreeNode {
      * 名称
      */
     private String title;
-
+    /**
+     * 图片地址
+     */
+    private String img;
     /**
      * 图标
      */

+ 3 - 1
platform-dao/src/main/java/com/platform/dao/util/TreeUtil.java

@@ -21,6 +21,7 @@ import com.platform.dao.vo.query.qykh.InformationVO;
 import com.platform.dao.vo.query.qykh.PlanVO;
 import com.platform.dao.vo.query.qykh.ProductVO;
 import com.platform.dao.vo.query.store.StoreVO;
+import com.platform.dao.vo.sb.SbPositionVO;
 import com.platform.dao.vo.sb.SbTypeVO;
 import lombok.experimental.UtilityClass;
 
@@ -337,7 +338,7 @@ public class TreeUtil {
      * @param root
      * @return
      */
-    public List<CommonTree> buildSbPositionTree(List<SbPosition> types, String root) {
+    public List<CommonTree> buildSbPositionTree(List<SbPositionVO> types, String root) {
         List<CommonTree> trees = new LinkedList<>();
         types.forEach(type -> {
             CommonTree node = new CommonTree();
@@ -345,6 +346,7 @@ public class TreeUtil {
             node.setKey(type.getId());
             node.setParentId(type.getParentId());
             node.setTitle(type.getName());
+            node.setImg(type.getOpcImg());
             node.setValue(type.getId());
             trees.add(node);
         });

+ 1 - 1
platform-rest/src/main/java/com/platform/rest/controller/sb/SbPositionController.java

@@ -158,7 +158,7 @@ public class SbPositionController {
      */
     @GetMapping("tree")
     public R queryTree(SbPositionDTO sbPositionDTO) {
-        return new R<>(TreeUtil.buildSbPositionTree(sbPositionService.getModelListByDTO(sbPositionDTO), ""));
+        return new R<>(TreeUtil.buildSbPositionTree(sbPositionService.selectVOList(sbPositionDTO), ""));
     }
 
     /**