package com.fullwisdom.scan;
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 ClassUtils {
/**
*
* @param clazz
* @return List<Class>
* @throws Exception
*/
@SuppressWarnings("unchecked")
public static List<Class> getInterfaceImplements(Class clazz) throws Exception{
List<Class> retList = new ArrayList<Class>();
try {
Class clazzs[] = getClasses(clazz.getPackage().getName());
int len = clazzs.length;
for(int i=0;i<len;i++){
if( clazz.isAssignableFrom(clazzs[i]) ){
if(!clazz.equals(clazzs[i]))
retList.add(clazzs[i]);
}
}
} catch (Exception e) {
e.printStackTrace();
throw new Exception("加载包错误...");
}
return retList;
}
/**
* 获得包下的类
*
* @param packageName
* The base package
* @return The classes
* @throws ClassNotFoundException
* @throws IOException
*/
@SuppressWarnings("unchecked")
private static Class[] getClasses(String packageName)throws ClassNotFoundException, IOException {
ClassLoader classLoader = Thread.currentThread().getContextClassLoader();
assert classLoader != null;
String path = packageName.replace('.', '/');
Enumeration<URL> resources = classLoader.getResources(path);
List<File> dirs = new ArrayList<File>();
ArrayList<Class> classes = new ArrayList<Class>();
while (resources.hasMoreElements()) {
URL resource = resources.nextElement();
String p = "";
if (resource.getFile().indexOf("!") >= 0) {// 在其他的jar文件中
p = resource.getFile().substring(0,
resource.getFile().indexOf("!")).replaceAll("%20", "");
} else {// 在classes目录中
p = resource.getFile();
}
if (p.startsWith("file:/"))
p = p.substring(6);
if (p.toLowerCase().endsWith(".jar")) {
JarFile jarFile = new JarFile(p);
Enumeration<JarEntry> enums = jarFile.entries();
while (enums.hasMoreElements()) {
JarEntry entry = (JarEntry) enums.nextElement();
String n = entry.getName();
if (n.endsWith(".class")) {
n = n.replaceAll("/", ".").substring(0, n.length() - 6);
if (n.startsWith(packageName)) {
classes.add(Class.forName(n));
}
}
}
} else {
dirs.add(new File(p));
}
}
for (File directory : dirs) {
classes.addAll(findClasses(directory, packageName));
}
return classes.toArray(new Class[classes.size()]);
}
/**
* 查找一个文件夹下的文件
*
* @param directory
* 文价夹目录名
* @param packageName
* 包名
* @return The classes
* @throws ClassNotFoundException
*/
@SuppressWarnings("unchecked")
private static List<Class> findClasses(File directory, String packageName)
throws ClassNotFoundException {
List<Class> classes = new ArrayList<Class>();
if (!directory.exists()) {
return classes;
}
File[] files = directory.listFiles();
for (File file : files) {
if (file.isDirectory()) {
assert !file.getName().contains(".");
classes.addAll(findClasses(file, packageName + "."
+ file.getName()));
} else if (file.getName().endsWith(".class")) {
classes.add(Class.forName(packageName
+ '.'
+ file.getName().substring(0,
file.getName().length() - 6)));
}
}
return classes;
}
}
扫描接口实现类
最新推荐文章于 2023-01-27 15:47:59 发布