diff --git a/laws-modules/src/main/java/com/jero/modules/split/controller/SarFileSplitMenuEOController.java b/laws-modules/src/main/java/com/jero/modules/split/controller/SarFileSplitMenuEOController.java index 13e93f6c..febc83ee 100644 --- a/laws-modules/src/main/java/com/jero/modules/split/controller/SarFileSplitMenuEOController.java +++ b/laws-modules/src/main/java/com/jero/modules/split/controller/SarFileSplitMenuEOController.java @@ -15,7 +15,10 @@ import org.slf4j.LoggerFactory; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.web.bind.annotation.*; +import java.util.ArrayList; +import java.util.HashMap; import java.util.List; +import java.util.Map; @RestController @RequestMapping("/split/sarFileSplitMenu") @@ -51,9 +54,40 @@ public class SarFileSplitMenuEOController extends JeroController getList = sarFileSplitMenuEOService.queryByList(page); + //转树结构 + List menuEOS = convertToTree(getList); + getList = menuEOS.get(0).getChildren(); return Result.OK(getList); } + public List convertToTree(List menuList) { + // 创建一个Map来保存id与SarFileSplitMenuEO对象的映射关系 + Map menuMap = new HashMap<>(); + for (SarFileSplitMenuEO menu : menuList) { + menuMap.put(menu.getId(), menu); + } + List rootNodes = new ArrayList<>(); + for (SarFileSplitMenuEO menu : menuList) { + String parentId = menu.getPId(); + if (parentId == null) { + // 如果pid为空,将其作为根节点添加到结果集 + rootNodes.add(menu); + } else { + // 如果pid不为空,将其作为子节点添加到父节点中 + SarFileSplitMenuEO parentMenu = menuMap.get(parentId); + if (parentMenu != null) { + List childList = parentMenu.getChildren(); + if (childList == null) { + childList = new ArrayList(); + parentMenu.setChildren(childList); + } + childList.add(menu); + } + } + } + return rootNodes; + } + @ApiOperation(value = "文档拆分条款目录-新增Y") @PostMapping("/addMenu") @RequiresPermissions("split:sarFileSplitMenu:add")