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 GenCodeGauss {
|
private static final String HOST = "113.250.189.120";
|
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 AUTHOR = "lin.liu";
|
|
private static Connection CONN = null;
|
private static final String DRIVER = "org.postgresql.Driver";
|
private static final String CONNECTION_STR = "jdbc:postgresql://" + HOST + ":" + PORT + "/" + DB_NAME;
|
|
//region 输出文件路径设置
|
class OutSet {
|
public static final String PO = "./src/main/java/com/lunhan/xxx/repository/po/";
|
public static final String VO = "./src/main/java/com/lunhan/xxx/repository/vo/";
|
public static final String CONVERT_MAPPER = "./src/main/java/com/lunhan/xxx/service/convert/";
|
public static final String MAPPER_IMPL = "./src/main/java/com/lunhan/xxx/repository/impl/";
|
public static final String MAPPER = "./src/main/java/com/lunhan/xxx/repository/mapper/";
|
public static final String SERVICE = "./src/main/java/com/lunhan/xxx/service/";
|
public static final String Controller = "./src/main/java/com/lunhan/xxx/host/controller/";
|
public static final String SEARCH = "./src/main/java/com/lunhan/xxx/entity/dto/search/";
|
public static final String RequestDTO = "./src/main/java/com/lunhan/xxx/entity/request/";
|
public static final String ResponseDTO = "./src/main/java/com/lunhan/xxx/entity/response/";
|
}
|
//endregion
|
|
//region 架包路径设置
|
class PackageSet {
|
public static final String PO = "com.lunhan.xxx.repository.po";
|
public static final String VO = "com.lunhan.xxx.repository.vo";
|
public static final String CONVERT_MAPPER = "com.lunhan.xxx.service.convert";
|
public static final String MAPPER_IMPL = "com.lunhan.xxx.repository.impl";
|
public static final String MAPPER = "com.lunhan.xxx.repository.mapper";
|
public static final String SERVICE = "com.lunhan.xxx.service";
|
public static final String Controller = "com.lunhan.xxx.host.controller";
|
public static final String SEARCH = "com.lunhan.xxx.entity.dto.search";
|
public static final String Request = "com.lunhan.xxx.entity.request";
|
public static final String Response = "com.lunhan.xxx.entity.response";
|
}
|
//endregion
|
|
//region 后缀名设置
|
class SuffixSet {
|
public static final String PO = "PO";
|
public static final String VO = "VO";
|
public static final String CONVERT_MAPPER = "Convert";
|
public static final String MAPPER_IMPL = "MapperImpl";
|
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 com.baomidou.mybatisplus.annotation.*;",
|
"",
|
"import java.io.Serializable;",
|
"import java.sql.Timestamp;"
|
};
|
//endregion
|
|
//region vo类 import 内容设置
|
private static final String[] SET_VO_IMPORT = new String[]{
|
"import com.lunhan.xxx.common.util.LocalDateTimeUtil;",
|
"import com.lunhan.xxx.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 dao类 import 内容设置
|
private static final String[] SET_DAO_IMPORT = new String[]{
|
"import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;",
|
"import com.baomidou.mybatisplus.extension.plugins.pagination.Page;",
|
"import com.lunhan.xxx.common.PagerResult;",
|
"import com.lunhan.xxx.common.enums.EYesOrNo;",
|
"import com.lunhan.xxx.common.util.ListUtil;",
|
"import com.lunhan.xxx.common.util.NumericUtil;",
|
"import com.lunhan.xxx.common.util.StringUtil;",
|
"import com.lunhan.xxx.entity.dto.OrderByDTO;",
|
"import com.lunhan.xxx.entity.dto.SearchBasicDTO;",
|
"import com.lunhan.xxx.entity.enums.EOrderBy;",
|
"import com.lunhan.xxx.repository.BasicMapperImpl;",
|
"import org.springframework.stereotype.Repository;"
|
};
|
//endregion
|
|
//region service类 import 内容设置
|
private static final String[] SET_Service_IMPORT = new String[]{
|
"import com.lunhan.xxx.common.*;",
|
"import com.lunhan.xxx.common.enums.*;",
|
"import com.lunhan.xxx.common.model.Tuple;",
|
"import com.lunhan.xxx.common.util.*;",
|
"import org.apache.commons.lang3.BooleanUtils;",
|
"import org.springframework.beans.factory.annotation.Autowired;",
|
"import org.springframework.stereotype.Service;",
|
"import java.util.*;",
|
"import java.util.stream.Collectors;"
|
};
|
//endregion
|
|
//region Controller类 import 内容设置
|
private static final String[] SET_Controller_IMPORT = new String[]{
|
"import com.lunhan.xxx.common.ExecutedResult;",
|
"import com.lunhan.xxx.common.PagerResult;",
|
"import com.lunhan.xxx.common.util.ParameterUtil;",
|
"import com.lunhan.xxx.common.validator.ParameterValidateResult;",
|
"import com.lunhan.xxx.common.validator.ParameterValidator;",
|
"import com.lunhan.xxx.entity.request.ReqListId;",
|
"import com.lunhan.xxx.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 " + GenCodeGauss.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(
|
"water_meter_apply"
|
);
|
|
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);
|
genCreateDto(columns, tableName, tableDesc);
|
|
genSearch(columns, tableDesc);
|
genMapper(columns, tableDesc);
|
genModifyDto(columns, tableName, tableDesc);
|
genDao(tableName, tableDesc);
|
genConvertMapper(tableName, tableDesc);
|
genService(tableName, tableDesc);
|
genController(tableName, 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 + GenCodeGauss.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(GenCodeGauss.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@TableId");
|
if (Objects.equals(isIdentity, 1)) {
|
listColumn.add("(type = IdType.AUTO)");
|
} else {
|
listColumn.add("(type = IdType.ASSIGN_ID)");
|
}
|
}
|
String columnAnnotation = "\t@TableField(\"" + 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("@TableName(\"" + 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(GenCodeGauss.OutSet.PO, modelName + GenCodeGauss.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 + GenCodeGauss.SuffixSet.PO;
|
String voName = modelName + GenCodeGauss.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(GenCodeGauss.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(GenCodeGauss.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(GenCodeGauss.OutSet.VO + modelName + GenCodeGauss.SuffixSet.VO + ".java").exists()) {
|
return;
|
}
|
writeFile(GenCodeGauss.OutSet.VO, modelName + GenCodeGauss.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(GenCodeGauss.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(GenCodeGauss.OutSet.SEARCH + "/" + searchName + ".java").exists()) {
|
return;
|
}
|
writeFile(GenCodeGauss.OutSet.SEARCH + "/", searchName + ".java", content.toString());
|
}
|
|
private static void genDao(String tableName, String tableDesc) {
|
String modelName = underline2Camel(tableName, true);
|
String mapperName = modelName + SuffixSet.MAPPER;
|
String daoName = modelName + GenCodeGauss.SuffixSet.MAPPER_IMPL;
|
String poName = modelName + GenCodeGauss.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(GenCodeGauss.PackageSet.MAPPER_IMPL);
|
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(GenCodeGauss.PackageSet.SEARCH + ".");
|
content.append(searchName + ";\n");
|
content.append("import ");
|
content.append(PackageSet.MAPPER);
|
content.append(".");
|
content.append(mapperName + ";\n");
|
content.append("import ");
|
content.append(GenCodeGauss.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);
|
content.append("@Repository\n");
|
content.append("public class ");
|
content.append(daoName);
|
content.append(" extends BasicMapperImpl<");
|
content.append(modelName + GenCodeGauss.SuffixSet.PO);
|
content.append("> {");
|
content.append(ConstantFactory.STR_NEWLINE);
|
content.append("\t" + daoName + "(" + mapperName + " mapper) {\n");
|
content.append("\t\tsuper(mapper);\n");
|
content.append("\t}\n");
|
content.append(ConstantFactory.STR_NEWLINE);
|
content.append("\t@Override\n" +
|
"\tpublic PagerResult<" + poName + "> search(" + searchName + " request) {\n" +
|
"\t\t// 还原查询条件真实类型\n" +
|
"\t\t" + searchName + " search = (" + searchName + ")request;\n" +
|
"\t\t// 查询条件\n" +
|
"\t\tLambdaQueryWrapper<" + poName + "> queryWrapper = this.query();\n" +
|
"\t\t// 非逻辑删除\n" +
|
"\t\tqueryWrapper.eq(" + poName + "::getIsDelete, EYesOrNo.NO.getValue());\n" +
|
"\t\t// 状态\n" +
|
"\t\t//queryWrapper.eq(NumericUtil.tryParseInt(search.getStatus()).compareTo(0) > 0, " + poName + "::getStatus, search.getStatus());\n" +
|
"\t\t// 状态列表\n" +
|
"\t\t//queryWrapper.in(ListUtil.isNotNullOrEmpty(search.getListStatus()), " + poName + "::getStatus, search.getListStatus());\n" +
|
"\n" +
|
"\t\t// 数据创建时间-起始\n" +
|
"\t\tqueryWrapper.ge(NumericUtil.tryParseLong(search.getCreateTimeStart()).compareTo(0L) > 0, " + poName + "::getCreateTime, search.getCreateTimeStart());\n" +
|
"\t\t// 数据创建时间-截止\n" +
|
"\t\tqueryWrapper.le(NumericUtil.tryParseLong(search.getCreateTimeEnd()).compareTo(0L) > 0, " + poName + "::getCreateTime, search.getCreateTimeEnd());\n" +
|
"\t\t// 关键字\n" +
|
"\t\tif (StringUtil.isNotNullOrEmpty(search.getKeywords())) {\n" +
|
"\t\t\tqueryWrapper.and(q ->\n" +
|
"\t\t\t\tq.like(" + poName + "::getName, search.getKeywords())\n" +
|
"\t\t\t\t.or().like(" + poName + "::getPhone, search.getKeywords())\n" +
|
"\t\t\t);\n" +
|
"\t\t}\n" +
|
"\n" +
|
"\t\t// 排序处理\n" +
|
"\t\tif (ListUtil.isNotNullOrEmpty(search.getOrderBy())) {\n" +
|
"\t\t\tfor (OrderByDTO item : search.getOrderBy()) {\n" +
|
"\t\t\t\tEOrderBy orderBy = EOrderBy.getByValue(item.getOrderBy());\n" +
|
"\t\t\t\t// 顺序排序\n" +
|
"\t\t\t\tif (item.getIsAsc()) {\n" +
|
"\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\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\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\tbreak;\n" +
|
"\t\t\t\t\t}\n" +
|
"\t\t\t\t} else {\n" +
|
"\t\t\t\t\t// 倒叙排序\n" +
|
"\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\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\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\tbreak;\n" +
|
"\t\t\t\t\t}\n" +
|
"\t\t\t\t}\n" +
|
"\t\t\t}\n" +
|
"\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\treturn new PagerResult<>(pageResult.getSize(), pageResult.getPages(), pageResult.getTotal(), pageResult.getRecords());\n" +
|
"\t}");
|
content.append("}");
|
|
if (new File(GenCodeGauss.OutSet.MAPPER_IMPL + daoName + ".java").exists()) {
|
return;
|
}
|
writeFile(GenCodeGauss.OutSet.MAPPER_IMPL, 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 + GenCodeGauss.SuffixSet.PO;
|
String createDTOName = GenCodeGauss.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(GenCodeGauss.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 (GenCodeGauss.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(GenCodeGauss.OutSet.RequestDTO + tableNamePackage + "/" + createDTOName + ".java").exists()) {
|
return;
|
}
|
writeFile(GenCodeGauss.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 + GenCodeGauss.SuffixSet.PO;
|
String modifyDTOName = GenCodeGauss.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(GenCodeGauss.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 (GenCodeGauss.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(GenCodeGauss.OutSet.RequestDTO + tableNamePackage + "/" + modifyDTOName + ".java").exists()) {
|
return;
|
}
|
writeFile(GenCodeGauss.OutSet.RequestDTO + tableNamePackage + "/", modifyDTOName + ".java", content.toString());
|
}
|
|
private static void genConvertMapper(String tableName, String tableDesc) {
|
String modelName = underline2Camel(tableName, true);
|
String poName = modelName + GenCodeGauss.SuffixSet.PO;
|
String voName = modelName + GenCodeGauss.SuffixSet.VO;
|
String createDTOName = GenCodeGauss.SuffixSet.Create + modelName;
|
String modifyDTOName = GenCodeGauss.SuffixSet.Modify + modelName;
|
String convertMapperName = modelName + SuffixSet.CONVERT_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(GenCodeGauss.PackageSet.CONVERT_MAPPER + ConstantFactory.STR_SEMICOLON);
|
content.append(ConstantFactory.STR_NEWLINE);
|
content.append(ConstantFactory.STR_NEWLINE);
|
|
content.append("import ");
|
content.append(GenCodeGauss.PackageSet.Request + "." + tableNamePackage + ".");
|
content.append(createDTOName + ConstantFactory.STR_SEMICOLON);
|
content.append(ConstantFactory.STR_NEWLINE);
|
|
content.append("import ");
|
content.append(GenCodeGauss.PackageSet.Request + "." + tableNamePackage + ".");
|
content.append(modifyDTOName + ConstantFactory.STR_SEMICOLON);
|
content.append(ConstantFactory.STR_NEWLINE);
|
|
content.append("import ");
|
content.append(GenCodeGauss.PackageSet.PO + ".");
|
content.append(poName + ConstantFactory.STR_SEMICOLON);
|
content.append(ConstantFactory.STR_NEWLINE);
|
|
content.append("import ");
|
content.append(GenCodeGauss.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(GenCodeGauss.OutSet.CONVERT_MAPPER + convertMapperName + ".java").exists()) {
|
return;
|
}
|
writeFile(GenCodeGauss.OutSet.CONVERT_MAPPER, convertMapperName + ".java", content.toString());
|
}
|
|
private static void genService(String tableName, String tableDesc) {
|
String modelName = underline2Camel(tableName, true);
|
String poName = modelName + GenCodeGauss.SuffixSet.PO;
|
String serviceName = modelName + GenCodeGauss.SuffixSet.SERVICE;
|
String daoName = modelName + GenCodeGauss.SuffixSet.MAPPER_IMPL;
|
String createDTOName = GenCodeGauss.SuffixSet.Create + modelName;
|
String modifyDTOName = GenCodeGauss.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(GenCodeGauss.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(GenCodeGauss.PackageSet.MAPPER_IMPL);
|
content.append(".");
|
content.append(daoName + ";\n");
|
content.append("import ");
|
content.append(GenCodeGauss.PackageSet.PO);
|
content.append(".");
|
content.append(poName + ";\n");
|
content.append("import ");
|
content.append(GenCodeGauss.PackageSet.Request);
|
content.append("." + tableNamePackage + ".");
|
content.append(createDTOName + ";\n");
|
content.append("import ");
|
content.append(GenCodeGauss.PackageSet.Request);
|
content.append("." + tableNamePackage + ".");
|
content.append(modifyDTOName + ";\n");
|
content.append("import ");
|
content.append(GenCodeGauss.PackageSet.SEARCH + ".");
|
content.append(searchName + ";\n");
|
content.append("import ");
|
content.append(GenCodeGauss.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" +
|
" // 设置状态\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(GenCodeGauss.OutSet.SERVICE + serviceName + ".java").exists()) {
|
return;
|
}
|
writeFile(GenCodeGauss.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 + GenCodeGauss.SuffixSet.PO;
|
String voName = modelName + GenCodeGauss.SuffixSet.VO;
|
String serviceName = modelName + GenCodeGauss.SuffixSet.SERVICE;
|
String ControllerName = modelName + GenCodeGauss.SuffixSet.Controller;
|
String createDTOName = GenCodeGauss.SuffixSet.Create + modelName;
|
String modifyDTOName = GenCodeGauss.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(GenCodeGauss.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(GenCodeGauss.PackageSet.SERVICE);
|
content.append(".");
|
content.append(serviceName + ";\n");
|
content.append("import ");
|
content.append(GenCodeGauss.PackageSet.Request);
|
content.append("." + tableNamePackage + ".");
|
content.append(createDTOName + ";\n");
|
content.append("import ");
|
content.append(GenCodeGauss.PackageSet.Request);
|
content.append("." + tableNamePackage + ".");
|
content.append(modifyDTOName + ";\n");
|
content.append("import ");
|
content.append(GenCodeGauss.PackageSet.SEARCH + ".");
|
content.append(searchName + ";\n");
|
content.append("import ");
|
content.append(GenCodeGauss.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(GenCodeGauss.OutSet.Controller + ControllerName + ".java").exists()) {
|
return;
|
}
|
writeFile(GenCodeGauss.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 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(GenCodeGauss.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;", GenCodeGauss.PackageSet.PO, modelName, GenCodeGauss.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, GenCodeGauss.SuffixSet.MAPPER, modelName, GenCodeGauss.SuffixSet.PO));
|
content.append(ConstantFactory.STR_NEWLINE);
|
content.append("}");
|
|
if (new File(GenCodeGauss.OutSet.MAPPER + modelName + GenCodeGauss.SuffixSet.MAPPER + ".java").exists()) {
|
return;
|
}
|
writeFile(GenCodeGauss.OutSet.MAPPER, modelName + GenCodeGauss.SuffixSet.MAPPER + ".java", content.toString());
|
}
|
|
//#endregion
|
|
|
private 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" +
|
" FROM information_schema.tables tb" +
|
" 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)) {
|
//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;
|
}
|
|
private 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 " +
|
"ordinal_position;";
|
try (PreparedStatement pStmt = CONN.prepareStatement(sql)) {
|
pStmt.setString(1, tableName);
|
// pStmt.setString(2, "public");
|
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;
|
}
|
|
private static String getJavaType(String typeName) {
|
String result = "String";
|
switch (typeName) {
|
case "integer":
|
case "tinyint":
|
result = "Integer";
|
break;
|
case "char":
|
case "varchar":
|
case "nvarchar":
|
case "text":
|
result = "String";
|
break;
|
case "datetime":
|
case "timestamp":
|
case "timestamp without time zone":
|
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;
|
}
|
|
private 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();
|
}
|
}
|
}
|
