You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
50 lines
1.9 KiB
50 lines
1.9 KiB
2 years ago
|
package Tools;
|
||
|
|
|
||
|
|
import cn.hutool.core.io.FileUtil;
|
||
|
|
import com.dsideal.QingLong.Start;
|
||
|
|
import com.dsideal.QingLong.Util.CommonUtil;
|
||
|
|
import com.dsideal.QingLong.Util.PgSqlUtil;
|
||
|
|
import com.jfinal.kit.PropKit;
|
||
|
|
import com.jfinal.plugin.activerecord.ActiveRecordPlugin;
|
||
|
|
import com.jfinal.plugin.activerecord.CaseInsensitiveContainerFactory;
|
||
|
|
import com.jfinal.plugin.activerecord.dialect.PostgreSqlDialect;
|
||
|
|
import com.jfinal.plugin.hikaricp.HikariCpPlugin;
|
||
|
|
|
||
|
|
import java.io.File;
|
||
|
|
import java.net.URISyntaxException;
|
||
|
|
|
||
|
|
public class ClearGtzz {
|
||
|
|
public static void main(String[] args) throws URISyntaxException {
|
||
|
|
//告之配置文件位置
|
||
|
|
PropKit.use("application.properties");
|
||
|
|
HikariCpPlugin hp = new HikariCpPlugin(PropKit.get("jdbcUrl"), PropKit.get("user"),
|
||
|
|
PropKit.get("password").trim(), PropKit.get("driverClassName"));
|
||
|
|
hp.start();
|
||
|
|
// 配置ActiveRecord插件
|
||
|
|
ActiveRecordPlugin arp = new ActiveRecordPlugin(hp);
|
||
|
|
//配置默认小写
|
||
|
|
arp.setContainerFactory(new CaseInsensitiveContainerFactory(true));
|
||
|
|
|
||
|
|
arp.setDialect(new PostgreSqlDialect());
|
||
|
|
//遍历sql目录下所有的sql文件
|
||
|
|
File sqlDir;
|
||
|
|
String basePath = Start.class.getClassLoader().getResource(".").getPath();
|
||
|
|
sqlDir = new File(basePath + "/Sql");
|
||
|
|
File[] sqlFiles = sqlDir.listFiles();
|
||
|
|
for (File sqlFile : sqlFiles != null ? sqlFiles : new File[0]) {
|
||
|
|
//只加载.sql文件
|
||
|
|
if (sqlFile.getName().indexOf(".sql") > 0) {
|
||
|
|
arp.addSqlTemplate("/Sql/" + sqlFile.getName());
|
||
|
|
}
|
||
|
|
}
|
||
|
|
arp.start();
|
||
|
|
|
||
|
|
String str = FileUtil.readUtf8String(new File(CommonUtil.getClassPath() + "gtzz_cleartable.txt"));
|
||
|
|
String[] array = str.split("\n");
|
||
|
|
|
||
|
|
for (int i = 0; i < array.length; i++) {
|
||
|
|
PgSqlUtil.Truncate(array[i]);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|