package net.loyin.jFinal.plugin;
import java.io.File;
import java.net.URL;
import java.util.ArrayList;
import java.util.List;
/**
* 类查找
* @author loyin
* 2012-9-4 上午11:42:47
*/
public class ClassSearcher {
private static List<File> classFiles = new ArrayList<File>();
/**
* 递归查找文件
* @param baseDirName
* 查找的文件夹路径
* @param targetFileName
* 需要查找的文件名
*/
public static List<File> findFiles(String baseDirName,
String targetFileName) {
/**
* 算法简述: 从某个给定的需查找的文件夹出发,搜索该文件夹的所有子文件夹及文件,
* 若为文件,则进行匹配,匹配成功则加入结果集,若为子文件夹,则进队列。 队列不空,重复上述操作,队列为空,程序结束,返回结果。
*/
String tempName = null;
// 判断目录是否存在
File baseDir = new File(baseDirName);
if (!baseDir.exists() || !baseDir.isDirectory()) {
System.out.println("文件查找失败:" + baseDirName + "不是一个目录!");
} else {
File[] filelist = baseDir.listFiles();
if(filelist!=null)
for(File f:filelist){
if(f.isDirectory()==true){
findFiles(f.getPath(),targetFileName);
}else{
tempName = f.getName();
if (ClassSearcher.wildcardMatch(targetFileName, tempName)) {
classFiles.add(f.getAbsoluteFile());
}
}
}
}
return classFiles;
}
/**
* 通过父类查找
* @param clazz 父类
* @return
*/
@SuppressWarnings("rawtypes")
public static List<Class> findClasses(Class clazz) {
List<Class> classList = new ArrayList<Class>();
URL classPathUrl = ClassSearcher.class.getResource("/");
List<File> classFileList = findFiles(classPathUrl.getFile(), "*.class");
// String lib = new File(classPathUrl.getFile()).getParent() + "/lib/";
for (File classFile : classFileList) {
String className = className(classFile, "/classes");
try {
Class<?> classInFile = Class.forName(className);
if (classInFile.getSuperclass() == clazz) {
classList.add(classInFile);
}
} catch (ClassNotFoundException e) {
e.printStackTrace();
}
}
return classList;
}
@SuppressWarnings("rawtypes")
public static List<Class> findClasses() {
List<Class> classList = new ArrayList<Class>();
URL classPathUrl = ClassSearcher.class.getResource("/");
List<File> classFileList = findFiles(classPathUrl.getFile(), "*.class");
// String lib = new File(classPathUrl.getFile()).getParent() + "/lib/";
for (File classFile : classFileList) {
String className = className(classFile, "/classes");
try {
Class<?> classInFile = Class.forName(className);
classList.add(classInFile);
} catch (ClassNotFoundException e) {
continue;
}
}
return classList;
}
private static String className(File classFile, String pre) {
String objStr = classFile.toString().replaceAll("\\\\", "/");
String className;
className = objStr.substring(objStr.indexOf(pre) + pre.length(),
objStr.indexOf(".class"));
if (className.startsWith("/")) {
className = className.substring(className.indexOf("/") + 1);
}
return className.replaceAll("/", ".");
}
/**
* 通配符匹配
* @param pattern
* 通配符模式
* @param str
* 待匹配的字符串
* @return 匹配成功则返回true,否则返回false
*/
private static boolean wildcardMatch(String pattern, String str) {
int patternLength = pattern.length();
int strLength = str.length();
int strIndex = 0;
char ch;
for (int patternIndex = 0; patternIndex < patternLength; patternIndex++) {
ch = pattern.charAt(patternIndex);
if (ch == '*') {
// 通配符星号*表示可以匹配任意多个字符
while (strIndex < strLength) {
if (wildcardMatch(pattern.substring(patternIndex + 1),
str.substring(strIndex))) {
return true;
}
strIndex++;
}
} else if (ch == '?') {
// 通配符问号?表示匹配任意一个字符
strIndex++;
if (strIndex > strLength) {
// 表示str中已经没有字符匹配?了。
return false;
}
} else {
if ((strIndex >= strLength) || (ch != str.charAt(strIndex))) {
return false;
}
strIndex++;
}
}
return (strIndex == strLength);
}
}