diff --git a/src/main/java/com/kwan/springbootkwan/utils/FileContainsUtil.java b/src/main/java/com/kwan/springbootkwan/utils/FileContainsUtil.java index 1bcb4f1bcadfe2f9863e721af3a0c987dede59f8..676c631cda437746056a46571e63683cbcbff0d7 100644 --- a/src/main/java/com/kwan/springbootkwan/utils/FileContainsUtil.java +++ b/src/main/java/com/kwan/springbootkwan/utils/FileContainsUtil.java @@ -1,6 +1,5 @@ package com.kwan.springbootkwan.utils; - import cn.hutool.core.collection.CollectionUtil; import cn.hutool.core.date.StopWatch; import com.alibaba.fastjson2.JSON; @@ -10,6 +9,7 @@ import com.google.common.io.Files; import java.io.File; import java.io.IOException; import java.util.ArrayList; +import java.util.Arrays; import java.util.List; /** @@ -20,7 +20,6 @@ import java.util.List; * @date : 2023/2/8 10:45 */ public class FileContainsUtil { - /** * 默认不存在 */ @@ -30,13 +29,21 @@ public class FileContainsUtil { */ private static final String PIC_PATH = "/Users/qinyingjie/Documents/idea-workspace/blogimg/"; /** - * 博客路径 + * 博客路径1 + */ + private static final String BLOG_FOLDER1 = "/Users/qinyingjie/Documents/idea-workspace/blog/"; + /** + * 博客路径2 */ - private static final String BLOG_FOLDER = "/Users/qinyingjie/Documents/idea-workspace/blog/"; + private static final String BLOG_FOLDER2 = "/Users/qinyingjie/Documents/idea-workspace/study/blog/"; /** * 白名单 */ - private static final String[] WHITELISTS = {"http://qinyingjie.top/blogImg/image-20230324112725149.png"}; + private static final List WHITELISTS = Arrays.asList( + "http://qinyingjie.top/blogImg/image-20230324112725149.png" + , "http://qinyingjie.top/blogImg/logo.png" + , "http://qinyingjie.top/blogImg/image-20230601124308164.png" + ); public static void main(String[] args) throws Exception { StopWatch stopWatch = new StopWatch(); @@ -49,14 +56,7 @@ public class FileContainsUtil { if (CollectionUtil.isNotEmpty(picNames)) { for (String picName : picNames) { //是白名单里面的图片,直接忽略 - boolean isWhite = false; - for (String whitelist : WHITELISTS) { - if (whitelist.contains(picName)) { - isWhite = true; - break; - } - } - if (isWhite) { + if (WHITELISTS.contains(picName)) { continue; } //默认不存在 @@ -65,7 +65,8 @@ public class FileContainsUtil { List suffix = new ArrayList<>(); suffix.add(".md"); //包含某个字符串 - traverseFolder(BLOG_FOLDER, suffix, picName); + traverseFolder(BLOG_FOLDER1, suffix, picName); + traverseFolder(BLOG_FOLDER2, suffix, picName); //文件不存在 if (!IS_EXIST) { isNotExist.add(picName); @@ -83,7 +84,7 @@ public class FileContainsUtil { * 获取文件 * * @param path - * @param suffix + * @param suffixs * @param word * @throws IOException */