前段时间 @绝望的八皮 写了个自动探测model注册插件,见:http://my.oschina.net/b1412/blog/67764 主要功能是为了省去在系统的config中每个model对应表名的配置,通过注解的方式绑定相应的表名,这样做也为将来实现jfinal组件化分离埋下了很好的伏笔,总体来说还是蛮有意义的。遗憾的是可能由于时间原因,八皮没能完成对jar包中class类绑定的扫描,私下联系过说会抽时间做,奈何需求紧急,自己先应付着写了个粗陋的实现,且等八皮优化版本。
基本事先没有修改,唯一修改的是对class扫描的类ClassSeacher,其余参照以上连接实现,就不一一发了。
package net.zfsy.db.plugin;
import java.io.File;
import java.io.IOException;
import java.net.MalformedURLException;
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 ClassSearcher {
/**
* 递归查找文件
*
* @param baseDirName
* 查找的文件夹路径
* @param targetFileName
* 需要查找的文件名
*/
private static List<String> findFiles(String baseDirName,
String targetFileName) {
/**
* 算法简述: 从某个给定的需查找的文件夹出发,搜索该文件夹的所有子文件夹及文件,
* 若为文件,则进行匹配,匹配成功则加入结果集,若为子文件夹,则进队列。 队列不空,重复上述操作,队列为空,程序结束,返回结果。
*/
List<String> classFiles = new ArrayList<String>();
String tempName = null;
// 判断目录是否存在
File baseDir = new File(baseDirName);
if (!baseDir.exists() || !baseDir.isDirectory()) {
System.out.println("文件查找失败:" + baseDirName + "不是一个目录!");
} else {
String[] filelist = baseDir.list();
for (int i = 0; i < filelist.length; i++) {
File readfile = new File(baseDirName + File.separator
+ filelist[i]);
if (!readfile.isDirectory()) {
tempName = readfile.getName();
if (ClassSearcher.wildcardMatch(targetFileName, tempName)) {
String classname;
String tem = readfile.getAbsoluteFile().toString()
.toString().replaceAll("\\\\", "/");
classname = tem.substring(tem.indexOf("/classes")
+ "/classes".length(), tem.indexOf(".class"));
if (classname.startsWith("/")) {
classname = classname.substring(classname
.indexOf("/") + 1);
}
classname = className(classname, "/classes");
classFiles.add(classname);
}
} else if (readfile.isDirectory()) {
classFiles.addAll(findFiles(baseDirName + File.separator
+ filelist[i], targetFileName));
}
}
}
return classFiles;
}
/**
* 查找jar包中的class
* @param baseDirName jar路径
* @param jarFileURL jar文件地址
* @return
*/
public static List<String> findjarFiles(String baseDirName,
URL... jarFileURL) {
List<String> classFiles = new ArrayList<String>();
try {
// 判断目录是否存在
File baseDir = new File(baseDirName);
if (!baseDir.exists() || !baseDir.isDirectory()) {
System.out.println("文件查找失败:" + baseDirName + "不是一个目录!");
} else {
String[] filelist = baseDir.list();
for (int i = 0; i < filelist.length; i++) {
if (filelist[i].contains("zfplugin")) {//查找含有zfplugin的jar包
JarFile localJarFile = new JarFile(new File(baseDirName
+ File.separator + filelist[i]));
Enumeration<JarEntry> entries = localJarFile.entries();
while (entries.hasMoreElements()) {
JarEntry jarEntry = entries.nextElement();
String entryName = jarEntry.getName();
if (jarEntry.isDirectory()) {
// System.out.println(entryName);
} else if (entryName.endsWith(".class")) {
String className = entryName.replaceAll("/",
".").substring(0,
entryName.length() - 6);
classFiles.add(className);
}
}
}
}
}
} catch (IOException e) {
e.printStackTrace();
}
return classFiles;
}
public static List<Class> findClasses(Class clazz)
throws MalformedURLException {
List<Class> classList = new ArrayList<Class>();
URL classPathUrl = ClassSearcher.class.getResource("/");
List<String> classFileList = findFiles(classPathUrl.getFile(),
"*.class");
String lib = new File(classPathUrl.getFile()).getParent() + "/lib/";
List<String> jarclassFiles = findjarFiles(lib, new File(lib).toURL());
classFileList.addAll(jarclassFiles);
for (String classFile : classFileList) {
// String className = className(classFile, "/classes");
System.out.println(classFile);
try {
Class<?> classInFile = Class.forName(classFile);
if (classInFile.getSuperclass() == clazz) {
classList.add(classInFile);
}
} catch (ClassNotFoundException e) {
e.printStackTrace();
}
}
return classList;
}
private static String className(String classFile, String pre) {
String objStr = classFile.replaceAll("\\\\", "/");
return objStr.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);
}
public static void main(String[] args) throws MalformedURLException {
// URL classPathUrl = ClassSearcher.class.getResource("/");
// String baseDirName = new File(classPathUrl.getFile()).getParent()
// + "/lib/";
// URL url = new File(baseDirName).toURL();
// ClassSearcher.findjarFiles(baseDirName, url);
//ClassSearcher.findClasses(Demo.class);
}
}