@ -1,5 +1,6 @@
package com.dsideal.base.Tools ;
import com.dsideal.base.DataEase.Model.DataEaseModel ;
import com.dsideal.base.Plugin.YamlProp ;
import com.jfinal.kit.Kv ;
import com.jfinal.kit.Prop ;
@ -15,8 +16,6 @@ import java.util.ArrayList;
import java.util.List ;
public class InitDataEaseDataSet {
//DataEase数据库名称
public static final String DB_NAME = "dataease" ;
/ * *
* 获 取 数 据 集 的 表 名
@ -67,7 +66,7 @@ public class InitDataEaseDataSet {
public static List < String > getNotNullColumns ( String tableName ) {
List < String > columns = new ArrayList < > ( ) ;
String sql = "SELECT COLUMN_NAME FROM information_schema.COLUMNS WHERE TABLE_SCHEMA = ? AND TABLE_NAME = ? AND IS_NULLABLE = 'NO'" ;
List < Record > results = Db . find ( sql , D B_NAME, tableName ) ;
List < Record > results = Db . find ( sql , D ataEaseModel. D B_NAME, tableName ) ;
for ( Record result : results ) {
columns . add ( result . get ( "COLUMN_NAME" ) . toString ( ) ) ;
}
@ -77,14 +76,14 @@ public class InitDataEaseDataSet {
// 检查表是否存在主键
public static boolean hasPrimaryKey ( String tableName ) {
String sql = "SELECT COUNT(*) as c FROM information_schema.TABLE_CONSTRAINTS WHERE TABLE_SCHEMA = ? AND TABLE_NAME = ? AND CONSTRAINT_TYPE = 'PRIMARY KEY'" ;
return Db . use ( "dataease" ) . queryInt ( sql , D B_NAME, tableName ) > 0 ;
return Db . use ( "dataease" ) . queryInt ( sql , D ataEaseModel. D B_NAME, tableName ) > 0 ;
}
// 添加主键列,并设置为主键
public static void addPrimaryKey ( String tableName ) {
// 添加 id 列
String sql = "ALTER TABLE `" + tableName + "` ADD COLUMN `id` int(11) primary key auto_increment first" ;
Db . use ( D B_NAME) . update ( sql ) ;
Db . use ( D ataEaseModel. D B_NAME) . update ( sql ) ;
}
/ * *
@ -95,7 +94,7 @@ public class InitDataEaseDataSet {
public static List < Record > getExcelTable ( ) {
// 查询所有以 excel_ 开头的表
String sql = "SELECT TABLE_NAME FROM information_schema.TABLES WHERE TABLE_SCHEMA = 'dataease' AND TABLE_NAME LIKE 'excel\\_%'" ;
return Db . use ( D B_NAME) . find ( sql ) ;
return Db . use ( D ataEaseModel. D B_NAME) . find ( sql ) ;
}
/ * *
@ -133,7 +132,7 @@ public class InitDataEaseDataSet {
System . out . println ( "列" + col + "非空,正在去掉不允许为空的限制..." ) ;
//去掉不允许为空的限制
String sql = "ALTER TABLE `" + tableName + "` MODIFY `" + col + "` VARCHAR(255) NULL" ;
Db . use ( D B_NAME) . update ( sql ) ;
Db . use ( D ataEaseModel. D B_NAME) . update ( sql ) ;
System . out . println ( "去掉不允许为空的限制成功" ) ;
}
}
@ -148,7 +147,7 @@ public class InitDataEaseDataSet {
PropKit . get ( "mysql.password" ) . trim ( ) , PropKit . get ( "mysql.driverClassName" ) ) ;
masterPlugin . start ( ) ;
HikariCpPlugin dataEasePlugin = new HikariCpPlugin ( PropKit . get ( "mysql.jdbcUrl" ) . replace ( "ds_db" , D B_NAME) , PropKit . get ( "mysql.user" ) ,
HikariCpPlugin dataEasePlugin = new HikariCpPlugin ( PropKit . get ( "mysql.jdbcUrl" ) . replace ( "ds_db" , D ataEaseModel. D B_NAME) , PropKit . get ( "mysql.user" ) ,
PropKit . get ( "mysql.password" ) . trim ( ) , PropKit . get ( "mysql.driverClassName" ) ) ;
dataEasePlugin . start ( ) ;
@ -156,7 +155,7 @@ public class InitDataEaseDataSet {
ActiveRecordPlugin arp = new ActiveRecordPlugin ( "master" , masterPlugin ) ;
arp . setDialect ( new MysqlDialect ( ) ) ;
ActiveRecordPlugin arpDataEase = new ActiveRecordPlugin ( D B_NAME, dataEasePlugin ) ;
ActiveRecordPlugin arpDataEase = new ActiveRecordPlugin ( D ataEaseModel. D B_NAME, dataEasePlugin ) ;
arpDataEase . setDialect ( new MysqlDialect ( ) ) ;
//遍历sql目录下所有的sql文件