mirror of
https://gitee.com/chinabugotech/hutool.git
synced 2025-05-09 23:51:34 +08:00
add setClassLoader
This commit is contained in:
Looly
parent
1ddc01453f
commit
476285b302
@ -8,6 +8,7 @@
|
||||
### 新特性
|
||||
* 【all】 修复注释中的错别字（issue#I12XE6@Gitee）
|
||||
* 【core】 CsvWriter支持其它类型的参数（issue#I12XE3@Gitee）
|
||||
* 【core】 ClassScaner支持自定义ClassLoader
|
||||
|
||||
### Bug修复
|
||||
* 【all】 修复阶乘计算错误bug（issue#I12XE4@Gitee）
|
||||
|
||||
@ -1,5 +1,12 @@
|
||||
package cn.hutool.core.lang;
|
||||
|
||||
import cn.hutool.core.collection.CollUtil;
|
||||
import cn.hutool.core.collection.EnumerationIter;
|
||||
import cn.hutool.core.io.FileUtil;
|
||||
import cn.hutool.core.io.IORuntimeException;
|
||||
import cn.hutool.core.io.resource.ResourceUtil;
|
||||
import cn.hutool.core.util.*;
|
||||
|
||||
import java.io.File;
|
||||
import java.io.IOException;
|
||||
import java.io.Serializable;
|
||||
@ -12,48 +19,54 @@ import java.util.Set;
|
||||
import java.util.jar.JarEntry;
|
||||
import java.util.jar.JarFile;
|
||||
|
||||
import cn.hutool.core.collection.CollUtil;
|
||||
import cn.hutool.core.collection.EnumerationIter;
|
||||
import cn.hutool.core.io.FileUtil;
|
||||
import cn.hutool.core.io.IORuntimeException;
|
||||
import cn.hutool.core.io.resource.ResourceUtil;
|
||||
import cn.hutool.core.util.CharUtil;
|
||||
import cn.hutool.core.util.CharsetUtil;
|
||||
import cn.hutool.core.util.ClassUtil;
|
||||
import cn.hutool.core.util.StrUtil;
|
||||
import cn.hutool.core.util.URLUtil;
|
||||
|
||||
/**
|
||||
* 类扫描器
|
||||
*
|
||||
*
|
||||
* @author looly
|
||||
* @since 4.1.5
|
||||
*
|
||||
*/
|
||||
public class ClassScaner implements Serializable{
|
||||
public class ClassScaner implements Serializable {
|
||||
private static final long serialVersionUID = 1L;
|
||||
|
||||
/** 包名 */
|
||||
/**
|
||||
* 包名
|
||||
*/
|
||||
private String packageName;
|
||||
/** 包名，最后跟一个点，表示包名，避免在检查前缀时的歧义 */
|
||||
/**
|
||||
* 包名，最后跟一个点，表示包名，避免在检查前缀时的歧义
|
||||
*/
|
||||
private String packageNameWithDot;
|
||||
/** 包路径，用于文件中对路径操作 */
|
||||
/**
|
||||
* 包路径，用于文件中对路径操作
|
||||
*/
|
||||
private String packageDirName;
|
||||
/** 包路径，用于jar中对路径操作，在Linux下与packageDirName一致 */
|
||||
/**
|
||||
* 包路径，用于jar中对路径操作，在Linux下与packageDirName一致
|
||||
*/
|
||||
private String packagePath;
|
||||
/** 过滤器 */
|
||||
/**
|
||||
* 过滤器
|
||||
*/
|
||||
private Filter<Class<?>> classFilter;
|
||||
/** 编码 */
|
||||
/**
|
||||
* 编码
|
||||
*/
|
||||
private Charset charset;
|
||||
/** 是否初始化类 */
|
||||
/**
|
||||
* 类加载器
|
||||
*/
|
||||
private ClassLoader classLoader;
|
||||
/**
|
||||
* 是否初始化类
|
||||
*/
|
||||
private boolean initialize;
|
||||
|
||||
private Set<Class<?>> classes = new HashSet<Class<?>>();
|
||||
|
||||
private Set<Class<?>> classes = new HashSet<>();
|
||||
|
||||
/**
|
||||
* 扫描指定包路径下所有包含指定注解的类
|
||||
*
|
||||
* @param packageName 包路径
|
||||
*
|
||||
* @param packageName 包路径
|
||||
* @param annotationClass 注解类
|
||||
* @return 类集合
|
||||
*/
|
||||
@ -68,9 +81,9 @@ public class ClassScaner implements Serializable{
|
||||
|
||||
/**
|
||||
* 扫描指定包路径下所有指定类或接口的子类或实现类
|
||||
*
|
||||
*
|
||||
* @param packageName 包路径
|
||||
* @param superClass 父类或接口
|
||||
* @param superClass 父类或接口
|
||||
* @return 类集合
|
||||
*/
|
||||
public static Set<Class<?>> scanPackageBySuper(String packageName, final Class<?> superClass) {
|
||||
@ -84,7 +97,7 @@ public class ClassScaner implements Serializable{
|
||||
|
||||
/**
|
||||
* 扫描该包路径下所有class文件
|
||||
*
|
||||
*
|
||||
* @return 类集合
|
||||
*/
|
||||
public static Set<Class<?>> scanPackage() {
|
||||
@ -93,7 +106,7 @@ public class ClassScaner implements Serializable{
|
||||
|
||||
/**
|
||||
* 扫描该包路径下所有class文件
|
||||
*
|
||||
*
|
||||
* @param packageName 包路径 com | com. | com.abs | com.abs.
|
||||
* @return 类集合
|
||||
*/
|
||||
@ -105,7 +118,7 @@ public class ClassScaner implements Serializable{
|
||||
* 扫描包路径下满足class过滤器条件的所有class文件，<br>
|
||||
* 如果包路径为 com.abs + A.class 但是输入 abs会产生classNotFoundException<br>
|
||||
* 因为className 应该为 com.abs.A 现在却成为abs.A,此工具类对该异常进行忽略处理<br>
|
||||
*
|
||||
*
|
||||
* @param packageName 包路径 com | com. | com.abs | com.abs.
|
||||
* @param classFilter class过滤器，过滤掉不需要的class
|
||||
* @return 类集合
|
||||
@ -123,7 +136,7 @@ public class ClassScaner implements Serializable{
|
||||
|
||||
/**
|
||||
* 构造，默认UTF-8编码
|
||||
*
|
||||
*
|
||||
* @param packageName 包名，所有包传入""或者null
|
||||
*/
|
||||
public ClassScaner(String packageName) {
|
||||
@ -132,7 +145,7 @@ public class ClassScaner implements Serializable{
|
||||
|
||||
/**
|
||||
* 构造，默认UTF-8编码
|
||||
*
|
||||
*
|
||||
* @param packageName 包名，所有包传入""或者null
|
||||
* @param classFilter 过滤器，无需传入null
|
||||
*/
|
||||
@ -142,10 +155,10 @@ public class ClassScaner implements Serializable{
|
||||
|
||||
/**
|
||||
* 构造
|
||||
*
|
||||
*
|
||||
* @param packageName 包名，所有包传入""或者null
|
||||
* @param classFilter 过滤器，无需传入null
|
||||
* @param charset 编码
|
||||
* @param charset 编码
|
||||
*/
|
||||
public ClassScaner(String packageName, Filter<Class<?>> classFilter, Charset charset) {
|
||||
packageName = StrUtil.nullToEmpty(packageName);
|
||||
@ -159,42 +172,51 @@ public class ClassScaner implements Serializable{
|
||||
|
||||
/**
|
||||
* 扫描包路径下满足class过滤器条件的所有class文件
|
||||
*
|
||||
*
|
||||
* @return 类集合
|
||||
*/
|
||||
public Set<Class<?>> scan() {
|
||||
for (URL url : ResourceUtil.getResourceIter(this.packagePath)) {
|
||||
switch (url.getProtocol()) {
|
||||
case "file":
|
||||
scanFile(new File(URLUtil.decode(url.getFile(), this.charset.name())), null);
|
||||
break;
|
||||
case "jar":
|
||||
scanJar(URLUtil.getJarFile(url));
|
||||
break;
|
||||
case "file":
|
||||
scanFile(new File(URLUtil.decode(url.getFile(), this.charset.name())), null);
|
||||
break;
|
||||
case "jar":
|
||||
scanJar(URLUtil.getJarFile(url));
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
if(CollUtil.isEmpty(this.classes)) {
|
||||
|
||||
if (CollUtil.isEmpty(this.classes)) {
|
||||
scanJavaClassPaths();
|
||||
}
|
||||
|
||||
|
||||
return Collections.unmodifiableSet(this.classes);
|
||||
}
|
||||
|
||||
/**
|
||||
* 设置是否在扫描到类时初始化类
|
||||
*
|
||||
*
|
||||
* @param initialize 是否初始化类
|
||||
*/
|
||||
public void setInitialize(boolean initialize) {
|
||||
this.initialize = initialize;
|
||||
}
|
||||
|
||||
/**
|
||||
* 设置自定义的类加载器
|
||||
*
|
||||
* @param classLoader 类加载器
|
||||
* @since 4.6.9
|
||||
*/
|
||||
public void setClassLoader(ClassLoader classLoader) {
|
||||
this.classLoader = classLoader;
|
||||
}
|
||||
|
||||
// --------------------------------------------------------------------------------------------------- Private method start
|
||||
|
||||
/**
|
||||
* 扫描Java指定的ClassPath路径
|
||||
*
|
||||
* @return 扫描到的类
|
||||
*/
|
||||
private void scanJavaClassPaths() {
|
||||
final String[] javaClassPaths = ClassUtil.getJavaClassPaths();
|
||||
@ -205,11 +227,11 @@ public class ClassScaner implements Serializable{
|
||||
scanFile(new File(classPath), null);
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
/**
|
||||
* 扫描文件或目录中的类
|
||||
*
|
||||
* @param file 文件或目录
|
||||
*
|
||||
* @param file 文件或目录
|
||||
* @param rootDir 包名对应classpath绝对路径
|
||||
*/
|
||||
private void scanFile(File file, String rootDir) {
|
||||
@ -238,7 +260,7 @@ public class ClassScaner implements Serializable{
|
||||
|
||||
/**
|
||||
* 扫描jar包
|
||||
*
|
||||
*
|
||||
* @param jar jar包
|
||||
*/
|
||||
private void scanJar(JarFile jar) {
|
||||
@ -258,14 +280,20 @@ public class ClassScaner implements Serializable{
|
||||
|
||||
/**
|
||||
* 加载类
|
||||
*
|
||||
*
|
||||
* @param className 类名
|
||||
* @return 加载的类
|
||||
*/
|
||||
private Class<?> loadClass(String className) {
|
||||
ClassLoader loader = this.classLoader;
|
||||
if (null == loader) {
|
||||
loader = ClassLoaderUtil.getClassLoader();
|
||||
this.classLoader = loader;
|
||||
}
|
||||
|
||||
Class<?> clazz = null;
|
||||
try {
|
||||
clazz = Class.forName(className, this.initialize, ClassUtil.getClassLoader());
|
||||
clazz = Class.forName(className, this.initialize, loader);
|
||||
} catch (NoClassDefFoundError e) {
|
||||
// 由于依赖库导致的类无法加载，直接跳过此类
|
||||
} catch (UnsupportedClassVersionError e) {
|
||||
@ -276,27 +304,26 @@ public class ClassScaner implements Serializable{
|
||||
}
|
||||
return clazz;
|
||||
}
|
||||
|
||||
|
||||
/**
|
||||
* 通过过滤器，是否满足接受此类的条件
|
||||
*
|
||||
* @param clazz 类
|
||||
* @return 是否接受
|
||||
*
|
||||
* @param className 类名
|
||||
*/
|
||||
private void addIfAccept(String className) {
|
||||
if(StrUtil.isBlank(className)) {
|
||||
if (StrUtil.isBlank(className)) {
|
||||
return;
|
||||
}
|
||||
int classLen = className.length();
|
||||
int packageLen = this.packageName.length();
|
||||
if(classLen == packageLen) {
|
||||
if (classLen == packageLen) {
|
||||
//类名和包名长度一致，用户可能传入的包名是类名
|
||||
if(className.equals(this.packageName)) {
|
||||
if (className.equals(this.packageName)) {
|
||||
addIfAccept(loadClass(className));
|
||||
}
|
||||
} else if(classLen > packageLen){
|
||||
} else if (classLen > packageLen) {
|
||||
//检查类名是否以指定包名为前缀，包名后加.（避免类似于cn.hutool.A和cn.hutool.ATest这类类名引起的歧义）
|
||||
if(className.startsWith(this.packageNameWithDot)) {
|
||||
if (className.startsWith(this.packageNameWithDot)) {
|
||||
addIfAccept(loadClass(className));
|
||||
}
|
||||
}
|
||||
@ -304,9 +331,8 @@ public class ClassScaner implements Serializable{
|
||||
|
||||
/**
|
||||
* 通过过滤器，是否满足接受此类的条件
|
||||
*
|
||||
*
|
||||
* @param clazz 类
|
||||
* @return 是否接受
|
||||
*/
|
||||
private void addIfAccept(Class<?> clazz) {
|
||||
if (null != clazz) {
|
||||
@ -319,7 +345,7 @@ public class ClassScaner implements Serializable{
|
||||
|
||||
/**
|
||||
* 截取文件绝对路径中包名之前的部分
|
||||
*
|
||||
*
|
||||
* @param file 文件
|
||||
* @return 包名之前的部分
|
||||
*/
|
||||
|
||||
@ -10,7 +10,7 @@ public class ClassScanerTest {
|
||||
@Test
|
||||
@Ignore
|
||||
public void scanTest() {
|
||||
ClassScaner scaner = new ClassScaner("cn.hutool.core.util.StrUtil", null);
|
||||
ClassScaner scaner = new ClassScaner("cn.hutool.core.util", null);
|
||||
Set<Class<?>> set = scaner.scan();
|
||||
for (Class<?> clazz : set) {
|
||||
Console.log(clazz.getName());
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user