package com.dsideal.base.Tools.Util;

import com.dsideal.base.DataEase.Model.DataEaseModel;
import com.dsideal.base.Plugin.YamlProp;
import com.dsideal.base.Tools.DataSetInit;
import com.jfinal.kit.Prop;
import com.jfinal.plugin.activerecord.ActiveRecordPlugin;
import com.jfinal.plugin.activerecord.dialect.MysqlDialect;
import com.jfinal.plugin.hikaricp.HikariCpPlugin;

import java.io.File;

public class LocalMysqlConnectUtil {
    public static Prop PropKit;

    public static void Init() {
        //加载配置文件
        String configFile = "application.yaml";
        PropKit = new YamlProp(configFile);

        HikariCpPlugin masterPlugin = new HikariCpPlugin(PropKit.get("mysql.jdbcUrl"), PropKit.get("mysql.user"),
                PropKit.get("mysql.password").trim(), PropKit.get("mysql.driverClassName"));
        HikariCpPlugin dataEasePlugin = new HikariCpPlugin(PropKit.get("mysql.jdbcUrl").replace("ds_db", "dataease"), PropKit.get("mysql.user"),
                PropKit.get("mysql.password").trim(), PropKit.get("mysql.driverClassName"));

        masterPlugin.start();

        dataEasePlugin.start();

        // 配置ActiveRecord插件
        ActiveRecordPlugin arp = new ActiveRecordPlugin(masterPlugin);
        arp.setDialect(new MysqlDialect());

        ActiveRecordPlugin arpDataEase = new ActiveRecordPlugin(DataEaseModel.DB_NAME, dataEasePlugin);
        arpDataEase.setDialect(new MysqlDialect());

        //遍历sql目录下所有的sql文件
        File sqlDir;
        String basePath = DataSetInit.class.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());
                arpDataEase.addSqlTemplate("/Sql/" + sqlFile.getName());
            }
        }
        arp.start();
        arpDataEase.start();
    }
}