-- coding: UTF-8 --
package com.cqcye.factorymethod;
import java.io.File;
import java.io.IOException;
import java.net.URL;
import java.util.ArrayList;
import java.util.Enumeration;
import java.util.List;
/**
* Created by Administrator on 2016/10/10.
*/
public class ClassUtils {
/**
* 给一个接口,返回这个接口的所有实现类
*
* @param c
* @return
*/
public static List<Class> getAllClassByInterface(Class c) {
List<Class> returnClassList = new ArrayList<>(); //返回结果
if (c.isInterface()) { //如果不是一个接口,不做处理
String packageName = c.getPackage().getName(); //获取当前的包名
try {
List<Class> allClass = getClasses(packageName);
for (int i = 0; i < allClass.size(); i++) {
if (c.isAssignableFrom(allClass.get(i))) { //判断是否是同一个接口
if (!c.equals(allClass.get(i))) { //本身不加进去
returnClassList.add(allClass.get(i));
}
}
}
} catch (IOException e) {
e.printStackTrace();
} catch (ClassNotFoundException e) {
e.printStackTrace();
}
}
return returnClassList;
}
/**
* 获取当前包名下的所有类及子类
*
* @param packageName
* @return
*/
private static List<Class> getClasses(String packageName) throws IOException, ClassNotFoundException {
ClassLoader classLoader = Thread.currentThread().getContextClassLoader();
String path = packageName.replace(".", "/");
Enumeration<URL> resourses = classLoader.getResources(path);
List<File> dirs = new ArrayList<File>();
while (resourses.hasMoreElements()) {
URL resourse = resourses.nextElement();
dirs.add(new File(resourse.getFile()));
}
ArrayList<Class> classes = new ArrayList<Class>();
for (File directory : dirs) {
classes.addAll(findClasses(directory, packageName));
}
return classes;
}
/**
* 获取所有的类
*
* @param directory
* @param packageName
* @return
* @throws ClassNotFoundException
*/
private static List<Class> findClasses(File directory, String packageName) throws ClassNotFoundException {
List<Class> classes = new ArrayList<>();
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;
}
}