From 39815aaeb317c656eac4c67555ffb1148fb081b3 Mon Sep 17 00:00:00 2001 From: liulin <lin.liu@aliyun.com> Date: 星期五, 02 八月 2024 17:00:24 +0800 Subject: [PATCH] mysql迁移数据库到gauss --- src/test/java/com/lunhan/xxx/host/GenCodeGauss.java | 264 +++++ /dev/null | 0 src/test/java/com/lunhan/xxx/host/Mysql2Gauss.java | 53 + src/test/resources/application-test.yml | 47 src/test/resources/application.yml | 35 pom.xml | 10 src/test/java/com/lunhan/xxx/host/GenCodeMysql.java | 1996 +++++++++++++++++++++++++++++++++++++++++++++++ src/test/resources/application-dev.yml | 39 src/test/resources/application-prod.yml | 40 9 files changed, 2,412 insertions(+), 72 deletions(-) diff --git a/pom.xml b/pom.xml index 6549552..c02c4c3 100644 --- a/pom.xml +++ b/pom.xml @@ -77,11 +77,11 @@ </dependency> <!-- mysql数据库驱动 --> -<!-- <dependency>--> -<!-- <groupId>mysql</groupId>--> -<!-- <artifactId>mysql-connector-java</artifactId>--> -<!-- <version>${mysql-connector.version}</version>--> -<!-- </dependency>--> + <dependency> + <groupId>mysql</groupId> + <artifactId>mysql-connector-java</artifactId> + <version>${mysql-connector.version}</version> + </dependency> <!-- open-gauss数据库驱动 --> <dependency> <groupId>org.opengauss</groupId> diff --git a/src/test/java/com/lunhan/xxx/host/GenCodeGauss.java b/src/test/java/com/lunhan/xxx/host/GenCodeGauss.java index de0f69b..86923dd 100644 --- a/src/test/java/com/lunhan/xxx/host/GenCodeGauss.java +++ b/src/test/java/com/lunhan/xxx/host/GenCodeGauss.java @@ -1,10 +1,7 @@ package com.lunhan.xxx.host; import com.lunhan.xxx.common.ConstantFactory; -import com.lunhan.xxx.common.util.ListUtil; -import com.lunhan.xxx.common.util.LocalDateTimeUtil; -import com.lunhan.xxx.common.util.NumericUtil; -import com.lunhan.xxx.common.util.StringUtil; +import com.lunhan.xxx.common.util.*; import org.apache.commons.lang3.BooleanUtils; import java.io.File; @@ -12,6 +9,9 @@ import java.io.IOException; import java.io.PrintWriter; import java.sql.*; +import java.time.LocalDate; +import java.time.LocalDateTime; +import java.util.Date; import java.util.*; public class GenCodeGauss { @@ -19,7 +19,7 @@ private static final String PORT = "57654"; private static final String USER = "lunhan"; private static final String PASSWORD = "lunhan.20240330"; - private static final String DB_NAME = "test_db1"; + private static final String DB_NAME = "water_basic"; private static final String AUTHOR = "lin.liu"; private static Connection CONN = null; @@ -1368,7 +1368,7 @@ //#endregion - private static Map<String, String> getTables(String dbSchema) { + public static Map<String, String> getTables(String dbSchema) { // dbSchema="public"; Map<String, String> result = new LinkedHashMap<>(); String sql = "SELECT tb.table_name, d.description AS TABLE_COMMENT" + @@ -1391,27 +1391,20 @@ return result; } - private static List<Map<String, Object>> getColumns(String tableName, String dbSchema) { + public static List<Map<String, Object>> getColumns(String tableName, String dbSchema) { List<Map<String, Object>> result = new ArrayList<>(); - String sql = "select " + - "col.table_name AS objName, " + - "col.column_name AS name, " + - "col.data_type AS type, " + - "col.character_maximum_length AS LENGTH, " + - "des.description, " + - "col.numeric_precision, " + - "col.numeric_scale," + - "CASE is_identity WHEN 'YES' THEN 1 ELSE 0 END AS isIdentity, " + - "CASE col.IS_NULLABLE WHEN 'YES' THEN 1 ELSE 0 END AS isNullable, " + - "col.column_default AS default " + - "from " + - "information_schema.columns col left join pg_description des on " + - "col.table_name::regclass = des.objoid " + - "and col.ordinal_position = des.objsubid " + - "where " + - "table_name = ? " + - "and table_schema = 'public' " + - "order by " + + 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" + + "col.numeric_scale,\n" + + "col.datetime_precision,\n" + + "CASE is_identity WHEN 'YES' THEN 1 ELSE 0 END AS isIdentity,\n" + + "CASE col.IS_NULLABLE WHEN 'YES' THEN 1 ELSE 0 END AS isNullable,\n" + + "col.column_default AS default\n" + + "FROM information_schema.columns col \n" + + "LEFT JOIN pg_description des ON col.table_name::regclass = des.objoid AND col.ordinal_position = des.objsubid\n" + + "WHERE table_name = ? " + + "AND table_schema = 'public' " + + "ORDER BY " + "ordinal_position;"; try (PreparedStatement pStmt = CONN.prepareStatement(sql)) { pStmt.setString(1, tableName); @@ -1437,9 +1430,11 @@ return result; } - private static String getJavaType(String typeName) { + public static String getJavaType(String typeName) { String result = "String"; switch (typeName) { + case "int2": + case "int4": case "integer": case "tinyint": result = "Integer"; @@ -1462,6 +1457,7 @@ case "decimal": result = "BigDecimal"; break; + case "int8": case "bigint": result = "Long"; break; @@ -1475,7 +1471,7 @@ return result; } - private static String underline2Camel(String line, boolean... firstIsUpperCase) { + public static String underline2Camel(String line, boolean... firstIsUpperCase) { String str = ""; if (StringUtil.isNullOrEmpty(line)) { return str; @@ -1535,4 +1531,216 @@ e.printStackTrace(); } } + + public static String toDBValue(Object value) { + String result; + if (value instanceof Number || value instanceof Boolean) { + result = value.toString(); + } else if (value instanceof Timestamp) { + result = String.valueOf(((Timestamp) value).getTime()); + } else if (value instanceof LocalDateTime) { + result = "'" + LocalDateTimeUtil.toFormatFullString((LocalDateTime) value) + "'"; + } else if (value instanceof LocalDate) { + result = "'" + LocalDateTimeUtil.toFormatString((LocalDate) value) + "'"; + } else if (value instanceof Date) { + result = "'" + LocalDateTimeUtil.toFormatFullString(((Date) value).getTime()) + "'"; + } else if (value instanceof Calendar) { + result = "'" + CalendarUtil.toDateTimeMSStr((Calendar) value) + "'"; + } else { + result = "'" + value.toString() + "'"; + } + 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"; + + Map<String, String> mapColumnDesc = new HashMap<>(); + int idx = 0; + // 遍历列,构建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(); + + mapColumnDesc.put(name,desc); + + sql += "\t\""+ name +"\" " + 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"; + 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"; + break; + case "datetime": + case "timestamp": + String datetime_precision = column.get("datetime_precision").toString(); + sql += "(" + datetime_precision +") NOT NULL DEFAULT PG_SYSTIMESTAMP()"; + break; + case "int": + case "tinyint": + case "bit": + case "int2": + case "int4": + case "int8": + sql += " NOT NULL DEFAULT 0"; + break; + + default: + sql += " NULL DEFAULT NULL"; + break; + } + if (idx < listColumn.size()) { + sql += ","; + } + sql += "\n"; + } + sql += ")\n" + + ";\n" + + "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"; + } + // 设置主键 + 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 (\"id\");\n"; + // 触发器:自动赋值最后更新时间 + sql += "CREATE TRIGGER \"TG_" + tableName + "\" BEFORE UPDATE OF \"update_time\" ON \"public\".\"" + tableName + "\"\n" + + "FOR EACH ROW\n" + + "EXECUTE PROCEDURE \"public\".\"update_timestamp_column\"();"; + // 遍历索引,构建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"; + } + } + try (PreparedStatement pStmt = CONN.prepareCall(sql)) { + int[] result = pStmt.executeBatch(); + return result != null; + } catch (Exception e) { + System.out.println(e); + return Boolean.FALSE; + } + } + + public static Map<String, String> getPK(String tableName) { + Map<String, String> mapPK = new HashMap<>(); + // 查询主键 + String pkSq = "select pg_attribute.attname as col_name,pg_constraint.conname as pk_name\n" + + "from pg_constraint\n" + + "inner join pg_class on pg_constraint.conrelid=pg_class.oid\n" + + "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 (ResultSet rs = pStmt.executeQuery()) { + //获取键名 + ResultSetMetaData md = rs.getMetaData(); + //获取列的数量 + int columnCount = md.getColumnCount(); + while (rs.next()) { + String pk_name = rs.getString("pk_name"); + String col_name = rs.getString("col_name"); + mapPK.put(pk_name, col_name); + } + } + } catch (Exception e) { + System.out.println(e); + } + return mapPK; + } + + public static Map<String, String> getIndex(String tableName) { + Map<String, String> mapPK = new HashMap<>(); + // 查询主键 + String pkSq = "SELECT * FROM pg_indexes WHERE tablename = '" + tableName +"';"; + try (PreparedStatement pStmt = CONN.prepareStatement(pkSq)) { + try (ResultSet rs = pStmt.executeQuery()) { + //获取键名 + ResultSetMetaData md = rs.getMetaData(); + while (rs.next()) { + String indexname = rs.getString("indexname"); + String indexdef = rs.getString("indexdef"); + mapPK.put(indexname, indexdef); + } + } + } catch (Exception e) { + System.out.println(e); + } + return mapPK; + } + + public static List<Map<String, Object>> getListAll(String tableName, String dbSchema) { + List<Map<String, Object>> result = new ArrayList<>(); + String sql = "SELECT * FROM " + dbSchema + "." + tableName; + try (PreparedStatement pStmt = CONN.prepareStatement(sql)) { + try (ResultSet rs = pStmt.executeQuery()) { + //获取键名 + ResultSetMetaData md = rs.getMetaData(); + //获取列的数量 + int columnCount = md.getColumnCount(); + while (rs.next()) { + Map<String, Object> column = new HashMap<>(); + for (int i = 1; i <= columnCount; i++) { + String keyName = md.getColumnLabel(i); + Object value = rs.getObject(i); + column.put(keyName, value); + } + result.add(column); + } + } + } catch (SQLException e) { + e.printStackTrace(); + } + 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 + "("; + List<String> listColumn = new ArrayList<>(); + for (String column : list.get(0).keySet()) { + listColumn.add(column); + } + insertSql += StringUtil.join(listColumn) + ") VALUES"; + + String sql = ""; + for (int i = 0; i < list.size(); i++) { + if (StringUtil.isNullOrEmpty(sql)) { + sql = insertSql; + } + Map<String, Object> row = list.get(i); + List<String> listValue = 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(); + } + sql = ""; + } + } + return result; + } } diff --git a/src/test/java/com/lunhan/xxx/host/GenCodeMysql.java b/src/test/java/com/lunhan/xxx/host/GenCodeMysql.java new file mode 100644 index 0000000..9fd7533 --- /dev/null +++ b/src/test/java/com/lunhan/xxx/host/GenCodeMysql.java @@ -0,0 +1,1996 @@ +package com.lunhan.xxx.host; + +import com.lunhan.xxx.common.ConstantFactory; +import com.lunhan.xxx.common.util.ListUtil; +import com.lunhan.xxx.common.util.LocalDateTimeUtil; +import com.lunhan.xxx.common.util.NumericUtil; +import com.lunhan.xxx.common.util.StringUtil; +import org.apache.commons.lang3.BooleanUtils; + +import java.io.File; +import java.io.FileWriter; +import java.io.IOException; +import java.io.PrintWriter; +import java.sql.*; +import java.util.*; + +public class GenCodeMysql { + private static final String HOST = "113.250.189.120"; + private static final String PORT = "23306"; + private static final String USER = "lunhan"; + private static final String PASSWORD = "lunhan123"; + private static final String DB_NAME = "water_basic"; + private static final String AUTHOR = "lin.liu"; + + public static Connection CONN = null; + private static final String DRIVER = "com.mysql.cj.jdbc.Driver"; + private static final String CONNECTION_STR = "jdbc:mysql://" + HOST + ":" + PORT + "/" + DB_NAME + "?serverTimezone=UTC&useUnicode=true&characterEncoding=utf8&useSSL=false"; + + //region 输出文件路径设置 + class OutSet { + public static final String PO = "./src/main/java/com/lunhan/water/repository/po/"; + public static final String VO = "./src/main/java/com/lunhan/water/repository/vo/"; + public static final String COMVERT_MAPPER = "./src/main/java/com/lunhan/water/service/convert/"; + public static final String COLUMNS_ENUM = "./src/main/java/com/lunhan/water/repository/columns/"; + public static final String DAO = "./src/main/java/com/lunhan/water/repository/dao/"; + public static final String MAPPER = "./src/main/java/com/lunhan/water/repository/"; + public static final String MAPPERXML = "./src/main/resources/mapper/"; + public static final String SERVICE = "./src/main/java/com/lunhan/water/service/"; + public static final String Controller = "./src/main/java/com/lunhan/water/host/controller/"; + public static final String SEARCH = "./src/main/java/com/lunhan/water/entity/dto/search/"; + public static final String RequestDTO = "./src/main/java/com/lunhan/water/entity/request/"; + public static final String ResponseDTO = "./src/main/java/com/lunhan/water/entity/response/"; + } + //endregion + + //region 架包路径设置 + class PackageSet { + public static final String PO = "com.lunhan.water.repository.po"; + public static final String VO = "com.lunhan.water.repository.vo"; + public static final String CONVERT_MAPPER = "com.lunhan.water.service.convert"; + public static final String DAO = "com.lunhan.water.repository.dao"; + public static final String COLUMNS_ENUM = "com.lunhan.water.repository.columns"; + public static final String MAPPER = "com.lunhan.water.repository"; + public static final String SERVICE = "com.lunhan.water.service"; + public static final String Controller = "com.lunhan.water.host.controller"; + public static final String SEARCH = "com.lunhan.water.entity.dto.search"; + public static final String Request = "com.lunhan.water.entity.request"; + public static final String Response = "com.lunhan.water.entity.response"; + } + //endregion + + //region 后缀名设置 + class SuffixSet { + public static final String PO = "PO"; + public static final String VO = "VO"; + public static final String COLUMNS_ENUM = "";//POColumns + public static final String DAO = "Dao"; + public static final String MAPPER = "Mapper"; + public static final String SERVICE = "Service"; + public static final String Controller = "Controller"; + public static final String Request = "Req"; + public static final String Response = "Res"; + public static final String Create = "ReqCreate"; + public static final String Modify = "ReqModify"; + } + //endregion + + //region po中getter/setter生成方式设置 + /** + * po中getter/setter生成方式 传统=1,lombok注解=2 + */ + private static final Integer SET_PO_Getter = 1; + //endregion + + //region po类头部文字设置 + private static final String SET_BEFORE_TEXT = + "/**\n" + + "# __----~~~~~~~~~~~------___\n" + + "# . . ~~//====...... __--~ ~~\n" + + "# -. \\_|// |||\\\\ ~~~~~~::::... /~\n" + + "# ___-==_ _-~o~ \\/ ||| \\\\ _/~~-\n" + + "# __---~~~.==~||\\=_ -_--~/_-~|- |\\\\ \\\\ _/~\n" + + "# _-~~ .=~ | \\\\-_ '-~7 /- / || \\ /\n" + + "# .~ .~ | \\\\ -_ / /- / || \\ /\n" + + "# / ____ / | \\\\ ~-_/ /|- _/ .|| \\ /\n" + + "# |~~ ~~|--~~~~--_ \\ ~==-/ | \\~--===~~ .\\\n" + + "# ' ~-| /| |-~\\~~ __--~~\n" + + "# |-~~-_/ | | ~\\_ _-~ /\\\n" + + "# / \\ \\__ \\/~ \\__\n" + + "# _--~ _/ | .-~~____--~-/ ~~==.\n" + + "# ((->/~ '.|||' -_| ~~-/ , . _||\n" + + "# -_ ~\\ ~~---l__i__i__i--~~_/\n" + + "# _-~-__ ~) \\--______________--~~\n" + + "# //.-~~~-~_--~- |-------~~~~~~~~\n" + + "# //.-~~~--\\\n" + + "# 神兽保佑\n" + + "# 永无BUG!\n" + + "*/"; + //endregion + + //region po类 import 内容设置 + private static final String[] SET_PO_IMPORT = new String[]{ + "import javax.persistence.*;", + "import java.io.Serializable;", + "import java.sql.Timestamp;" + }; + //endregion + + //region vo类 import 内容设置 + private static final String[] SET_VO_IMPORT = new String[]{ + "import com.lunhan.water.common.util.LocalDateTimeUtil;", + "import com.lunhan.water.common.util.NumericUtil;", + "import java.util.Objects;" + }; + //endregion + + //region 实体映射类 import 内容设置 + private static final String[] SET_CONVERT_IMPORT = new String[]{ + "import org.mapstruct.Mapper;" + + "import org.mapstruct.factory.Mappers;" + + "" + + "import java.util.List;" + }; + //endregion + + //region 字段名枚举类 import 内容设置 + private static final String[] SET_COLUMNS_ENUM_IMPORT = new String[]{ + "import com.lunhan.water.common.orm2.enums.ColumnBasic;", + "import java.util.Arrays;", + "import java.util.Objects;" + }; + //endregion + + //region dao类 import 内容设置 + private static final String[] SET_DAO_IMPORT = new String[]{ + "import com.lunhan.water.common.ConstantFactory;", + "import com.lunhan.water.common.PagerResult;", + "import com.lunhan.water.common.orm2.SqlBuilder;", + "import com.lunhan.water.common.util.*;", + "import com.lunhan.water.entity.dto.OrderByDTO;", + "import com.lunhan.water.entity.enums.*;", + "import com.lunhan.water.repository.BasicDao;", + "import org.springframework.stereotype.Repository;", + "import java.util.*;", + "import org.springframework.transaction.annotation.Transactional;" + }; + //endregion + + //region service类 import 内容设置 + private static final String[] SET_Service_IMPORT = new String[]{ + "import com.lunhan.water.common.*;", + "import com.lunhan.water.common.model.Tuple;", + "import com.lunhan.water.common.util.*;", + "import org.apache.commons.lang3.BooleanUtils;", + "import org.springframework.beans.factory.annotation.Autowired;", + "import org.springframework.stereotype.Service;", + "import java.util.ArrayList;", + "import java.util.List;", + "import java.util.Objects;" + }; + //endregion + + //region Controller类 import 内容设置 + private static final String[] SET_Controller_IMPORT = new String[]{ + "import com.lunhan.water.common.ExecutedResult;", + "import com.lunhan.water.common.PagerResult;", + "import com.lunhan.water.common.util.ParameterUtil;", + "import com.lunhan.water.common.validator.ParameterValidateResult;", + "import com.lunhan.water.common.validator.ParameterValidator;", + "import com.lunhan.water.entity.request.ReqListId;", + "import com.lunhan.water.host.BasicController;", + "import org.springframework.beans.factory.annotation.Autowired;", + "import org.springframework.web.bind.annotation.*;", + "import java.util.List;" + }; + //endregion + + //region search类 import 内容设置 + private static final String[] SET_SEARCH_IMPORT = new String[]{ + "import " + PackageSet.SEARCH.replaceAll("\\.search", "") + ".SearchBasicDTO;", + }; + //endregion + + //region 类注释模版 + private static final String SET_CLASS_DESC_TPL = + "/**\n" + + " * {#=desc}\n" + + " * @author {#=author}\n" + + " * @description {#=desc}\n" + + " */"; + private static final String SET_METHOD_DESC_TPL = + " /**\n" + + " * {#=desc}\n" + + " * @author {#=author}\n" + + " * @description {#=desc}\n" + + " */"; + private static final String SET_PRO_DESC_TPL = + "\t/**\n" + + "\t * {#=desc}\n" + + "\t */"; + //endregion + + //#region 创建接口忽略字段配置 + private static final List<String> NO_CREATE_COLUMNS = Arrays.asList( + "id", + "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", "status", "state" + ); + //#endregion + + //#region 编辑接口忽略字段配置 + 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( + "update_time", "datachange_lasttime", "data_change_last_time" + ); + //#endregion + + static { + //加载驱动 + try { + Class.forName(DRIVER); + } catch (ClassNotFoundException e) { + e.printStackTrace(); + } + //获得连接 + try { + CONN = DriverManager.getConnection(CONNECTION_STR, USER, PASSWORD); + } catch (SQLException e) { + e.printStackTrace(); + } + } + + /** + * 只生成以下配置的表 + */ + private static final List<String> ONLY_TABLES = Arrays.asList( + "test_info" + ); + + public static void main(String[] args) { + if (null == CONN) { + System.out.println("conn is null."); + return; + } + Map<String, String> tables = getTables(DB_NAME); + for (Map.Entry<String, String> table : tables.entrySet()) { + String tableName = table.getKey(); + String tableDesc = table.getValue(); + System.out.println(tableName + " - " + tableDesc); + if (ONLY_TABLES.size() > 0 && BooleanUtils.isFalse(ONLY_TABLES.contains(tableName))) { + continue; + } + List<Map<String, Object>> columns = getColumns(tableName, DB_NAME); + genPO(columns, tableDesc); + genVO(columns, tableDesc); + genColumnsEnum(columns, tableDesc); + genCreateDto(columns, tableName, tableDesc); + genModifyDto(columns, tableName, tableDesc); + genDao(tableName, tableDesc); + genSearch(columns, tableDesc); + genConvertMapper(tableName, tableDesc); + genService(tableName, tableDesc); + genController(tableName, tableDesc); + + //genMapperXml(columns, tableDesc); + //genMapper(columns, tableDesc); + } + } + + private static void genPO(List<Map<String, Object>> columns, String tableDesc) { + if (ListUtil.isNullOrEmpty(columns)) { + return; + } + String tableName = columns.get(0).get("objName").toString(); + String modelName = underline2Camel(tableName, true); + String poName = modelName + SuffixSet.PO; + boolean isData = Objects.equals(SET_PO_Getter, 2); + + StringBuilder content = new StringBuilder(); + content.append(SET_BEFORE_TEXT); + content.append(ConstantFactory.STR_NEWLINE); + content.append("package "); + content.append(PackageSet.PO); + content.append(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_PO_IMPORT.length > 0) { + content.append(ConstantFactory.STR_NEWLINE); + for (String s : SET_PO_IMPORT) { + content.append(s); + content.append(ConstantFactory.STR_NEWLINE); + } + } + + boolean hasBigDecimal = false; + List<String> listColumn = new ArrayList<>(); + List<String> listGetterSetter = new ArrayList<>(); + for (Map<String, Object> column : columns) { + String name = column.get("name").toString(); + String desc = String.valueOf(column.get("description")); + String type = column.get("type").toString(); + int length = NumericUtil.tryParseInt(column.get("length"), 0); + String defValue = String.valueOf(column.get("default")); + int isPK = NumericUtil.tryParseInt(column.get("isPK"), 0); + int isIdentity = NumericUtil.tryParseInt(column.get("isIdentity"), 0); + int isNullable = NumericUtil.tryParseInt(column.get("isNullable"), 0); + + String javaType = getJavaType(type); + if ("BigDecimal".equalsIgnoreCase(javaType)) { + hasBigDecimal = true; + } + String propName = underline2Camel(name); + String columnNameUpper = underline2Camel(name, true); + + listColumn.add(SET_PRO_DESC_TPL.replaceAll("\\{\\#\\=desc\\}", desc)); + if (Objects.equals(isPK, 1)) { + listColumn.add("\t@Id"); + } + if (Objects.equals(isIdentity, 1)) { + listColumn.add("\t@GeneratedValue(strategy = GenerationType.AUTO)"); + } + String columnAnnotation = "\t@Column(name = \"" + name + "\""; + //忽略最后修改时间这一列的新增和编辑 + if (UPDATE_TIME_COLUMNS.stream().anyMatch(c -> c.equalsIgnoreCase(name))) { + columnAnnotation += ", insertable = false, updatable = false"; + } + if ("datachange_lasttime".equalsIgnoreCase(name)) { + propName = "dataChangeLastTime"; + columnNameUpper = "DataChangeLastTime"; + } + listColumn.add(columnAnnotation + ")"); + //追加列 + listColumn.add(String.format("\tprivate %s %s;", javaType, propName)); + + listGetterSetter.add(""); + listGetterSetter.add(String.format("\tpublic %s get%s() {", javaType, columnNameUpper)); + listGetterSetter.add(String.format("\t\treturn this.%s;", propName)); + listGetterSetter.add("\t}"); + listGetterSetter.add(String.format("\tpublic void set%s(%s %s) {", columnNameUpper, javaType, propName)); + listGetterSetter.add(String.format("\t\tthis.%s = %s;", propName, propName)); + listGetterSetter.add("\t}"); + } + if (hasBigDecimal) { + content.append("import java.math.BigDecimal;"); + content.append(ConstantFactory.STR_NEWLINE); + } + content.append(ConstantFactory.STR_NEWLINE); + content.append(SET_CLASS_DESC_TPL + .replaceAll("\\{\\#\\=desc\\}", 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("@Entity"); + content.append(ConstantFactory.STR_NEWLINE); + content.append("@Table(name = \"" + tableName); + content.append("\")"); + content.append(ConstantFactory.STR_NEWLINE); + content.append("public class "); + content.append(poName); + content.append(" implements Serializable {"); + 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); + content.append("}"); + + writeFile(OutSet.PO, modelName + SuffixSet.PO + ".java", content.toString()); + } + + private static void genVO(List<Map<String, Object>> columns, String tableDesc) { + if (ListUtil.isNullOrEmpty(columns)) { + return; + } + String tableName = columns.get(0).get("objName").toString(); + String modelName = underline2Camel(tableName, true); + String poName = modelName + SuffixSet.PO; + String voName = modelName + SuffixSet.VO; + boolean isData = Objects.equals(SET_PO_Getter, 2); + + StringBuilder content = new StringBuilder(); + content.append(SET_BEFORE_TEXT); + content.append(ConstantFactory.STR_NEWLINE); + content.append("package "); + content.append(PackageSet.VO); + content.append(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_VO_IMPORT.length > 0) { + content.append(ConstantFactory.STR_NEWLINE); + for (String s : SET_VO_IMPORT) { + content.append(s); + content.append(ConstantFactory.STR_NEWLINE); + } + } + content.append("import "); + content.append(PackageSet.PO); + content.append("."); + content.append(poName + ";\n"); + + content.append(ConstantFactory.STR_NEWLINE); + content.append(SET_CLASS_DESC_TPL + .replaceAll("\\{\\#\\=desc\\}", 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 "); + content.append(voName); + content.append(" extends " + poName + " implements BasicVO {\n"); + content.append(ConstantFactory.STR_NEWLINE); + content.append(" @Override\n" + + " public String getCreateTimeView() {\n" + + " if (NumericUtil.tryParseLong(this.getCreateTime()).compareTo(0L) > 0) {\n" + + " return LocalDateTimeUtil.toFormatString(this.getCreateTime());\n" + + " }\n" + + " return \"\";\n" + + " }\n" + + "\n" + + " @Override\n" + + " public String getUpdateTimeView() {\n" + + " if (Objects.isNull(this.getUpdateTime())) {\n" + + " return \"\";\n" + + " }\n" + + " return LocalDateTimeUtil.toFormatFullString(this.getUpdateTime());\n" + + " }"); + content.append(ConstantFactory.STR_NEWLINE); + content.append("}"); + + if (new File(OutSet.VO + modelName + SuffixSet.VO + ".java").exists()) { + return; + } + writeFile(OutSet.VO, modelName + SuffixSet.VO + ".java", content.toString()); + } + + private static void genSearch(List<Map<String, Object>> columns, String tableDesc) { + if (ListUtil.isNullOrEmpty(columns)) { + return; + } + String tableName = columns.get(0).get("objName").toString(); + String modelName = underline2Camel(tableName, true); + String searchName = "Search" + modelName; + String tableNamePackage = tableName.toLowerCase().replaceAll("_", ""); + + StringBuilder content = new StringBuilder(); + content.append("package "); + content.append(PackageSet.SEARCH + ConstantFactory.STR_SEMICOLON); + content.append(ConstantFactory.STR_NEWLINE); + if (SET_SEARCH_IMPORT.length > 0) { + content.append(ConstantFactory.STR_NEWLINE); + for (String s : SET_SEARCH_IMPORT) { + content.append(s); + content.append(ConstantFactory.STR_NEWLINE); + } + } + content.append(ConstantFactory.STR_NEWLINE); + content.append(SET_CLASS_DESC_TPL + .replaceAll("\\{\\#\\=desc\\}", tableDesc) + .replaceAll("\\{\\#\\=author\\}", AUTHOR) + .replaceAll("\\{\\#\\=date\\}", LocalDateTimeUtil.todayStr().replaceAll("\\-", "/")) + ); + 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}"); + content.append(ConstantFactory.STR_NEWLINE); + content.append("}"); + + if (new File(OutSet.SEARCH + "/" + searchName + ".java").exists()) { + return; + } + writeFile(OutSet.SEARCH + "/", searchName + ".java", content.toString()); + } + + private static void genColumnsEnum(List<Map<String, Object>> columns, String tableDesc) { + if (ListUtil.isNullOrEmpty(columns)) { + return; + } + String tableName = columns.get(0).get("objName").toString(); + String modelName = underline2Camel(tableName, true); + String enumName = "E" + modelName + SuffixSet.COLUMNS_ENUM; + + StringBuilder content = new StringBuilder(); + content.append(SET_BEFORE_TEXT); + content.append(ConstantFactory.STR_NEWLINE); + content.append("package "); + content.append(PackageSet.COLUMNS_ENUM); + content.append(ConstantFactory.STR_SEMICOLON); + content.append(ConstantFactory.STR_NEWLINE); + if (SET_COLUMNS_ENUM_IMPORT.length > 0) { + content.append(ConstantFactory.STR_NEWLINE); + for (String s : SET_COLUMNS_ENUM_IMPORT) { + content.append(s); + content.append(ConstantFactory.STR_NEWLINE); + } + } + content.append(ConstantFactory.STR_NEWLINE); + content.append(SET_CLASS_DESC_TPL + .replaceAll("\\{\\#\\=desc\\}", tableDesc) + .replaceAll("\\{\\#\\=author\\}", AUTHOR) + .replaceAll("\\{\\#\\=date\\}", LocalDateTimeUtil.todayStr().replaceAll("\\-", "/")) + ); + content.append(ConstantFactory.STR_NEWLINE); + content.append("public enum "); + content.append(enumName); + content.append(" implements ColumnBasic {"); + + List<String> listColumn = new ArrayList<>(); + String lastStr = ""; + int idx = 0; + for (Map<String, Object> column : columns) { + idx++; + String name = column.get("name").toString(); + String desc = String.valueOf(column.get("description")); + String type = column.get("type").toString(); + int length = NumericUtil.tryParseInt(column.get("length"), 0); + String defValue = String.valueOf(column.get("default")); + int isPK = NumericUtil.tryParseInt(column.get("isPK"), 0); + int isIdentity = NumericUtil.tryParseInt(column.get("isIdentity"), 0); + int isNullable = NumericUtil.tryParseInt(column.get("isNullable"), 0); + + String propName = underline2Camel(name); + + listColumn.add(SET_PRO_DESC_TPL.replaceAll("\\{\\#\\=desc\\}", desc)); + String str = String.format("\t%s(\"%s\", %s),", name.toUpperCase(), name, idx); + //追加列 + listColumn.add(str); + lastStr = str; + } + content.append(ConstantFactory.STR_NEWLINE); + content.append(StringUtil.join(listColumn, ConstantFactory.STR_NEWLINE).replace(lastStr, lastStr.replace("),", ");"))); + content.append(ConstantFactory.STR_NEWLINE); + content.append(ConstantFactory.STR_NEWLINE); + content.append("\tprivate String columnName;\n"); + content.append("\tprivate Integer value;\n"); + content.append("\tpublic Integer getValue() {\n"); + content.append("\t\treturn value;\n"); + content.append("\t}\n"); + content.append("\tpublic String getColumnName() {\n"); + content.append("\t\treturn columnName;\n"); + content.append("\t}\n"); + content.append("\tstatic final String TABLE_NAME = \"" + tableName + "\";\n"); + content.append("\tpublic String getTableName() {\n"); + content.append("\t\treturn TABLE_NAME;\n"); + content.append("\t}\n"); + content.append(ConstantFactory.STR_NEWLINE); + content.append("\t"); + content.append(enumName); + content.append("(String columnName, Integer value) {\n"); + content.append("\t\tthis.columnName = columnName;\n"); + content.append("\t\tthis.value = value;\n"); + content.append("\t}\n"); + content.append(ConstantFactory.STR_NEWLINE); + content.append("\tpublic static " + enumName + " getByValue(Integer value) {\n"); + content.append("\t\treturn Arrays.stream(" + enumName + ".values())\n"); + content.append("\t\t\t\t.filter(e -> Objects.equals(e.getValue(), value))\n"); + content.append("\t\t\t\t.findAny()\n"); + content.append("\t\t\t\t.orElse(null);\n"); + content.append("\t}\n"); + content.append(ConstantFactory.STR_NEWLINE); + content.append("\t@Override\n"); + content.append("\tpublic String toString() {\n"); + content.append("\t\treturn \"" + enumName + "{\" +\n"); + content.append("\t\t\t\t\"columnName='\" + columnName + '\\'' +\n"); + content.append("\t\t\t\t\", value=\" + value +\n"); + content.append("\t\t\t\t'}';\n"); + content.append("\t}\n"); + content.append("}"); + + writeFile(OutSet.COLUMNS_ENUM, enumName + ".java", content.toString()); + } + + private static void genDao(String tableName, String tableDesc) { + String modelName = underline2Camel(tableName, true); + String daoName = modelName + SuffixSet.DAO; + String poName = modelName + SuffixSet.PO; + String enumName = "E" + modelName; + String searchName = "Search" + modelName; + String tableNamePackage = tableName.toLowerCase().replaceAll("_", ""); + + StringBuilder content = new StringBuilder(); + content.append(SET_BEFORE_TEXT); + content.append(ConstantFactory.STR_NEWLINE); + content.append("package "); + content.append(PackageSet.DAO); + content.append(ConstantFactory.STR_SEMICOLON); + content.append(ConstantFactory.STR_NEWLINE); + if (SET_DAO_IMPORT.length > 0) { + content.append(ConstantFactory.STR_NEWLINE); + for (String s : SET_DAO_IMPORT) { + content.append(s); + content.append(ConstantFactory.STR_NEWLINE); + } + } + content.append("import "); + content.append(PackageSet.COLUMNS_ENUM); + content.append("."); + content.append("E" + modelName + SuffixSet.COLUMNS_ENUM + ";\n"); + content.append("import "); + content.append(PackageSet.PO); + content.append("."); + content.append(poName + ";\n"); + content.append("import "); + content.append(PackageSet.SEARCH); + content.append("."); + content.append(searchName + ";\n"); + content.append(ConstantFactory.STR_NEWLINE); + content.append(SET_CLASS_DESC_TPL + .replaceAll("\\{\\#\\=desc\\}", tableDesc) + .replaceAll("\\{\\#\\=author\\}", AUTHOR) + .replaceAll("\\{\\#\\=date\\}", LocalDateTimeUtil.todayStr().replaceAll("\\-", "/")) + ); + content.append(ConstantFactory.STR_NEWLINE); + content.append("@Repository\n"); + content.append("public class "); + content.append(daoName); + content.append(" extends BasicDao<"); + content.append(modelName + SuffixSet.PO); + content.append("> {"); + content.append(ConstantFactory.STR_NEWLINE); + content.append("\t" + daoName + "() {\n"); + content.append("\t\tsuper(E" + modelName + SuffixSet.COLUMNS_ENUM + ".values());\n"); + content.append("\t}\n"); + content.append(ConstantFactory.STR_NEWLINE); + content.append(" /**\n" + + " * 批量删除数据\n" + + " * @param listId 主键id列表\n" + + " * @return 是否成功\n" + + " */\n" + + " public Boolean removeByListId(List<Long> listId) {\n" + + " return super.remove(new SqlBuilder().where(" + enumName + ".ID.in(listId)));\n" + + " }\n\n"); + content.append(" /**\n" + + " * 根据主键获取数据列表\n" + + " * @param listId 主键id列表\n" + + " * @return 数据实体\n" + + " */\n" + + " @Transactional(readOnly = true)\n" + + " public List<" + poName + "> getListById(List<Long> listId) {\n" + + " return super.getList(new SqlBuilder()\n" + + " .where(" + enumName + ".ID.in(listId))\n" + + " );\n" + + " }\n\n"); + content.append("\tpublic PagerResult<" + poName + "> search(" + searchName + " search) {\n" + + "\t\tSqlBuilder builder = new SqlBuilder()\n" + + "\t\t\t\t// 分页参数\n" + + "\t\t\t\t.page(NumericUtil.tryParseInt(search.getPage(), 1), NumericUtil.tryParseInt(search.getLimit(), 1))\n" + + "\n" + + "\t\t\t\t// 数据非逻辑删除\n" + + "\t\t\t\t//.and(" + enumName + ".IS_DELETE.equal(EYesOrNo.NO.getValue()))\n" + + "\t\t\t\t// 状态非删除状态\n" + + "\t\t\t\t//.and(" + enumName + ".STATUS.notEqual(EState.DELETED.getValue()))\n" + + "\n" + + "\t\t\t\t// 状态\n" + + "\t\t\t\t//.andIf(NumericUtil.tryParseInt(search.getStatus()).compareTo(0) > 0, " + enumName + ".STATUS.equal(search.getStatus()))\n" + + "\t\t\t\t// 状态列表\n" + + "\t\t\t\t//.andIf(ListUtil.isNotNullOrEmpty(search.getListStatus()), " + enumName + ".STATUS.in(search.getListStatus()))\n" + + "\t\t\t\t// id列表\n" + + "\t\t\t\t.andIf(ListUtil.isNotNullOrEmpty(search.getListId()), " + enumName + ".ID.in(search.getListId()))\n" + + "\t\t\t\t// 关键字\n" + + "\t\t\t\t//.andIf(StringUtil.isNotNullOrEmpty(search.getKeywords()), " + enumName + ".NAME.like(search.getKeywords()))\n" + + "\t\t\t\t// 创建时间-起始\n" + + "\t\t\t\t.andIf(NumericUtil.tryParseLong(search.getCreateTimeStart()).compareTo(0L) > 0, " + enumName + ".CREATE_TIME.greaterOrEqual(search.getCreateTimeStart()))\n" + + "\t\t\t\t// 创建时间-截止\n" + + "\t\t\t\t.andIf(NumericUtil.tryParseLong(search.getCreateTimeEnd()).compareTo(0L) > 0, " + enumName + ".CREATE_TIME.lessOrEqual(search.getCreateTimeEnd()))\n" + + "\t\t\t\t;\n" + + "\t\tif (ListUtil.isNotNullOrEmpty(search.getOrderBy())) {\n" + + "\t\t\tList<String> listOrderBy = new ArrayList<>();\n" + + "\t\t\tfor (OrderByDTO item : search.getOrderBy()) {\n" + + "\t\t\t\tEOrderBy orderBy = EOrderBy.getByValue(NumericUtil.tryParseInt(item.getOrderBy(), 1));\n" + + "\t\t\t\tif (Objects.isNull(orderBy)) {\n" + + "\t\t\t\t\tcontinue;\n" + + "\t\t\t\t}\n" + + "\t\t\t\tswitch (orderBy) {\n" + + "\t\t\t\t\tcase ID:\n" + + "\t\t\t\t\t\tlistOrderBy.add(" + enumName + ".ID.getColumnName() +\n" + + "\t\t\t\t\t\t\t\t(item.getIsAsc() ? ConstantFactory.SQL_ASC : ConstantFactory.SQL_DESC)\n" + + "\t\t\t\t\t\t);\n" + + "\t\t\t\t\t\tbreak;\n" + + "\n" + + "\t\t\t\t\tcase CREATE_TIME:\n" + + "\t\t\t\t\t\tlistOrderBy.add(" + enumName + ".CREATE_TIME.getColumnName() +\n" + + "\t\t\t\t\t\t\t\t(item.getIsAsc() ? ConstantFactory.SQL_ASC : ConstantFactory.SQL_DESC)\n" + + "\t\t\t\t\t\t);\n" + + "\t\t\t\t\t\tbreak;\n" + + "\n" + + "\t\t\t\t\tcase UPDATE_TIME:\n" + + "\t\t\t\t\t\tlistOrderBy.add(" + enumName + ".UPDATE_TIME.getColumnName() +\n" + + "\t\t\t\t\t\t\t\t(item.getIsAsc() ? ConstantFactory.SQL_ASC : ConstantFactory.SQL_DESC)\n" + + "\t\t\t\t\t\t);\n" + + "\t\t\t\t\t\tbreak;\n" + + "\n" + + "\t\t\t\t\tdefault:\n" + + "\t\t\t\t\t\tbreak;\n" + + "\t\t\t\t}\n" + + "\t\t\t}\n" + + "\t\t\tbuilder.orderBy(ListUtil.toArray(listOrderBy));\n" + + "\t\t}\n" + + "\t\treturn super.getPageList(builder);\n" + + "\t} else {\n" + + "\t\t\tbuilder.orderBy(" + enumName + ".ID.getColumnName() + ConstantFactory.SQL_DESC);\n" + + "\t\t}"); + content.append("}"); + + if (new File(OutSet.DAO + daoName + ".java").exists()) { + return; + } + writeFile(OutSet.DAO, daoName + ".java", content.toString()); + } + + private static Boolean IsNoCreate(String name, Integer isPK, Integer isIdentity) { + if (Objects.equals(isPK, 1) || Objects.equals(isIdentity, 1) || NO_CREATE_COLUMNS.stream().anyMatch(c -> c.equalsIgnoreCase(name))) { + return Boolean.TRUE; + } + return Boolean.FALSE; + } + + private static Boolean IsNoModify(String name) { + if (NO_UPDATE_COLUMNS.stream().anyMatch(c -> c.equalsIgnoreCase(name))) { + return Boolean.TRUE; + } + return Boolean.FALSE; + } + + private static void genCreateDto(List<Map<String, Object>> columns, String tabName, String tableDesc) { + if (ListUtil.isNullOrEmpty(columns)) { + return; + } + String tableName = columns.get(0).get("objName").toString(); + String modelName = underline2Camel(tableName, true); + String poName = modelName + SuffixSet.PO; + String createDTOName = SuffixSet.Create + modelName; + boolean isData = Objects.equals(SET_PO_Getter, 2); + String tableNamePackage = tableName.toLowerCase().replaceAll("_", ""); + + StringBuilder content = new StringBuilder(); + content.append(SET_BEFORE_TEXT); + content.append(ConstantFactory.STR_NEWLINE); + content.append("package "); + content.append(PackageSet.Request + "." + tableNamePackage); + content.append(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); + } + + boolean hasBigDecimal = false; + List<String> listColumn = new ArrayList<>(); + List<String> listGetterSetter = new ArrayList<>(); + for (Map<String, Object> column : columns) { + String name = column.get("name").toString(); + String desc = String.valueOf(column.get("description")); + String type = column.get("type").toString(); + int length = NumericUtil.tryParseInt(column.get("length"), 0); + String defValue = String.valueOf(column.get("default")); + int isPK = NumericUtil.tryParseInt(column.get("isPK"), 0); + int isIdentity = NumericUtil.tryParseInt(column.get("isIdentity"), 0); + int isNullable = NumericUtil.tryParseInt(column.get("isNullable"), 0); + + String javaType = getJavaType(type); + if ("BigDecimal".equalsIgnoreCase(javaType)) { + hasBigDecimal = true; + } + String propName = underline2Camel(name); + String columnNameUpper = underline2Camel(name, true); + + // 主键和自增列不加入新增接口的入参 + if (GenCodeMysql.IsNoCreate(name, isPK, isIdentity)) { + continue; + } + listColumn.add(SET_PRO_DESC_TPL.replaceAll("\\{\\#\\=desc\\}", desc)); + //追加列 + listColumn.add(String.format("\tprivate %s %s;", javaType, propName)); + + listGetterSetter.add(""); + listGetterSetter.add(String.format("\tpublic %s get%s() {", javaType, columnNameUpper)); + listGetterSetter.add(String.format("\t\treturn this.%s;", propName)); + listGetterSetter.add("\t}"); + listGetterSetter.add(String.format("\tpublic void set%s(%s %s) {", columnNameUpper, javaType, propName)); + listGetterSetter.add(String.format("\t\tthis.%s = %s;", propName, propName)); + listGetterSetter.add("\t}"); + } + if (hasBigDecimal) { + content.append("import java.math.BigDecimal;"); + content.append(ConstantFactory.STR_NEWLINE); + } + content.append(ConstantFactory.STR_NEWLINE); + content.append(SET_CLASS_DESC_TPL + .replaceAll("\\{\\#\\=desc\\}", 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 "); + content.append(createDTOName); + content.append(" {"); + 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); + content.append("}"); + + if (new File(OutSet.RequestDTO + tableNamePackage + "/" + createDTOName + ".java").exists()) { + return; + } + writeFile(OutSet.RequestDTO + tableNamePackage + "/", createDTOName + ".java", content.toString()); + } + + private static void genModifyDto(List<Map<String, Object>> columns, String tabName, String tableDesc) { + if (ListUtil.isNullOrEmpty(columns)) { + return; + } + String tableName = columns.get(0).get("objName").toString(); + String modelName = underline2Camel(tableName, true); + String poName = modelName + SuffixSet.PO; + String modifyDTOName = SuffixSet.Modify + modelName; + boolean isData = Objects.equals(SET_PO_Getter, 2); + String tableNamePackage = tableName.toLowerCase().replaceAll("_", ""); + + StringBuilder content = new StringBuilder(); + content.append(SET_BEFORE_TEXT); + content.append(ConstantFactory.STR_NEWLINE); + content.append("package "); + content.append(PackageSet.Request + "." + tableNamePackage); + content.append(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); + } + + boolean hasBigDecimal = false; + List<String> listColumn = new ArrayList<>(); + List<String> listGetterSetter = new ArrayList<>(); + for (Map<String, Object> column : columns) { + String name = column.get("name").toString(); + String desc = String.valueOf(column.get("description")); + String type = column.get("type").toString(); + int length = NumericUtil.tryParseInt(column.get("length"), 0); + String defValue = String.valueOf(column.get("default")); + int isPK = NumericUtil.tryParseInt(column.get("isPK"), 0); + int isIdentity = NumericUtil.tryParseInt(column.get("isIdentity"), 0); + int isNullable = NumericUtil.tryParseInt(column.get("isNullable"), 0); + + String javaType = getJavaType(type); + if ("BigDecimal".equalsIgnoreCase(javaType)) { + hasBigDecimal = true; + } + String propName = underline2Camel(name); + String columnNameUpper = underline2Camel(name, true); + + // 主键和自增列不加入新增接口的入参 + if (GenCodeMysql.IsNoModify(name)) { + continue; + } + listColumn.add(SET_PRO_DESC_TPL.replaceAll("\\{\\#\\=desc\\}", desc)); + //追加列 + listColumn.add(String.format("\tprivate %s %s;", javaType, propName)); + + listGetterSetter.add(""); + listGetterSetter.add(String.format("\tpublic %s get%s() {", javaType, columnNameUpper)); + listGetterSetter.add(String.format("\t\treturn this.%s;", propName)); + listGetterSetter.add("\t}"); + listGetterSetter.add(String.format("\tpublic void set%s(%s %s) {", columnNameUpper, javaType, propName)); + listGetterSetter.add(String.format("\t\tthis.%s = %s;", propName, propName)); + listGetterSetter.add("\t}"); + } + if (hasBigDecimal) { + content.append("import java.math.BigDecimal;"); + content.append(ConstantFactory.STR_NEWLINE); + } + content.append(ConstantFactory.STR_NEWLINE); + content.append(SET_CLASS_DESC_TPL + .replaceAll("\\{\\#\\=desc\\}", 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 "); + content.append(modifyDTOName); + content.append(" {"); + 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); + content.append("}"); + + if (new File(OutSet.RequestDTO + tableNamePackage + "/" + modifyDTOName + ".java").exists()) { + return; + } + writeFile(OutSet.RequestDTO + tableNamePackage + "/", modifyDTOName + ".java", content.toString()); + } + + private static void genConvertMapper(String tableName, String tableDesc) { + String modelName = underline2Camel(tableName, true); + String poName = modelName + SuffixSet.PO; + String voName = modelName + SuffixSet.VO; + String createDTOName = SuffixSet.Create + modelName; + String modifyDTOName = SuffixSet.Modify + modelName; + String convertMapperName = modelName + "Mapper"; + String tableNamePackage = tableName.toLowerCase().replaceAll("_", ""); + + StringBuilder content = new StringBuilder(); + content.append(SET_BEFORE_TEXT); + content.append(ConstantFactory.STR_NEWLINE); + content.append("package "); + content.append(PackageSet.CONVERT_MAPPER + ConstantFactory.STR_SEMICOLON); + content.append(ConstantFactory.STR_NEWLINE); + content.append(ConstantFactory.STR_NEWLINE); + + content.append("import "); + content.append(PackageSet.Request + "." + tableNamePackage + "."); + content.append(createDTOName + ConstantFactory.STR_SEMICOLON); + content.append(ConstantFactory.STR_NEWLINE); + + content.append("import "); + content.append(PackageSet.Request + "." + tableNamePackage + "."); + content.append(modifyDTOName + ConstantFactory.STR_SEMICOLON); + content.append(ConstantFactory.STR_NEWLINE); + + content.append("import "); + content.append(PackageSet.PO + "."); + content.append(poName + ConstantFactory.STR_SEMICOLON); + content.append(ConstantFactory.STR_NEWLINE); + + content.append("import "); + content.append(PackageSet.VO + "."); + content.append(voName + ConstantFactory.STR_SEMICOLON); + content.append(ConstantFactory.STR_NEWLINE); + + if (SET_CONVERT_IMPORT.length > 0) { + content.append(ConstantFactory.STR_NEWLINE); + for (String s : SET_CONVERT_IMPORT) { + content.append(s); + content.append(ConstantFactory.STR_NEWLINE); + } + } + + content.append(ConstantFactory.STR_NEWLINE); + content.append(SET_CLASS_DESC_TPL + .replaceAll("\\{\\#\\=desc\\}", tableDesc) + ); + content.append(ConstantFactory.STR_NEWLINE); + content.append("@Mapper"); + content.append(ConstantFactory.STR_NEWLINE); + content.append("public interface "); + content.append(convertMapperName); + content.append(" {"); + content.append(ConstantFactory.STR_NEWLINE); + content.append(" " + convertMapperName + " INSTANCE = Mappers.getMapper(" + convertMapperName + ".class);\n" + + "\n" + + " " + poName + " toCreate(" + createDTOName + " request);\n" + + " " + poName + " toModify(" + modifyDTOName + " request);\n" + + "\n" + + " " + voName + " toVo(" + poName + " item);\n" + + " List<" + voName + "> toVo(List<" + poName + "> list);"); + content.append(ConstantFactory.STR_NEWLINE); + content.append("}"); + + if (new File(OutSet.COMVERT_MAPPER + convertMapperName + ".java").exists()) { + return; + } + writeFile(OutSet.COMVERT_MAPPER, convertMapperName + ".java", content.toString()); + } + + private static void genService(String tableName, String tableDesc) { + String modelName = underline2Camel(tableName, true); + String poName = modelName + SuffixSet.PO; + String serviceName = modelName + SuffixSet.SERVICE; + String daoName = modelName + SuffixSet.DAO; + String createDTOName = SuffixSet.Create + modelName; + String modifyDTOName = SuffixSet.Modify + modelName; + String responseDTOName = modelName + "VO"; + String searchName = "Search" + modelName; + String tableNamePackage = tableName.toLowerCase().replaceAll("_", ""); + + StringBuilder content = new StringBuilder(); + content.append(SET_BEFORE_TEXT); + content.append(ConstantFactory.STR_NEWLINE); + content.append("package "); + content.append(PackageSet.SERVICE); + content.append(ConstantFactory.STR_SEMICOLON); + content.append(ConstantFactory.STR_NEWLINE); + if (SET_Service_IMPORT.length > 0) { + content.append(ConstantFactory.STR_NEWLINE); + for (String s : SET_Service_IMPORT) { + content.append(s); + content.append(ConstantFactory.STR_NEWLINE); + } + } + content.append("import "); + content.append(PackageSet.DAO); + content.append("."); + content.append(daoName + ";\n"); + content.append("import "); + content.append(PackageSet.PO); + content.append("."); + content.append(poName + ";\n"); + content.append("import "); + content.append(PackageSet.Request); + content.append("." + tableNamePackage + "."); + content.append(createDTOName + ";\n"); + content.append("import "); + content.append(PackageSet.Request); + content.append("." + tableNamePackage + "."); + content.append(modifyDTOName + ";\n"); + content.append("import "); + content.append(PackageSet.SEARCH); + content.append("."); + content.append(searchName + ";\n"); + content.append("import "); + content.append(PackageSet.VO); + content.append("."); + content.append(responseDTOName + ";\n"); + content.append(ConstantFactory.STR_NEWLINE); + content.append(SET_CLASS_DESC_TPL + .replaceAll("\\{\\#\\=desc\\}", tableDesc) + .replaceAll("\\{\\#\\=author\\}", AUTHOR) + .replaceAll("\\{\\#\\=date\\}", LocalDateTimeUtil.todayStr().replaceAll("\\-", "/")) + ); + content.append(ConstantFactory.STR_NEWLINE); + content.append("@Service\n"); + content.append("public class "); + content.append(serviceName); + content.append(" extends BaseService {\n"); + content.append(" @Autowired\n" + + " private " + daoName + " dao;\n"); + content.append(ConstantFactory.STR_NEWLINE); + content.append(" public ExecutedResult<Long> create(" + createDTOName + " request) {\n" + + " // 转换po\n" + + " " + poName + " item = CopierUtil.mapTo(request, " + poName + ".class);\n" + + " // 设置主键id\n" + + " //item.setId(SnowFlakeUtil.getId());\n" + + " // 设置状态\n" + + " //item.setStatus(EState.NORMAL.getValue());\n" + + " // 设置记录创建时间\n" + + " item.setCreateTime(LocalDateTimeUtil.nowTimeStamp());\n" + + " // 是否删除(逻辑删除)初始值\n" + + " //item.setIsDelete(EYesOrNo.NO.getValue());\n" + + "\n" + + " Boolean result = this.dao.add(item);\n" + + " if (BooleanUtils.isFalse(result)) {\n" + + " return ExecutedResult.failed(\"创建[" + tableDesc + "]失败。\");\n" + + " }\n" + + " return ExecutedResult.success(item.getId());\n" + + " }\n" + + "\n" + + " public ExecutedResult<String> modify(" + modifyDTOName + " request) {\n" + + " // 验证记录是否存在\n" + + " ExecutedResult<" + poName + "> checkExists = this.check4Id(request.getId());\n" + + " if (checkExists.isFailed()) {\n" + + " return ExecutedResult.failed(checkExists.getMsg());\n" + + " }\n" + + " // 转换po\n" + + " " + poName + " item = CopierUtil.mapTo(request, " + poName + ".class);\n" + + "\n" + + " Boolean result = this.dao.modify(item);\n" + + " if (BooleanUtils.isFalse(result)) {\n" + + " return ExecutedResult.failed(\"编辑[" + tableDesc + "]失败。\");\n" + + " }\n" + + " return ExecutedResult.success();\n" + + " }\n" + + "\n" + + " public ExecutedResult<" + responseDTOName + "> get(Long id) {\n" + + " " + responseDTOName + " result = new " + responseDTOName + "();\n" + + "\n" + + " " + poName + " find = dao.getById(id);\n" + + " if (null != find) {\n" + + " result = CopierUtil.mapTo(find, " + responseDTOName + ".class);\n" + + " }\n" + + " return ExecutedResult.success(result);\n" + + " }\n" + + "\n" + + "// public ExecutedResult<String> stop(Long id) {\n" + + "// // 验证记录是否存在\n" + + "// ExecutedResult<" + poName + "> checkExists = this.check4Id(id);\n" + + "// if (checkExists.isFailed()) {\n" + + "// return ExecutedResult.failed(checkExists.getMsg());\n" + + "// }\n" + + "// " + poName + " item = new " + poName + "();\n" + + "// item.setId(id);\n" + + "// item.setStatus(EState.DISABLED.getValue());\n" + + "//\n" + + "// Boolean result = this.dao.modify(item);\n" + + "// if (BooleanUtils.isFalse(result)) {\n" + + "// return ExecutedResult.failed(\"停用[" + tableDesc + "]失败。\");\n" + + "// }\n" + + "// return ExecutedResult.success();\n" + + "// }\n" + + "//\n" + + "// public ExecutedResult<String> enable(Long id) {\n" + + "// // 验证记录是否存在\n" + + "// ExecutedResult<" + poName + "> checkExists = this.check4Id(id);\n" + + "// if (checkExists.isFailed()) {\n" + + "// return ExecutedResult.failed(checkExists.getMsg());\n" + + "// }\n" + + "// " + poName + " item = new " + poName + "();\n" + + "// item.setId(id);\n" + + "// item.setStatus(EState.NORMAL.getValue());\n" + + "//\n" + + "// Boolean result = this.dao.modify(item);\n" + + "// if (BooleanUtils.isFalse(result)) {\n" + + "// return ExecutedResult.failed(\"启用[" + tableDesc + "]失败。\");\n" + + "// }\n" + + "// return ExecutedResult.success();\n" + + "// }\n" + + "//\n" + + "// public ExecutedResult<String> setSort(ReqSetSort request) {\n" + + "// // 验证记录是否存在\n" + + "// ExecutedResult<" + poName + "> checkExists = this.check4Id(request.getId());\n" + + "// if (checkExists.isFailed()) {\n" + + "// return ExecutedResult.failed(checkExists.getMsg());\n" + + "// }\n" + + "// " + poName + " item = new " + poName + "();\n" + + "// item.setId(request.getId());\n" + + "// item.setSort(request.getSort());\n" + + "//\n" + + "// Boolean result = this.dao.modify(item);\n" + + "// if (BooleanUtils.isFalse(result)) {\n" + + "// return ExecutedResult.failed(\"设置[" + tableDesc + "]排序值失败。\");\n" + + "// }\n" + + "// return ExecutedResult.success();\n" + + "// }\n" + + "//\n" + + "// public ExecutedResult<String> remove(Long id) {\n" + + "// Boolean result = this.dao.remove(id);\n" + + "// if (BooleanUtils.isFalse(result)) {\n" + + "// return ExecutedResult.failed(\"删除[" + tableDesc + "]失败。\");\n" + + "// }\n" + + "// return ExecutedResult.success();\n" + + "// }\n" + + "//\n" + + "// public ExecutedResult<String> removeList(List<Long> ids) {\n" + + "// Boolean result = this.dao.removeByListId(ids);\n" + + "// if (BooleanUtils.isFalse(result)) {\n" + + "// return ExecutedResult.failed(\"删除[" + tableDesc + "]失败。\");\n" + + "// }\n" + + "// return ExecutedResult.success();\n" + + "// }\n" + + "\n" + + " public ExecutedResult<List<" + responseDTOName + ">> getList(List<Long> listId) {\n" + + " List<" + responseDTOName + "> result = new ArrayList<>();\n" + + "\n" + + " List<" + poName + "> list = this.dao.getListById(listId);\n" + + " if (ListUtil.isNotNullOrEmpty(list)) {\n" + + " result = CopierUtil.mapTo(list, " + responseDTOName + ".class);\n" + + " }\n" + + " return ExecutedResult.success(result);\n" + + " }\n" + + "\n" + + " public ExecutedResult<PagerResult<" + responseDTOName + ">> search(" + searchName + " search) {\n" + + " // 处理创建时间范围-查询参数\n" + + " Tuple<String, String> createTimeRange = ParameterUtil.getTimeRange(search.getCreateTimeRange());\n" + + " if (StringUtil.isNotNullOrEmpty(createTimeRange.getItem1())) {\n" + + " search.setCreateTimeStart(LocalDateTimeUtil.getTimeStamp(createTimeRange.getItem1()).getTime());\n" + + " }\n" + + " if (StringUtil.isNotNullOrEmpty(createTimeRange.getItem2())) {\n" + + " search.setCreateTimeEnd(LocalDateTimeUtil.getTimeStamp(createTimeRange.getItem2()).getTime());\n" + + " }\n" + + "\n" + + " PagerResult<" + poName + "> pageList = dao.search(search);\n" + + " List<" + responseDTOName + "> listVo = new ArrayList<>();\n" + + " List<" + poName + "> list = pageList.getList();\n" + + " if (ListUtil.isNotNullOrEmpty(list)) {\n" + + " pageList.setLastId(list.get(list.size() - 1).getId());\n" + + " // 转换vo\n" + + " listVo = CopierUtil.mapTo(list, " + responseDTOName + ".class);\n" + + " }\n" + + " PagerResult<" + responseDTOName + "> result = new PagerResult<>(pageList.getLimit(), pageList.getPage(), pageList.getTotal(), listVo);\n" + + " result.setLastId(pageList.getLastId());\n" + + " return ExecutedResult.success(result);\n" + + " }\n" + + "\n" + + " protected ExecutedResult<" + poName + "> check4Id(Long id) {\n" + + " " + poName + " exists = dao.getById(id);\n" + + " if (Objects.isNull(exists)) {\n" + + " return ExecutedResult.failed(\"[" + tableDesc + "]不存在:\" + id);\n" + + " }\n" + + " return ExecutedResult.success(exists);\n" + + " }\n"); + content.append("}"); + + if (new File(OutSet.SERVICE + serviceName + ".java").exists()) { + return; + } + writeFile(OutSet.SERVICE, serviceName + ".java", content.toString()); + } + + private static void genController(String tableName, String tableDesc) { + String apiName = underline2Camel(tableName, false); + String modelName = underline2Camel(tableName, true); + String poName = modelName + SuffixSet.PO; + String voName = modelName + SuffixSet.VO; + String serviceName = modelName + SuffixSet.SERVICE; + String ControllerName = modelName + SuffixSet.Controller; + String createDTOName = SuffixSet.Create + modelName; + String modifyDTOName = SuffixSet.Modify + modelName; + String responseDTOName = modelName + "VO"; + String searchName = "Search" + modelName; + String tableNamePackage = tableName.toLowerCase().replaceAll("_", ""); + + StringBuilder content = new StringBuilder(); + content.append(SET_BEFORE_TEXT); + content.append(ConstantFactory.STR_NEWLINE); + content.append("package "); + content.append(PackageSet.Controller); + content.append(ConstantFactory.STR_SEMICOLON); + content.append(ConstantFactory.STR_NEWLINE); + if (SET_Controller_IMPORT.length > 0) { + content.append(ConstantFactory.STR_NEWLINE); + for (String s : SET_Controller_IMPORT) { + content.append(s); + content.append(ConstantFactory.STR_NEWLINE); + } + } + content.append("import "); + content.append(PackageSet.SERVICE); + content.append("."); + content.append(serviceName + ";\n"); + content.append("import "); + content.append(PackageSet.Request); + content.append("." + tableNamePackage + "."); + content.append(createDTOName + ";\n"); + content.append("import "); + content.append(PackageSet.Request); + content.append("." + tableNamePackage + "."); + content.append(modifyDTOName + ";\n"); + content.append("import "); + content.append(PackageSet.SEARCH); + content.append("."); + content.append(searchName + ";\n"); + content.append("import "); + content.append(PackageSet.VO); + content.append("."); + content.append(responseDTOName + ";\n"); + content.append(ConstantFactory.STR_NEWLINE); + content.append(SET_CLASS_DESC_TPL + .replaceAll("\\{\\#\\=desc\\}", tableDesc) + .replaceAll("\\{\\#\\=author\\}", AUTHOR) + .replaceAll("\\{\\#\\=date\\}", LocalDateTimeUtil.todayStr().replaceAll("\\-", "/")) + ); + content.append(ConstantFactory.STR_NEWLINE); + content.append("@RestController\n" + + "@RequestMapping(value = \"" + apiName + "\")\n"); + content.append("public class "); + content.append(ControllerName); + content.append(" extends BasicController {\n"); + content.append(" @Autowired\n" + + " private " + serviceName + " service;\n"); + content.append(ConstantFactory.STR_NEWLINE); + content.append(SET_METHOD_DESC_TPL + .replaceAll("\\{\\#\\=desc\\}", "创建[" + tableDesc + "]") + .replaceAll("\\{\\#\\=author\\}", AUTHOR) + .replaceAll("\\{\\#\\=date\\}", LocalDateTimeUtil.todayStr().replaceAll("\\-", "/")) + ); + content.append(ConstantFactory.STR_NEWLINE); + content.append(" @PostMapping(value = \"create\")\n" + + " public ExecutedResult<Long> create(@RequestBody " + createDTOName + " request) {\n" + + " //#region 参数验证\n" + + " ParameterValidator validator = new ParameterValidator()\n" + + " // 非空\n" + + " //.addNotNullOrEmpty(ParameterUtil.named(\"名称\"), request.getName())\n" + + " // 限制最大长度\n" + + " //.addLengthMax(ParameterUtil.named(\"名称\"), request.getName(), ConstantFactory.LENGTH_MAX50)\n" + + " ;\n" + + " ParameterValidateResult result = validator.validate();\n" + + " if (result.getIsFiled()) {\n" + + " return failed(result.getErrorMsg());\n" + + " }\n" + + " //#endregion\n" + + " return this.service.create(request);\n" + + " }\n" + + "\n"); + content.append(SET_METHOD_DESC_TPL + .replaceAll("\\{\\#\\=desc\\}", "编辑[" + tableDesc + "]") + .replaceAll("\\{\\#\\=author\\}", AUTHOR) + .replaceAll("\\{\\#\\=date\\}", LocalDateTimeUtil.todayStr().replaceAll("\\-", "/")) + ); + content.append(ConstantFactory.STR_NEWLINE); + content.append(" @PostMapping(value = \"modify\")\n" + + " public ExecutedResult<String> modify(@RequestBody " + modifyDTOName + " request) {\n" + + " //#region 参数验证\n" + + " ParameterValidator validator = new ParameterValidator()\n" + + " // 必须大于0\n" + + " .addGreater(ParameterUtil.named(\"[" + tableDesc + "]id\"), request.getId(), 0L)\n" + + " // 非空\n" + + " //.addNotNullOrEmpty(ParameterUtil.named(\"名称\"), request.getName())\n" + + " // 限制最大长度\n" + + " //.addLengthMax(ParameterUtil.named(\"名称\"), request.getName(), ConstantFactory.LENGTH_MAX50)\n" + + " ;\n" + + " ParameterValidateResult result = validator.validate();\n" + + " if (result.getIsFiled()) {\n" + + " return failed(result.getErrorMsg());\n" + + " }\n" + + " //#endregion\n" + + " return this.service.modify(request);\n" + + " }\n\n"); + content.append(SET_METHOD_DESC_TPL + .replaceAll("\\{\\#\\=desc\\}", "获取[" + tableDesc + "]") + .replaceAll("\\{\\#\\=author\\}", AUTHOR) + .replaceAll("\\{\\#\\=date\\}", LocalDateTimeUtil.todayStr().replaceAll("\\-", "/")) + ); + content.append(ConstantFactory.STR_NEWLINE); + content.append(" @GetMapping(value = \"get\")\n" + + " public ExecutedResult<" + voName + "> get(@RequestParam Long id) {\n" + + " return this.service.get(id);\n" + + " }\n\n"); + content.append(SET_METHOD_DESC_TPL + .replaceAll("\\{\\#\\=desc\\}", "根据id批量获取[" + tableDesc + "]") + .replaceAll("\\{\\#\\=author\\}", AUTHOR) + .replaceAll("\\{\\#\\=date\\}", LocalDateTimeUtil.todayStr().replaceAll("\\-", "/")) + ); + content.append(ConstantFactory.STR_NEWLINE); + content.append(" @PostMapping(value = \"getList\")\n" + + " public ExecutedResult<List<" + voName + ">> getList(@RequestBody ReqListId request) {\n" + + " return this.service.getList(request.getListId());\n" + + " }\n\n"); + content.append(SET_METHOD_DESC_TPL + .replaceAll("\\{\\#\\=desc\\}", "查询[" + tableDesc + "]") + .replaceAll("\\{\\#\\=author\\}", AUTHOR) + .replaceAll("\\{\\#\\=date\\}", LocalDateTimeUtil.todayStr().replaceAll("\\-", "/")) + ); + content.append(ConstantFactory.STR_NEWLINE); + content.append(" @PostMapping(value = \"search\")\n" + + " public ExecutedResult<PagerResult<" + voName + ">> search(@RequestBody " + searchName + " request) {\n" + + " return this.service.search(request);\n" + + " }\n"); + content.append("}"); + + if (new File(OutSet.Controller + ControllerName + ".java").exists()) { + return; + } + writeFile(OutSet.Controller, ControllerName + ".java", content.toString()); + } + + private static String genResultMap(Map<String, String> mapResultMap, Map<String, String> mapResultMapDesc) { + List<String> result = new ArrayList<>(); + for (Map.Entry<String, String> entry : mapResultMap.entrySet()) { + String name = entry.getKey(); + result.add("\t\t<!-- " + mapResultMapDesc.get(entry.getKey()) + " -->"); + result.add("\t\t<result property=\"" + entry.getValue() + "\" column=\"" + name + "\" />"); + } + return StringUtil.join(result, ConstantFactory.STR_NEWLINE); + } + + + //#region gen MyBatis mapper + + private static void genMapperXml(List<Map<String, Object>> columns, String tableDesc) { + if (ListUtil.isNullOrEmpty(columns)) { + return; + } + String tableName = columns.get(0).get("objName").toString(); + String modelName = underline2Camel(tableName, true); + + StringBuilder content = new StringBuilder(); + content.append("<?xml version=\"1.0\" encoding=\"UTF-8\" ?>\n" + + "<!DOCTYPE mapper\n" + + " PUBLIC \"-//mybatis.org//DTD Mapper 3.0//EN\"\n" + + " \"http://mybatis.org/dtd/mybatis-3-mapper.dtd\">\n"); + content.append("<!-- "); + content.append(tableDesc); + content.append(" mapper -->\n"); + content.append(String.format("<mapper namespace=\"%s.%s%s\">\n", PackageSet.MAPPER, modelName, SuffixSet.MAPPER)); + + Map<String, String> mapResultMap = new HashMap<>(); + Map<String, String> mapResultMapDesc = new HashMap<>(); + for (Map<String, Object> column : columns) { + String name = column.get("name").toString(); + String desc = String.valueOf(column.get("description")); + + String propName = underline2Camel(name); + //忽略最后修改时间这一列的新增和编辑 + if ("datachange_lasttime".equalsIgnoreCase(name)) { + propName = "dataChangeLastTime"; + } + + mapResultMap.put(name, propName); + mapResultMapDesc.put(name, desc); + } + + //追加 resultMap + content.append(String.format("\t<resultMap type=\"%s.%s%s\" id=\"resultMap\">\n", PackageSet.PO, modelName, SuffixSet.PO)); + content.append(genResultMap(mapResultMap, mapResultMapDesc)); + content.append("\n\t</resultMap>\n\n"); + + //追加 add 的方法 + content.append(String.format("\t<insert id=\"add\" parameterType=\"%s.%s%s\" keyProperty=\"id\" useGeneratedKeys=\"true\">\n", PackageSet.PO, modelName, SuffixSet.PO)); + content.append(genInsert(tableName, mapResultMap, mapResultMapDesc)); + content.append("\n\t</insert>\n\n"); + + //追加 addBatch 的方法 + content.append(String.format("\t<insert id=\"addBatch\" parameterType=\"%s.%s%s\" useGeneratedKeys=\"true\" keyProperty=\"id\">\n", PackageSet.PO, modelName, SuffixSet.PO)); + content.append(genInsertBath(tableName, mapResultMap, mapResultMapDesc)); + content.append("\n\t</insert>\n\n"); + + //追加 update 的方法 + content.append(String.format("\t<update id=\"update\" parameterType=\"%s.%s%s\" keyProperty=\"id\">\n", PackageSet.PO, modelName, SuffixSet.PO)); + content.append(genUpdate(tableName, mapResultMap, mapResultMapDesc)); + content.append("\n\t</update>\n\n"); + + //追加 updateBatch 的方法 + content.append("\t<update id=\"updateBatch\" parameterType=\"java.util.List\">\n"); + content.append(genUpdateBatch(tableName, mapResultMap, mapResultMapDesc)); + content.append("\n\t</update>\n\n"); + + //追加 delete 的方法 + content.append("\t<delete id=\"delete\">\n"); + content.append("\t\tDELETE FROM `"); + content.append(tableName); + content.append("` WHERE `id` = #{value}\n"); + content.append("\t</delete>\n\n"); + + //追加 deleteBatch 的方法 + content.append("\t<delete id=\"deleteBatch\" parameterType=\"java.util.List\">\n"); + content.append("\t\tDELETE FROM `"); + content.append(tableName); + content.append("` WHERE `id` IN"); + content.append("\t\t<!-- “collection”固定值为list,“item”为循环变量取的别名 -->\n"); + content.append("\t\t<foreach collection=\"list\" item=\"id\" open=\"(\" separator=\",\" close=\")\">\n"); + content.append("\t\t\t#{id}\n"); + content.append("\t\t</foreach>\n"); + content.append("\t</delete>\n\n"); + + //追加 get 的方法 + content.append("\t<select id=\"get\" resultMap=\"resultMap\">\n"); + content.append(genGet(tableName, mapResultMap, mapResultMapDesc)); + content.append("\n\t</select>\n\n"); + + //追加 getCount 的方法 + content.append(String.format("\t<select id=\"getCount\" resultType=\"Long\" parameterType=\"%s.Search%s\">\n", PackageSet.SEARCH, modelName)); + content.append(genGetCount(tableName)); + content.append("\n\t</select>\n\n"); + + //追加 getList 的方法 + content.append(String.format("\t<select id=\"getList\" resultMap=\"resultMap\" parameterType=\"%s.Search%s\">\n", PackageSet.SEARCH, modelName)); + content.append(genGetList(tableName, mapResultMap, mapResultMapDesc)); + content.append("\n\t</select>\n\n"); + + //追加 getListBySql 的方法 + content.append("\t<select id=\"getListBySql\" parameterType=\"java.util.Map\" resultType=\"HashMap\">\n"); + content.append("\t\t${sql}\n"); + content.append("\t</select>\n"); + + content.append("</mapper>"); + + if (new File(OutSet.MAPPERXML + modelName + SuffixSet.MAPPER + ".xml").exists()) { + return; + } + writeFile(OutSet.MAPPERXML, modelName + SuffixSet.MAPPER + ".xml", content.toString()); + } + + private static void genMapper(List<Map<String, Object>> columns, String tableDesc) { + if (ListUtil.isNullOrEmpty(columns)) { + return; + } + String tableName = columns.get(0).get("objName").toString(); + String modelName = underline2Camel(tableName, true); + + StringBuilder content = new StringBuilder(); + content.append("package "); + content.append(PackageSet.MAPPER); + content.append(ConstantFactory.STR_SEMICOLON); + content.append(ConstantFactory.STR_NEWLINE); + content.append(ConstantFactory.STR_NEWLINE); + content.append(String.format("import %s.%s%s;", PackageSet.PO, modelName, SuffixSet.PO)); + content.append(ConstantFactory.STR_NEWLINE); + content.append(ConstantFactory.STR_NEWLINE); + content.append(SET_CLASS_DESC_TPL + .replaceAll("\\{\\#\\=desc\\}", tableDesc + " mapper") + .replaceAll("\\{\\#\\=author\\}", AUTHOR) + .replaceAll("\\{\\#\\=date\\}", LocalDateTimeUtil.todayStr().replaceAll("\\-", "/")) + ); + content.append(ConstantFactory.STR_NEWLINE); + content.append(String.format("public interface %s%s extends BasicMapper<%s%s> {", modelName, SuffixSet.MAPPER, modelName, SuffixSet.PO)); + content.append(ConstantFactory.STR_NEWLINE); + content.append("}"); + + if (new File(OutSet.MAPPER + modelName + SuffixSet.MAPPER + ".java").exists()) { + return; + } + writeFile(OutSet.MAPPER, modelName + SuffixSet.MAPPER + ".java", content.toString()); + } + + private static String genInsert(String tableName, Map<String, String> mapResultMap, Map<String, String> mapResultMapDesc) { + List<String> result = new ArrayList<>(); + result.add("\t\tINSERT INTO `" + tableName + "`("); + + List<String> listColumn = new ArrayList<>(); + List<String> listValue = new ArrayList<>(); + for (Map.Entry<String, String> entry : mapResultMap.entrySet()) { + String name = entry.getKey(); + if (NO_CREATE_COLUMNS.stream().anyMatch(c -> c.equalsIgnoreCase(name))) { + continue; + } + listColumn.add("\t\t<!-- " + mapResultMapDesc.get(entry.getKey()) + " -->\n\t\t`" + name + "`"); + + listValue.add("\t\t#{" + entry.getValue() + "}"); + } + result.add(StringUtil.join(listColumn, ",\r\n")); + result.add("\t\t) VALUES("); + result.add(StringUtil.join(listValue, ",\r\n")); + result.add("\t);"); + return StringUtil.join(result, ConstantFactory.STR_NEWLINE); + } + + private static String genInsertBath(String tableName, Map<String, String> mapResultMap, Map<String, String> mapResultMapDesc) { + List<String> result = new ArrayList<>(); + result.add("\t\tINSERT INTO `" + tableName + "`("); + + List<String> listColumn = new ArrayList<>(); + List<String> listValue = new ArrayList<>(); + for (Map.Entry<String, String> entry : mapResultMap.entrySet()) { + String name = entry.getKey(); + if (NO_CREATE_COLUMNS.stream().anyMatch(c -> c.equalsIgnoreCase(name))) { + continue; + } + listColumn.add("\t\t<!-- " + mapResultMapDesc.get(entry.getKey()) + " -->\n\t\t`" + name + "`"); + + listValue.add("\t\t\t#{item." + entry.getValue() + "}"); + } + result.add(StringUtil.join(listColumn, ",\r\n")); + result.add("\t\t) VALUES("); + result.add("\t\t<!-- “collection”固定值为list,“item”为循环变量取的别名 -->"); + result.add("\t\t<foreach collection=\"list\" item=\"item\" separator=\",\">"); + result.add("\t\t\t("); + result.add(StringUtil.join(listValue, ",\r\n")); + result.add("\t\t\t)"); + result.add("\t\t</foreach>"); + result.add("\t\t);"); + return StringUtil.join(result, ConstantFactory.STR_NEWLINE); + } + + private static String genUpdate(String tableName, Map<String, String> mapResultMap, Map<String, String> mapResultMapDesc) { + List<String> result = new ArrayList<>(); + result.add("\t\tUPDATE `" + tableName + "`"); + result.add("\t\t<trim prefix=\"SET\" suffixOverrides=\",\">"); + for (Map.Entry<String, String> entry : mapResultMap.entrySet()) { + String name = entry.getKey(); + String propName = entry.getValue(); + if (NO_UPDATE_COLUMNS.stream().anyMatch(c -> c.equalsIgnoreCase(name))) { + continue; + } + result.add("\t\t\t<!-- " + mapResultMapDesc.get(entry.getKey()) + " -->"); + result.add("\t\t\t<if test=\"null != " + propName + "\">`" + name + "` = #{" + propName + "},</if>"); + } + result.add("\t\t</trim>"); + result.add("\t\tWHERE `id` = #{id}"); + return StringUtil.join(result, ConstantFactory.STR_NEWLINE); + } + + private static String genUpdateBatch(String tableName, Map<String, String> mapResultMap, Map<String, String> mapResultMapDesc) { + List<String> result = new ArrayList<>(); + result.add("\t\t<!-- “collection”固定值为list,“item”为循环变量取的别名 -->"); + result.add("\t\t<foreach collection=\"list\" item=\"item\" index=\"index\" open=\"\" close=\"\" separator=\";\">"); + result.add("\t\t\tUPDATE `" + tableName + "`"); + result.add("\t\t\t<trim prefix=\"SET\" suffixOverrides=\",\">"); + for (Map.Entry<String, String> entry : mapResultMap.entrySet()) { + String name = entry.getKey(); + String propName = entry.getValue(); + if (NO_UPDATE_COLUMNS.stream().anyMatch(c -> c.equalsIgnoreCase(name))) { + continue; + } + result.add("\t\t\t\t<!-- " + mapResultMapDesc.get(entry.getKey()) + " -->"); + result.add("\t\t\t\t<if test=\"null != item." + propName + "\">`" + name + "` = #{item." + propName + "},</if>"); + } + result.add("\t\t\t</trim>"); + result.add("\t\t\tWHERE `id` = #{id}"); + result.add("\t\t</foreach>"); + return StringUtil.join(result, ConstantFactory.STR_NEWLINE); + } + + private static String genGet(String tableName, Map<String, String> mapResultMap, Map<String, String> mapResultMapDesc) { + List<String> result = new ArrayList<>(); + result.add("\t\t<![CDATA[SELECT"); + + List<String> listColumn = new ArrayList<>(); + for (Map.Entry<String, String> entry : mapResultMap.entrySet()) { + String name = entry.getKey(); + listColumn.add("\t\t\t`" + name + "`"); + } + result.add(StringUtil.join(listColumn, ",\r\n")); + result.add("\t\tFROM `" + tableName + "`]]>"); + result.add("\t\tWHERE `id` = #{id}"); + return StringUtil.join(result, ConstantFactory.STR_NEWLINE); + } + + private static String genGetCount(String tableName) { + List<String> result = new ArrayList<>(); + result.add("\t\t<![CDATA[SELECT COUNT(*) FROM `" + tableName + "`]]>"); + result.add("\t\t<!-- 加上“WHERE”前缀,去掉第一个“AND” -->"); + result.add("\t\t<trim prefix=\"WHERE\" prefixOverrides=\"AND \">"); + + result.add("\t\t\t<!-- 自增id -->"); + result.add("\t\t\t<if test=\"null != id and id > 0\">"); + result.add("\t\t\t\tAND `id` = #{id}"); + result.add("\t\t\t</if>"); + + result.add("\t\t\t<!-- 自增id列表 -->"); + result.add("\t\t\t<if test=\"null != listId and listId.size() > 0\">"); + result.add("\t\t\t\tAND `id` IN"); + result.add("\t\t\t\t<foreach collection=\"listId\" item=\"item\" open=\"(\" separator=\",\" close=\")\">"); + result.add("\t\t\t\t\t#{item}"); + result.add("\t\t\t\t</foreach>"); + result.add("\t\t\t</if>"); + + result.add("\t\t\t<!-- 数据状态 -->"); + result.add("\t\t\t<if test=\"null != status and status > 0\">"); + result.add("\t\t\t\tAND `status` = #{status}"); + result.add("\t\t\t</if>"); + + result.add("\t\t\t<!-- 数据状态列表 -->"); + result.add("\t\t\t<if test=\"null != listStatus and listStatus.size() > 0\">"); + result.add("\t\t\t\tAND `status` IN"); + result.add("\t\t\t\t<foreach collection=\"listStatus\" item=\"item\" open=\"(\" separator=\",\" close=\")\">"); + result.add("\t\t\t\t\t#{item}"); + result.add("\t\t\t\t</foreach>"); + result.add("\t\t\t</if>"); + + result.add("\t\t\t<!-- 关键字模糊搜索 -->"); + result.add("\t\t\t<!-- <if test=\"null != keywords and '' != keywords.trim()\">"); + result.add("\t\t\t\tAND (`user_name` LIKE #{keywords} OR `nick_name` LIKE #{keywords})"); + result.add("\t\t\t</if> -->"); + + result.add("\t\t\t<!-- 数据创建时间-起始 -->"); + result.add("\t\t\t<if test=\"null != createTimeStart and createTimeStart > 0\">"); + result.add("\t\t\t\tAND `create_time` >= #{createTimeStart}"); + result.add("\t\t\t</if>"); + + result.add("\t\t\t<!-- 数据创建时间-截止 -->"); + result.add("\t\t\t<if test=\"null != createTimeEnd and createTimeEnd > 0\">"); + result.add("\t\t\t\tAND `create_time` <= #{createTimeEnd}"); + result.add("\t\t\t</if>"); + + result.add("\t\t</trim>"); + return StringUtil.join(result, ConstantFactory.STR_NEWLINE); + } + + private static String genGetList(String tableName, Map<String, String> mapResultMap, Map<String, String> mapResultMapDesc) { + List<String> result = new ArrayList<>(); + result.add("\t\t<![CDATA[SELECT"); + + List<String> listColumn = new ArrayList<>(); + for (Map.Entry<String, String> entry : mapResultMap.entrySet()) { + String name = entry.getKey(); + listColumn.add("\t\t`" + name + "`"); + } + result.add(StringUtil.join(listColumn, ",\r\n")); + result.add("\t\tFROM `" + tableName + "`]]>"); + result.add("\t\t<!-- 加上“WHERE”前缀,去掉第一个“AND” -->"); + result.add("\t\t<trim prefix=\"WHERE\" prefixOverrides=\"AND \">"); + + result.add("\t\t\t<!-- 自增id -->"); + result.add("\t\t\t<if test=\"null != id and id > 0\">"); + result.add("\t\t\t\tAND `id` = #{id}"); + result.add("\t\t\t</if>"); + + result.add("\t\t\t<!-- 自增id列表 -->"); + result.add("\t\t\t<if test=\"null != listId and listId.size() > 0\">"); + result.add("\t\t\t\tAND `id` IN"); + result.add("\t\t\t\t<foreach collection=\"listId\" item=\"item\" open=\"(\" separator=\",\" close=\")\">"); + result.add("\t\t\t\t\t#{item}"); + result.add("\t\t\t\t</foreach>"); + result.add("\t\t\t</if>"); + + result.add("\t\t\t<!-- 数据状态 -->"); + result.add("\t\t\t<if test=\"null != status and status > 0\">"); + result.add("\t\t\t\tAND `status` = #{status}"); + result.add("\t\t\t</if>"); + + result.add("\t\t\t<!-- 数据状态列表 -->"); + result.add("\t\t\t<if test=\"null != listStatus and listStatus.size() > 0\">"); + result.add("\t\t\t\tAND `status` IN"); + result.add("\t\t\t\t<foreach collection=\"listStatus\" item=\"item\" open=\"(\" separator=\",\" close=\")\">"); + result.add("\t\t\t\t\t#{item}"); + result.add("\t\t\t\t</foreach>"); + result.add("\t\t\t</if>"); + + result.add("\t\t\t<!-- 关键字模糊搜索 -->"); + result.add("\t\t\t<!-- <if test=\"null != keywords and '' != keywords.trim()\">"); + result.add("\t\t\t\tAND (`user_name` LIKE #{keywords} OR `nick_name` LIKE #{keywords})"); + result.add("\t\t\t</if> -->"); + + result.add("\t\t\t<!-- 数据创建时间-起始 -->"); + result.add("\t\t\t<if test=\"null != createTimeStart and createTimeStart > 0\">"); + result.add("\t\t\t\tAND `create_time` >= #{createTimeStart}"); + result.add("\t\t\t</if>"); + + result.add("\t\t\t<!-- 数据创建时间-截止 -->"); + result.add("\t\t\t<if test=\"null != createTimeEnd and createTimeEnd > 0\">"); + result.add("\t\t\t\tAND `create_time` <= #{createTimeEnd}"); + result.add("\t\t\t</if>"); + + result.add("\t\t</trim>"); + result.add("\t\t<choose>"); + result.add("\t\t\t<when test=\"null!=orderBy and orderBy.trim()!=''\">"); + result.add("\t\t\t\tORDER BY ${orderBy}"); + result.add("\t\t\t</when>"); + result.add("\t\t\t<otherwise>"); + result.add("\t\t\t\tORDER BY id DESC"); + result.add("\t\t\t</otherwise>"); + result.add("\t\t</choose>"); + result.add("\t\t<if test=\"null!=pageSize and pageSize>0 and null!=lastRowNo and lastRowNo>=0\">"); + result.add("\t\t\tLIMIT ${lastRowNo}, ${pageSize}"); + result.add("\t\t</if>"); + return StringUtil.join(result, ConstantFactory.STR_NEWLINE); + } + + //#endregion + + + public static Map<String, String> getTables(String dbSchema) { + Map<String, String> result = new LinkedHashMap<>(); + String sql = " SELECT TABLE_NAME ,TABLE_COMMENT " + + " FROM information_schema.`TABLES` " + + " WHERE TABLE_SCHEMA = ? "; + try (PreparedStatement pStmt = CONN.prepareStatement(sql)) { + pStmt.setString(1, dbSchema); + try (ResultSet rs = pStmt.executeQuery()) { + while (rs.next()) { + String tableName = rs.getString("TABLE_NAME"); + String tableComment = rs.getString("TABLE_COMMENT"); + result.put(tableName, tableComment); + } + } + } catch (SQLException e) { + e.printStackTrace(); + } + return result; + } + + public static List<Map<String, Object>> getColumns(String tableName, String dbSchema) { + List<Map<String, Object>> result = new ArrayList<>(); + String sql = "SELECT `TABLE_NAME` AS'objName', `COLUMN_NAME`AS'name', `DATA_TYPE` AS'type', `CHARACTER_MAXIMUM_LENGTH` AS'length', `COLUMN_COMMENT` AS'description',\n" + + "`COLUMN_DEFAULT` AS'default',NUMERIC_PRECISION AS'numeric_precision',NUMERIC_SCALE AS'numeric_scale',DATETIME_PRECISION AS'datetime_precision',\n" + + "CASE `EXTRA` WHEN 'auto_increment' THEN CONVERT(1,UNSIGNED) ELSE CONVERT(0,UNSIGNED) END AS'isIdentity',\n" + + "CASE `COLUMN_KEY` WHEN 'PRI' THEN CONVERT(1,UNSIGNED) ELSE CONVERT(0,UNSIGNED) END AS'isPK', \n" + + "CASE `IS_NULLABLE` WHEN 'YES' THEN CONVERT(1,UNSIGNED) ELSE CONVERT(0,UNSIGNED) END AS'isNullable'\n" + + "FROM `information_schema`.`COLUMNS`\n" + + "WHERE `table_name` = ? AND `table_schema` = ?; "; + try (PreparedStatement pStmt = CONN.prepareStatement(sql)) { + pStmt.setString(1, tableName); + pStmt.setString(2, dbSchema); + try (ResultSet rs = pStmt.executeQuery()) { + //获取键名 + ResultSetMetaData md = rs.getMetaData(); + //获取列的数量 + int columnCount = md.getColumnCount(); + while (rs.next()) { + Map<String, Object> column = new HashMap<>(); + for (int i = 1; i <= columnCount; i++) { + String keyName = md.getColumnLabel(i); + Object value = rs.getObject(i); + column.put(keyName, value); + } + result.add(column); + } + } + } catch (SQLException e) { + e.printStackTrace(); + } + return result; + } + + public static List<Map<String, Object>> getListAll(String sql) { + List<Map<String, Object>> result = new ArrayList<>(); + try (PreparedStatement pStmt = CONN.prepareStatement(sql)) { + try (ResultSet rs = pStmt.executeQuery()) { + //获取键名 + ResultSetMetaData md = rs.getMetaData(); + //获取列的数量 + int columnCount = md.getColumnCount(); + while (rs.next()) { + Map<String, Object> column = new HashMap<>(); + for (int i = 1; i <= columnCount; i++) { + String keyName = md.getColumnLabel(i); + Object value = rs.getObject(i); + column.put(keyName, value); + } + result.add(column); + } + } + } catch (SQLException e) { + e.printStackTrace(); + } + return result; + } + + public static String getJavaType(String typeName) { + String result = "String"; + switch (typeName) { + case "int": + case "tinyint": + result = "Integer"; + break; + case "char": + case "varchar": + case "nvarchar": + case "text": + result = "String"; + break; + case "datetime": + case "timestamp": + result = "Timestamp"; + break; + case "bit": + result = "Boolean"; + break; + case "numeric": + case "decimal": + result = "BigDecimal"; + break; + case "bigint": + result = "Long"; + break; + case "date": + result = "Date"; + break; + + default: + break; + } + return result; + } + + public static String underline2Camel(String line, boolean... firstIsUpperCase) { + String str = ""; + if (StringUtil.isNullOrEmpty(line)) { + return str; + } else { + StringBuilder sb = new StringBuilder(); + String[] strArr; + // 不包含下划线,且第二个参数是空的 + if (!line.contains("_") && firstIsUpperCase.length == 0) { + sb.append(line.substring(0, 1).toLowerCase()).append(line.substring(1)); + str = sb.toString(); + } else if (!line.contains("_") && firstIsUpperCase.length != 0) { + if (!firstIsUpperCase[0]) { + sb.append(line.substring(0, 1).toLowerCase()).append(line.substring(1)); + str = sb.toString(); + } else { + sb.append(line.substring(0, 1).toUpperCase()).append(line.substring(1)); + str = sb.toString(); + } + } else if (line.contains("_") && firstIsUpperCase.length == 0) { + strArr = line.split("_"); + for (String s : strArr) { + sb.append(s.substring(0, 1).toUpperCase()).append(s.substring(1)); + } + str = sb.toString(); + str = str.substring(0, 1).toLowerCase() + str.substring(1); + } else if (line.contains("_") && firstIsUpperCase.length != 0) { + strArr = line.split("_"); + for (String s : strArr) { + sb.append(s.substring(0, 1).toUpperCase()).append(s.substring(1)); + } + if (!firstIsUpperCase[0]) { + str = sb.toString(); + str = str.substring(0, 1).toLowerCase() + str.substring(1); + } else { + str = sb.toString(); + } + } + } + return str; + } + + private static void writeFile(String path, String fileName, String content) { + File directory = new File(path); + //如果路径不存在,新建 + if (!directory.exists() && !directory.isDirectory()) { + directory.mkdirs(); + } + File columnFile = new File(path + fileName); + try { + columnFile.createNewFile(); + try (FileWriter fw = new FileWriter(columnFile.getAbsolutePath()); + PrintWriter pw = new PrintWriter(fw)) { + pw.println(content); + pw.flush(); + } + } catch (IOException e) { + e.printStackTrace(); + } + } + + public static Map<String, String> getPK(String tableName) { + Map<String, String> result = new LinkedHashMap<>(); + String sql = "SHOW INDEX FROM ? WHERE Key_name='PRIMARY'"; + try (PreparedStatement pStmt = CONN.prepareStatement(sql)) { + pStmt.setString(1, tableName); + try (ResultSet rs = pStmt.executeQuery()) { + while (rs.next()) { + String Key_name = rs.getString("Key_name"); + String Column_name = rs.getString("Column_name"); + result.put(Key_name, Column_name); + } + } + } catch (SQLException e) { + e.printStackTrace(); + } + return result; + } + + public static Map<String, String> getIndex(String tableName) { + Map<String, String> result = new LinkedHashMap<>(); + String sql = "SHOW INDEX FROM ? WHERE Key_name!='PRIMARY'"; + try (PreparedStatement pStmt = CONN.prepareStatement(sql)) { + pStmt.setString(1, tableName); + try (ResultSet rs = pStmt.executeQuery()) { + while (rs.next()) { + String Key_name = rs.getString("Key_name"); + String Column_name = rs.getString("Column_name"); + result.put(Key_name, Column_name); + } + } + } catch (SQLException e) { + e.printStackTrace(); + } + return result; + } +} diff --git a/src/test/java/com/lunhan/xxx/host/Mysql2Gauss.java b/src/test/java/com/lunhan/xxx/host/Mysql2Gauss.java new file mode 100644 index 0000000..d7217f8 --- /dev/null +++ b/src/test/java/com/lunhan/xxx/host/Mysql2Gauss.java @@ -0,0 +1,53 @@ +package com.lunhan.xxx.host; + +import com.lunhan.xxx.common.util.ListUtil; +import com.lunhan.xxx.common.util.SerializeUtil; + +import java.util.List; +import java.util.Map; +import java.util.Objects; + +public class Mysql2Gauss { + private static final String Mysql_DB = "water_basic"; + private static final String Gauss_DB = "water_basic"; + public static void main(String[] args) { + // 获取mysql所有的表名称 + Map<String, String> tables = GenCodeMysql.getTables(Mysql_DB); + // 遍历表名,获取表结构 + for (Map.Entry<String, String> table : tables.entrySet()) { + // 获取mysql表结构 + List<Map<String, Object>> columns = GenCodeMysql.getColumns(table.getKey(), Mysql_DB); + // 获取mysql表的主键 + Map<String, String> mapPK = GenCodeMysql.getPK(table.getKey()); + // 获取mysql表的索引 + Map<String, String> mapIndex = GenCodeMysql.getIndex(table.getKey()); + + System.out.println("从mysql查询表结构:" + table.getKey()); + // 调用gauss创建表结构 + boolean genTale = GenCodeGauss.genTable(table.getKey(), table.getValue(), columns, mapPK, mapIndex); + System.out.println("向gauss数据库创建表结果:" + genTale); + + // 获取mysql表所有数据 + String sqlSelect = "SELECT * FROM " + Mysql_DB + "." + table.getKey(); + Integer count = 1; + Integer pageSize = 500; + 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 = GenCodeMysql.getListAll(sql); + if (ListUtil.isNullOrEmpty(list)) { + continue; + } + count = list.size(); + lastId = list.get(list.size() - 1).get(orderBy); + + System.out.println("从mysql获取到数据:" + count); + // 批量插入gauss数据库 + List<Boolean> insertAll = GenCodeGauss.insertAll(list, table.getKey(), pageSize); + System.out.println("向gauss数据库插入数据结果:" + SerializeUtil.toJson(insertAll)); + } + } + } +} diff --git a/src/test/resources/application-dev.yml b/src/test/resources/application-dev.yml index 37875df..c58bd19 100644 --- a/src/test/resources/application-dev.yml +++ b/src/test/resources/application-dev.yml @@ -1,13 +1,14 @@ server: - port: 8014 + port: 8027 spring: jackson: date-format: yyyy-MM-dd HH:mm:ss datasource: - url: jdbc:mysql://113.250.189.120:23366/water_detective?useUnicode=true&characterEncoding=utf-8&serverTimezone=Asia/Shanghai&allowMultiQueries=true + driver-class-name: org.postgresql.Driver + url: jdbc:postgresql://113.250.189.120:57654/test_db1 username: lunhan - password: lunhan123 + password: lunhan.20240330 druid: initial-size: 4 #连接池初始化大小 min-idle: 10 #最小空闲连接数 @@ -17,6 +18,8 @@ stat-view-servlet: #访问监控网页的登录用户名和密码 login-username: druid login-password: lunhan123 + jpa: + database-platform: org.hibernate.dialect.openGaussDialect redis: host: 127.0.0.1 # Redis服务器地址 database: 2 # Redis数据库索引(默认为0) @@ -35,10 +38,30 @@ root: info com.lunhan.xxx: info -file: - uploadBasicPath: ./logs/upload/ - fileBasicUrl: http://localhost/file - disturbStr: com.lunhan.xxx - jdbc-set: sql-log: true + +# xxl-job配置 +xxl-job: + ### 调度中心部署根地址 [选填]:如调度中心集群部署存在多个地址则用逗号分隔。执行器将会使用该地址进行"执行器心跳注册"和"任务结果回调";为空则关闭自动注册; + admin-addresses: http://113.250.189.120:10003/xxl-job-admin + ### 执行器通讯TOKEN [选填]:非空时启用; + access-token: lunhan.20231017 + ### 执行器AppName [选填]:执行器心跳注册分组依据;为空则关闭自动注册 + executor-app-name: xxx-yyy-service + ### 执行器注册 [选填]:优先使用该配置作为注册地址,为空时使用内嵌服务 ”IP:PORT“ 作为注册地址。从而更灵活的支持容器类型执行器动态IP和动态映射端口问题。 + ### 【!!!需要注意的是:如果调度服务和执行器不在同一内网,此地址需要部署xxl-job服务所在机器能访问到的外网地址!!!】 + #executor-address: 113.250.189.120:1803 + ### 执行器IP [选填]:默认为空表示自动获取IP,多网卡时可手动设置指定IP,该IP不会绑定Host仅作为通讯实用;地址信息用于 "执行器注册" 和 "调度中心请求并触发任务"; + #executor-ip: 113.250.189.120 + ### 执行器端口号 [选填]:小于等于0则自动获取;默认端口为9999,单机部署多个执行器时,注意要配置不同执行器端口; + ### 【!!!建议配置(随机端口可能会占用其他服务,造成无法启动),且不能和Srping启动端口一样(启动报错端口被占用)!!!】 + executor-port: 1803 + ### 执行器运行日志文件存储磁盘路径 [选填] :需要对该路径拥有读写权限;为空则使用默认路径; + executor-log-path: ./logs/xxl-job/jobhandler + ### 执行器日志文件保存天数 [选填] : 过期日志自动清理, 限制值大于等于3时生效; 否则, 如-1, 关闭自动清理功能; + executor-log-retention-days: 15 + +api-set: + security_key: 3d54aa3834df + water_basic_url: http://113.250.189.120:8016 diff --git a/src/test/resources/application-prod.yml b/src/test/resources/application-prod.yml index 2be9aa7..cd241d7 100644 --- a/src/test/resources/application-prod.yml +++ b/src/test/resources/application-prod.yml @@ -5,10 +5,10 @@ jackson: date-format: yyyy-MM-dd HH:mm:ss datasource: - url: jdbc:mysql://113.250.189.120:3306/water_detective?useUnicode=true&characterEncoding=utf-8&serverTimezone=Asia/Shanghai&allowMultiQueries=true - username: root - password: LunHan.20210809 - driver-class-name: com.mysql.cj.jdbc.Driver + driver-class-name: org.postgresql.Driver + url: jdbc:postgresql://113.250.189.120:57654/water_trade_center + username: lunhan + password: lunhan.20240330 druid: initial-size: 4 #连接池初始化大小 min-idle: 10 #最小空闲连接数 @@ -34,12 +34,32 @@ path: ./logs level: root: info - com.lunhan.xxx: info - -file: - uploadBasicPath: /usr/local/nginx/huiwuyun/data - fileBasicUrl: https://www.huiwuyuntong.com/data - disturbStr: com.lunhan.xxx + com.lunhan.xxx.host: debug jdbc-set: sql-log: true + +# xxl-job配置 +xxl-job: + ### 调度中心部署根地址 [选填]:如调度中心集群部署存在多个地址则用逗号分隔。执行器将会使用该地址进行"执行器心跳注册"和"任务结果回调";为空则关闭自动注册; + admin-addresses: http://113.250.189.120:10003/xxl-job-admin + ### 执行器通讯TOKEN [选填]:非空时启用; + access-token: lunhan.20231017 + ### 执行器AppName [选填]:执行器心跳注册分组依据;为空则关闭自动注册 + executor-app-name: xxx-yyy-service + ### 执行器注册 [选填]:优先使用该配置作为注册地址,为空时使用内嵌服务 ”IP:PORT“ 作为注册地址。从而更灵活的支持容器类型执行器动态IP和动态映射端口问题。 + ### 【!!!需要注意的是:如果调度服务和执行器不在同一内网,此地址需要部署xxl-job服务所在机器能访问到的外网地址!!!】 + executor-address: 113.250.189.120:1803 + ### 执行器IP [选填]:默认为空表示自动获取IP,多网卡时可手动设置指定IP,该IP不会绑定Host仅作为通讯实用;地址信息用于 "执行器注册" 和 "调度中心请求并触发任务"; + executor-ip: 113.250.189.120 + ### 执行器端口号 [选填]:小于等于0则自动获取;默认端口为9999,单机部署多个执行器时,注意要配置不同执行器端口; + ### 【!!!建议配置(随机端口可能会占用其他服务,造成无法启动),且不能和Srping启动端口一样(启动报错端口被占用)!!!】 + executor-port: 1803 + ### 执行器运行日志文件存储磁盘路径 [选填] :需要对该路径拥有读写权限;为空则使用默认路径; + executor-log-path: ./logs/xxl-job/jobhandler + ### 执行器日志文件保存天数 [选填] : 过期日志自动清理, 限制值大于等于3时生效; 否则, 如-1, 关闭自动清理功能; + executor-log-retention-days: 15 + + api-set: + security_key: 3d54aa3834df + water_basic_url: http://localhost:9010 diff --git a/src/test/resources/application-test.yml b/src/test/resources/application-test.yml index 46ccd4e..a06d4b6 100644 --- a/src/test/resources/application-test.yml +++ b/src/test/resources/application-test.yml @@ -1,14 +1,14 @@ server: - port: 8014 + port: 8027 spring: jackson: date-format: yyyy-MM-dd HH:mm:ss datasource: - url: jdbc:mysql://113.250.189.120:3306/water_detective?useUnicode=true&characterEncoding=utf-8&serverTimezone=Asia/Shanghai&allowMultiQueries=true - username: hardware - password: hardware.admin.2023 - driver-class-name: com.mysql.cj.jdbc.Driver + driver-class-name: org.postgresql.Driver + url: jdbc:postgresql://113.250.189.120:57654/water_trade_center + username: lunhan + password: lunhan.20240330 druid: initial-size: 4 #连接池初始化大小 min-idle: 10 #最小空闲连接数 @@ -19,10 +19,10 @@ login-username: druid login-password: lunhan123 redis: - host: 113.250.189.120 # Redis服务器地址 + host: 127.0.0.1 # Redis服务器地址 database: 2 # Redis数据库索引(默认为0) port: 6379 # Redis服务器连接端口 - password: luhan123456 # Redis服务器连接密码(默认为空) + password: # Redis服务器连接密码(默认为空) timeout: 300 # 连接超时时间(毫秒) # invalid in2.4 # maxActive: 1024 # 可用连接实例的最大数目,默认为8,如果赋值为-1,则表示不限制,如果pool已经分配了maxActive个jedis实例,则此时pool的状态为exhausted(耗尽)。 maxIdle: 200 # 制一个pool最多有多少个状态为idle(空闲的)的jedis实例 @@ -34,12 +34,33 @@ path: ./logs level: root: info - com.lunhan.xxx: info - -file: - uploadBasicPath: /usr/local/nginx/huiwuyun/data - fileBasicUrl: https://www.huiwuyuntong.com/data - disturbStr: com.lunhan.xxx + com.lunhan.xxx.host: DEBUG + com.lunhan.xxx.service: DEBUG jdbc-set: sql-log: true + +# xxl-job配置 +xxl-job: + ### 调度中心部署根地址 [选填]:如调度中心集群部署存在多个地址则用逗号分隔。执行器将会使用该地址进行"执行器心跳注册"和"任务结果回调";为空则关闭自动注册; + admin-addresses: http://113.250.189.120:10003/xxl-job-admin + ### 执行器通讯TOKEN [选填]:非空时启用; + access-token: lunhan.20231017 + ### 执行器AppName [选填]:执行器心跳注册分组依据;为空则关闭自动注册 + executor-app-name: xxx-yyy-service + ### 执行器注册 [选填]:优先使用该配置作为注册地址,为空时使用内嵌服务 ”IP:PORT“ 作为注册地址。从而更灵活的支持容器类型执行器动态IP和动态映射端口问题。 + ### 【!!!需要注意的是:如果调度服务和执行器不在同一内网,此地址需要部署xxl-job服务所在机器能访问到的外网地址!!!】 + #executor-address: 113.250.189.120:1803 + ### 执行器IP [选填]:默认为空表示自动获取IP,多网卡时可手动设置指定IP,该IP不会绑定Host仅作为通讯实用;地址信息用于 "执行器注册" 和 "调度中心请求并触发任务"; + #executor-ip: 113.250.189.120 + ### 执行器端口号 [选填]:小于等于0则自动获取;默认端口为9999,单机部署多个执行器时,注意要配置不同执行器端口; + ### 【!!!建议配置(随机端口可能会占用其他服务,造成无法启动),且不能和Srping启动端口一样(启动报错端口被占用)!!!】 + executor-port: 1803 + ### 执行器运行日志文件存储磁盘路径 [选填] :需要对该路径拥有读写权限;为空则使用默认路径; + executor-log-path: ./logs/xxl-job/jobhandler + ### 执行器日志文件保存天数 [选填] : 过期日志自动清理, 限制值大于等于3时生效; 否则, 如-1, 关闭自动清理功能; + executor-log-retention-days: 15 + +api-set: + security_key: 3d54aa3834df + water_basic_url: http://192.168.91.200:8016 diff --git a/src/test/resources/application.yml b/src/test/resources/application.yml index ac25426..6cb0af0 100644 --- a/src/test/resources/application.yml +++ b/src/test/resources/application.yml @@ -2,15 +2,24 @@ application: name: water-detective profiles: - active: test #默认为开发环境 + # 通过mvn打包传递参数,指定配置文件环境 + active: @package.environment@ + # 开发环境使用 mvn -P dev(默认) + # 测试环境使用 mvn -P test + # 生产环境使用 mvn -P prod servlet: multipart: enabled: true #开启文件上传 max-file-size: 10MB #限制文件上传大小为10M + mail: + password: ennnntqxzijsdjdh + username: 2413500195@qq.com + host: smtp.qq.com + default-encoding: UTF-8 jwt: token_header: Token #JWT存储的请求头 - secret: hardware-lunhan-secret #JWT加解密密钥(所有硬件项目通用,公用一个用户中心登录,其他项目不写登录接口,直接用token) + secret: water-lunhan-secret #hardware-lunhan-secret #JWT加解密密钥(所有硬件项目通用,公用一个用户中心登录,其他项目不写登录接口,直接用token) expiration: 15811200 #JWT的超期限时间(60*60*24*183) 单位:秒 head: 'Bearer ' #JWT负载中拿到开头 @@ -19,9 +28,19 @@ secret: 883d188be8593bd14c9b637ca5405c04 grant_type: authorization_code -mail: - smtp: smtp.aliyun.com - port: 465 - userName: lin.liu@aliyun.com - password: - sendEmail: lin.liu@aliyun.com +wx: + appId: wx08c82dac406bee56 + secret: efb703be421ca95fc5191c07576e8938 + miniapp: + #上面拿到的很重要的那个模板ID + template-id: kuVF5aAUw4LRVszKBm9pMxoDrOmXnvbsZYoeLNfdj10 + configs: + #微信小程序的appid 开发者工具拿到 + appid: wx51878700809c2e5c + #开发者工具拿到Secret + secret: 8b475e0b7905ef72dd014956f42cb558 + #微信小程序消息服务器配置的token + token: 123456 + #微信小程序消息服务器配置的EncodingAESKey + aesKey: + msgDataFormat: JSON diff --git a/src/test/resources/lib/jave-1.0.2.jar b/src/test/resources/lib/jave-1.0.2.jar deleted file mode 100644 index a00a3db..0000000 --- a/src/test/resources/lib/jave-1.0.2.jar +++ /dev/null Binary files differ -- Gitblit v1.9.3