package org.apache.ibatis.thread;
|
|
import java.io.File;
|
import java.io.FileInputStream;
|
import java.io.FileNotFoundException;
|
import java.util.ArrayList;
|
import java.util.List;
|
|
import org.apache.commons.lang3.StringUtils;
|
import org.apache.ibatis.session.Configuration;
|
import org.apache.log4j.Logger;
|
import org.mybatis.spring.SqlSessionFactoryBean;
|
import org.springframework.core.NestedIOException;
|
|
/**
|
* 刷新使用进程
|
*
|
* @author liubaoquan
|
*
|
*/
|
public class Runnable implements java.lang.Runnable {
|
|
public static Logger log = Logger.getLogger(Runnable.class);
|
|
private String location;
|
private Configuration configuration;
|
|
private Long beforeTime = 0L; // 上一次刷新时间
|
private static boolean refresh = false; // 是否执行刷新
|
|
private static String mappingPath = "mappings"; // xml文件夹匹配字符串，需要根据需要修改
|
private static int delaySeconds = 10;// 延迟刷新秒数
|
private static int sleepSeconds = 1;// 休眠时间
|
|
private static boolean enabled = false;
|
|
static {
|
delaySeconds = PropertiesUtil.getInt("delaySeconds");
|
sleepSeconds = PropertiesUtil.getInt("sleepSeconds");
|
mappingPath = PropertiesUtil.getString("mappingPath");
|
enabled = "true".equals(PropertiesUtil.getString("enabled"));
|
|
delaySeconds = delaySeconds == 0 ? 50 : delaySeconds;
|
sleepSeconds = sleepSeconds == 0 ? 1 : sleepSeconds;
|
mappingPath = StringUtils.isBlank(mappingPath) ? "mappings"
|
: mappingPath;
|
|
log.debug("[delaySeconds] " + delaySeconds);
|
log.debug("[sleepSeconds] " + sleepSeconds);
|
log.debug("[mappingPath] " + mappingPath);
|
|
}
|
|
public static boolean isRefresh() {
|
return refresh;
|
}
|
|
public Runnable(String location, Configuration configuration) {
|
this.location = location.replaceAll("\\\\", "/");
|
this.configuration = configuration;
|
}
|
|
@Override
|
public void run() {
|
location = location.substring("file [".length(),
|
location.lastIndexOf(mappingPath) + mappingPath.length());
|
beforeTime = System.currentTimeMillis();
|
|
log.debug("[location] " + location);
|
log.debug("[configuration] " + configuration);
|
|
if (enabled){
|
start(this);
|
}
|
}
|
|
public void start(final Runnable runnable) {
|
|
new Thread(new java.lang.Runnable() {
|
|
@Override
|
public void run() {
|
|
try {
|
Thread.sleep(delaySeconds * 1000);
|
} catch (InterruptedException e2) {
|
e2.printStackTrace();
|
}
|
refresh = true;
|
|
System.out.println("========= Enabled refresh mybatis mapper =========");
|
|
while (true) {
|
try {
|
runnable.refresh(location, beforeTime);
|
} catch (Exception e1) {
|
e1.printStackTrace();
|
}
|
|
try {
|
Thread.sleep(sleepSeconds * 1000);
|
} catch (InterruptedException e) {
|
e.printStackTrace();
|
}
|
|
}
|
}
|
}).start();
|
}
|
|
/**
|
* 执行刷新
|
*
|
* @param filePath
|
* 刷新目录
|
* @param beforeTime
|
* 上次刷新时间
|
* @throws NestedIOException
|
* 解析异常
|
* @throws FileNotFoundException
|
* 文件未找到
|
*/
|
public void refresh(String filePath, Long beforeTime) throws Exception {
|
|
// 本次刷新时间
|
Long refrehTime = System.currentTimeMillis();
|
|
List<File> refreshs = this.getRefreshFile(new File(filePath),
|
beforeTime);
|
if (refreshs.size() > 0) {
|
log.debug("refresh files:" + refreshs.size());
|
}
|
for (int i = 0; i < refreshs.size(); i++) {
|
System.out.println("Refresh file: "
|
+ mappingPath
|
+ StringUtils.substringAfterLast(refreshs.get(i)
|
.getAbsolutePath(), mappingPath));
|
log.debug("refresh file:" + refreshs.get(i).getAbsolutePath());
|
log.debug("refresh filename:" + refreshs.get(i).getName());
|
SqlSessionFactoryBean.refresh(new FileInputStream(refreshs.get(i)),
|
refreshs.get(i).getAbsolutePath(), configuration);
|
}
|
// 如果刷新了文件，则修改刷新时间，否则不修改
|
if (refreshs.size() > 0) {
|
this.beforeTime = refrehTime;
|
}
|
}
|
|
/**
|
* 获取需要刷新的文件列表
|
*
|
* @param dir
|
* 目录
|
* @param beforeTime
|
* 上次刷新时间
|
* @return 刷新文件列表
|
*/
|
public List<File> getRefreshFile(File dir, Long beforeTime) {
|
List<File> refreshs = new ArrayList<File>();
|
|
File[] files = dir.listFiles();
|
for (int i = 0; i < files.length; i++) {
|
File file = files[i];
|
if (file.isDirectory()) {
|
refreshs.addAll(this.getRefreshFile(file, beforeTime));
|
} else if (file.isFile()) {
|
if (this.check(file, beforeTime)) {
|
refreshs.add(file);
|
}
|
} else {
|
System.out.println("error file." + file.getName());
|
}
|
}
|
|
return refreshs;
|
}
|
|
/**
|
* 判断文件是否需要刷新
|
*
|
* @param file
|
* 文件
|
* @param beforeTime
|
* 上次刷新时间
|
* @return 需要刷新返回true，否则返回false
|
*/
|
public boolean check(File file, Long beforeTime) {
|
if (file.lastModified() > beforeTime) {
|
return true;
|
}
|
return false;
|
}
|
|
}
|
