QingLong/src/main/java/Tools/ClearGtzz.java

package Tools;

import cn.hutool.core.io.FileUtil;
import com.dsideal.QingLong.Const.GtzzConst;
import com.dsideal.QingLong.Gtzz.Model.GtzzModel;
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++) {
            System.out.println("正在执行清空表:" + array[i].replace("\r", ""));
            PgSqlUtil.Truncate(array[i].replace("\r", ""));
        }

        //增加两个默认的验收任务
        GtzzModel gm = new GtzzModel();
        int id = gm.addYs("【初审专家分配系统保留任务】", "0203"); //为“技术方案初审” 服务
        System.out.println("添加默认验收任务，id为：" + id);
        gm.saveYsExpert(id, GtzzConst.DEFAULT_EXPERT, GtzzConst.DEFAULT_EXPERT);
        id = gm.addYs("【验收专家分配系统保留任务】", "0307"); //为"市级验收"服务
        gm.saveYsExpert(id, GtzzConst.DEFAULT_EXPERT, GtzzConst.DEFAULT_EXPERT);
        System.out.println("添加默认验收任务，id为：" + id);
    }
}