之前发过一个
JFinal 自动探测Model 注册插件 AutoScanPlugin
http://my.oschina.net/b1412/blog/67764
然后参考了 @Flio 对我的代码做出的扩展
jfinal的Model自动绑定表插件
http://www.oschina.net/code/snippet_591384_12022
现在整合下代码采用@Flio的插件命名，并且重构了一下搜索类的代码。
类探测api如下方便以后其类探测的使用.
List<Class> classList = ClassSearcher.findClasses(Model.class);插件注册代码,由于AutoTableBindPlugin继承自ActiveRecordPlugin,所以不需要单独注册它了.
主要数据库连接池插件要在AutoTableBindPlugin之前注册.
public void configPlugin(Plugins me) {
DruidPlugin druidPlugin = new DruidPlugin.DruidBuilder(
getProperty("url"), getProperty("username"),
getProperty("password")).build();
AutoTableBindPlugin autoTableBindPlugin = new AutoTableBindPlugin(
druidPlugin, TableNameStyle.LOWER);
autoTableBindPlugin.setShowSql(true);
SqlReporter.setLogger(true);
SqlInXmlPlugin sqlInXmlPlugin = new SqlInXmlPlugin();
QuartzPlugin quartzPlugin = new QuartzPlugin();
me.add(druidPlugin).add(sqlInXmlPlugin).add(autoTableBindPlugin).add(quartzPlugin);
}
目前之内探测/classes下的类，如果是/lib下jar包中的类还不能探测。
之前做过运行动态修改字节码的工具，是吧lib下的jar解压出来然后修改了打包回去。如果需要探测jar中类是否为Model子类是否必须坐jar包解压的工作？大家有什么好的方案么？这个问题弄清楚了再加上扫描jar包中Model的工作。
目前如果是有项目吧class文件打成jar包部署的朋友暂时无法使用自动探测绑定的功能...
package com.jfinal.plugin.tablebind;
import java.util.List;
import javax.sql.DataSource;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import com.jfinal.plugin.activerecord.ActiveRecordPlugin;
import com.jfinal.plugin.activerecord.IDataSourceProvider;
import com.jfinal.plugin.activerecord.Model;
import com.jfinal.util.StringKit;
public class AutoTableBindPlugin extends ActiveRecordPlugin {
private TableNameStyle tableNameStyle;
private Logger logger = LoggerFactory.getLogger(getClass());
public AutoTableBindPlugin(DataSource dataSource) {
super(dataSource);
}
public AutoTableBindPlugin(IDataSourceProvider dataSourceProvider, TableNameStyle tableNameStyle) {
super(dataSourceProvider);
this.tableNameStyle = tableNameStyle;
}
@Override
public boolean start() {
try {
List<Class> modelClasses = ClassSearcher.findClasses(Model.class);
logger.debug("modelClasses.size {}",modelClasses.size());
TableBind tb = null;
for (Class modelClass : modelClasses) {
tb = (TableBind) modelClass.getAnnotation(TableBind.class);
if (tb == null) {
this.addMapping(tableName(modelClass), modelClass);
logger.debug("auto bindTable: addMapping({}, {})", tableName(modelClass), modelClass.getName());
} else {
if (StringKit.notBlank(tb.pkName())) {
this.addMapping(tb.tableName(), tb.pkName(), modelClass);
logger.debug("auto bindTable: addMapping({}, {},{})", new Object[]{tb.tableName(),tb.pkName(), modelClass.getName()});
} else {
this.addMapping(tb.tableName(), modelClass);
logger.debug("auto bindTable: addMapping({}, {})", tb.tableName(), modelClass.getName());
}
}
}
} catch (Exception e) {
throw new RuntimeException(e);
}
return super.start();
}
@Override
public boolean stop() {
return super.stop();
}
private String tableName(Class<?> clazz) {
String tableName = clazz.getSimpleName();
if (tableNameStyle == TableNameStyle.UP) {
tableName = tableName.toUpperCase();
} else if (tableNameStyle == TableNameStyle.LOWER) {
tableName = tableName.toLowerCase();
} else {
tableName = StringKit.firstCharToLowerCase(tableName);
}
return tableName;
}
}
package com.jfinal.plugin.tablebind;
import java.lang.annotation.ElementType;
import java.lang.annotation.Inherited;
import java.lang.annotation.Retention;
import java.lang.annotation.RetentionPolicy;
import java.lang.annotation.Target;
@Inherited
@Retention(RetentionPolicy.RUNTIME)
@Target({ElementType.TYPE})
public @ interface TableBind {
String tableName();
String pkName() default "";
}
package com.jfinal.plugin.tablebind;
import java.io.File;
import java.net.URL;
import java.util.ArrayList;
import java.util.List;
public class ClassSearcher {
private static List<File> classFiles;
/**
* 递归查找文件
*
* @param baseDirName
* 查找的文件夹路径
* @param targetFileName
* 需要查找的文件名
*/
private static List<File> findFiles(String baseDirName, String targetFileName) {
/**
* 算法简述： 从某个给定的需查找的文件夹出发，搜索该文件夹的所有子文件夹及文件，
* 若为文件，则进行匹配，匹配成功则加入结果集，若为子文件夹，则进队列。 队列不空，重复上述操作，队列为空，程序结束，返回结果。
*/
classFiles = new ArrayList<File>();
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)) {
classFiles.add(readfile.getAbsoluteFile());
}
} else if (readfile.isDirectory()) {
classFiles.addAll(findFiles(baseDirName + File.separator
+ filelist[i], targetFileName));
}
}
}
return classFiles;
}
public static List<Class> findClasses(Class clazz) {
List<Class> classList = new ArrayList<>();
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;
}
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);
}
}
package com.jfinal.plugin.tablebind;
public enum TableNameStyle {
UP, LOWER, UP_UNDERLINE, LOWER_UNDERLINE
}
