导读

由于项目需要，需要写一个工具类生成思维脑图。

使用

工具类

引用依赖

  1. <dependency>
  2.      <groupId>com.github.eljah</groupId>
  3.      <artifactId>xmindjbehaveplugin</artifactId>
  4.      <version>0.8</version>
  5. </dependency>
  8. <dependency>
  9.      <groupId>com.google.guava</groupId>
  10.      <artifactId>guava</artifactId>
  11.      <version>29.0-jre</version>
  12. </dependency>

工具类Tutis

  1. import cn.hutool.system.SystemUtil;
  2. import com.google.common.collect.Lists;
  3. import org.xmind.core.*;
  5. import java.io.IOException;
  6. import java.util.ArrayList;
  7. import java.util.List;
  9. /**
  10.  * 生成思维脑图工具类
  11.  *
  12.  * @date 2020/10/16
  13.  */
  14. public class Tutis {
  15.     /**
  16.      * 当前类路径
  17.      */
  18.     public static final String CLASS_PATH = GeneratorDoubanXmind.class.getResource("/").getPath();
  19.     /**
  20.      * 文件分隔符
  21.      */
  22.     public static final String FILE_SEPARATOR = SystemUtil.getOsInfo().getFileSeparator();
  24.     /**
  25.      * 生成思维脑图
  26.      *
  27.      * @param list     集合数据
  28.      * @param topicName 中心主题名称
  29.      * @param fileName 文件名.xmind
  30.      * @throws IOException
  31.      * @throws CoreException
  32.      */
  33.     public void exportXmind(List<Tree> list,String topicName, String fileName) throws IOException, CoreException {
  34.         // 创建思维导图的工作空间
  35.         IWorkbookBuilder workbookBuilder = Core.getWorkbookBuilder();
  36.         IWorkbook workbook = workbookBuilder.createWorkbook();
  38.         // 获得默认sheet
  39.         ISheet primarySheet = workbook.getPrimarySheet();
  41.         // 获得根主题
  42.         ITopic rootTopic = primarySheet.getRootTopic();
  44.         // 设置根主题的标题
  45.         rootTopic.setTitleText(topicName);
  47.         // 章节 topic 的列表
  48.         ArrayList<ITopic> chapterTopics = Lists.newArrayList();
  49.         recycle(chapterTopics, workbook, list);
  51.         // 把章节节点添加到要节点上
  52.         chapterTopics.forEach(it -> rootTopic.add(it, ITopic.ATTACHED));
  54.         // 保存
  55.         workbook.save(CLASS_PATH + FILE_SEPARATOR + fileName + ".xmind");
  56.     }
  58.     /**
  59.      * 思维导图递归调用
  60.      *
  61.      * @param chapterTopics
  62.      * @param workbook
  63.      * @param childrens     集合数据
  64.      */
  65.     public void recycle(ArrayList<ITopic> chapterTopics, IWorkbook workbook, List<Tree> childrens) {
  66.         for (Tree children : childrens) {
  67.             if (children != null) {
  68.                 if (children.getChildrens() != null && children.getChildrens().size() > 0) {
  69.                     // 创建章节节点
  70.                     ITopic topic = workbook.createTopic();
  71.                     topic.setTitleText(children.getName());
  72.                     chapterTopics.add(topic);
  73.                     List<Tree> list = children.getChildrens();
  74.                     //如果还有子节点，那么采用地柜调用继续生成
  75.                     if (list != null && list.size() > 0) {
  76.                         recycle(chapterTopics, workbook, list);
  77.                     } else {
  78.                         //否则的结束当前循环
  79.                         break;
  80.                     }
  81.                 } else {
  82.                     // 创建小节节点
  83.                     ITopic topic = workbook.createTopic();
  84.                     topic.setTitleText(children.getName());
  85.                     chapterTopics.get(chapterTopics.size() - 1).add(topic, ITopic.ATTACHED);
  86.                 }
  87.             }
  88.         }
  89.     }
  91. }

树状实体类Tree

  1. import java.util.List;
  3. /**
  4.  * 树状类
  5.  */
  6. public class Tree {
  7.     /**
  8.      * id
  9.      */
  10.     private int id;
  12.     /**
  13.      * 标题
  14.      */
  15.     private String name;
  17.     /**
  18.      * 父级Id
  19.      */
  20.     private int parentId;
  22.     /**
  23.      * 子集
  24.      */
  25.     private List<Tree> childrens;
  27.     public int getId() {
  28.         return id;
  29.     }
  31.     public void setId(int id) {
  32.         this.id = id;
  33.     }
  35.     public String getName() {
  36.         return name;
  37.     }
  39.     public void setName(String name) {
  40.         this.name = name;
  41.     }
  43.     public int getParentId() {
  44.         return parentId;
  45.     }
  47.     public void setParentId(int parentId) {
  48.         this.parentId = parentId;
  49.     }
  51.     public List<Tree> getChildrens() {
  52.         return childrens;
  53.     }
  55.     public void setChildrens(List<Tree> childrens) {
  56.         this.childrens = childrens;
  57.     }
  59.     public Tree(int id, String name, int parentId) {
  60.         this.id = id;
  61.         this.name = name;
  62.         this.parentId = parentId;
  63.     }
  65.     @Override
  66.     public String toString() {
  67.         return "Tree [id=" + id + ", name=" + name + ", parentId=" + parentId
  68.                 + ", childrens=" + childrens + "]";
  69.     }
  70. }

测试数据

  1. import java.util.ArrayList;
  2. import java.util.List;
  4. public class Tes {
  5.     public static void main(String[] args) {
  6.         Tree tree1 = new Tree(1, "顶层节点1", 0);
  7.         Tree tree2 = new Tree(2, "顶层节点2", 0);
  8.         Tree tree3 = new Tree(3, "顶层节点3", 0);
  10.         Tree tree4 = new Tree(4, "二级节点4", 1);
  11.         Tree tree5 = new Tree(5, "二级节点5", 2);
  12.         Tree tree6 = new Tree(6, "二级节点6", 3);
  14.         Tree tree7 = new Tree(7, "三级节点7", 4);
  15.         Tree tree8 = new Tree(8, "三级节点8", 4);
  16.         Tree tree9 = new Tree(9, "三级节点9", 5);
  18.         Tree m7 = new Tree(10, "M7", 4);
  19.         Tree m8 = new Tree(11, "M8", 4);
  20.         Tree m9 = new Tree(12, "M9", 5);
  22.         Tree k7 = new Tree(13, "k7", 10);
  23.         Tree k8 = new Tree(14, "k8", 10);
  24.         Tree k9 = new Tree(15, "k9", 5);
  26.         List<Tree> trees = new ArrayList<Tree>();
  27.         trees.add(tree9);
  28.         trees.add(tree8);
  29.         trees.add(tree7);
  30.         trees.add(tree6);
  31.         trees.add(tree5);
  32.         trees.add(tree4);
  33.         trees.add(tree3);
  34.         trees.add(tree2);
  35.         trees.add(tree1);
  36.         trees.add(m7);
  37.         trees.add(m8);
  38.         trees.add(m9);
  39.         trees.add(k7);
  40.         trees.add(k8);
  41.         trees.add(k9);
  43.         List<Tree> rootTrees = new ArrayList<Tree>();
  44.         for (Tree tree : trees) {
  45.             if (tree.getParentId() == 0) {
  46.                 rootTrees.add(tree);
  47.             }
  48.             for (Tree t : trees) {
  49.                 if (t.getParentId() == tree.getId()) {
  50.                     if (tree.getChildrens() == null) {
  51.                         List<Tree> myChildrens = new ArrayList<Tree>();
  52.                         myChildrens.add(t);
  53.                         tree.setChildrens(myChildrens);
  54.                     } else {
  55.                         tree.getChildrens().add(t);
  56.                     }
  57.                 }
  58.             }
  59.         }
  61.         for (Tree tree : rootTrees) {
  62.             System.out.println(tree.toString());
  63.         }
  64.         try {
  65.             Tutis tutis = new Tutis();
  66.             tutis.exportXmind(rootTrees,"根节点描述","测试思维导出");
  67.         } catch (Exception e) {
  68.             e.printStackTrace();
  69.         }
  70.     }
  71. }

运行结果

控制台数据

  1. Tree [id=3, name=顶层节点3, parentId=0, childrens=[Tree [id=6, name=二级节点6, parentId=3, childrens=null]]]
  2. Tree [id=2, name=顶层节点2, parentId=0, childrens=[Tree [id=5, name=二级节点5, parentId=2, childrens=[Tree [id=9, name=三级节点9, parentId=5, childrens=null], Tree [id=12, name=M9, parentId=5, childrens=null], Tree [id=15, name=k9, parentId=5, childrens=null]]]]]
  3. Tree [id=1, name=顶层节点1, parentId=0, childrens=[Tree [id=4, name=二级节点4, parentId=1, childrens=[Tree [id=8, name=三级节点8, parentId=4, childrens=null], Tree [id=7, name=三级节点7, parentId=4, childrens=null], Tree [id=10, name=M7, parentId=4, childrens=[Tree [id=13, name=k7, parentId=10, childrens=null], Tree [id=14, name=k8, parentId=10, childrens=null]]], Tree [id=11, name=M8, parentId=4, childrens=null]]]]]

image.png

生成的脑图

image.png

END

参考：java生成思维导图

搞定~