|
|
|
|
@ -16,8 +16,11 @@ import org.slf4j.LoggerFactory;
|
|
|
|
|
import java.io.*;
|
|
|
|
|
import java.net.JarURLConnection;
|
|
|
|
|
import java.net.MalformedURLException;
|
|
|
|
|
import java.net.URISyntaxException;
|
|
|
|
|
import java.net.URL;
|
|
|
|
|
import java.nio.charset.StandardCharsets;
|
|
|
|
|
import java.security.CodeSource;
|
|
|
|
|
import java.security.ProtectionDomain;
|
|
|
|
|
import java.text.ParseException;
|
|
|
|
|
import java.text.SimpleDateFormat;
|
|
|
|
|
import java.util.*;
|
|
|
|
|
@ -57,25 +60,32 @@ public class CommonUtil {
|
|
|
|
|
return IoUtil.read(is, "UTF-8");
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
/**
|
|
|
|
|
* 获取所有sql文件，并返回一个list集合
|
|
|
|
|
*
|
|
|
|
|
* @return
|
|
|
|
|
*/
|
|
|
|
|
public static List<String> getAllSql() throws IOException {
|
|
|
|
|
public static List<String> getAllSql() throws IOException, URISyntaxException {
|
|
|
|
|
List<String> list = new ArrayList<>();
|
|
|
|
|
if (isRunInJar()) {
|
|
|
|
|
URL url = CommonUtil.class.getClassLoader().getResource("Sql/");
|
|
|
|
|
String jarPath = url.toString().substring(0, url.toString().indexOf("!/") + 2);
|
|
|
|
|
URL jarURL = new URL(jarPath);
|
|
|
|
|
JarURLConnection jarCon = (JarURLConnection) jarURL.openConnection();
|
|
|
|
|
JarFile jarFile = jarCon.getJarFile();
|
|
|
|
|
Enumeration<JarEntry> jarEntrys = jarFile.entries();
|
|
|
|
|
while (jarEntrys.hasMoreElements()) {
|
|
|
|
|
JarEntry entry = jarEntrys.nextElement();
|
|
|
|
|
String name = entry.getName();
|
|
|
|
|
if (name.startsWith("Sql/") && !entry.isDirectory()) {
|
|
|
|
|
list.add(name);
|
|
|
|
|
// 获取当前类的ProtectionDomain
|
|
|
|
|
ProtectionDomain domain = CommonUtil.class.getProtectionDomain();
|
|
|
|
|
// 获取CodeSource
|
|
|
|
|
CodeSource source = domain.getCodeSource();
|
|
|
|
|
// 获取Jar文件的URL
|
|
|
|
|
URL jarUrl = source.getLocation();
|
|
|
|
|
// 获取Jar文件的路径
|
|
|
|
|
String jarPath = jarUrl.toURI().getSchemeSpecificPart();
|
|
|
|
|
// 打开Jar文件
|
|
|
|
|
JarFile jarFile = new JarFile(jarPath);
|
|
|
|
|
// 遍历Jar中的所有条目
|
|
|
|
|
Enumeration<JarEntry> entries = jarFile.entries();
|
|
|
|
|
while (entries.hasMoreElements()) {
|
|
|
|
|
JarEntry entry = entries.nextElement();
|
|
|
|
|
// 检查条目是否是SQL文件
|
|
|
|
|
if (entry.getName().startsWith("Sql/") && entry.getName().toLowerCase().endsWith(".sql")) {
|
|
|
|
|
list.add(entry.getName());
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
} else {//如果运行在文件系统中，直接加载sql文件
|
|
|
|
|
|
