在当前ClassPath中搜索类
2023-02-18 15:31:11 时间
Spring 2.5提供了自动在当前ClassPath搜索被标注有特定注解的类,这个特性非常有用,跟踪了一下源码,发现其实核心代码就是利用ClassLoader的方法:
public Enumeration<URL> getResources(String name)
于是自己动手,也写了一个能在ClassPath下搜索特定类的Scanner:
package com.liaoxuefeng.util;
import java.io.File;
import java.io.IOException;
import java.net.URL;
import java.util.ArrayList;
import java.util.Enumeration;
import java.util.List;
import java.util.jar.JarEntry;
import java.util.jar.JarFile;
public class ClassPathScanner {
private static final String PROTOCOL_FILE = "file";
private static final String PROTOCOL_JAR = "jar";
private static final String PREFIX_FILE = "file:";
private static final String JAR_URL_SEPERATOR = "!/";
private static final String CLASS_FILE = ".class";
private final String packageName;
private final ClassFilter filter;
public ClassPathScanner(String packageName) {
this(packageName, null);
}
public ClassPathScanner(String packageName, ClassFilter filter) {
this.packageName = packageName;
this.filter = filter;
}
public List<Class<?>> scan() {
List<Class<?>> list = new ArrayList<Class<?>>();
Enumeration<URL> en = null;
try {
en = getClass().getClassLoader().getResources(dotToPath(packageName));
}
catch(IOException e) {
e.printStackTrace();
}
while (en.hasMoreElements()) {
URL url = en.nextElement();
if (PROTOCOL_FILE.equals(url.getProtocol())) {
File root = new File(url.getFile());
findInDirectory(list, root, root, packageName);
}
else if (PROTOCOL_JAR.equals(url.getProtocol())) {
findInJar(list, getJarFile(url), packageName);
}
}
return list;
}
public File getJarFile(URL url) {
String file = url.getFile();
if (file.startsWith(PREFIX_FILE))
file = file.substring(PREFIX_FILE.length());
int end = file.indexOf(JAR_URL_SEPERATOR);
if (end!=(-1))
file = file.substring(0, end);
return new File(file);
}
void findInJar(List<Class<?>> results, File file, String packageName) {
JarFile jarFile = null;
String packagePath = dotToPath(packageName) + "/";
try {
jarFile = new JarFile(file);
Enumeration<JarEntry> en = jarFile.entries();
while (en.hasMoreElements()) {
JarEntry je = en.nextElement();
String name = je.getName();
if (name.startsWith(packagePath) && name.endsWith(CLASS_FILE)) {
String className = name.substring(0, name.length() - CLASS_FILE.length());
add(results, pathToDot(className));
}
}
}
catch(IOException e) {
e.printStackTrace();
}
finally {
if (jarFile!=null) {
try {
jarFile.close();
}
catch(IOException e) {}
}
}
}
void findInDirectory(List<Class<?>> results, File rootDir, File dir, String packageName) {
File[] files = dir.listFiles();
String rootPath = rootDir.getPath();
for (File file : files) {
if (file.isFile()) {
String classFileName = file.getPath();
if (classFileName.endsWith(CLASS_FILE)) {
String className = classFileName.substring(rootPath.length() - packageName.length(), classFileName.length() - CLASS_FILE.length());
add(results, pathToDot(className));
}
}
else if (file.isDirectory()) {
findInDirectory(results, rootDir, file, packageName);
}
}
}
void add(List<Class<?>> results, String className) {
Class<?> clazz = null;
try {
clazz = Class.forName(className);
}
catch(ClassNotFoundException e) {
return;
}
if (filter==null || filter.accept(clazz))
results.add(clazz);
}
String dotToPath(String s) {
return s.replace('.', '/');
}
String pathToDot(String s) {
return s.replace('/', '.').replace('\\', '.');
}
}
ClassFilter用来过滤是否要返回这一个类:
public interface ClassFilter {
boolean accept(Class<?> clazz);
}
一个完整的示例如下:
public static void main(String[] args) {
List<Class<?>> list = new ClassPathScanner(
"org.springframework.core.io",
new ClassFilter() {
public boolean accept(Class<?> clazz) {
return clazz.isInterface(); // 返回接口类
}
}
).scan();
for (Class<?> clazz : list) {
System.out.println(clazz);
}
}
相关文章
- Java-什么是面向对象
- 苹果MacOS_虚拟机安装详细过程
- 21条最佳实践,全面保障 GitHub 使用安全
- 字节后端实习一面凉经
- 面向接口编程的好处-介绍API数据接口的好处!
- C语言大学生考试全面题库
- 数据治理很抽象吗
- Elasticsearch互联网主流分布式全文检索框架实战-ElasticStack(上)v7.14.0
- 分布式全局ID生成器原理剖析及非常齐全开源方案应用示例
- ElasticJob分布式任务调度应用v2.5.2
- 云原生概念你了解多少
- 主流微服务一站式解决方案Spring Cloud Alibaba入门看这篇就足够了-开篇v2.2.1.RELEASE
- Pulsar云原生分布式消息和流平台v2.8.0
- 回顾2022,展望2023,一个普通98年程序员的自述和分享
- SpringBoot+Mybatis-plus整合easyExcel批量导入Excel到数据库+导出Excel
- Springboot整合策略模式概念->使用场景->优缺点->企业级实战
- 三台服务器使用docker搭建redis一主二从三哨兵,概念-搭建-整合springboot
- IDEA中给源码添加自己注释——private-notes插件安装使用
- docker搭建Elasticsearch、Kibana、Logstash 同步mysql数据到ES
- Spark简单介绍,Windows下安装Scala+Hadoop+Spark运行环境,集成到IDEA中