package com.dic.factory;
import java.io.File;
import java.io.IOException;
import java.net.URL;
import java.util.ArrayList;
import java.util.Enumeration;
import java.util.List;
public class ClassUtils {
/**
* 返回接口所以的实现类.
* @param c
* @return
*/
@SuppressWarnings({ "rawtypes", "unchecked" })
public static List getAllClassByInterface(Class c) {
List returnClassList = null;
if(c.isInterface()) {
// 获取当前的包名
String packageName = c.getPackage().getName();
try {
// 获取当前包下以及子包下所以的类
List allClass = getClasses(packageName);
if(allClass != null) {
returnClassList = new ArrayList();
for(Class classes : allClass) {
// 判断是否是同一个接口
if(c.isAssignableFrom(classes)) {
// 本身不加入进去
if(!c.equals(classes)) {
returnClassList.add(classes);
}
}
}
}
} catch (ClassNotFoundException e) {
e.printStackTrace();
} catch (IOException e) {
e.printStackTrace();
}
}
return returnClassList;
}
@SuppressWarnings("rawtypes")
private static List getClasses(String packageName) throws ClassNotFoundException, IOException {
ClassLoader classLoader = Thread.currentThread().getContextClassLoader();
String path = packageName.replace(".", "/");
Enumeration resources = classLoader.getResources(path);
List dirs = null;
if(resources != null) {
dirs = new ArrayList();
while(resources.hasMoreElements()) {
URL resource = resources.nextElement();
dirs.add(new File(resource.getFile()));
}
}
List classes = null;
if(dirs != null) {
classes = new ArrayList();
for(File directory : dirs) {
classes.addAll(findClasses(directory,packageName));
}
}
return classes;
}
@SuppressWarnings("rawtypes")
private static List findClasses(File directory,String packageName) throws ClassNotFoundException {
List classes = null;
if(!directory.exists()) {
return classes;
}
File[] files = directory.listFiles();
if(files != null) {
classes = new ArrayList();
for(File file : files) {
if(file.isDirectory()) {
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;
}
}