main
黄海 9 months ago
parent 48350b120d
commit dec3ee1dc6

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

@ -1,5 +1,6 @@
package com.dsideal.base.Tools; package com.dsideal.base.Tools;
import com.dsideal.base.DataEase.Model.DataEaseModel;
import com.dsideal.base.Plugin.YamlProp; import com.dsideal.base.Plugin.YamlProp;
import com.jfinal.kit.Kv; import com.jfinal.kit.Kv;
import com.jfinal.kit.Prop; import com.jfinal.kit.Prop;
@ -15,8 +16,6 @@ import java.util.ArrayList;
import java.util.List; import java.util.List;
public class InitDataEaseDataSet { 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) { public static List<String> getNotNullColumns(String tableName) {
List<String> columns = new ArrayList<>(); List<String> columns = new ArrayList<>();
String sql = "SELECT COLUMN_NAME FROM information_schema.COLUMNS WHERE TABLE_SCHEMA = ? AND TABLE_NAME = ? AND IS_NULLABLE = 'NO'"; 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) { for (Record result : results) {
columns.add(result.get("COLUMN_NAME").toString()); columns.add(result.get("COLUMN_NAME").toString());
} }
@ -77,14 +76,14 @@ public class InitDataEaseDataSet {
// 检查表是否存在主键 // 检查表是否存在主键
public static boolean hasPrimaryKey(String tableName) { 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'"; 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) { public static void addPrimaryKey(String tableName) {
// 添加 id 列 // 添加 id 列
String sql = "ALTER TABLE `" + tableName + "` ADD COLUMN `id` int(11) primary key auto_increment first"; 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() { public static List<Record> getExcelTable() {
// 查询所有以 excel_ 开头的表 // 查询所有以 excel_ 开头的表
String sql = "SELECT TABLE_NAME FROM information_schema.TABLES WHERE TABLE_SCHEMA = 'dataease' AND TABLE_NAME LIKE '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 + "非空,正在去掉不允许为空的限制..."); System.out.println("列" + col + "非空,正在去掉不允许为空的限制...");
//去掉不允许为空的限制 //去掉不允许为空的限制
String sql = "ALTER TABLE `" + tableName + "` MODIFY `" + col + "` VARCHAR(255) NULL"; 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("去掉不允许为空的限制成功"); System.out.println("去掉不允许为空的限制成功");
} }
} }
@ -148,7 +147,7 @@ public class InitDataEaseDataSet {
PropKit.get("mysql.password").trim(), PropKit.get("mysql.driverClassName")); PropKit.get("mysql.password").trim(), PropKit.get("mysql.driverClassName"));
masterPlugin.start(); 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")); PropKit.get("mysql.password").trim(), PropKit.get("mysql.driverClassName"));
dataEasePlugin.start(); dataEasePlugin.start();
@ -156,7 +155,7 @@ public class InitDataEaseDataSet {
ActiveRecordPlugin arp = new ActiveRecordPlugin("master", masterPlugin); ActiveRecordPlugin arp = new ActiveRecordPlugin("master", masterPlugin);
arp.setDialect(new MysqlDialect()); arp.setDialect(new MysqlDialect());
ActiveRecordPlugin arpDataEase = new ActiveRecordPlugin(DB_NAME, dataEasePlugin); ActiveRecordPlugin arpDataEase = new ActiveRecordPlugin(DataEaseModel.DB_NAME, dataEasePlugin);
arpDataEase.setDialect(new MysqlDialect()); arpDataEase.setDialect(new MysqlDialect());
//遍历sql目录下所有的sql文件 //遍历sql目录下所有的sql文件

Loading…
Cancel
Save