diff --git a/src/main/java/com/dsideal/base/Tools/ClearDataBase.java b/src/main/java/com/dsideal/base/Tools/ClearDataBase.java index 49f7a249..075cfc75 100644 --- a/src/main/java/com/dsideal/base/Tools/ClearDataBase.java +++ b/src/main/java/com/dsideal/base/Tools/ClearDataBase.java @@ -1,20 +1,22 @@ package com.dsideal.base.Tools; -import com.jfinal.kit.PathKit; -import com.jfinal.kit.PropKit; +import com.dsideal.base.Plugin.YamlProp; +import com.jfinal.kit.Prop; import com.jfinal.plugin.activerecord.ActiveRecordPlugin; import com.jfinal.plugin.activerecord.Db; -import com.jfinal.plugin.druid.DruidPlugin; +import com.jfinal.plugin.hikaricp.HikariCpPlugin; -import java.io.File; +import static com.dsideal.base.BaseApplication.getEnvPrefix; public class ClearDataBase { - + public static Prop PropKit; public static void main(String[] args) { //1、配置数据库 - PropKit.use("application_dev.properties"); - DruidPlugin druid = new DruidPlugin(PropKit.get("jdbcUrl"), PropKit.get("user"), - PropKit.get("password").trim(), PropKit.get("driverClassName")); + String configFile = "application_{?}.yaml".replace("{?}", getEnvPrefix()); + PropKit = new YamlProp(configFile); + + HikariCpPlugin druid = new HikariCpPlugin(PropKit.get("mysql.jdbcUrl"), PropKit.get("mysql.user"), + PropKit.get("mysql.password").trim(), PropKit.get("mysql.driverClassName")); druid.start(); // 配置ActiveRecord插件 ActiveRecordPlugin arp = new ActiveRecordPlugin(druid);