From 6b7b2214ef457a953381226dc42354d237d6b295 Mon Sep 17 00:00:00 2001
From: liulin <lin.liu@aliyun.com>
Date: 星期二, 13 八月 2024 00:39:26 +0800
Subject: [PATCH] 导出mysql
---
src/test/java/com/lunhan/xxx/host/GenCodeGauss.java | 449 +++++++++++++++++++++++++++++++++++++++++--------------
1 files changed, 333 insertions(+), 116 deletions(-)
diff --git a/src/test/java/com/lunhan/xxx/host/GenCodeGauss.java b/src/test/java/com/lunhan/xxx/host/GenCodeGauss.java
index 86923dd..8b089c9 100644
--- a/src/test/java/com/lunhan/xxx/host/GenCodeGauss.java
+++ b/src/test/java/com/lunhan/xxx/host/GenCodeGauss.java
@@ -13,6 +13,7 @@
import java.time.LocalDateTime;
import java.util.Date;
import java.util.*;
+import java.util.stream.Collectors;
public class GenCodeGauss {
private static final String HOST = "113.250.189.120";
@@ -22,7 +23,7 @@
private static final String DB_NAME = "water_basic";
private static final String AUTHOR = "lin.liu";
- private static Connection CONN = null;
+ public static Connection CONN = null;
private static final String DRIVER = "org.postgresql.Driver";
private static final String CONNECTION_STR = "jdbc:postgresql://" + HOST + ":" + PORT + "/" + DB_NAME;
@@ -143,7 +144,9 @@
"import com.lunhan.xxx.entity.dto.*;",
"import com.lunhan.xxx.entity.enums.*;",
"import com.lunhan.xxx.repository.BasicMapperImpl;",
- "import org.springframework.stereotype.Repository;"
+ "import org.springframework.stereotype.Repository;",
+ "",
+ "import java.util.List;"
};
//endregion
@@ -213,12 +216,12 @@
//#endregion
//#region 编辑接口忽略字段配置
- private static final List<String> NO_UPDATE_COLUMNS = Arrays.asList(
+ private static final List<String> NO_UPDATE_COLUMNS = Arrays.asList(
"create_time", "create_user", "create_user_id", "create_user_name",
"update_time", "update_user", "update_user_id", "update_user_name",
"datachange_lasttime", "data_change_last_time", "is_delete"
);
- private static final List<String> UPDATE_TIME_COLUMNS = Arrays.asList(
+ private static final List<String> UPDATE_TIME_COLUMNS = Arrays.asList(
"update_time", "datachange_lasttime", "data_change_last_time"
);
//#endregion
@@ -242,7 +245,7 @@
* 只生成以下配置的表
*/
private static final List<String> ONLY_TABLES = Arrays.asList(
- "admin_menus"
+
);
public static void main(String[] args) {
@@ -250,7 +253,7 @@
System.out.println("conn is null.");
return;
}
- Map<String, String> tables = getTables(DB_NAME);
+ Map<String, String> tables = getTables(CONN, DB_NAME);
for (Map.Entry<String, String> table : tables.entrySet()) {
String tableName = table.getKey();
String tableDesc = table.getValue();
@@ -258,7 +261,7 @@
if (ONLY_TABLES.size() > 0 && BooleanUtils.isFalse(ONLY_TABLES.contains(tableName))) {
continue;
}
- List<Map<String, Object>> columns = getColumns(tableName, DB_NAME);
+ List<Map<String, Object>> columns = getColumns(CONN, tableName);
genPO(columns, tableDesc);
genVO(columns, tableDesc);
genCreateDto(columns, tableName, tableDesc);
@@ -331,16 +334,18 @@
listColumn.add("(type = IdType.ASSIGN_ID)");
}
}
- String columnAnnotation = "\t@TableField(\"" + name + "\"";
- //忽略最后修改时间这一列的新增和编辑
+ if (!"id".equals(name)) {
+ String columnAnnotation = "\t@TableField(\"" + name + "\"";
+ //忽略最后修改时间这一列的新增和编辑
// if (UPDATE_TIME_COLUMNS.stream().anyMatch(c -> c.equalsIgnoreCase(name))) {
// columnAnnotation += ", insertable = false, updatable = false";
// }
+ listColumn.add(columnAnnotation + ")");
+ }
if ("datachange_lasttime".equalsIgnoreCase(name)) {
propName = "dataChangeLastTime";
columnNameUpper = "DataChangeLastTime";
}
- listColumn.add(columnAnnotation + ")");
//追加列
listColumn.add(String.format("\tprivate %s %s;", javaType, propName));
@@ -358,7 +363,7 @@
}
content.append(ConstantFactory.STR_NEWLINE);
content.append(SET_CLASS_DESC_TPL
- .replaceAll("\\{\\#\\=desc\\}", tableDesc)
+ .replaceAll("\\{\\#\\=desc\\}", tableDesc == null ? modelName : tableDesc)
.replaceAll("\\{\\#\\=author\\}", AUTHOR)
.replaceAll("\\{\\#\\=date\\}", LocalDateTimeUtil.todayStr().replaceAll("\\-", "/"))
);
@@ -376,8 +381,10 @@
content.append(ConstantFactory.STR_NEWLINE);
content.append(StringUtil.join(listColumn, ConstantFactory.STR_NEWLINE));
content.append(ConstantFactory.STR_NEWLINE);
- content.append(StringUtil.join(listGetterSetter, ConstantFactory.STR_NEWLINE));
- content.append(ConstantFactory.STR_NEWLINE);
+ if (!isData) {
+ content.append(StringUtil.join(listGetterSetter, ConstantFactory.STR_NEWLINE));
+ content.append(ConstantFactory.STR_NEWLINE);
+ }
content.append("}");
writeFile(GenCodeGauss.OutSet.PO, modelName + GenCodeGauss.SuffixSet.PO + ".java", content.toString());
@@ -419,14 +426,13 @@
content.append(ConstantFactory.STR_NEWLINE);
content.append(SET_CLASS_DESC_TPL
- .replaceAll("\\{\\#\\=desc\\}", tableDesc)
+ .replaceAll("\\{\\#\\=desc\\}", tableDesc == null ? modelName : tableDesc)
.replaceAll("\\{\\#\\=author\\}", AUTHOR)
.replaceAll("\\{\\#\\=date\\}", LocalDateTimeUtil.todayStr().replaceAll("\\-", "/"))
);
content.append(ConstantFactory.STR_NEWLINE);
if (isData) {
content.append("@Data");
- content.append(ConstantFactory.STR_NEWLINE);
}
content.append(ConstantFactory.STR_NEWLINE);
content.append("public class ");
@@ -465,11 +471,17 @@
String modelName = underline2Camel(tableName, true);
String searchName = "Search" + modelName;
String tableNamePackage = tableName.toLowerCase().replaceAll("_", "");
+ boolean isData = Objects.equals(SET_PO_Getter, 2);
StringBuilder content = new StringBuilder();
content.append("package ");
content.append(GenCodeGauss.PackageSet.SEARCH + ConstantFactory.STR_SEMICOLON);
content.append(ConstantFactory.STR_NEWLINE);
+ if (isData) {
+ content.append(ConstantFactory.STR_NEWLINE);
+ content.append("import lombok.Data;");
+ content.append(ConstantFactory.STR_NEWLINE);
+ }
if (SET_SEARCH_IMPORT.length > 0) {
content.append(ConstantFactory.STR_NEWLINE);
for (String s : SET_SEARCH_IMPORT) {
@@ -479,40 +491,46 @@
}
content.append(ConstantFactory.STR_NEWLINE);
content.append(SET_CLASS_DESC_TPL
- .replaceAll("\\{\\#\\=desc\\}", tableDesc)
+ .replaceAll("\\{\\#\\=desc\\}", tableDesc == null ? modelName : tableDesc)
.replaceAll("\\{\\#\\=author\\}", AUTHOR)
.replaceAll("\\{\\#\\=date\\}", LocalDateTimeUtil.todayStr().replaceAll("\\-", "/"))
);
+ if (isData) {
+ content.append(ConstantFactory.STR_NEWLINE);
+ content.append("@Data");
+ }
content.append(ConstantFactory.STR_NEWLINE);
content.append("public class " + searchName + " extends SearchBasicDTO {");
content.append(ConstantFactory.STR_NEWLINE);
- content.append("\t@Override");
- content.append(ConstantFactory.STR_NEWLINE);
- content.append("\tpublic String toString() {");
- content.append(ConstantFactory.STR_NEWLINE);
- content.append("\t\treturn \"" + searchName + "{\" + \"'\" +");
- content.append(ConstantFactory.STR_NEWLINE);
- content.append("\t\t\t\"keywords='\" + getKeywords() +");
- content.append(ConstantFactory.STR_NEWLINE);
- content.append("\t\t\t\", listId =\" + getListId() +");
- content.append(ConstantFactory.STR_NEWLINE);
- content.append("\t\t\t\", status =\" + getStatus() +");
- content.append(ConstantFactory.STR_NEWLINE);
- content.append("\t\t\t\", listStatus =\" + getListStatus() +");
- content.append(ConstantFactory.STR_NEWLINE);
- content.append("\t\t\t\", createTimeRange='\" + getCreateTimeRange() + \"'\" +");
- content.append(ConstantFactory.STR_NEWLINE);
- content.append("\t\t\t\", createTimeStart =\" + getCreateTimeEnd() +");
- content.append(ConstantFactory.STR_NEWLINE);
- content.append("\t\t\t\", createTimeEnd =\" + getCreateTimeEnd() +");
- content.append(ConstantFactory.STR_NEWLINE);
- content.append("\t\t\t\", lastId =\" + getListId() +");
- content.append(ConstantFactory.STR_NEWLINE);
- content.append("\t\t\t\", orderBy='\" + getOrderBy() + \"'\" +");
- content.append(ConstantFactory.STR_NEWLINE);
- content.append("\t\t'}';");
- content.append(ConstantFactory.STR_NEWLINE);
- content.append("\t}");
+ if (!isData) {
+ content.append("\t@Override");
+ content.append(ConstantFactory.STR_NEWLINE);
+ content.append("\tpublic String toString() {");
+ content.append(ConstantFactory.STR_NEWLINE);
+ content.append("\t\treturn \"" + searchName + "{\" + \"'\" +");
+ content.append(ConstantFactory.STR_NEWLINE);
+ content.append("\t\t\t\"keywords='\" + getKeywords() +");
+ content.append(ConstantFactory.STR_NEWLINE);
+ content.append("\t\t\t\", listId =\" + getListId() +");
+ content.append(ConstantFactory.STR_NEWLINE);
+ content.append("\t\t\t\", status =\" + getStatus() +");
+ content.append(ConstantFactory.STR_NEWLINE);
+ content.append("\t\t\t\", listStatus =\" + getListStatus() +");
+ content.append(ConstantFactory.STR_NEWLINE);
+ content.append("\t\t\t\", createTimeRange='\" + getCreateTimeRange() + \"'\" +");
+ content.append(ConstantFactory.STR_NEWLINE);
+ content.append("\t\t\t\", createTimeStart =\" + getCreateTimeEnd() +");
+ content.append(ConstantFactory.STR_NEWLINE);
+ content.append("\t\t\t\", createTimeEnd =\" + getCreateTimeEnd() +");
+ content.append(ConstantFactory.STR_NEWLINE);
+ content.append("\t\t\t\", lastId =\" + getListId() +");
+ content.append(ConstantFactory.STR_NEWLINE);
+ content.append("\t\t\t\", orderBy='\" + getOrderBy() + \"'\" +");
+ content.append(ConstantFactory.STR_NEWLINE);
+ content.append("\t\t'}';");
+ content.append(ConstantFactory.STR_NEWLINE);
+ content.append("\t}");
+ }
content.append(ConstantFactory.STR_NEWLINE);
content.append("}");
@@ -558,7 +576,7 @@
content.append(poName + ";\n");
content.append(ConstantFactory.STR_NEWLINE);
content.append(SET_CLASS_DESC_TPL
- .replaceAll("\\{\\#\\=desc\\}", tableDesc)
+ .replaceAll("\\{\\#\\=desc\\}", tableDesc == null ? modelName : tableDesc)
.replaceAll("\\{\\#\\=author\\}", AUTHOR)
.replaceAll("\\{\\#\\=date\\}", LocalDateTimeUtil.todayStr().replaceAll("\\-", "/"))
);
@@ -568,7 +586,7 @@
content.append(daoName);
content.append(" extends BasicMapperImpl<");
content.append(modelName + GenCodeGauss.SuffixSet.PO);
- content.append("> {");
+ content.append(", " + mapperName + "> {");
content.append(ConstantFactory.STR_NEWLINE);
content.append("\t" + daoName + "(" + mapperName + " mapper) {\n");
content.append("\t\tsuper(mapper);\n");
@@ -624,15 +642,15 @@
"\t\t\t\t\tswitch (orderBy) {\n" +
"\t\t\t\t\t\t// 主键\n" +
"\t\t\t\t\t\tcase ID:\n" +
- "\t\t\t\t\t\t\tqueryWrapper.orderByAsc(" + poName + "::getId);\n" +
+ "\t\t\t\t\t\t\tqueryWrapper.orderByDesc(" + poName + "::getId);\n" +
"\t\t\t\t\t\t\tbreak;\n" +
"\t\t\t\t\t\t// 数据创建时间\n" +
"\t\t\t\t\t\tcase CREATE_TIME:\n" +
- "\t\t\t\t\t\t\tqueryWrapper.orderByAsc(" + poName + "::getCreateTime);\n" +
+ "\t\t\t\t\t\t\tqueryWrapper.orderByDesc(" + poName + "::getCreateTime);\n" +
"\t\t\t\t\t\t\tbreak;\n" +
"\t\t\t\t\t\t// 最后更新时间\n" +
"\t\t\t\t\t\tcase UPDATE_TIME:\n" +
- "\t\t\t\t\t\t\tqueryWrapper.orderByAsc(" + poName + "::getUpdateTime);\n" +
+ "\t\t\t\t\t\t\tqueryWrapper.orderByDesc(" + poName + "::getUpdateTime);\n" +
"\t\t\t\t\t\t\tbreak;\n" +
"\t\t\t\t\t}\n" +
"\t\t\t\t}\n" +
@@ -640,9 +658,28 @@
"\t\t} else {\n" +
"\t\t\tqueryWrapper.orderByDesc(" + poName + "::getId);\n" +
"\t\t}\n" +
- "\t\tPage<" + poName + "> pageResult = DB.selectPage(new Page<>(search.getPage(), search.getLimit()), queryWrapper);\n" +
+ "\t\tPage<" + poName + "> pageResult = super.selectPage(new Page<>(search.getPage(), search.getLimit()), queryWrapper);\n" +
"\t\treturn new PagerResult<>(pageResult.getSize(), pageResult.getPages(), pageResult.getTotal(), pageResult.getRecords());\n" +
- "\t}");
+ "\t}\n" +
+ "\n" +
+ "\tpublic Boolean add(" + poName + " item) {\n" +
+ "\t\tint rowCount = super.insert(item);\n" +
+ "\t\treturn rowCount == 1;\n" +
+ "\t}\n" +
+ "\n" +
+ "\tpublic Boolean addNotIncrement(" + poName + " item) {\n" +
+ "\t\tint rowCount = super.insert(item);\n" +
+ "\t\treturn rowCount == 1;\n" +
+ "\t}\n" +
+ "\n" +
+ "\tpublic " + poName + " getById(Long id) {\n" +
+ "\t\treturn super.get(id);\n" +
+ "\t}\n" +
+ "\n" +
+ "\tpublic List<" + poName + "> getListById(List<Long> listId) {\n" +
+ "\t\treturn super.getList(listId);\n" +
+ "\t}\n"
+ );
content.append("}");
if (new File(GenCodeGauss.OutSet.MAPPER_IMPL + daoName + ".java").exists()) {
@@ -731,14 +768,13 @@
}
content.append(ConstantFactory.STR_NEWLINE);
content.append(SET_CLASS_DESC_TPL
- .replaceAll("\\{\\#\\=desc\\}", tableDesc)
+ .replaceAll("\\{\\#\\=desc\\}", tableDesc == null ? modelName : tableDesc)
.replaceAll("\\{\\#\\=author\\}", AUTHOR)
.replaceAll("\\{\\#\\=date\\}", LocalDateTimeUtil.todayStr().replaceAll("\\-", "/"))
);
content.append(ConstantFactory.STR_NEWLINE);
if (isData) {
content.append("@Data");
- content.append(ConstantFactory.STR_NEWLINE);
}
content.append(ConstantFactory.STR_NEWLINE);
content.append("public class ");
@@ -747,8 +783,10 @@
content.append(ConstantFactory.STR_NEWLINE);
content.append(StringUtil.join(listColumn, ConstantFactory.STR_NEWLINE));
content.append(ConstantFactory.STR_NEWLINE);
- content.append(StringUtil.join(listGetterSetter, ConstantFactory.STR_NEWLINE));
- content.append(ConstantFactory.STR_NEWLINE);
+ if (!isData) {
+ content.append(StringUtil.join(listGetterSetter, ConstantFactory.STR_NEWLINE));
+ content.append(ConstantFactory.STR_NEWLINE);
+ }
content.append("}");
if (new File(GenCodeGauss.OutSet.RequestDTO + tableNamePackage + "/" + createDTOName + ".java").exists()) {
@@ -823,14 +861,13 @@
}
content.append(ConstantFactory.STR_NEWLINE);
content.append(SET_CLASS_DESC_TPL
- .replaceAll("\\{\\#\\=desc\\}", tableDesc)
+ .replaceAll("\\{\\#\\=desc\\}", tableDesc == null ? modelName : tableDesc)
.replaceAll("\\{\\#\\=author\\}", AUTHOR)
.replaceAll("\\{\\#\\=date\\}", LocalDateTimeUtil.todayStr().replaceAll("\\-", "/"))
);
content.append(ConstantFactory.STR_NEWLINE);
if (isData) {
content.append("@Data");
- content.append(ConstantFactory.STR_NEWLINE);
}
content.append(ConstantFactory.STR_NEWLINE);
content.append("public class ");
@@ -839,8 +876,10 @@
content.append(ConstantFactory.STR_NEWLINE);
content.append(StringUtil.join(listColumn, ConstantFactory.STR_NEWLINE));
content.append(ConstantFactory.STR_NEWLINE);
- content.append(StringUtil.join(listGetterSetter, ConstantFactory.STR_NEWLINE));
- content.append(ConstantFactory.STR_NEWLINE);
+ if (!isData) {
+ content.append(StringUtil.join(listGetterSetter, ConstantFactory.STR_NEWLINE));
+ content.append(ConstantFactory.STR_NEWLINE);
+ }
content.append("}");
if (new File(GenCodeGauss.OutSet.RequestDTO + tableNamePackage + "/" + modifyDTOName + ".java").exists()) {
@@ -896,7 +935,7 @@
content.append(ConstantFactory.STR_NEWLINE);
content.append(SET_CLASS_DESC_TPL
- .replaceAll("\\{\\#\\=desc\\}", tableDesc)
+ .replaceAll("\\{\\#\\=desc\\}", tableDesc == null ? modelName : tableDesc)
);
content.append(ConstantFactory.STR_NEWLINE);
content.append("@Mapper");
@@ -971,7 +1010,7 @@
content.append(responseDTOName + ";\n");
content.append(ConstantFactory.STR_NEWLINE);
content.append(SET_CLASS_DESC_TPL
- .replaceAll("\\{\\#\\=desc\\}", tableDesc)
+ .replaceAll("\\{\\#\\=desc\\}", tableDesc == null ? modelName : tableDesc)
.replaceAll("\\{\\#\\=author\\}", AUTHOR)
.replaceAll("\\{\\#\\=date\\}", LocalDateTimeUtil.todayStr().replaceAll("\\-", "/"))
);
@@ -991,7 +1030,7 @@
" // 设置记录创建时间\n" +
" item.setCreateTime(LocalDateTimeUtil.nowTimeStamp());\n" +
" // 是否删除(逻辑删除)初始值\n" +
- " //item.setIsDelete(EYesOrNo.NO.getValue());\n" +
+ " item.setIsDelete(EYesOrNo.NO.getValue());\n" +
"\n" +
" int rowCount = mapper.insert(item);\n" +
" if (rowCount != 1) {\n" +
@@ -1156,7 +1195,7 @@
" return ExecutedResult.failed(\"[" + tableDesc + "]不存在：\" + id);\n" +
" }\n" +
" return ExecutedResult.success(exists);\n" +
- " }\n"+
+ " }\n" +
" protected ExecutedResult<List<" + poName + ">> check4Id(List<Long> listId) {\n" +
" // 从数据库查找" + tableDesc + "\n" +
" List<" + poName + "> list = mapper.getList(listId);\n" +
@@ -1231,7 +1270,7 @@
content.append(responseDTOName + ";\n");
content.append(ConstantFactory.STR_NEWLINE);
content.append(SET_CLASS_DESC_TPL
- .replaceAll("\\{\\#\\=desc\\}", tableDesc)
+ .replaceAll("\\{\\#\\=desc\\}", tableDesc == null ? modelName : tableDesc)
.replaceAll("\\{\\#\\=author\\}", AUTHOR)
.replaceAll("\\{\\#\\=date\\}", LocalDateTimeUtil.todayStr().replaceAll("\\-", "/"))
);
@@ -1350,7 +1389,7 @@
content.append(ConstantFactory.STR_NEWLINE);
content.append(ConstantFactory.STR_NEWLINE);
content.append(SET_CLASS_DESC_TPL
- .replaceAll("\\{\\#\\=desc\\}", tableDesc + " mapper")
+ .replaceAll("\\{\\#\\=desc\\}", tableDesc == null ? modelName : tableDesc + " mapper")
.replaceAll("\\{\\#\\=author\\}", AUTHOR)
.replaceAll("\\{\\#\\=date\\}", LocalDateTimeUtil.todayStr().replaceAll("\\-", "/"))
);
@@ -1368,7 +1407,7 @@
//#endregion
- public static Map<String, String> getTables(String dbSchema) {
+ public static Map<String, String> getTables(Connection connect, String dbSchema) {
// dbSchema="public";
Map<String, String> result = new LinkedHashMap<>();
String sql = "SELECT tb.table_name, d.description AS TABLE_COMMENT" +
@@ -1376,7 +1415,7 @@
" JOIN pg_class c ON c.relname = tb.table_name" +
" LEFT JOIN pg_description d ON d.objoid = c.oid AND d.objsubid = '0'" +
" WHERE tb.table_schema = 'public' ";
- try (PreparedStatement pStmt = CONN.prepareStatement(sql)) {
+ try (PreparedStatement pStmt = connect.prepareStatement(sql)) {
//pStmt.setString(1, dbSchema);
try (ResultSet rs = pStmt.executeQuery()) {
while (rs.next()) {
@@ -1391,7 +1430,7 @@
return result;
}
- public static List<Map<String, Object>> getColumns(String tableName, String dbSchema) {
+ public static List<Map<String, Object>> getColumns(Connection connect, String tableName) {
List<Map<String, Object>> result = new ArrayList<>();
String sql = "SELECT col.table_name AS objName, col.column_name AS name, col.udt_name AS type, col.character_maximum_length AS LENGTH, des.description,\n" +
"col.numeric_precision, \n" +
@@ -1406,7 +1445,7 @@
"AND table_schema = 'public' " +
"ORDER BY " +
"ordinal_position;";
- try (PreparedStatement pStmt = CONN.prepareStatement(sql)) {
+ try (PreparedStatement pStmt = connect.prepareStatement(sql)) {
pStmt.setString(1, tableName);
// pStmt.setString(2, "public");
try (ResultSet rs = pStmt.executeQuery()) {
@@ -1534,6 +1573,9 @@
public static String toDBValue(Object value) {
String result;
+ if (null == value) {
+ return "";
+ }
if (value instanceof Number || value instanceof Boolean) {
result = value.toString();
} else if (value instanceof Timestamp) {
@@ -1552,40 +1594,61 @@
return result;
}
- public static boolean genTable(String tableName, String tableDesc, List<Map<String, Object>> listColumn, Map<String, String> mapPK, Map<String, String> mapIndex) {
- String sql = "DROP TABLE IF EXISTS \"public\".\"" + tableName + "\";\n" +
- "CREATE TABLE \"public\".\"" + tableName + "\" (\n";
+ public static List<String> genTable(String tableName, String tableDesc, List<Map<String, Object>> listColumn, Map<String, String> mapPK, Map<String, String> mapIndex) {
+ List<String> list = new ArrayList<>();
+
+ String sql = "CREATE TABLE \"public\".\"" + tableName + "\" (\n";
Map<String, String> mapColumnDesc = new HashMap<>();
int idx = 0;
+
+ list.add("-- " + tableName + " - " + tableDesc + "\nDROP TABLE IF EXISTS \"public\".\"" + tableName + "\";\n");
+
// 遍历列，构建sql
for (Map<String, Object> column : listColumn) {
idx++;
String name = column.get("name").toString();
String type = column.get("type").toString();
- String desc = column.get("description").toString();
+ Object desc = column.get("description");
+ Object isNullable = column.get("isNullable");
- mapColumnDesc.put(name,desc);
+ mapColumnDesc.put(name, ParameterUtil.dealNullStr(desc));
- sql += "\t\""+ name +"\" " + type;
+ sql += "\t\"" + name + "\" " + GenCodeGauss.dbType2Gauss(type);
switch (type) {
case "char":
case "varchar":
case "nvarchar":
- case "text":
- String length = column.get("length").toString();
- sql += "(" + length+ ") NOT NULL DEFAULT NULL::character varying";
+ if (Objects.nonNull(column.get("length"))) {
+ String length = column.get("length").toString();
+ sql += "(" + length + ")";
+ }
+ if ("1".equals(isNullable)) {
+ sql += " NULL DEFAULT NULL";
+ } else {
+ sql += " NOT NULL DEFAULT NULL::character varying";
+ }
break;
case "numeric":
case "decimal":
String numeric_precision = column.get("numeric_precision").toString();
String numeric_scale = column.get("numeric_scale").toString();
- sql += "(" + numeric_precision+ ","+ numeric_scale +") NOT NULL DEFAULT 0";
+ sql += "(" + numeric_precision + "," + numeric_scale + ")";
+ if ("1".equals(isNullable)) {
+ sql += " NULL DEFAULT NULL";
+ } else {
+ sql += " NOT NULL DEFAULT 0";
+ }
break;
case "datetime":
case "timestamp":
String datetime_precision = column.get("datetime_precision").toString();
- sql += "(" + datetime_precision +") NOT NULL DEFAULT PG_SYSTIMESTAMP()";
+ sql += "(" + datetime_precision + ")";
+ if ("1".equals(isNullable)) {
+ sql += " NULL DEFAULT NULL";
+ } else {
+ sql += " NOT NULL DEFAULT pg_systimestamp()";
+ }
break;
case "int":
case "tinyint":
@@ -1593,7 +1656,12 @@
case "int2":
case "int4":
case "int8":
- sql += " NOT NULL DEFAULT 0";
+ case "bigint":
+ if ("1".equals(isNullable)) {
+ sql += " NULL DEFAULT NULL";
+ } else {
+ sql += " NOT NULL DEFAULT 0";
+ }
break;
default:
@@ -1605,40 +1673,91 @@
}
sql += "\n";
}
- sql += ")\n" +
- ";\n" +
- "COMMENT ON TABLE \"public\".\"" + tableName + "\" IS '" + tableDesc + "';\n";
+ sql += ");\n";
+ list.add(sql);
+ sql = "";
+
+ sql += "COMMENT ON TABLE \"public\".\"" + tableName + "\" IS '" + tableDesc + "';\n";
// 构建字段描述sql
for (Map.Entry<String, String> desc : mapColumnDesc.entrySet()) {
- sql += "COMMENT ON COLUMN \"public\".\"" + tableName + "\".\"" + desc.getKey() +"\" IS '" + desc.getValue().replace("\r\n", " ").replace("\n", " ") + "';\n";
+ sql += "COMMENT ON COLUMN \"public\".\"" + tableName + "\".\"" + desc.getKey() + "\" IS '" + desc.getValue().replace("\r\n", " ").replace("\n", " ") + "';\n";
}
+ list.add("-- 字段描述\n" + sql);
+ sql = "";
+
// 设置主键
if (!mapPK.isEmpty()) {
for (Map.Entry<String, String> key : mapPK.entrySet()) {
- sql += "ALTER TABLE \"public\".\"" + tableName + "\" ADD CONSTRAINT \"PK_" + tableName + "\" PRIMARY KEY (\"" + key.getValue() +"\");\n";
+ sql += "ALTER TABLE \"public\".\"" + tableName + "\" ADD CONSTRAINT \"PK_" + tableName + "\" PRIMARY KEY (\"" + key.getValue() + "\");\n";
}
}
sql += "ALTER TABLE \"public\".\"" + tableName + "\" ADD CONSTRAINT \"PK_" + tableName + "\" PRIMARY KEY (\"id\");\n";
- // 触发器：自动赋值最后更新时间
- sql += "CREATE TRIGGER \"TG_" + tableName + "\" BEFORE UPDATE OF \"update_time\" ON \"public\".\"" + tableName + "\"\n" +
+ list.add("-- 设置主键\n" + sql);
+ sql = "";
+
+ // 触发器：自动赋值最后更新时间
+ sql += "CREATE TRIGGER \"TG_" + tableName + "\" BEFORE INSERT OR UPDATE ON \"public\".\"" + tableName + "\"\n" +
"FOR EACH ROW\n" +
"EXECUTE PROCEDURE \"public\".\"update_timestamp_column\"();";
+ list.add("-- 触发器：自动赋值最后更新时间\n" + sql);
+ sql = "";
+ list.add("-- 创建索引\n");
// 遍历索引，构建sql
for (Map.Entry<String, String> index : mapIndex.entrySet()) {
if (!mapPK.containsKey(index.getKey())) {
sql += "CREATE INDEX \"" + index.getKey() + "\" ON \"public\".\"test_info\"(\"" + index.getValue() + "\");\n";
+ list.add(sql);
+ sql = "";
}
}
- try (PreparedStatement pStmt = CONN.prepareCall(sql)) {
- int[] result = pStmt.executeBatch();
- return result != null;
+ return list;
+ }
+
+ private static String dbType2Gauss(String type) {
+ String result = "";
+ switch (type) {
+ case "char":
+ case "nvarchar":
+ result = "varchar";
+ break;
+
+ case "longtext":
+ result = "text";
+ break;
+
+ case "decimal":
+ result = "numeric";
+ break;
+ case "datetime":
+ result = "timestamp";
+ break;
+
+ case "bit":
+ case "tinyint":
+ result = "int2";
+ break;
+
+ case "int":
+ result = "int4";
+ break;
+
+ default:
+ result = type;
+ break;
+ }
+ return result;
+ }
+
+ public static boolean execSql(Connection connect, String sql) {
+ try(PreparedStatement pst = connect.prepareStatement(sql)) {
+ int rowCount = pst.executeUpdate();
+ return rowCount > 0;
} catch (Exception e) {
- System.out.println(e);
- return Boolean.FALSE;
+ return false;
}
}
- public static Map<String, String> getPK(String tableName) {
+ public static Map<String, String> getPK(Connection connect, String tableName) {
Map<String, String> mapPK = new HashMap<>();
// 查询主键
String pkSq = "select pg_attribute.attname as col_name,pg_constraint.conname as pk_name\n" +
@@ -1647,7 +1766,7 @@
"inner join pg_attribute on pg_attribute.attrelid=pg_class.oid and pg_attribute.attnum=pg_constraint.conkey[1]\n" +
"where pg_class.relname='" + tableName + "'\n" +
"and pg_constraint.contype='p'";
- try (PreparedStatement pStmt = CONN.prepareStatement(pkSq)) {
+ try (PreparedStatement pStmt = connect.prepareStatement(pkSq)) {
try (ResultSet rs = pStmt.executeQuery()) {
//获取键名
ResultSetMetaData md = rs.getMetaData();
@@ -1665,11 +1784,11 @@
return mapPK;
}
- public static Map<String, String> getIndex(String tableName) {
+ public static Map<String, String> getIndex(Connection connect, String tableName) {
Map<String, String> mapPK = new HashMap<>();
// 查询主键
- String pkSq = "SELECT * FROM pg_indexes WHERE tablename = '" + tableName +"';";
- try (PreparedStatement pStmt = CONN.prepareStatement(pkSq)) {
+ String pkSq = "SELECT * FROM pg_indexes WHERE tablename = '" + tableName + "';";
+ try (PreparedStatement pStmt = connect.prepareStatement(pkSq)) {
try (ResultSet rs = pStmt.executeQuery()) {
//获取键名
ResultSetMetaData md = rs.getMetaData();
@@ -1685,10 +1804,9 @@
return mapPK;
}
- public static List<Map<String, Object>> getListAll(String tableName, String dbSchema) {
+ public static List<Map<String, Object>> getListAll(Connection connect, String sql) {
List<Map<String, Object>> result = new ArrayList<>();
- String sql = "SELECT * FROM " + dbSchema + "." + tableName;
- try (PreparedStatement pStmt = CONN.prepareStatement(sql)) {
+ try (PreparedStatement pStmt = connect.prepareStatement(sql)) {
try (ResultSet rs = pStmt.executeQuery()) {
//获取键名
ResultSetMetaData md = rs.getMetaData();
@@ -1710,14 +1828,17 @@
return result;
}
- public static List<Boolean> insertAll(List<Map<String, Object>> list, String tableName, Integer onceCount) {
- List<Boolean> result = new ArrayList<>();
- String insertSql = "INSERT INTO public." + tableName + "(";
+ public static List<String> insertAll(List<Map<String, Object>> list, String tableName, Integer onceCount) {
+ List<String> result = new ArrayList<>();
+ String insertSql = "INSERT INTO public.\"" + tableName + "\"(\"";
List<String> listColumn = new ArrayList<>();
for (String column : list.get(0).keySet()) {
- listColumn.add(column);
+ if ("update_time".equalsIgnoreCase(column)) {
+ continue;
+ }
+ listColumn.add("\"" + column + "\"");
}
- insertSql += StringUtil.join(listColumn) + ") VALUES";
+ insertSql += StringUtil.join(listColumn, "\",\"") + "\") VALUES";
String sql = "";
for (int i = 0; i < list.size(); i++) {
@@ -1725,22 +1846,118 @@
sql = insertSql;
}
Map<String, Object> row = list.get(i);
- List<String> listValue = new ArrayList<>();
+
+ List<Object> listColValue = new ArrayList<>();
for (Map.Entry<String, Object> item : row.entrySet()) {
- listValue.add(GenCodeGauss.toDBValue(item.getValue()));
- }
- sql += "(" + StringUtil.join(listValue) + ")";
- if (i > 0 && i % onceCount == 0) {
- try (PreparedStatement pStmt = CONN.prepareStatement(sql)) {
- boolean insertList = pStmt.execute();
- result.add(insertList);
- System.out.println("插入数据" + Math.max(i - onceCount, 0) + " - " + i + "，结果：" + insertList);
- } catch (SQLException e) {
- e.printStackTrace();
+ if ("update_time".equalsIgnoreCase(item.getKey())) {
+ continue;
}
+ listColValue.add(item.getValue());
+ }
+ sql += "\n(" + StringUtil.join(listColValue.stream().map(GenCodeGauss::getSqlValue).collect(Collectors.toList())) + "),";
+
+ if (i > 0 && i % onceCount == 0) {
+ sql = sql.substring(0, sql.length() - 1);
+
+ result.add(sql + ";");
+
sql = "";
}
}
+ if (StringUtil.isNotNullOrEmpty(sql)) {
+ sql = sql.substring(0, sql.length() - 1);
+
+ result.add(sql + ";");
+ }
return result;
}
+ private static String getSqlValue(Object value) {
+ if (Objects.isNull(value)) {
+ return "null";
+ }
+ if (value instanceof String) {
+ return "'" + value + "'";
+ }
+ if (value instanceof Date) {
+ return "'" + LocalDateTimeUtil.toFormatFullString(((Date)value).getTime()) + "'";
+ }
+ if (value instanceof LocalDate) {
+ return "'" + LocalDateTimeUtil.toFormatString((LocalDate)value) + "'";
+ }
+ if (value instanceof LocalDateTime) {
+ return "'" + LocalDateTimeUtil.toFormatFullString((LocalDateTime)value) + "'";
+ }
+ if (value instanceof Calendar) {
+ return "'" + CalendarUtil.toDateTimeMSStr((Calendar)value) + "'";
+ }
+ return value.toString();
+ }
+
+ public static String getAutoUpdateTimeFunction() {
+ return "CREATE OR REPLACE FUNCTION \"public\".\"update_timestamp_column\"()\n" +
+ " RETURNS \"pg_catalog\".\"trigger\" AS $BODY$\n" +
+ "BEGIN\n" +
+ "NEW.update_time := current_timestamp;\n" +
+ "RETURN NEW;\n" +
+ "END;\n" +
+ "$BODY$\n" +
+ " LANGUAGE plpgsql VOLATILE\n" +
+ " COST 100";
+ }
+
+ public static List<String> export(Connection connect, String dbSchema, boolean isData, boolean isAutoUpdateTimeFunc) {
+ List<String> listTable = new ArrayList<>();
+ List<String> listData = new ArrayList<>();
+ if (isAutoUpdateTimeFunc) {
+ listTable.add(GenCodeGauss.getAutoUpdateTimeFunction());
+ }
+ // 获取所有表
+ Map<String, String> tables = GenCodeGauss.getTables(connect, dbSchema);
+ // 遍历所有表
+ for (Map.Entry<String, String> table : tables.entrySet()) {
+ String tabName = table.getKey();
+ String tabDesc = table.getValue();
+
+ // 获取所有列
+ List<Map<String, Object>> listCol = GenCodeGauss.getColumns(connect, tabName);
+
+ Map<String, String> mapPK = GenCodeGauss.getPK(connect, tabName);
+
+ Map<String, String> mapIndex = GenCodeGauss.getIndex(connect, tabName);
+
+ // 生成表的sql
+ List<String> listGenTable = GenCodeGauss.genTable(tabName, tabDesc, listCol, mapPK, mapIndex);
+ listTable.addAll(listGenTable);
+
+ if (isData) {
+ // 获取mysql表所有数据
+ String sqlSelect = "SELECT * FROM " + table.getKey();
+ Integer count = 1;
+ Integer pageSize = ConstantFactory.NUM500;
+ String orderBy = "id";
+ Object lastId = null;
+ while (count > 0) {
+ count = 0;
+ String sql = sqlSelect + (Objects.isNull(lastId) ? "" : " WHERE " + orderBy + " > " + GenCodeGauss.toDBValue(lastId) + "") + " ORDER BY " + orderBy + " LIMIT " + pageSize;
+ List<Map<String, Object>> list = GenCodeGauss.getListAll(connect, sql);
+ if (ListUtil.isNullOrEmpty(list)) {
+ continue;
+ }
+ count = list.size();
+ lastId = list.get(list.size() - 1).get(orderBy);
+
+ // 批量插入gauss数据库
+ List<String> insertAll = GenCodeGauss.insertAll(list, table.getKey(), ConstantFactory.NUM50);
+ listData.add(StringUtil.join(insertAll, "\n"));
+ }
+ }
+ }
+ if (ListUtil.isNotNullOrEmpty(listData)) {
+ listTable.addAll(listData);
+ }
+ return listTable;
+ }
+ public static List<String> export(Connection connect, String dbSchema) {
+ return GenCodeGauss.export(connect, dbSchema, Boolean.TRUE, Boolean.TRUE);
+ }
}
--
Gitblit v1.9.3