自动生成代码逻辑修改

liulin

2024-08-05 896e1cf5e9b195163f76632ca75675ffd7187e54

 src/test/java/com/lunhan/xxx/host/GenCodeGauss.java

@@ -13,6 +13,7 @@
import java.time.LocalDateTime;
import java.util.Date;
import java.util.*;
import java.util.stream.Collectors;

public class GenCodeGauss {
    private static final String HOST = "113.250.189.120";
@@ -22,7 +23,7 @@
    private static final String DB_NAME = "water_basic";
    private static final String AUTHOR = "lin.liu";

    private static Connection CONN = null;
    public static Connection CONN = null;
    private static final String DRIVER = "org.postgresql.Driver";
    private static final String CONNECTION_STR = "jdbc:postgresql://" + HOST + ":" + PORT + "/" + DB_NAME;

@@ -213,12 +214,12 @@
    //#endregion

    //#region 编辑接口忽略字段配置
    private static final  List<String> NO_UPDATE_COLUMNS = Arrays.asList(
    private static final List<String> NO_UPDATE_COLUMNS = Arrays.asList(
            "create_time", "create_user", "create_user_id", "create_user_name",
            "update_time", "update_user", "update_user_id", "update_user_name",
            "datachange_lasttime", "data_change_last_time", "is_delete"
    );
    private static final  List<String> UPDATE_TIME_COLUMNS = Arrays.asList(
    private static final List<String> UPDATE_TIME_COLUMNS = Arrays.asList(
            "update_time", "datachange_lasttime", "data_change_last_time"
    );
    //#endregion
@@ -242,7 +243,7 @@
     * 只生成以下配置的表
     */
    private static final List<String> ONLY_TABLES = Arrays.asList(
            "admin_menus"

    );

    public static void main(String[] args) {
@@ -250,7 +251,7 @@
            System.out.println("conn is null.");
            return;
        }
        Map<String, String> tables = getTables(DB_NAME);
        Map<String, String> tables = getTables(CONN, DB_NAME);
        for (Map.Entry<String, String> table : tables.entrySet()) {
            String tableName = table.getKey();
            String tableDesc = table.getValue();
@@ -258,7 +259,7 @@
            if (ONLY_TABLES.size() > 0 && BooleanUtils.isFalse(ONLY_TABLES.contains(tableName))) {
                continue;
            }
            List<Map<String, Object>> columns = getColumns(tableName, DB_NAME);
            List<Map<String, Object>> columns = getColumns(CONN, tableName);
            genPO(columns, tableDesc);
            genVO(columns, tableDesc);
            genCreateDto(columns, tableName, tableDesc);
@@ -331,16 +332,18 @@
                    listColumn.add("(type = IdType.ASSIGN_ID)");
                }
            }
            String columnAnnotation = "\t@TableField(\"" + name + "\"";
            //忽略最后修改时间这一列的新增和编辑
            if (!"id".equals(name)) {
                String columnAnnotation = "\t@TableField(\"" + name + "\"";
                //忽略最后修改时间这一列的新增和编辑
//            if (UPDATE_TIME_COLUMNS.stream().anyMatch(c -> c.equalsIgnoreCase(name))) {
//                columnAnnotation += ", insertable = false, updatable = false";
//            }
                listColumn.add(columnAnnotation + ")");
            }
            if ("datachange_lasttime".equalsIgnoreCase(name)) {
                propName = "dataChangeLastTime";
                columnNameUpper = "DataChangeLastTime";
            }
            listColumn.add(columnAnnotation + ")");
            //追加列
            listColumn.add(String.format("\tprivate %s %s;", javaType, propName));

@@ -358,7 +361,7 @@
        }
        content.append(ConstantFactory.STR_NEWLINE);
        content.append(SET_CLASS_DESC_TPL
                .replaceAll("\\{\\#\\=desc\\}", tableDesc)
                .replaceAll("\\{\\#\\=desc\\}", tableDesc == null ? modelName : tableDesc)
                .replaceAll("\\{\\#\\=author\\}", AUTHOR)
                .replaceAll("\\{\\#\\=date\\}", LocalDateTimeUtil.todayStr().replaceAll("\\-", "/"))
        );
@@ -376,8 +379,10 @@
        content.append(ConstantFactory.STR_NEWLINE);
        content.append(StringUtil.join(listColumn, ConstantFactory.STR_NEWLINE));
        content.append(ConstantFactory.STR_NEWLINE);
        content.append(StringUtil.join(listGetterSetter, ConstantFactory.STR_NEWLINE));
        content.append(ConstantFactory.STR_NEWLINE);
        if (!isData) {
            content.append(StringUtil.join(listGetterSetter, ConstantFactory.STR_NEWLINE));
            content.append(ConstantFactory.STR_NEWLINE);
        }
        content.append("}");

        writeFile(GenCodeGauss.OutSet.PO, modelName + GenCodeGauss.SuffixSet.PO + ".java", content.toString());
@@ -419,14 +424,13 @@

        content.append(ConstantFactory.STR_NEWLINE);
        content.append(SET_CLASS_DESC_TPL
                .replaceAll("\\{\\#\\=desc\\}", tableDesc)
                .replaceAll("\\{\\#\\=desc\\}", tableDesc == null ? modelName : tableDesc)
                .replaceAll("\\{\\#\\=author\\}", AUTHOR)
                .replaceAll("\\{\\#\\=date\\}", LocalDateTimeUtil.todayStr().replaceAll("\\-", "/"))
        );
        content.append(ConstantFactory.STR_NEWLINE);
        if (isData) {
            content.append("@Data");
            content.append(ConstantFactory.STR_NEWLINE);
        }
        content.append(ConstantFactory.STR_NEWLINE);
        content.append("public class ");
@@ -465,11 +469,17 @@
        String modelName = underline2Camel(tableName, true);
        String searchName = "Search" + modelName;
        String tableNamePackage = tableName.toLowerCase().replaceAll("_", "");
        boolean isData = Objects.equals(SET_PO_Getter, 2);

        StringBuilder content = new StringBuilder();
        content.append("package ");
        content.append(GenCodeGauss.PackageSet.SEARCH + ConstantFactory.STR_SEMICOLON);
        content.append(ConstantFactory.STR_NEWLINE);
        if (isData) {
            content.append(ConstantFactory.STR_NEWLINE);
            content.append("import lombok.Data;");
            content.append(ConstantFactory.STR_NEWLINE);
        }
        if (SET_SEARCH_IMPORT.length > 0) {
            content.append(ConstantFactory.STR_NEWLINE);
            for (String s : SET_SEARCH_IMPORT) {
@@ -479,40 +489,46 @@
        }
        content.append(ConstantFactory.STR_NEWLINE);
        content.append(SET_CLASS_DESC_TPL
                .replaceAll("\\{\\#\\=desc\\}", tableDesc)
                .replaceAll("\\{\\#\\=desc\\}", tableDesc == null ? modelName : tableDesc)
                .replaceAll("\\{\\#\\=author\\}", AUTHOR)
                .replaceAll("\\{\\#\\=date\\}", LocalDateTimeUtil.todayStr().replaceAll("\\-", "/"))
        );
        if (isData) {
            content.append(ConstantFactory.STR_NEWLINE);
            content.append("@Data");
        }
        content.append(ConstantFactory.STR_NEWLINE);
        content.append("public class " + searchName + " extends SearchBasicDTO {");
        content.append(ConstantFactory.STR_NEWLINE);
        content.append("\t@Override");
        content.append(ConstantFactory.STR_NEWLINE);
        content.append("\tpublic String toString() {");
        content.append(ConstantFactory.STR_NEWLINE);
        content.append("\t\treturn \"" + searchName + "{\" + \"'\" +");
        content.append(ConstantFactory.STR_NEWLINE);
        content.append("\t\t\t\"keywords='\" + getKeywords() +");
        content.append(ConstantFactory.STR_NEWLINE);
        content.append("\t\t\t\", listId =\" + getListId() +");
        content.append(ConstantFactory.STR_NEWLINE);
        content.append("\t\t\t\", status =\" + getStatus() +");
        content.append(ConstantFactory.STR_NEWLINE);
        content.append("\t\t\t\", listStatus =\" + getListStatus() +");
        content.append(ConstantFactory.STR_NEWLINE);
        content.append("\t\t\t\", createTimeRange='\" + getCreateTimeRange() + \"'\" +");
        content.append(ConstantFactory.STR_NEWLINE);
        content.append("\t\t\t\", createTimeStart =\" + getCreateTimeEnd() +");
        content.append(ConstantFactory.STR_NEWLINE);
        content.append("\t\t\t\", createTimeEnd =\" + getCreateTimeEnd() +");
        content.append(ConstantFactory.STR_NEWLINE);
        content.append("\t\t\t\", lastId =\" + getListId() +");
        content.append(ConstantFactory.STR_NEWLINE);
        content.append("\t\t\t\", orderBy='\" + getOrderBy() + \"'\" +");
        content.append(ConstantFactory.STR_NEWLINE);
        content.append("\t\t'}';");
        content.append(ConstantFactory.STR_NEWLINE);
        content.append("\t}");
        if (!isData) {
            content.append("\t@Override");
            content.append(ConstantFactory.STR_NEWLINE);
            content.append("\tpublic String toString() {");
            content.append(ConstantFactory.STR_NEWLINE);
            content.append("\t\treturn \"" + searchName + "{\" + \"'\" +");
            content.append(ConstantFactory.STR_NEWLINE);
            content.append("\t\t\t\"keywords='\" + getKeywords() +");
            content.append(ConstantFactory.STR_NEWLINE);
            content.append("\t\t\t\", listId =\" + getListId() +");
            content.append(ConstantFactory.STR_NEWLINE);
            content.append("\t\t\t\", status =\" + getStatus() +");
            content.append(ConstantFactory.STR_NEWLINE);
            content.append("\t\t\t\", listStatus =\" + getListStatus() +");
            content.append(ConstantFactory.STR_NEWLINE);
            content.append("\t\t\t\", createTimeRange='\" + getCreateTimeRange() + \"'\" +");
            content.append(ConstantFactory.STR_NEWLINE);
            content.append("\t\t\t\", createTimeStart =\" + getCreateTimeEnd() +");
            content.append(ConstantFactory.STR_NEWLINE);
            content.append("\t\t\t\", createTimeEnd =\" + getCreateTimeEnd() +");
            content.append(ConstantFactory.STR_NEWLINE);
            content.append("\t\t\t\", lastId =\" + getListId() +");
            content.append(ConstantFactory.STR_NEWLINE);
            content.append("\t\t\t\", orderBy='\" + getOrderBy() + \"'\" +");
            content.append(ConstantFactory.STR_NEWLINE);
            content.append("\t\t'}';");
            content.append(ConstantFactory.STR_NEWLINE);
            content.append("\t}");
        }
        content.append(ConstantFactory.STR_NEWLINE);
        content.append("}");

@@ -558,7 +574,7 @@
        content.append(poName + ";\n");
        content.append(ConstantFactory.STR_NEWLINE);
        content.append(SET_CLASS_DESC_TPL
                .replaceAll("\\{\\#\\=desc\\}", tableDesc)
                .replaceAll("\\{\\#\\=desc\\}", tableDesc == null ? modelName : tableDesc)
                .replaceAll("\\{\\#\\=author\\}", AUTHOR)
                .replaceAll("\\{\\#\\=date\\}", LocalDateTimeUtil.todayStr().replaceAll("\\-", "/"))
        );
@@ -568,7 +584,7 @@
        content.append(daoName);
        content.append(" extends BasicMapperImpl<");
        content.append(modelName + GenCodeGauss.SuffixSet.PO);
        content.append("> {");
        content.append(", " + mapperName + "> {");
        content.append(ConstantFactory.STR_NEWLINE);
        content.append("\t" + daoName + "(" + mapperName + " mapper) {\n");
        content.append("\t\tsuper(mapper);\n");
@@ -624,15 +640,15 @@
                "\t\t\t\t\tswitch (orderBy) {\n" +
                "\t\t\t\t\t\t// 主键\n" +
                "\t\t\t\t\t\tcase ID:\n" +
                "\t\t\t\t\t\t\tqueryWrapper.orderByAsc(" + poName + "::getId);\n" +
                "\t\t\t\t\t\t\tqueryWrapper.orderByDesc(" + poName + "::getId);\n" +
                "\t\t\t\t\t\t\tbreak;\n" +
                "\t\t\t\t\t\t// 数据创建时间\n" +
                "\t\t\t\t\t\tcase CREATE_TIME:\n" +
                "\t\t\t\t\t\t\tqueryWrapper.orderByAsc(" + poName + "::getCreateTime);\n" +
                "\t\t\t\t\t\t\tqueryWrapper.orderByDesc(" + poName + "::getCreateTime);\n" +
                "\t\t\t\t\t\t\tbreak;\n" +
                "\t\t\t\t\t\t// 最后更新时间\n" +
                "\t\t\t\t\t\tcase UPDATE_TIME:\n" +
                "\t\t\t\t\t\t\tqueryWrapper.orderByAsc(" + poName + "::getUpdateTime);\n" +
                "\t\t\t\t\t\t\tqueryWrapper.orderByDesc(" + poName + "::getUpdateTime);\n" +
                "\t\t\t\t\t\t\tbreak;\n" +
                "\t\t\t\t\t}\n" +
                "\t\t\t\t}\n" +
@@ -640,9 +656,28 @@
                "\t\t} else {\n" +
                "\t\t\tqueryWrapper.orderByDesc(" + poName + "::getId);\n" +
                "\t\t}\n" +
                "\t\tPage<" + poName + "> pageResult = DB.selectPage(new Page<>(search.getPage(), search.getLimit()), queryWrapper);\n" +
                "\t\tPage<" + poName + "> pageResult = super.selectPage(new Page<>(search.getPage(), search.getLimit()), queryWrapper);\n" +
                "\t\treturn new PagerResult<>(pageResult.getSize(), pageResult.getPages(), pageResult.getTotal(), pageResult.getRecords());\n" +
                "\t}");
                "\t}\n" +
                "\n" +
                "\tpublic Boolean add(" + poName + " item) {\n" +
                "\t\tint rowCount = super.insert(item);\n" +
                "\t\treturn rowCount == 1;\n" +
                "\t}\n" +
                "\n" +
                "\tpublic Boolean addNotIncrement(" + poName + " item) {\n" +
                "\t\tint rowCount = super.insert(item);\n" +
                "\t\treturn rowCount == 1;\n" +
                "\t}\n" +
                "\n" +
                "\tpublic " + poName + " getById(Long id) {\n" +
                "\t\treturn super.get(id);\n" +
                "\t}\n" +
                "\n" +
                "\tpublic List<" + poName + "> getListById(List<Long> listId) {\n" +
                "\t\treturn super.getList(listId);\n" +
                "\t}\n"
        );
        content.append("}");

        if (new File(GenCodeGauss.OutSet.MAPPER_IMPL + daoName + ".java").exists()) {
@@ -731,14 +766,13 @@
        }
        content.append(ConstantFactory.STR_NEWLINE);
        content.append(SET_CLASS_DESC_TPL
                .replaceAll("\\{\\#\\=desc\\}", tableDesc)
                .replaceAll("\\{\\#\\=desc\\}", tableDesc == null ? modelName : tableDesc)
                .replaceAll("\\{\\#\\=author\\}", AUTHOR)
                .replaceAll("\\{\\#\\=date\\}", LocalDateTimeUtil.todayStr().replaceAll("\\-", "/"))
        );
        content.append(ConstantFactory.STR_NEWLINE);
        if (isData) {
            content.append("@Data");
            content.append(ConstantFactory.STR_NEWLINE);
        }
        content.append(ConstantFactory.STR_NEWLINE);
        content.append("public class ");
@@ -747,8 +781,10 @@
        content.append(ConstantFactory.STR_NEWLINE);
        content.append(StringUtil.join(listColumn, ConstantFactory.STR_NEWLINE));
        content.append(ConstantFactory.STR_NEWLINE);
        content.append(StringUtil.join(listGetterSetter, ConstantFactory.STR_NEWLINE));
        content.append(ConstantFactory.STR_NEWLINE);
        if (!isData) {
            content.append(StringUtil.join(listGetterSetter, ConstantFactory.STR_NEWLINE));
            content.append(ConstantFactory.STR_NEWLINE);
        }
        content.append("}");

        if (new File(GenCodeGauss.OutSet.RequestDTO + tableNamePackage + "/" + createDTOName + ".java").exists()) {
@@ -823,14 +859,13 @@
        }
        content.append(ConstantFactory.STR_NEWLINE);
        content.append(SET_CLASS_DESC_TPL
                .replaceAll("\\{\\#\\=desc\\}", tableDesc)
                .replaceAll("\\{\\#\\=desc\\}", tableDesc == null ? modelName : tableDesc)
                .replaceAll("\\{\\#\\=author\\}", AUTHOR)
                .replaceAll("\\{\\#\\=date\\}", LocalDateTimeUtil.todayStr().replaceAll("\\-", "/"))
        );
        content.append(ConstantFactory.STR_NEWLINE);
        if (isData) {
            content.append("@Data");
            content.append(ConstantFactory.STR_NEWLINE);
        }
        content.append(ConstantFactory.STR_NEWLINE);
        content.append("public class ");
@@ -839,8 +874,10 @@
        content.append(ConstantFactory.STR_NEWLINE);
        content.append(StringUtil.join(listColumn, ConstantFactory.STR_NEWLINE));
        content.append(ConstantFactory.STR_NEWLINE);
        content.append(StringUtil.join(listGetterSetter, ConstantFactory.STR_NEWLINE));
        content.append(ConstantFactory.STR_NEWLINE);
        if (!isData) {
            content.append(StringUtil.join(listGetterSetter, ConstantFactory.STR_NEWLINE));
            content.append(ConstantFactory.STR_NEWLINE);
        }
        content.append("}");

        if (new File(GenCodeGauss.OutSet.RequestDTO + tableNamePackage + "/" + modifyDTOName + ".java").exists()) {
@@ -896,7 +933,7 @@

        content.append(ConstantFactory.STR_NEWLINE);
        content.append(SET_CLASS_DESC_TPL
                .replaceAll("\\{\\#\\=desc\\}", tableDesc)
                .replaceAll("\\{\\#\\=desc\\}", tableDesc == null ? modelName : tableDesc)
        );
        content.append(ConstantFactory.STR_NEWLINE);
        content.append("@Mapper");
@@ -971,7 +1008,7 @@
        content.append(responseDTOName + ";\n");
        content.append(ConstantFactory.STR_NEWLINE);
        content.append(SET_CLASS_DESC_TPL
                .replaceAll("\\{\\#\\=desc\\}", tableDesc)
                .replaceAll("\\{\\#\\=desc\\}", tableDesc == null ? modelName : tableDesc)
                .replaceAll("\\{\\#\\=author\\}", AUTHOR)
                .replaceAll("\\{\\#\\=date\\}", LocalDateTimeUtil.todayStr().replaceAll("\\-", "/"))
        );
@@ -991,7 +1028,7 @@
                "        // 设置记录创建时间\n" +
                "        item.setCreateTime(LocalDateTimeUtil.nowTimeStamp());\n" +
                "        // 是否删除(逻辑删除)初始值\n" +
                "        //item.setIsDelete(EYesOrNo.NO.getValue());\n" +
                "        item.setIsDelete(EYesOrNo.NO.getValue());\n" +
                "\n" +
                "        int rowCount = mapper.insert(item);\n" +
                "        if (rowCount != 1) {\n" +
@@ -1156,7 +1193,7 @@
                "            return ExecutedResult.failed(\"[" + tableDesc + "]不存在：\" + id);\n" +
                "        }\n" +
                "        return ExecutedResult.success(exists);\n" +
                "    }\n"+
                "    }\n" +
                "    protected ExecutedResult<List<" + poName + ">> check4Id(List<Long> listId) {\n" +
                "        // 从数据库查找" + tableDesc + "\n" +
                "        List<" + poName + "> list = mapper.getList(listId);\n" +
@@ -1231,7 +1268,7 @@
        content.append(responseDTOName + ";\n");
        content.append(ConstantFactory.STR_NEWLINE);
        content.append(SET_CLASS_DESC_TPL
                .replaceAll("\\{\\#\\=desc\\}", tableDesc)
                .replaceAll("\\{\\#\\=desc\\}", tableDesc == null ? modelName : tableDesc)
                .replaceAll("\\{\\#\\=author\\}", AUTHOR)
                .replaceAll("\\{\\#\\=date\\}", LocalDateTimeUtil.todayStr().replaceAll("\\-", "/"))
        );
@@ -1350,7 +1387,7 @@
        content.append(ConstantFactory.STR_NEWLINE);
        content.append(ConstantFactory.STR_NEWLINE);
        content.append(SET_CLASS_DESC_TPL
                .replaceAll("\\{\\#\\=desc\\}", tableDesc + " mapper")
                .replaceAll("\\{\\#\\=desc\\}", tableDesc == null ? modelName : tableDesc + " mapper")
                .replaceAll("\\{\\#\\=author\\}", AUTHOR)
                .replaceAll("\\{\\#\\=date\\}", LocalDateTimeUtil.todayStr().replaceAll("\\-", "/"))
        );
@@ -1368,7 +1405,7 @@
    //#endregion


    public static Map<String, String> getTables(String dbSchema) {
    public static Map<String, String> getTables(Connection connect, String dbSchema) {
        //  dbSchema="public";
        Map<String, String> result = new LinkedHashMap<>();
        String sql = "SELECT tb.table_name, d.description AS TABLE_COMMENT" +
@@ -1376,7 +1413,7 @@
                " JOIN pg_class c ON c.relname = tb.table_name" +
                " LEFT JOIN pg_description d ON d.objoid = c.oid AND d.objsubid = '0'" +
                " WHERE tb.table_schema = 'public' ";
        try (PreparedStatement pStmt = CONN.prepareStatement(sql)) {
        try (PreparedStatement pStmt = connect.prepareStatement(sql)) {
            //pStmt.setString(1, dbSchema);
            try (ResultSet rs = pStmt.executeQuery()) {
                while (rs.next()) {
@@ -1391,7 +1428,7 @@
        return result;
    }

    public static List<Map<String, Object>> getColumns(String tableName, String dbSchema) {
    public static List<Map<String, Object>> getColumns(Connection connect, String tableName) {
        List<Map<String, Object>> result = new ArrayList<>();
        String sql = "SELECT col.table_name AS objName, col.column_name AS name, col.udt_name AS type, col.character_maximum_length AS LENGTH, des.description,\n" +
                "col.numeric_precision, \n" +
@@ -1406,7 +1443,7 @@
                "AND table_schema = 'public' " +
                "ORDER BY " +
                "ordinal_position;";
        try (PreparedStatement pStmt = CONN.prepareStatement(sql)) {
        try (PreparedStatement pStmt = connect.prepareStatement(sql)) {
            pStmt.setString(1, tableName);
//            pStmt.setString(2, "public");
            try (ResultSet rs = pStmt.executeQuery()) {
@@ -1534,6 +1571,9 @@

    public static String toDBValue(Object value) {
        String result;
        if (null == value) {
            return "";
        }
        if (value instanceof Number || value instanceof Boolean) {
            result = value.toString();
        } else if (value instanceof Timestamp) {
@@ -1552,40 +1592,58 @@
        return result;
    }

    public static boolean genTable(String tableName, String tableDesc, List<Map<String, Object>> listColumn, Map<String, String> mapPK, Map<String, String> mapIndex) {
        String sql = "DROP TABLE IF EXISTS \"public\".\"" + tableName + "\";\n" +
                "CREATE TABLE \"public\".\"" + tableName + "\" (\n";
    public static boolean genTable(Connection connect, String tableName, String tableDesc, List<Map<String, Object>> listColumn, Map<String, String> mapPK, Map<String, String> mapIndex) throws SQLException {
        String sql = "CREATE TABLE \"public\".\"" + tableName + "\" (\n";

        Map<String, String> mapColumnDesc = new HashMap<>();
        int idx = 0;
        boolean result = Boolean.FALSE;

        result = GenCodeGauss.execSql(connect, "DROP TABLE IF EXISTS \"public\".\"" + tableName + "\";\n");

        // 遍历列，构建sql
        for (Map<String, Object> column : listColumn) {
            idx++;
            String name = column.get("name").toString();
            String type = column.get("type").toString();
            String desc = column.get("description").toString();
            String isNullable = column.get("isNullable").toString();

            mapColumnDesc.put(name,desc);
            mapColumnDesc.put(name, desc);

            sql += "\t\""+ name +"\" " + type;
            sql += "\t\"" + name + "\" " + GenCodeGauss.mysqlType2Gauss(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";
                    sql += "(" + length + ")";
                    if ("1".equals(isNullable)) {
                        sql += " NULL DEFAULT NULL";
                    } else {
                        sql += " NOT NULL DEFAULT NULL::character varying";
                    }
                    break;
                case "numeric":
                case "decimal":
                    String numeric_precision = column.get("numeric_precision").toString();
                    String numeric_scale = column.get("numeric_scale").toString();
                    sql += "(" + numeric_precision+ ","+ numeric_scale +") NOT NULL DEFAULT 0";
                    sql += "(" + numeric_precision + "," + numeric_scale + ")";
                    if ("1".equals(isNullable)) {
                        sql += " NULL DEFAULT NULL";
                    } else {
                        sql += " NOT NULL DEFAULT 0";
                    }
                    break;
                case "datetime":
                case "timestamp":
                    String datetime_precision = column.get("datetime_precision").toString();
                    sql += "(" + datetime_precision +") NOT NULL DEFAULT PG_SYSTIMESTAMP()";
                    sql += "(" + datetime_precision + ")";
                    if ("1".equals(isNullable)) {
                        sql += " NULL DEFAULT NULL";
                    } else {
                        sql += " NOT NULL DEFAULT pg_systimestamp()";
                    }
                    break;
                case "int":
                case "tinyint":
@@ -1593,7 +1651,12 @@
                case "int2":
                case "int4":
                case "int8":
                    sql += " NOT NULL DEFAULT 0";
                case "bigint":
                    if ("1".equals(isNullable)) {
                        sql += " NULL DEFAULT NULL";
                    } else {
                        sql += " NOT NULL DEFAULT 0";
                    }
                    break;

                default:
@@ -1606,39 +1669,91 @@
            sql += "\n";
        }
        sql += ")\n" +
                ";\n" +
                "COMMENT ON TABLE \"public\".\"" + tableName + "\" IS '" + tableDesc + "';\n";
                ";\n";
        result = GenCodeGauss.execSql(connect, sql);
        sql = "";

        sql += "COMMENT ON TABLE \"public\".\"" + tableName + "\" IS '" + tableDesc + "';\n";
        // 构建字段描述sql
        for (Map.Entry<String, String> desc : mapColumnDesc.entrySet()) {
            sql += "COMMENT ON COLUMN \"public\".\"" + tableName + "\".\"" + desc.getKey() +"\" IS '" + desc.getValue().replace("\r\n", " ").replace("\n", " ") + "';\n";
            sql += "COMMENT ON COLUMN \"public\".\"" + tableName + "\".\"" + desc.getKey() + "\" IS '" + desc.getValue().replace("\r\n", " ").replace("\n", " ") + "';\n";
        }
        result = GenCodeGauss.execSql(connect, sql);
        sql = "";

        // 设置主键
        if (!mapPK.isEmpty()) {
            for (Map.Entry<String, String> key : mapPK.entrySet()) {
                sql += "ALTER TABLE \"public\".\"" + tableName + "\" ADD CONSTRAINT \"PK_" + tableName + "\" PRIMARY KEY (\"" + key.getValue() +"\");\n";
                sql += "ALTER TABLE \"public\".\"" + tableName + "\" ADD CONSTRAINT \"PK_" + tableName + "\" PRIMARY KEY (\"" + key.getValue() + "\");\n";
            }
        }
        sql += "ALTER TABLE \"public\".\"" + tableName + "\" ADD CONSTRAINT \"PK_" + tableName + "\" PRIMARY KEY (\"id\");\n";
                // 触发器：自动赋值最后更新时间
        sql += "CREATE TRIGGER \"TG_" + tableName + "\" BEFORE UPDATE OF \"update_time\" ON \"public\".\"" + tableName + "\"\n" +
        result = GenCodeGauss.execSql(connect, sql);
        sql = "";

        // 触发器：自动赋值最后更新时间
        sql += "CREATE TRIGGER \"TG_" + tableName + "\" BEFORE INSERT OR UPDATE ON \"public\".\"" + tableName + "\"\n" +
                "FOR EACH ROW\n" +
                "EXECUTE PROCEDURE \"public\".\"update_timestamp_column\"();";
        result = GenCodeGauss.execSql(connect, sql);
        sql = "";

        // 遍历索引，构建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";
                result = GenCodeGauss.execSql(connect, sql);
                sql = "";
            }
        }
        try (PreparedStatement pStmt = CONN.prepareCall(sql)) {
            int[] result = pStmt.executeBatch();
            return result != null;
        return result;
    }

    private static String mysqlType2Gauss(String type) {
        String result = "";
        switch (type) {
            case "char":
            case "nvarchar":
                result = "varchar";
                break;

            case "longtext":
                result = "text";
                break;

            case "decimal":
                result = "numeric";
                break;
            case "datetime":
                result = "timestamp";
                break;

            case "bit":
            case "tinyint":
                result = "int2";
                break;

            case "int":
                result = "int4";
                break;

            default:
                result = type;
                break;
        }
        return result;
    }

    public static boolean execSql(Connection connect, String sql) {
        try(PreparedStatement pst = connect.prepareStatement(sql)) {
            int rowCount = pst.executeUpdate();
            return rowCount > 0;
        } catch (Exception e) {
            System.out.println(e);
            return Boolean.FALSE;
            return false;
        }
    }

    public static Map<String, String> getPK(String tableName) {
    public static Map<String, String> getPK(Connection connect, String tableName) {
        Map<String, String> mapPK = new HashMap<>();
        // 查询主键
        String pkSq = "select pg_attribute.attname as col_name,pg_constraint.conname as pk_name\n" +
@@ -1647,7 +1762,7 @@
                "inner join pg_attribute on pg_attribute.attrelid=pg_class.oid and pg_attribute.attnum=pg_constraint.conkey[1]\n" +
                "where pg_class.relname='" + tableName + "'\n" +
                "and pg_constraint.contype='p'";
        try (PreparedStatement pStmt = CONN.prepareStatement(pkSq)) {
        try (PreparedStatement pStmt = connect.prepareStatement(pkSq)) {
            try (ResultSet rs = pStmt.executeQuery()) {
                //获取键名
                ResultSetMetaData md = rs.getMetaData();
@@ -1665,11 +1780,11 @@
        return mapPK;
    }

    public static Map<String, String> getIndex(String tableName) {
    public static Map<String, String> getIndex(Connection connect, String tableName) {
        Map<String, String> mapPK = new HashMap<>();
        // 查询主键
        String pkSq = "SELECT * FROM pg_indexes WHERE tablename = '" + tableName +"';";
        try (PreparedStatement pStmt = CONN.prepareStatement(pkSq)) {
        String pkSq = "SELECT * FROM pg_indexes WHERE tablename = '" + tableName + "';";
        try (PreparedStatement pStmt = connect.prepareStatement(pkSq)) {
            try (ResultSet rs = pStmt.executeQuery()) {
                //获取键名
                ResultSetMetaData md = rs.getMetaData();
@@ -1685,10 +1800,10 @@
        return mapPK;
    }

    public static List<Map<String, Object>> getListAll(String tableName, String dbSchema) {
    public static List<Map<String, Object>> getListAll(Connection connect, String tableName, String dbSchema) {
        List<Map<String, Object>> result = new ArrayList<>();
        String sql = "SELECT * FROM " + dbSchema + "." + tableName;
        try (PreparedStatement pStmt = CONN.prepareStatement(sql)) {
        try (PreparedStatement pStmt = connect.prepareStatement(sql)) {
            try (ResultSet rs = pStmt.executeQuery()) {
                //获取键名
                ResultSetMetaData md = rs.getMetaData();
@@ -1710,28 +1825,42 @@
        return result;
    }

    public static List<Boolean> insertAll(List<Map<String, Object>> list, String tableName, Integer onceCount) {
    public static List<Boolean> insertAll(Connection connect, List<Map<String, Object>> list, String tableName, Integer onceCount) {
        List<Boolean> result = new ArrayList<>();
        String insertSql = "INSERT INTO public." + tableName + "(";
        String insertSql = "INSERT INTO public." + tableName + "(\"";
        List<String> listColumn = new ArrayList<>();
        for (String column : list.get(0).keySet()) {
            if ("update_time".equalsIgnoreCase(column)) {
                continue;
            }
            listColumn.add(column);
        }
        insertSql += StringUtil.join(listColumn) + ") VALUES";
        insertSql += StringUtil.join(listColumn, "\",\"") + "\") VALUES";

        String sql = "";
        List<Object> listValue = new ArrayList<>();
        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<>();

            List<Object> listColValue = new ArrayList<>();
            for (Map.Entry<String, Object> item : row.entrySet()) {
                listValue.add(GenCodeGauss.toDBValue(item.getValue()));
                if ("update_time".equalsIgnoreCase(item.getKey())) {
                    continue;
                }
                listColValue.add(item.getValue());
            }
            sql += "(" + StringUtil.join(listValue) + ")";
            sql += "\n(" + StringUtil.join(listColValue.stream().map(c -> "?").collect(Collectors.toList())) + "),";
            listValue.addAll(listColValue);
            if (i > 0 && i % onceCount == 0) {
                try (PreparedStatement pStmt = CONN.prepareStatement(sql)) {
                sql = sql.substring(0, sql.length() - 1);
                try (PreparedStatement pStmt = connect.prepareStatement(sql)) {
                    for (int j = 0; j < listValue.size(); j++) {
                        pStmt.setObject(j + 1, listValue.get(j));
                    }
                    listValue = new ArrayList<>();
                    boolean insertList = pStmt.execute();
                    result.add(insertList);
                    System.out.println("插入数据" + Math.max(i - onceCount, 0) + " - " + i + "，结果：" + insertList);
@@ -1741,6 +1870,25 @@
                sql = "";
            }
        }
        if (StringUtil.isNotNullOrEmpty(sql)) {
            sql = sql.substring(0, sql.length() - 1);
            try (PreparedStatement pStmt = connect.prepareStatement(sql)) {
                for (int j = 0; j < listValue.size(); j++) {
                    Object value = listValue.get(j);
                    if (Objects.equals("", value)) {
                        // TODO 神奇的gauss，''会判定为null，如果有非空约束，会插入失败
                        value = " ";
                    }
                    pStmt.setObject(j + 1, value);
                }
                listValue = new ArrayList<>();
                boolean insertList = pStmt.execute();
                result.add(insertList);
                System.out.println("插入数据" + (list.size() - list.size() % onceCount) + " - " + list.size() + "，结果：" + insertList);
            } catch (SQLException e) {
                e.printStackTrace();
            }
        }
        return result;
    }
}