mirror of
https://gitee.com/chinabugotech/hutool.git
synced 2025-05-09 23:51:34 +08:00
add ResourceFinder
This commit is contained in:
parent
b6bae921ba
commit
6345b1d054
@ -12,12 +12,12 @@
|
|||||||
|
|
||||||
package org.dromara.hutool.core.io.resource;
|
package org.dromara.hutool.core.io.resource;
|
||||||
|
|
||||||
|
import org.dromara.hutool.core.classloader.ClassLoaderUtil;
|
||||||
import org.dromara.hutool.core.collection.iter.EnumerationIter;
|
import org.dromara.hutool.core.collection.iter.EnumerationIter;
|
||||||
import org.dromara.hutool.core.compress.ZipUtil;
|
import org.dromara.hutool.core.compress.ZipUtil;
|
||||||
import org.dromara.hutool.core.exception.HutoolException;
|
import org.dromara.hutool.core.exception.HutoolException;
|
||||||
import org.dromara.hutool.core.io.IORuntimeException;
|
import org.dromara.hutool.core.io.IORuntimeException;
|
||||||
import org.dromara.hutool.core.io.IoUtil;
|
import org.dromara.hutool.core.io.IoUtil;
|
||||||
import org.dromara.hutool.core.io.file.FileNameUtil;
|
|
||||||
import org.dromara.hutool.core.io.file.FileUtil;
|
import org.dromara.hutool.core.io.file.FileUtil;
|
||||||
import org.dromara.hutool.core.net.url.UrlUtil;
|
import org.dromara.hutool.core.net.url.UrlUtil;
|
||||||
import org.dromara.hutool.core.text.AntPathMatcher;
|
import org.dromara.hutool.core.text.AntPathMatcher;
|
||||||
@ -34,10 +34,33 @@ import java.util.jar.JarFile;
|
|||||||
import java.util.zip.ZipException;
|
import java.util.zip.ZipException;
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* 资源查找器
|
* 资源查找器<br>
|
||||||
|
* <p>
|
||||||
|
* 参考Spring的PathMatchingResourcePatternResolver,实现classpath资源查找,利用{@link AntPathMatcher}筛选资源。
|
||||||
|
*
|
||||||
|
* @author Spring, Looly
|
||||||
*/
|
*/
|
||||||
public class ResourceFinder {
|
public class ResourceFinder {
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 构建新的ResourceFinder,使用当前环境的类加载器
|
||||||
|
*
|
||||||
|
* @return ResourceFinder
|
||||||
|
*/
|
||||||
|
public static ResourceFinder of() {
|
||||||
|
return of(ClassLoaderUtil.getClassLoader());
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 构建新的ResourceFinder
|
||||||
|
*
|
||||||
|
* @param classLoader 类加载器,用于限定查找范围
|
||||||
|
* @return ResourceFinder
|
||||||
|
*/
|
||||||
|
public static ResourceFinder of(final ClassLoader classLoader) {
|
||||||
|
return new ResourceFinder(classLoader);
|
||||||
|
}
|
||||||
|
|
||||||
private final ClassLoader classLoader;
|
private final ClassLoader classLoader;
|
||||||
private final AntPathMatcher pathMatcher;
|
private final AntPathMatcher pathMatcher;
|
||||||
|
|
||||||
@ -163,16 +186,16 @@ public class ResourceFinder {
|
|||||||
protected MultiResource findInDir(final FileResource resource, final String subPattern) {
|
protected MultiResource findInDir(final FileResource resource, final String subPattern) {
|
||||||
final MultiResource result = new MultiResource();
|
final MultiResource result = new MultiResource();
|
||||||
final File rootDir = resource.getFile();
|
final File rootDir = resource.getFile();
|
||||||
if(!rootDir.exists() || !rootDir.isDirectory() || !rootDir.canRead()){
|
if (!rootDir.exists() || !rootDir.isDirectory() || !rootDir.canRead()) {
|
||||||
// 保证给定文件存在、为目录且可读
|
// 保证给定文件存在、为目录且可读
|
||||||
return result;
|
return result;
|
||||||
}
|
}
|
||||||
|
|
||||||
final String rootPath = rootDir.getAbsolutePath();
|
final String fullPattern = replaceBackSlash(rootDir.getAbsolutePath() + StrUtil.SLASH + subPattern);
|
||||||
final String fullPattern = FileNameUtil.normalize(rootPath + StrUtil.SLASH + subPattern);
|
|
||||||
FileUtil.walkFiles(rootDir, (file -> {
|
FileUtil.walkFiles(rootDir, (file -> {
|
||||||
final String currentPath = StrUtil.normalize(file.getAbsolutePath());
|
final String currentPath = replaceBackSlash(file.getAbsolutePath());
|
||||||
if(file.isDirectory()){
|
if (file.isDirectory()) {
|
||||||
// 检查目录是否满足表达式开始规则,满足则继续向下查找,否则跳过
|
// 检查目录是否满足表达式开始规则,满足则继续向下查找,否则跳过
|
||||||
return pathMatcher.matchStart(fullPattern, StrUtil.addSuffixIfNot(currentPath, StrUtil.SLASH));
|
return pathMatcher.matchStart(fullPattern, StrUtil.addSuffixIfNot(currentPath, StrUtil.SLASH));
|
||||||
}
|
}
|
||||||
@ -205,4 +228,13 @@ public class ResourceFinder {
|
|||||||
}
|
}
|
||||||
return location.substring(0, rootDirEnd);
|
return location.substring(0, rootDirEnd);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 替换'\'为'/'
|
||||||
|
* @param path 路径
|
||||||
|
* @return 替换后的路径
|
||||||
|
*/
|
||||||
|
private static String replaceBackSlash(final String path){
|
||||||
|
return StrUtil.isEmpty(path) ? path : path.replace(CharUtil.BACKSLASH, CharUtil.SLASH);
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
@ -0,0 +1,40 @@
|
|||||||
|
/*
|
||||||
|
* Copyright (c) 2023. looly(loolly@aliyun.com)
|
||||||
|
* Hutool is licensed under Mulan PSL v2.
|
||||||
|
* You can use this software according to the terms and conditions of the Mulan PSL v2.
|
||||||
|
* You may obtain a copy of Mulan PSL v2 at:
|
||||||
|
* https://license.coscl.org.cn/MulanPSL2
|
||||||
|
* THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND,
|
||||||
|
* EITHER EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT,
|
||||||
|
* MERCHANTABILITY OR FIT FOR A PARTICULAR PURPOSE.
|
||||||
|
* See the Mulan PSL v2 for more details.
|
||||||
|
*/
|
||||||
|
|
||||||
|
package org.dromara.hutool.core.io.resource;
|
||||||
|
|
||||||
|
import org.dromara.hutool.core.lang.Console;
|
||||||
|
import org.junit.jupiter.api.Disabled;
|
||||||
|
import org.junit.jupiter.api.Test;
|
||||||
|
|
||||||
|
public class ResourceFinderTest {
|
||||||
|
@Test
|
||||||
|
@Disabled
|
||||||
|
void findAllTest() {
|
||||||
|
final MultiResource resources = ResourceFinder.of()
|
||||||
|
.find("**/*");
|
||||||
|
|
||||||
|
Console.log("===== result =====");
|
||||||
|
for (final Resource resource : resources) {
|
||||||
|
Console.log(resource);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
@Disabled
|
||||||
|
void getResourcesTest() {
|
||||||
|
final MultiResource resources = ResourceUtil.getResources("");
|
||||||
|
for (final Resource resource : resources) {
|
||||||
|
Console.log(resource);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
Loading…
x
Reference in New Issue
Block a user