main
黄海 9 months ago
parent 48350b120d
commit dec3ee1dc6

@ -8,9 +8,11 @@ import net.sf.json.JSONObject;
import java.util.ArrayList;
import java.util.List;
import static com.dsideal.base.Tools.InitDataEaseDataSet.DB_NAME;
public class DataEaseModel {
//DataEase数据库名称
public static final String DB_NAME = "dataease";
/**
*
*

@ -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, DB_NAME, tableName);
List<Record> results = Db.find(sql, DataEaseModel.DB_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, DB_NAME, tableName) > 0;
return Db.use("dataease").queryInt(sql, DataEaseModel.DB_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(DB_NAME).update(sql);
Db.use(DataEaseModel.DB_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(DB_NAME).find(sql);
return Db.use(DataEaseModel.DB_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(DB_NAME).update(sql);
Db.use(DataEaseModel.DB_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", DB_NAME), PropKit.get("mysql.user"),
HikariCpPlugin dataEasePlugin = new HikariCpPlugin(PropKit.get("mysql.jdbcUrl").replace("ds_db", DataEaseModel.DB_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(DB_NAME, dataEasePlugin);
ActiveRecordPlugin arpDataEase = new ActiveRecordPlugin(DataEaseModel.DB_NAME, dataEasePlugin);
arpDataEase.setDialect(new MysqlDialect());
//遍历sql目录下所有的sql文件

Loading…
Cancel
Save