是否生成DTO

This commit is contained in:
wangw 2025-02-17 14:12:34 +08:00
parent 919f727049
commit 00f83f0e2d
1 changed files with 14 additions and 10 deletions

View File

@ -23,13 +23,15 @@ public class Main {
// packageName 指定生成代码项目 // packageName 指定生成代码项目
// tableName 指定需要生成的表 // tableName 指定需要生成的表
// String servicePackageName = "system"; // String packageName = "system";
// String servicePackageName = "account";
// String servicePackageName = "product";
String packageName = "account"; String packageName = "account";
// String packageName = "product";
// String packageName = "order";
String tableName = "tb_shop_table"; String tableName = "tb_shop_coupon_product";
String author = "zs"; String author = "ww";
//是否生成DTO实体 默认生成
boolean isGenerateDto = true;
//配置数据源 //配置数据源
HikariDataSource dataSource = new HikariDataSource(); HikariDataSource dataSource = new HikariDataSource();
@ -53,11 +55,13 @@ public class Main {
serviceGenerator.generate(); serviceGenerator.generate();
//默认生成全参数DTO //默认生成全参数DTO
GlobalConfig dtoConfig = createDtoGlobalConfig(currentWorkingDirectory + "/cash-common/cash-common-service", if(isGenerateDto){
basePackage + packageName, "tb", tableName); GlobalConfig dtoConfig = createDtoGlobalConfig(currentWorkingDirectory + "/cash-common/cash-common-service",
Generator dtoGenerator = new Generator(dataSource, dtoConfig); basePackage + packageName, "tb", tableName);
dtoConfig.setAuthor(author); Generator dtoGenerator = new Generator(dataSource, dtoConfig);
dtoGenerator.generate(); dtoConfig.setAuthor(author);
dtoGenerator.generate();
}
} }
public static GlobalConfig createGlobalConfigUseStyle(String projectPath, String packageName, String tablePrefix, String... tableNames) { public static GlobalConfig createGlobalConfigUseStyle(String projectPath, String packageName, String tablePrefix, String... tableNames) {