用Java写一个输出纯文本的目录树

之前在用HUI的模板时看他的目录结构时他给了一个纯文本格式的目录结构,像这样.

以前不是这样的,老版本好看多了,反正搞个截图吧.后面自己搞一些项目的目录结构时,都是手搓控制格式.后面没事自己整了个目录扫描工具类,后面再看目录的结构什么的就一目了然了.

直接上自己写的一个目录扫描工具类吧:

package com.gxlm.temp.tools; import java.io.File; import java.util.ArrayList; import java.util.List; import org.gxlm.core.util.CollectionUtil; import org.gxlm.core.util.FileUtil; import org.gxlm.core.util.IOUtil; import org.gxlm.core.util.Resource; import org.gxlm.core.util.StringUtil; /** * Description 类描述: <br/> * 目录树 * * @version 创建时间：2019年10月23日 上午10:10:16 */ public class DirecTree { // 空白分隔符 private static final String BLANK_SPLIT = " "; // 竖线分隔符 private static final String VERTICAL_BLANK_SPLIT = "│ "; // 节点符 private static final String NODE = "├──"; // 文件图标 private static final String FILE_ICON = "▶ ○ "; // 非空目录图标 private static final String DIREC_ICON = "▼ ◇ "; // 空目录图标 private static final String EMPTY_DIREC_ICON = "▶ ◇ "; // 文件节点 private static String FILE_NODE = NODE; // 非空目录节点 private static String DIREC_NODE = NODE; // 空目录节点 private static String EMPTY_DIREC_NODE = NODE; // 扫描目录 private File direc; /** * 目录树 * * @version 日期:2019年10月25日 * @param direc 扫描目录 */ public DirecTree(String direc) { this(direc, false); } /** * 目录树 * * @version 日期:2019年10月25日 * @param direc 扫描目录 * @param showIcon 是否显示区分文件目录图标 */ public DirecTree(String direc, boolean showIcon) { this(new File(direc), showIcon); } /** * 目录树 * * @version 日期:2019年10月25日 * @param direc 扫描目录 */ public DirecTree(File direc) { this(direc, false); } /** * 目录树 * * @version 日期:2019年10月25日 * @param direc 扫描目录 * @param showIcon 是否显示区分文件目录图标 */ public DirecTree(File direc, boolean showIcon) { this.direc = direc; if (showIcon) { FILE_NODE = NODE + FILE_ICON; DIREC_NODE = NODE + DIREC_ICON; EMPTY_DIREC_NODE = NODE + EMPTY_DIREC_ICON; } } /** * 绘制目录树 * * @version 日期:2019年10月25日 */ public void draw() { drawDirecTree(direc, "", false, 1, null, null); } /** * 绘制目录树 * * @version 日期:2019年10月25日 * @param outFile 绘制树保存文件 * @param limitLevel 限制绘制深度 * @param likeEclipse 按eclipse的包结构绘制目录树(目录下只有一个子目录,子目录不占用一个分支) */ public void draw(File outFile, Integer limitLevel, boolean likeEclipse) { StringBuilder builder = outFile == null ? null : new StringBuilder(); if (likeEclipse) { File parentFile = direc.getParentFile(); if (parentFile == null)// 根目录 drawDirecTreeLikeEclipse(direc, null, "", false, 1, limitLevel, builder); else drawDirecTreeLikeEclipse(direc, parentFile.getAbsolutePath(), "", false, 1, limitLevel, builder); } else { drawDirecTree(direc, "", false, 1, limitLevel, builder); } if (builder != null) IOUtil.writeText(builder.toString(), outFile, "UTF-8", true); } /** * 绘制目录树-以eclipse的package视图风格显示目录文件 * * @version 日期:2019年3月19日 * @param file * @param basePath * @param prefix 节点前缀 * @param hasNext file是否有兄弟file * @param level 目录树深度 * @param limitLevel 限定显示目录树深度上限 * @param builder */ private void drawDirecTreeLikeEclipse(File file, String basePath, String prefix, boolean hasNext, int level, Integer limitLevel, StringBuilder builder) { if (limitLevel != null && level > limitLevel.intValue()) return; // 根目录路径以"\"字符结尾,而普通目录不以"\"字符结尾,这里统一下都不以"\"字符 if (StringUtil.isNotBlank(basePath) && basePath.lastIndexOf(File.separatorChar) == basePath.length() - 1) basePath = basePath.substring(0, basePath.length() - 1); // 文件 if (file.isFile()) { String nodeInfo = prefix + FILE_NODE + file.getName() + "(" + FileUtil.formatFileSize(file.length()) + ")"; if (builder == null) System.out.println(nodeInfo); else builder.append(nodeInfo).append("\r\n"); return; } File[] fileArr = file.listFiles(); // 空目录 if (CollectionUtil.isEmpty(fileArr)) { String nodeInfo = prefix + EMPTY_DIREC_NODE; if (StringUtil.isBlank(basePath)) nodeInfo += file.getAbsolutePath().replace(File.separator, file.getParentFile() == null? "": "."); else nodeInfo += file.getAbsolutePath().substring(basePath.length() + 1).replace(File.separatorChar, '.'); if (builder == null) System.out.println(nodeInfo); else builder.append(nodeInfo).append("\r\n"); return; } // 目录下有且仅有一个子目录 if (fileArr.length == 1 && fileArr[0].isDirectory()) { File f = fileArr[0]; while (true) { File[] tempFileArr = f.listFiles(); if (CollectionUtil.isEmpty(tempFileArr) || tempFileArr.length > 1 || tempFileArr[0].isFile()) break; f = tempFileArr[0]; } drawDirecTreeLikeEclipse(f, basePath, prefix, hasNext, level + 1, limitLevel, builder); return; } // 打印当前目录 String nodeInfo = prefix + DIREC_NODE; if (StringUtil.isBlank(basePath)) nodeInfo += file.getAbsolutePath().replace(File.separator, file.getParentFile() == null? "": "."); else nodeInfo += file.getAbsolutePath().substring(basePath.length() + 1).replace(File.separatorChar, '.'); if (builder == null) System.out.println(nodeInfo); else builder.append(nodeInfo).append("\r\n"); // 子目录及文件 String p = prefix + (hasNext ? VERTICAL_BLANK_SPLIT : BLANK_SPLIT); List<File> direcList = new ArrayList<File>(); for (File item : fileArr) { if (item.isFile()) drawDirecTreeLikeEclipse(item, file.getAbsolutePath(), p, false, level + 1, limitLevel, builder); else direcList.add(item); } if (CollectionUtil.isEmpty(direcList)) return; for (int i = 0; i < direcList.size(); i++) drawDirecTreeLikeEclipse(direcList.get(i), file.getAbsolutePath(), p, i != direcList.size() - 1, level + 1, limitLevel, builder); } /** * 绘制目录树-每个目录均占用一个位置(即只有一个子目录的也要展开) * * @version 日期:2019年10月23日 * @param file * @param prefix 节点前缀 * @param hasNext file是否有兄弟file * @param level 目录树深度 * @param limitLevel 限定显示目录树深度上限 * @param builder */ private void drawDirecTree(File file, String prefix, boolean hasNext, int level, Integer limitLevel, StringBuilder builder) { if (limitLevel != null && level > limitLevel.intValue()) return; // 文件 if (file.isFile()) { String nodeInfo = prefix + FILE_NODE + file.getName() + "(" + FileUtil.formatFileSize(file.length()) + ")"; if (builder == null) System.out.println(nodeInfo); else builder.append(nodeInfo).append("\r\n"); return; } File[] fileArr = file.listFiles(); // 空目录 if (CollectionUtil.isEmpty(fileArr)) { String nodeInfo = prefix + EMPTY_DIREC_NODE + (file.getParentFile() == null? file.getPath().replace(File.separator, ""): file.getName()); if (builder == null) System.out.println(nodeInfo); else builder.append(nodeInfo).append("\r\n"); return; } // 打印当前目录 String nodeInfo = prefix + DIREC_NODE + (file.getParentFile() == null? file.getPath().replace(File.separator, ""): file.getName()); if (builder == null) System.out.println(nodeInfo); else builder.append(nodeInfo).append("\r\n"); // 递归打印子目录及文件 String p = prefix + (hasNext ? VERTICAL_BLANK_SPLIT : BLANK_SPLIT); List<File> direcList = new ArrayList<File>(); for (File item : fileArr) { if (item.isFile()) drawDirecTree(item, p, false, level + 1, limitLevel, builder); else direcList.add(item); } if (CollectionUtil.isEmpty(direcList)) return; for (int i = 0; i < direcList.size(); i++) drawDirecTree(direcList.get(i), p, i != direcList.size() - 1, level + 1, limitLevel, builder); } /** * 测试绘制目树 * * @version 日期:2019年10月23日 * @param args */ public static void main(String[] args) { // 目录树,所有目录均展开,不限定目录深度,结果输出到控制台上 //new DirecStruct("E:\\Recently").draw(); // 目录树,像eclipse展示包结构一样展示目录树,不限定目录深度,结果输出到控制台上 // new DirecStruct("E:\\Recently").draw(null, null, true); // 目录树,像eclipse展示包结构一样展示目录树,不限定目录深度,结果输出到桌面上 new DirecTree("F:\\").draw(new File(Resource.getDesktopPath() + File.separatorChar + "西数-F盘.txt"), null, false); } }

生成效果:

FileUtil.formatFileSize(file.length())是格式化文件大小单位,这里也给下

public static String formatFileSize(long fileSize) { if (fileSize < 1024) return fileSize + "Byte"; else if (fileSize < 1024 * 1024) return ((long) (fileSize / 1024.0 * 100)) / 100.0 + "KB"; else if (fileSize < 1024 * 1024 * 1024) return ((long) (fileSize / 1048576.0 * 100)) / 100.0 + "MB"; else return ((long) (fileSize / 1073741824.0 * 100)) / 100.0 + "GB"; }