添加章节树形结构

sstudy/src/main/java/com/xhkjedu/sstudy/controller/subjectbook/DirectorController.java

@@ -26,83 +26,86 @@ public class DirectorController {
     private DirectorService directorService;
 
     /**
-     *功能描述  教材下章节树形结构
+     * 功能描述  教材下章节树形结构
+     *
+     * @param * @param director
+     * @return com.xhkjedu.vo.ResultVo
      * @author WN
      * @date 2020/6/3
-     * @param  * @param director
-     * @return com.xhkjedu.vo.ResultVo
      */
     @PostMapping("/list")
-    public ResultVo list(@RequestBody TDirector director){
-        N_Utils.validation(new Object[]{director.getLsbid(),"教材id",1});
+    public ResultVo list(@RequestBody TDirector director) {
+        N_Utils.validation(new Object[]{director.getLsbid(), "教材id", 1});
         List<TDirector> lst = directorService.lisTDirectors(director);
-        return new ResultVo(0,"获取教材下章节成功",lst);
+        return new ResultVo(0, "获取教材下章节成功", lst);
     }
 
     /**
-     *功能描述  添加
+     * 功能描述  添加
+     *
+     * @param * @param director
+     * @return com.xhkjedu.vo.ResultVo
      * @author WN
      * @date 2020/6/3
-     * @param  * @param director
-     * @return com.xhkjedu.vo.ResultVo
      */
     @PostMapping("/add")
-    public ResultVo add(@RequestBody TDirector director){
-        N_Utils.validation(new Object[]{director.getDirectorname(),"章节名称",2,director.getLsbid(),"教材id",1,
-        director.getBelong(),"章节归属",1,director.getDirectorlevel(),"归属层级",1});
+    public ResultVo add(@RequestBody TDirector director) {
+        N_Utils.validation(new Object[]{director.getDirectorname(), "章节名称", 2, director.getLsbid(), "教材id", 1,
+                director.getBelong(), "章节归属", 1, director.getDirectorlevel(), "归属层级", 1});
         if (director.getDirectorlevel() > 4) {
-            return new ResultVo(1,"超过层级限制");
+            return new ResultVo(1, "超过层级限制");
         }
         int result = directorService.addDirector(director);
         if (result > 0) {
             return new ResultVo(0, "保存成功");
-        }else{
+        } else {
             return new ResultVo(1, "保存失败");
         }
     }
 
     /**
-     *功能描述  修改章节
+     * 功能描述  修改章节
+     *
+     * @param * @param director
+     * @return com.xhkjedu.vo.ResultVo
      * @author WN
      * @date 2020/6/1
-     * @param  * @param director
-     * @return com.xhkjedu.vo.ResultVo
      */
     @PostMapping("/update")
-    public ResultVo update(@RequestBody TDirector director){
-        N_Utils.validation(new Object[]{director.getDirectorid(),"章节id",2,director.getDirectorname(),"章节名称",2});
+    public ResultVo update(@RequestBody TDirector director) {
+        N_Utils.validation(new Object[]{director.getDirectorid(), "章节id", 2, director.getDirectorname(), "章节名称", 2});
         int result = directorService.updateDirector(director);
         if (result > 0) {
             directorService.updateDirectorAll(director.getDirectorid());
             return new ResultVo(0, "修改成功");
-        }
-        else return new ResultVo(1,"修改失败");
+        } else return new ResultVo(1, "修改失败");
     }
 
     /**
-     *功能描述 删除
+     * 功能描述 删除
+     *
+     * @param * @param director
+     * @return com.xhkjedu.vo.ResultVo
      * @author WN
      * @date 2020/6/3
-     * @param  * @param director
-     * @return com.xhkjedu.vo.ResultVo
      */
     @PostMapping("/del")
-    public ResultVo del(@RequestBody TDirector director){
-        N_Utils.validation(new Object[]{director.getDirectorid(),"章节id",2});
+    public ResultVo del(@RequestBody TDirector director) {
+        N_Utils.validation(new Object[]{director.getDirectorid(), "章节id", 2});
         String useModule = directorService.findUseModule(director.getDirectorid());
         if (N_Utils.isNotEmpty(useModule)) {
 //            return new ResultVo(1,"使用章节模块【"+useModule+"】，禁止删除");
             String[] strs = useModule.split(",");
             for (String str : strs) {
-                if (N_Utils.isNotEmpty(str)){
+                if (N_Utils.isNotEmpty(str)) {
                     useModule = str;
                     break;
                 }
             }
-            return new ResultVo(1,"此章节下已有"+useModule+"，禁止删除");
+            return new ResultVo(1, "此章节下已有" + useModule + "，禁止删除");
         }
         int result = directorService.delDirector(director);
-        if (result > 0) return new ResultVo(0,"删除成功");
-        else return new ResultVo(1,"删除失败");
+        if (result > 0) return new ResultVo(0, "删除成功");
+        else return new ResultVo(1, "删除失败");
     }
 }

sstudy/src/main/java/com/xhkjedu/sstudy/service/subjectbook/DirectorService.java

@@ -11,6 +11,8 @@ import org.springframework.stereotype.Service;
 
 import java.util.ArrayList;
 import java.util.List;
+import java.util.Objects;
+import java.util.stream.Collectors;
 
 /**
  * @ClassName DirectorService
@@ -24,13 +26,14 @@ public class DirectorService {
     private DirectorMapper directorMapper;
 
     /**
-     *功能描述 新增
+     * 功能描述 新增
+     *
+     * @param * @param director
+     * @return java.lang.Integer
      * @author WN
      * @date 2020/6/3
-     * @param  * @param director
-     * @return java.lang.Integer
      */
-    public Integer addDirector(TDirector director){
+    public Integer addDirector(TDirector director) {
         String id = StudyUtil.getId();
         director.setDirectorid(id);
         if ("".equals(director.getDirectorpid())) {
@@ -42,12 +45,13 @@ public class DirectorService {
 
     /**
      * 添加并返回主键
+     *
+     * @return java.lang.Integer
      * @Param [director]
      * @Author ywx
      * @Date 2020/7/6 15:00
-     * @return java.lang.Integer
      **/
-    public String addAndReturnKey(TDirector director){
+    public String addAndReturnKey(TDirector director) {
         String id = StudyUtil.getId();
         director.setDirectorid(id);
         director.setCreatetime(N_Utils.getSecondTimestamp());
@@ -56,42 +60,45 @@ public class DirectorService {
     }
 
     /**
-     *功能描述 修改
+     * 功能描述 修改
+     *
+     * @param * @param director
+     * @return java.lang.Integer
      * @author WN
      * @date 2020/6/3
-     * @param  * @param director
-     * @return java.lang.Integer
      */
-    public Integer updateDirector(TDirector director){
+    public Integer updateDirector(TDirector director) {
         director.setCreatetime(N_Utils.getSecondTimestamp());
         return directorMapper.updateByPrimaryKeySelective(director);
     }
 
     /**
-     *功能描述  删除
+     * 功能描述  删除
+     *
+     * @param * @param director
+     * @return java.lang.Integer
      * @author WN
      * @date 2020/6/3
-     * @param  * @param director
-     * @return java.lang.Integer
      */
-    public Integer delDirector(TDirector director){
+    public Integer delDirector(TDirector director) {
         return directorMapper.deleteByPrimaryKey(director.getDirectorid());
     }
 
     /**
-     *功能描述  章节树形结构
+     * 功能描述  章节树形结构
+     *
+     * @param * @param director
+     * @return java.util.List<com.xhkjedu.model.subjectbook.TDirector>
      * @author WN
      * @date 2020/6/3
-     * @param  * @param director
-     * @return java.util.List<com.xhkjedu.model.subjectbook.TDirector>
      */
-    public List<TDirector> lisTDirectors(TDirector director){
+    public List<TDirector> lisTDirectors(TDirector director) {
         List<TDirector> lst = directorMapper.listDirectors(director);
 
         List<TDirector> directors = new ArrayList<>();
-        for (TDirector d:lst){
-            if(d.getDirectorpid() == null){
-                List<TDirector> childDirectors = listChildedirector(d,lst);
+        for (TDirector d : lst) {
+            if (d.getDirectorpid() == null) {
+                List<TDirector> childDirectors = listChildedirector(d, lst);
                 d.setChildren(childDirectors);
                 directors.add(d);
             }
@@ -101,11 +108,11 @@ public class DirectorService {
     }
 
     //获取子章节
-    private List<TDirector> listChildedirector(TDirector director,List<TDirector> directors){
+    private List<TDirector> listChildedirector(TDirector director, List<TDirector> directors) {
         List<TDirector> rtndirectors = new ArrayList<>();
-        for(TDirector d : directors){
-            if(director.getDirectorid().equals(d.getDirectorpid())){
-                List<TDirector> childDirectors = listChildedirector(d,directors);
+        for (TDirector d : directors) {
+            if (director.getDirectorid().equals(d.getDirectorpid())) {
+                List<TDirector> childDirectors = listChildedirector(d, directors);
                 d.setChildren(childDirectors);
                 rtndirectors.add(d);
             }
@@ -113,12 +120,39 @@ public class DirectorService {
         return rtndirectors;
     }
 
+    //2022-03-16使用stream递归
+    public List<TDirector> lisTDirectors2(TDirector director) {
+        List<TDirector> lst = directorMapper.listDirectors(director);
+
+        List<TDirector> collect = lst.stream().filter(d -> d.getDirectorpid() == null).map(
+                (d) -> {
+                    d.setChildren(getChildrens(d, lst));
+                    return d;
+                }
+        ).collect(Collectors.toList());
+        return collect;
+    }
+
+    //递归查询子节点
+    private List<TDirector> getChildrens(TDirector root, List<TDirector> all) {
+        List<TDirector> children = all.stream().filter(d -> {
+            return Objects.equals(d.getDirectorpid(), root.getDirectorid());
+        }).map(
+                (d) -> {
+                    d.setChildren(getChildrens(d, all));
+                    return d;
+                }
+        ).collect(Collectors.toList());
+        return children;
+    }
+
     /**
      * 教师收藏教材下的章节数量
+     *
+     * @return java.lang.Integer
      * @Param [teacherid, lsbid]
      * @Author ywx
      * @Date 2020/7/8 14:49
-     * @return java.lang.Integer
      **/
     public Integer countByTeacherIdLsbId(Integer teacherid, String lsbid) {
         return directorMapper.countByTeacherIdLsbId(teacherid, lsbid);
@@ -135,10 +169,11 @@ public class DirectorService {
 
     /**
      * 修改层级章节
+     *
+     * @return void
      * @Param [directorid, directorname]
      * @Author ywx
      * @Date 2022/3/14 15:14
-     * @return void
      **/
     public void updateDirectorAll(String directorid) {
         DirectorVo vo = directorMapper.getNameAndLevelById(directorid);