ソースを参照

:zap: 简化代码

Aron 6 年 前
コミット
c6e5ba0122

+ 6 - 11
src/main/java/com/ifast/common/utils/BuildTree.java

@@ -70,17 +70,12 @@ public class BuildTree {
 		return topNodes;
 	}
 
-    private static <T> void children(List<Tree<T>> nodes, Tree<T> children, String pid) {
-        for (Tree<T> parent : nodes) {
-            String id = parent.getId();
-            if (id != null && id.equals(pid)) {
-                parent.getChildren().add(children);
-                children.setHasParent(true);
-                parent.setChildren(true);
-
-                continue;
-            }
-        }
+    private static <T> void children(List<Tree<T>> nodes, final Tree<T> children, final String pId) {
+		nodes.stream().filter(node -> node.getId() != null && node.getId().equals(pId)).forEach(node -> {
+			node.getChildren().add(children);
+			node.setHasParent(true);
+			node.setChildren(true);
+		});
     }
 
 }

+ 4 - 4
src/main/java/com/ifast/common/utils/FileType.java

@@ -16,7 +16,7 @@ public class FileType {
             // 获取文件后缀名并转化为写,用于后续比较
             String fileType = fileName.substring(fileName.lastIndexOf(".") + 1, fileName.length()).toLowerCase();
             // 创建图片类型数组0
-            String[] img = {"bmp", "jpg", "jpeg", "png", "tiff", "gif", "pcx", "tga", "exif", "fpx", "svg", "psd", "cdr", "pcd", "dxf", "ufo", "eps", "ai", "raw", "wmf"};
+            String[] img = {"bmp", "jpg", "jpeg", "png", "fpx", "svg", "psd", "cdr", "pcd", "tiff", "gif", "pcx", "tga", "exif", "dxf", "ufo", "eps", "ai", "raw", "wmf"};
             for (int i = 0; i < img.length; i++) {
                 if (img[i].equals(fileType)) {
                     return 0;
@@ -24,21 +24,21 @@ public class FileType {
             }
 
             // 创建文档类型数组1
-            String[] document = {"txt", "doc", "docx", "xls", "htm", "html", "jsp", "rtf", "wpd", "pdf", "ppt"};
+            String[] document = {"txt", "htm", "html", "jsp", "rtf", "doc", "docx", "xls", "wpd", "pdf", "ppt"};
             for (int i = 0; i < document.length; i++) {
                 if (document[i].equals(fileType)) {
                     return 1;
                 }
             }
             // 创建视频类型数组2
-            String[] video = {"mp4", "avi", "mov", "wmv", "asf", "navi", "3gp", "mkv", "f4v", "rmvb", "webm"};
+            String[] video = {"mp4", "avi", "navi", "3gp", "mkv", "f4v", "mov", "wmv", "asf", "rmvb", "webm"};
             for (int i = 0; i < video.length; i++) {
                 if (video[i].equals(fileType)) {
                     return 2;
                 }
             }
             // 创建音乐类型数组3
-            String[] music = {"mp3", "wma", "wav", "mod", "ra", "cd", "md", "asf", "aac", "vqf", "ape", "mid", "ogg", "m4a", "vqf"};
+            String[] music = {"mp3", "wma", "wav", "mod", "asf", "aac", "vqf", "ape", "mid", "ogg", "m4a", "ra", "cd", "md", "vqf"};
             for (int i = 0; i < music.length; i++) {
                 if (music[i].equals(fileType)) {
                     return 3;