14 changed files with 125 additions and 2351 deletions
@ -1,246 +0,0 @@ |
|||
package com.gxwebsoft.generator; |
|||
|
|||
import com.baomidou.mybatisplus.annotation.IdType; |
|||
import com.baomidou.mybatisplus.core.toolkit.StringPool; |
|||
import com.baomidou.mybatisplus.generator.AutoGenerator; |
|||
import com.baomidou.mybatisplus.generator.InjectionConfig; |
|||
import com.baomidou.mybatisplus.generator.config.*; |
|||
import com.baomidou.mybatisplus.generator.config.po.TableInfo; |
|||
import com.baomidou.mybatisplus.generator.config.rules.NamingStrategy; |
|||
import com.gxwebsoft.generator.engine.BeetlTemplateEnginePlus; |
|||
|
|||
import java.util.ArrayList; |
|||
import java.util.HashMap; |
|||
import java.util.List; |
|||
import java.util.Map; |
|||
|
|||
/** |
|||
* 小程序模块-代码生成工具 |
|||
* |
|||
* @author WebSoft |
|||
* @since 2021-09-05 00:31:14 |
|||
*/ |
|||
public class DocsGenerator { |
|||
// 输出位置
|
|||
private static final String OUTPUT_LOCATION = System.getProperty("user.dir"); |
|||
//private static final String OUTPUT_LOCATION = "D:/codegen"; // 不想生成到项目中可以写磁盘路径
|
|||
// 输出目录
|
|||
private static final String OUTPUT_DIR = "/src/main/java"; |
|||
// Vue文件输出位置
|
|||
private static final String OUTPUT_LOCATION_VUE = "/Users/gxwebsoft/VUE/oa"; |
|||
// Vue文件输出目录
|
|||
private static final String OUTPUT_LOCATION_UNIAPP = "/Users/gxwebsoft/APP/oa"; |
|||
// Vue文件输出目录
|
|||
private static final String OUTPUT_DIR_VUE = "/src"; |
|||
// 作者名称
|
|||
private static final String AUTHOR = "科技小王子"; |
|||
// 是否在xml中添加二级缓存配置
|
|||
private static final boolean ENABLE_CACHE = false; |
|||
// 数据库连接配置
|
|||
private static final String DB_URL = "jdbc:mysql://47.119.165.234:3308/modules?useUnicode=true&characterEncoding=utf8&useSSL=false&serverTimezone=GMT%2B8"; |
|||
private static final String DB_DRIVER = "com.mysql.cj.jdbc.Driver"; |
|||
private static final String DB_USERNAME = "modules"; |
|||
private static final String DB_PASSWORD = "8YdLnk7KsPAyDXGA"; |
|||
// 包名
|
|||
private static final String PACKAGE_NAME = "com.gxwebsoft"; |
|||
// 模块名
|
|||
private static final String MODULE_NAME = "docs"; |
|||
// 需要生成的表
|
|||
private static final String[] TABLE_NAMES = new String[]{ |
|||
"docs", |
|||
"docs_user", |
|||
"docs_content" |
|||
}; |
|||
// 需要去除的表前缀
|
|||
private static final String[] TABLE_PREFIX = new String[]{ |
|||
"tb_" |
|||
}; |
|||
// 不需要作为查询参数的字段
|
|||
private static final String[] PARAM_EXCLUDE_FIELDS = new String[]{ |
|||
"tenant_id", |
|||
"create_time", |
|||
"update_time" |
|||
}; |
|||
// 查询参数使用String的类型
|
|||
private static final String[] PARAM_TO_STRING_TYPE = new String[]{ |
|||
"Date", |
|||
"LocalDate", |
|||
"LocalTime", |
|||
"LocalDateTime" |
|||
}; |
|||
// 查询参数使用EQ的类型
|
|||
private static final String[] PARAM_EQ_TYPE = new String[]{ |
|||
"Integer", |
|||
"Boolean", |
|||
"BigDecimal" |
|||
}; |
|||
// 是否添加权限注解
|
|||
private static final boolean AUTH_ANNOTATION = true; |
|||
// 是否添加日志注解
|
|||
private static final boolean LOG_ANNOTATION = true; |
|||
// controller的mapping前缀
|
|||
private static final String CONTROLLER_MAPPING_PREFIX = "/api"; |
|||
// 模板所在位置
|
|||
private static final String TEMPLATES_DIR = "/src/test/java/com/gxwebsoft/generator/templates"; |
|||
|
|||
public static void main(String[] args) { |
|||
// 代码生成器
|
|||
AutoGenerator mpg = new AutoGenerator(); |
|||
|
|||
// 全局配置
|
|||
GlobalConfig gc = new GlobalConfig(); |
|||
gc.setOutputDir(OUTPUT_LOCATION + OUTPUT_DIR); |
|||
gc.setAuthor(AUTHOR); |
|||
gc.setOpen(false); |
|||
gc.setFileOverride(true); |
|||
gc.setEnableCache(ENABLE_CACHE); |
|||
gc.setSwagger2(true); |
|||
gc.setIdType(IdType.AUTO); |
|||
gc.setServiceName("%sService"); |
|||
mpg.setGlobalConfig(gc); |
|||
|
|||
// 数据源配置
|
|||
DataSourceConfig dsc = new DataSourceConfig(); |
|||
dsc.setUrl(DB_URL); |
|||
// dsc.setSchemaName("public");
|
|||
dsc.setDriverName(DB_DRIVER); |
|||
dsc.setUsername(DB_USERNAME); |
|||
dsc.setPassword(DB_PASSWORD); |
|||
mpg.setDataSource(dsc); |
|||
|
|||
// 包配置
|
|||
PackageConfig pc = new PackageConfig(); |
|||
pc.setModuleName(MODULE_NAME); |
|||
pc.setParent(PACKAGE_NAME); |
|||
mpg.setPackageInfo(pc); |
|||
|
|||
// 策略配置
|
|||
StrategyConfig strategy = new StrategyConfig(); |
|||
strategy.setNaming(NamingStrategy.underline_to_camel); |
|||
strategy.setColumnNaming(NamingStrategy.underline_to_camel); |
|||
strategy.setInclude(TABLE_NAMES); |
|||
strategy.setTablePrefix(TABLE_PREFIX); |
|||
strategy.setSuperControllerClass(PACKAGE_NAME + ".common.core.web.BaseController"); |
|||
strategy.setEntityLombokModel(true); |
|||
strategy.setRestControllerStyle(true); |
|||
strategy.setControllerMappingHyphenStyle(true); |
|||
strategy.setLogicDeleteFieldName("deleted"); |
|||
mpg.setStrategy(strategy); |
|||
|
|||
// 模板配置
|
|||
TemplateConfig templateConfig = new TemplateConfig(); |
|||
templateConfig.setController(TEMPLATES_DIR + "/controller.java"); |
|||
templateConfig.setEntity(TEMPLATES_DIR + "/entity.java"); |
|||
templateConfig.setMapper(TEMPLATES_DIR + "/mapper.java"); |
|||
templateConfig.setXml(TEMPLATES_DIR + "/mapper.xml"); |
|||
templateConfig.setService(TEMPLATES_DIR + "/service.java"); |
|||
templateConfig.setServiceImpl(TEMPLATES_DIR + "/serviceImpl.java"); |
|||
mpg.setTemplate(templateConfig); |
|||
mpg.setTemplateEngine(new BeetlTemplateEnginePlus()); |
|||
|
|||
// 自定义模板配置
|
|||
InjectionConfig cfg = new InjectionConfig() { |
|||
@Override |
|||
public void initMap() { |
|||
Map<String, Object> map = new HashMap<>(); |
|||
map.put("packageName", PACKAGE_NAME); |
|||
map.put("paramExcludeFields", PARAM_EXCLUDE_FIELDS); |
|||
map.put("paramToStringType", PARAM_TO_STRING_TYPE); |
|||
map.put("paramEqType", PARAM_EQ_TYPE); |
|||
map.put("authAnnotation", AUTH_ANNOTATION); |
|||
map.put("logAnnotation", LOG_ANNOTATION); |
|||
map.put("controllerMappingPrefix", CONTROLLER_MAPPING_PREFIX); |
|||
this.setMap(map); |
|||
} |
|||
}; |
|||
String templatePath = TEMPLATES_DIR + "/param.java.btl"; |
|||
List<FileOutConfig> focList = new ArrayList<>(); |
|||
focList.add(new FileOutConfig(templatePath) { |
|||
@Override |
|||
public String outputFile(TableInfo tableInfo) { |
|||
return OUTPUT_LOCATION + OUTPUT_DIR + "/" |
|||
+ PACKAGE_NAME.replace(".", "/") |
|||
+ "/" + pc.getModuleName() + "/param/" |
|||
+ tableInfo.getEntityName() + "Param" + StringPool.DOT_JAVA; |
|||
} |
|||
}); |
|||
/** |
|||
* 以下是生成VUE项目代码 |
|||
* 生成文件的路径 /api/shop/goods/index.ts |
|||
*/ |
|||
templatePath = TEMPLATES_DIR + "/index.ts.btl"; |
|||
|
|||
focList.add(new FileOutConfig(templatePath) { |
|||
@Override |
|||
public String outputFile(TableInfo tableInfo) { |
|||
return OUTPUT_LOCATION_VUE + OUTPUT_DIR_VUE |
|||
+ "/api/" + pc.getModuleName() + "/" |
|||
+ tableInfo.getEntityPath() + "/" + "index.ts"; |
|||
} |
|||
}); |
|||
focList.add(new FileOutConfig() { |
|||
@Override |
|||
public String outputFile(TableInfo tableInfo) { |
|||
return OUTPUT_LOCATION_UNIAPP + OUTPUT_DIR_VUE |
|||
+ "/api/" + pc.getModuleName() + "/" |
|||
+ tableInfo.getEntityPath() + "/" + "index.ts"; |
|||
} |
|||
}); |
|||
// 生成TS文件 (/api/shop/goods/model/index.ts)
|
|||
templatePath = TEMPLATES_DIR + "/model.ts.btl"; |
|||
focList.add(new FileOutConfig(templatePath) { |
|||
@Override |
|||
public String outputFile(TableInfo tableInfo) { |
|||
return OUTPUT_LOCATION_VUE + OUTPUT_DIR_VUE |
|||
+ "/api/" + pc.getModuleName() + "/" |
|||
+ tableInfo.getEntityPath() + "/model/" + "index.ts"; |
|||
} |
|||
}); |
|||
focList.add(new FileOutConfig(templatePath) { |
|||
@Override |
|||
public String outputFile(TableInfo tableInfo) { |
|||
return OUTPUT_LOCATION_UNIAPP + OUTPUT_DIR_VUE |
|||
+ "/api/" + pc.getModuleName() + "/" |
|||
+ tableInfo.getEntityPath() + "/model/" + "index.ts"; |
|||
} |
|||
}); |
|||
// 生成Vue文件(/views/shop/goods/index.vue)
|
|||
templatePath = TEMPLATES_DIR + "/index.vue.btl"; |
|||
focList.add(new FileOutConfig(templatePath) { |
|||
@Override |
|||
public String outputFile(TableInfo tableInfo) { |
|||
return OUTPUT_LOCATION_VUE + OUTPUT_DIR_VUE |
|||
+ "/views/" + pc.getModuleName() + "/" |
|||
+ tableInfo.getEntityPath() + "/" + "index.vue"; |
|||
} |
|||
}); |
|||
|
|||
// 生成components文件(/views/shop/goods/components/edit.vue)
|
|||
templatePath = TEMPLATES_DIR + "/components.edit.vue.btl"; |
|||
focList.add(new FileOutConfig(templatePath) { |
|||
@Override |
|||
public String outputFile(TableInfo tableInfo) { |
|||
return OUTPUT_LOCATION_VUE + OUTPUT_DIR_VUE |
|||
+ "/views/" + pc.getModuleName() + "/" |
|||
+ tableInfo.getEntityPath() + "/components/" + tableInfo.getEntityPath() + "Edit.vue"; |
|||
} |
|||
}); |
|||
|
|||
// 生成components文件(/views/shop/goods/components/search.vue)
|
|||
templatePath = TEMPLATES_DIR + "/components.search.vue.btl"; |
|||
focList.add(new FileOutConfig(templatePath) { |
|||
@Override |
|||
public String outputFile(TableInfo tableInfo) { |
|||
return OUTPUT_LOCATION_VUE + OUTPUT_DIR_VUE |
|||
+ "/views/" + pc.getModuleName() + "/" |
|||
+ tableInfo.getEntityPath() + "/components/" + "search.vue"; |
|||
} |
|||
}); |
|||
|
|||
cfg.setFileOutConfigList(focList); |
|||
mpg.setCfg(cfg); |
|||
|
|||
mpg.execute(); |
|||
} |
|||
|
|||
} |
|||
@ -1,248 +0,0 @@ |
|||
package com.gxwebsoft.generator; |
|||
|
|||
import com.baomidou.mybatisplus.annotation.IdType; |
|||
import com.baomidou.mybatisplus.core.toolkit.StringPool; |
|||
import com.baomidou.mybatisplus.generator.AutoGenerator; |
|||
import com.baomidou.mybatisplus.generator.InjectionConfig; |
|||
import com.baomidou.mybatisplus.generator.config.*; |
|||
import com.baomidou.mybatisplus.generator.config.po.TableInfo; |
|||
import com.baomidou.mybatisplus.generator.config.rules.NamingStrategy; |
|||
import com.gxwebsoft.generator.engine.BeetlTemplateEnginePlus; |
|||
|
|||
import java.util.ArrayList; |
|||
import java.util.HashMap; |
|||
import java.util.List; |
|||
import java.util.Map; |
|||
|
|||
/** |
|||
* 独立应用模块-代码生成工具 |
|||
* |
|||
* @author WebSoft |
|||
* @since 2021-09-05 00:31:14 |
|||
*/ |
|||
public class HouseGenerator { |
|||
// 输出位置
|
|||
private static final String OUTPUT_LOCATION = System.getProperty("user.dir"); |
|||
//private static final String OUTPUT_LOCATION = "D:/codegen"; // 不想生成到项目中可以写磁盘路径
|
|||
// 输出目录
|
|||
private static final String OUTPUT_DIR = "/src/main/java"; |
|||
// Vue文件输出位置
|
|||
private static final String OUTPUT_LOCATION_VUE = "/Users/gxwebsoft/VUE/site"; |
|||
// Vue文件输出目录
|
|||
private static final String OUTPUT_LOCATION_UNIAPP = "/Users/gxwebsoft/APP/site"; |
|||
// Vue文件输出目录
|
|||
private static final String OUTPUT_DIR_VUE = "/src"; |
|||
// 作者名称
|
|||
private static final String AUTHOR = "科技小王子"; |
|||
// 是否在xml中添加二级缓存配置
|
|||
private static final boolean ENABLE_CACHE = false; |
|||
// 数据库连接配置
|
|||
private static final String DB_URL = "jdbc:mysql://47.119.165.234:3308/modules?useUnicode=true&characterEncoding=utf8&useSSL=false&serverTimezone=GMT%2B8"; |
|||
private static final String DB_DRIVER = "com.mysql.cj.jdbc.Driver"; |
|||
private static final String DB_USERNAME = "modules"; |
|||
private static final String DB_PASSWORD = "8YdLnk7KsPAyDXGA"; |
|||
// 包名
|
|||
private static final String PACKAGE_NAME = "com.gxwebsoft"; |
|||
// 模块名
|
|||
private static final String MODULE_NAME = "house"; |
|||
// 需要生成的表
|
|||
private static final String[] TABLE_NAMES = new String[]{ |
|||
// "house_info",
|
|||
// "house_like_log",
|
|||
// "house_reservation",
|
|||
// "house_views_log",
|
|||
// "house_user"
|
|||
}; |
|||
// 需要去除的表前缀
|
|||
private static final String[] TABLE_PREFIX = new String[]{ |
|||
"tb_" |
|||
}; |
|||
// 不需要作为查询参数的字段
|
|||
private static final String[] PARAM_EXCLUDE_FIELDS = new String[]{ |
|||
"tenant_id", |
|||
"create_time", |
|||
"update_time" |
|||
}; |
|||
// 查询参数使用String的类型
|
|||
private static final String[] PARAM_TO_STRING_TYPE = new String[]{ |
|||
"Date", |
|||
"LocalDate", |
|||
"LocalTime", |
|||
"LocalDateTime" |
|||
}; |
|||
// 查询参数使用EQ的类型
|
|||
private static final String[] PARAM_EQ_TYPE = new String[]{ |
|||
"Integer", |
|||
"Boolean", |
|||
"BigDecimal" |
|||
}; |
|||
// 是否添加权限注解
|
|||
private static final boolean AUTH_ANNOTATION = false; |
|||
// 是否添加日志注解
|
|||
private static final boolean LOG_ANNOTATION = false; |
|||
// controller的mapping前缀
|
|||
private static final String CONTROLLER_MAPPING_PREFIX = "/api"; |
|||
// 模板所在位置
|
|||
private static final String TEMPLATES_DIR = "/src/test/java/com/gxwebsoft/generator/templates"; |
|||
|
|||
public static void main(String[] args) { |
|||
// 代码生成器
|
|||
AutoGenerator mpg = new AutoGenerator(); |
|||
|
|||
// 全局配置
|
|||
GlobalConfig gc = new GlobalConfig(); |
|||
gc.setOutputDir(OUTPUT_LOCATION + OUTPUT_DIR); |
|||
gc.setAuthor(AUTHOR); |
|||
gc.setOpen(false); |
|||
gc.setFileOverride(true); |
|||
gc.setEnableCache(ENABLE_CACHE); |
|||
gc.setSwagger2(true); |
|||
gc.setIdType(IdType.AUTO); |
|||
gc.setServiceName("%sService"); |
|||
mpg.setGlobalConfig(gc); |
|||
|
|||
// 数据源配置
|
|||
DataSourceConfig dsc = new DataSourceConfig(); |
|||
dsc.setUrl(DB_URL); |
|||
// dsc.setSchemaName("public");
|
|||
dsc.setDriverName(DB_DRIVER); |
|||
dsc.setUsername(DB_USERNAME); |
|||
dsc.setPassword(DB_PASSWORD); |
|||
mpg.setDataSource(dsc); |
|||
|
|||
// 包配置
|
|||
PackageConfig pc = new PackageConfig(); |
|||
pc.setModuleName(MODULE_NAME); |
|||
pc.setParent(PACKAGE_NAME); |
|||
mpg.setPackageInfo(pc); |
|||
|
|||
// 策略配置
|
|||
StrategyConfig strategy = new StrategyConfig(); |
|||
strategy.setNaming(NamingStrategy.underline_to_camel); |
|||
strategy.setColumnNaming(NamingStrategy.underline_to_camel); |
|||
strategy.setInclude(TABLE_NAMES); |
|||
strategy.setTablePrefix(TABLE_PREFIX); |
|||
strategy.setSuperControllerClass(PACKAGE_NAME + ".common.core.web.BaseController"); |
|||
strategy.setEntityLombokModel(true); |
|||
strategy.setRestControllerStyle(true); |
|||
strategy.setControllerMappingHyphenStyle(true); |
|||
strategy.setLogicDeleteFieldName("deleted"); |
|||
mpg.setStrategy(strategy); |
|||
|
|||
// 模板配置
|
|||
TemplateConfig templateConfig = new TemplateConfig(); |
|||
templateConfig.setController(TEMPLATES_DIR + "/controller.java"); |
|||
templateConfig.setEntity(TEMPLATES_DIR + "/entity.java"); |
|||
templateConfig.setMapper(TEMPLATES_DIR + "/mapper.java"); |
|||
templateConfig.setXml(TEMPLATES_DIR + "/mapper.xml"); |
|||
templateConfig.setService(TEMPLATES_DIR + "/service.java"); |
|||
templateConfig.setServiceImpl(TEMPLATES_DIR + "/serviceImpl.java"); |
|||
mpg.setTemplate(templateConfig); |
|||
mpg.setTemplateEngine(new BeetlTemplateEnginePlus()); |
|||
|
|||
// 自定义模板配置
|
|||
InjectionConfig cfg = new InjectionConfig() { |
|||
@Override |
|||
public void initMap() { |
|||
Map<String, Object> map = new HashMap<>(); |
|||
map.put("packageName", PACKAGE_NAME); |
|||
map.put("paramExcludeFields", PARAM_EXCLUDE_FIELDS); |
|||
map.put("paramToStringType", PARAM_TO_STRING_TYPE); |
|||
map.put("paramEqType", PARAM_EQ_TYPE); |
|||
map.put("authAnnotation", AUTH_ANNOTATION); |
|||
map.put("logAnnotation", LOG_ANNOTATION); |
|||
map.put("controllerMappingPrefix", CONTROLLER_MAPPING_PREFIX); |
|||
this.setMap(map); |
|||
} |
|||
}; |
|||
String templatePath = TEMPLATES_DIR + "/param.java.btl"; |
|||
List<FileOutConfig> focList = new ArrayList<>(); |
|||
focList.add(new FileOutConfig(templatePath) { |
|||
@Override |
|||
public String outputFile(TableInfo tableInfo) { |
|||
return OUTPUT_LOCATION + OUTPUT_DIR + "/" |
|||
+ PACKAGE_NAME.replace(".", "/") |
|||
+ "/" + pc.getModuleName() + "/param/" |
|||
+ tableInfo.getEntityName() + "Param" + StringPool.DOT_JAVA; |
|||
} |
|||
}); |
|||
/** |
|||
* 以下是生成VUE项目代码 |
|||
* 生成文件的路径 /api/shop/goods/index.ts |
|||
*/ |
|||
templatePath = TEMPLATES_DIR + "/index.ts.btl"; |
|||
|
|||
focList.add(new FileOutConfig(templatePath) { |
|||
@Override |
|||
public String outputFile(TableInfo tableInfo) { |
|||
return OUTPUT_LOCATION_VUE + OUTPUT_DIR_VUE |
|||
+ "/api/" + pc.getModuleName() + "/" |
|||
+ tableInfo.getEntityPath() + "/" + "index.ts"; |
|||
} |
|||
}); |
|||
focList.add(new FileOutConfig() { |
|||
@Override |
|||
public String outputFile(TableInfo tableInfo) { |
|||
return OUTPUT_LOCATION_UNIAPP + OUTPUT_DIR_VUE |
|||
+ "/api/" + pc.getModuleName() + "/" |
|||
+ tableInfo.getEntityPath() + "/" + "index.ts"; |
|||
} |
|||
}); |
|||
// 生成TS文件 (/api/shop/goods/model/index.ts)
|
|||
templatePath = TEMPLATES_DIR + "/model.ts.btl"; |
|||
focList.add(new FileOutConfig(templatePath) { |
|||
@Override |
|||
public String outputFile(TableInfo tableInfo) { |
|||
return OUTPUT_LOCATION_VUE + OUTPUT_DIR_VUE |
|||
+ "/api/" + pc.getModuleName() + "/" |
|||
+ tableInfo.getEntityPath() + "/model/" + "index.ts"; |
|||
} |
|||
}); |
|||
focList.add(new FileOutConfig(templatePath) { |
|||
@Override |
|||
public String outputFile(TableInfo tableInfo) { |
|||
return OUTPUT_LOCATION_UNIAPP + OUTPUT_DIR_VUE |
|||
+ "/api/" + pc.getModuleName() + "/" |
|||
+ tableInfo.getEntityPath() + "/model/" + "index.ts"; |
|||
} |
|||
}); |
|||
// 生成Vue文件(/views/shop/goods/index.vue)
|
|||
templatePath = TEMPLATES_DIR + "/index.vue.btl"; |
|||
focList.add(new FileOutConfig(templatePath) { |
|||
@Override |
|||
public String outputFile(TableInfo tableInfo) { |
|||
return OUTPUT_LOCATION_VUE + OUTPUT_DIR_VUE |
|||
+ "/views/" + pc.getModuleName() + "/" |
|||
+ tableInfo.getEntityPath() + "/" + "index.vue"; |
|||
} |
|||
}); |
|||
|
|||
// 生成components文件(/views/shop/goods/components/edit.vue)
|
|||
templatePath = TEMPLATES_DIR + "/components.edit.vue.btl"; |
|||
focList.add(new FileOutConfig(templatePath) { |
|||
@Override |
|||
public String outputFile(TableInfo tableInfo) { |
|||
return OUTPUT_LOCATION_VUE + OUTPUT_DIR_VUE |
|||
+ "/views/" + pc.getModuleName() + "/" |
|||
+ tableInfo.getEntityPath() + "/components/" + tableInfo.getEntityPath() + "Edit.vue"; |
|||
} |
|||
}); |
|||
|
|||
// 生成components文件(/views/shop/goods/components/search.vue)
|
|||
templatePath = TEMPLATES_DIR + "/components.search.vue.btl"; |
|||
focList.add(new FileOutConfig(templatePath) { |
|||
@Override |
|||
public String outputFile(TableInfo tableInfo) { |
|||
return OUTPUT_LOCATION_VUE + OUTPUT_DIR_VUE |
|||
+ "/views/" + pc.getModuleName() + "/" |
|||
+ tableInfo.getEntityPath() + "/components/" + "search.vue"; |
|||
} |
|||
}); |
|||
|
|||
cfg.setFileOutConfigList(focList); |
|||
mpg.setCfg(cfg); |
|||
|
|||
mpg.execute(); |
|||
} |
|||
|
|||
} |
|||
@ -1,248 +0,0 @@ |
|||
package com.gxwebsoft.generator; |
|||
|
|||
import com.baomidou.mybatisplus.annotation.IdType; |
|||
import com.baomidou.mybatisplus.core.toolkit.StringPool; |
|||
import com.baomidou.mybatisplus.generator.AutoGenerator; |
|||
import com.baomidou.mybatisplus.generator.InjectionConfig; |
|||
import com.baomidou.mybatisplus.generator.config.*; |
|||
import com.baomidou.mybatisplus.generator.config.po.TableInfo; |
|||
import com.baomidou.mybatisplus.generator.config.rules.NamingStrategy; |
|||
import com.gxwebsoft.generator.engine.BeetlTemplateEnginePlus; |
|||
|
|||
import java.util.ArrayList; |
|||
import java.util.HashMap; |
|||
import java.util.List; |
|||
import java.util.Map; |
|||
|
|||
/** |
|||
* 小程序模块-代码生成工具 |
|||
* |
|||
* @author WebSoft |
|||
* @since 2021-09-05 00:31:14 |
|||
*/ |
|||
public class MpGenerator { |
|||
// 输出位置
|
|||
private static final String OUTPUT_LOCATION = System.getProperty("user.dir"); |
|||
//private static final String OUTPUT_LOCATION = "D:/codegen"; // 不想生成到项目中可以写磁盘路径
|
|||
// 输出目录
|
|||
private static final String OUTPUT_DIR = "/src/main/java"; |
|||
// Vue文件输出位置
|
|||
private static final String OUTPUT_LOCATION_VUE = "/Users/gxwebsoft/VUE/mp"; |
|||
// Vue文件输出目录
|
|||
private static final String OUTPUT_LOCATION_UNIAPP = "/Users/gxwebsoft/APP/mp"; |
|||
// Vue文件输出目录
|
|||
private static final String OUTPUT_DIR_VUE = "/src"; |
|||
// 作者名称
|
|||
private static final String AUTHOR = "科技小王子"; |
|||
// 是否在xml中添加二级缓存配置
|
|||
private static final boolean ENABLE_CACHE = false; |
|||
// 数据库连接配置
|
|||
private static final String DB_URL = "jdbc:mysql://47.119.165.234:3308/modules?useUnicode=true&characterEncoding=utf8&useSSL=false&serverTimezone=GMT%2B8"; |
|||
private static final String DB_DRIVER = "com.mysql.cj.jdbc.Driver"; |
|||
private static final String DB_USERNAME = "modules"; |
|||
private static final String DB_PASSWORD = "8YdLnk7KsPAyDXGA"; |
|||
// 包名
|
|||
private static final String PACKAGE_NAME = "com.gxwebsoft"; |
|||
// 模块名
|
|||
private static final String MODULE_NAME = "mp"; |
|||
// 需要生成的表
|
|||
private static final String[] TABLE_NAMES = new String[]{ |
|||
"mp", |
|||
"mp_ad", |
|||
"mp_field", |
|||
"mp_menu", |
|||
"mp_pages" |
|||
}; |
|||
// 需要去除的表前缀
|
|||
private static final String[] TABLE_PREFIX = new String[]{ |
|||
"tb_" |
|||
}; |
|||
// 不需要作为查询参数的字段
|
|||
private static final String[] PARAM_EXCLUDE_FIELDS = new String[]{ |
|||
"tenant_id", |
|||
"create_time", |
|||
"update_time" |
|||
}; |
|||
// 查询参数使用String的类型
|
|||
private static final String[] PARAM_TO_STRING_TYPE = new String[]{ |
|||
"Date", |
|||
"LocalDate", |
|||
"LocalTime", |
|||
"LocalDateTime" |
|||
}; |
|||
// 查询参数使用EQ的类型
|
|||
private static final String[] PARAM_EQ_TYPE = new String[]{ |
|||
"Integer", |
|||
"Boolean", |
|||
"BigDecimal" |
|||
}; |
|||
// 是否添加权限注解
|
|||
private static final boolean AUTH_ANNOTATION = true; |
|||
// 是否添加日志注解
|
|||
private static final boolean LOG_ANNOTATION = true; |
|||
// controller的mapping前缀
|
|||
private static final String CONTROLLER_MAPPING_PREFIX = "/api"; |
|||
// 模板所在位置
|
|||
private static final String TEMPLATES_DIR = "/src/test/java/com/gxwebsoft/generator/templates"; |
|||
|
|||
public static void main(String[] args) { |
|||
// 代码生成器
|
|||
AutoGenerator mpg = new AutoGenerator(); |
|||
|
|||
// 全局配置
|
|||
GlobalConfig gc = new GlobalConfig(); |
|||
gc.setOutputDir(OUTPUT_LOCATION + OUTPUT_DIR); |
|||
gc.setAuthor(AUTHOR); |
|||
gc.setOpen(false); |
|||
gc.setFileOverride(true); |
|||
gc.setEnableCache(ENABLE_CACHE); |
|||
gc.setSwagger2(true); |
|||
gc.setIdType(IdType.AUTO); |
|||
gc.setServiceName("%sService"); |
|||
mpg.setGlobalConfig(gc); |
|||
|
|||
// 数据源配置
|
|||
DataSourceConfig dsc = new DataSourceConfig(); |
|||
dsc.setUrl(DB_URL); |
|||
// dsc.setSchemaName("public");
|
|||
dsc.setDriverName(DB_DRIVER); |
|||
dsc.setUsername(DB_USERNAME); |
|||
dsc.setPassword(DB_PASSWORD); |
|||
mpg.setDataSource(dsc); |
|||
|
|||
// 包配置
|
|||
PackageConfig pc = new PackageConfig(); |
|||
pc.setModuleName(MODULE_NAME); |
|||
pc.setParent(PACKAGE_NAME); |
|||
mpg.setPackageInfo(pc); |
|||
|
|||
// 策略配置
|
|||
StrategyConfig strategy = new StrategyConfig(); |
|||
strategy.setNaming(NamingStrategy.underline_to_camel); |
|||
strategy.setColumnNaming(NamingStrategy.underline_to_camel); |
|||
strategy.setInclude(TABLE_NAMES); |
|||
strategy.setTablePrefix(TABLE_PREFIX); |
|||
strategy.setSuperControllerClass(PACKAGE_NAME + ".common.core.web.BaseController"); |
|||
strategy.setEntityLombokModel(true); |
|||
strategy.setRestControllerStyle(true); |
|||
strategy.setControllerMappingHyphenStyle(true); |
|||
strategy.setLogicDeleteFieldName("deleted"); |
|||
mpg.setStrategy(strategy); |
|||
|
|||
// 模板配置
|
|||
TemplateConfig templateConfig = new TemplateConfig(); |
|||
templateConfig.setController(TEMPLATES_DIR + "/controller.java"); |
|||
templateConfig.setEntity(TEMPLATES_DIR + "/entity.java"); |
|||
templateConfig.setMapper(TEMPLATES_DIR + "/mapper.java"); |
|||
templateConfig.setXml(TEMPLATES_DIR + "/mapper.xml"); |
|||
templateConfig.setService(TEMPLATES_DIR + "/service.java"); |
|||
templateConfig.setServiceImpl(TEMPLATES_DIR + "/serviceImpl.java"); |
|||
mpg.setTemplate(templateConfig); |
|||
mpg.setTemplateEngine(new BeetlTemplateEnginePlus()); |
|||
|
|||
// 自定义模板配置
|
|||
InjectionConfig cfg = new InjectionConfig() { |
|||
@Override |
|||
public void initMap() { |
|||
Map<String, Object> map = new HashMap<>(); |
|||
map.put("packageName", PACKAGE_NAME); |
|||
map.put("paramExcludeFields", PARAM_EXCLUDE_FIELDS); |
|||
map.put("paramToStringType", PARAM_TO_STRING_TYPE); |
|||
map.put("paramEqType", PARAM_EQ_TYPE); |
|||
map.put("authAnnotation", AUTH_ANNOTATION); |
|||
map.put("logAnnotation", LOG_ANNOTATION); |
|||
map.put("controllerMappingPrefix", CONTROLLER_MAPPING_PREFIX); |
|||
this.setMap(map); |
|||
} |
|||
}; |
|||
String templatePath = TEMPLATES_DIR + "/param.java.btl"; |
|||
List<FileOutConfig> focList = new ArrayList<>(); |
|||
focList.add(new FileOutConfig(templatePath) { |
|||
@Override |
|||
public String outputFile(TableInfo tableInfo) { |
|||
return OUTPUT_LOCATION + OUTPUT_DIR + "/" |
|||
+ PACKAGE_NAME.replace(".", "/") |
|||
+ "/" + pc.getModuleName() + "/param/" |
|||
+ tableInfo.getEntityName() + "Param" + StringPool.DOT_JAVA; |
|||
} |
|||
}); |
|||
/** |
|||
* 以下是生成VUE项目代码 |
|||
* 生成文件的路径 /api/shop/goods/index.ts |
|||
*/ |
|||
templatePath = TEMPLATES_DIR + "/index.ts.btl"; |
|||
|
|||
focList.add(new FileOutConfig(templatePath) { |
|||
@Override |
|||
public String outputFile(TableInfo tableInfo) { |
|||
return OUTPUT_LOCATION_VUE + OUTPUT_DIR_VUE |
|||
+ "/api/" + pc.getModuleName() + "/" |
|||
+ tableInfo.getEntityPath() + "/" + "index.ts"; |
|||
} |
|||
}); |
|||
focList.add(new FileOutConfig() { |
|||
@Override |
|||
public String outputFile(TableInfo tableInfo) { |
|||
return OUTPUT_LOCATION_UNIAPP + OUTPUT_DIR_VUE |
|||
+ "/api/" + pc.getModuleName() + "/" |
|||
+ tableInfo.getEntityPath() + "/" + "index.ts"; |
|||
} |
|||
}); |
|||
// 生成TS文件 (/api/shop/goods/model/index.ts)
|
|||
templatePath = TEMPLATES_DIR + "/model.ts.btl"; |
|||
focList.add(new FileOutConfig(templatePath) { |
|||
@Override |
|||
public String outputFile(TableInfo tableInfo) { |
|||
return OUTPUT_LOCATION_VUE + OUTPUT_DIR_VUE |
|||
+ "/api/" + pc.getModuleName() + "/" |
|||
+ tableInfo.getEntityPath() + "/model/" + "index.ts"; |
|||
} |
|||
}); |
|||
focList.add(new FileOutConfig(templatePath) { |
|||
@Override |
|||
public String outputFile(TableInfo tableInfo) { |
|||
return OUTPUT_LOCATION_UNIAPP + OUTPUT_DIR_VUE |
|||
+ "/api/" + pc.getModuleName() + "/" |
|||
+ tableInfo.getEntityPath() + "/model/" + "index.ts"; |
|||
} |
|||
}); |
|||
// 生成Vue文件(/views/shop/goods/index.vue)
|
|||
templatePath = TEMPLATES_DIR + "/index.vue.btl"; |
|||
focList.add(new FileOutConfig(templatePath) { |
|||
@Override |
|||
public String outputFile(TableInfo tableInfo) { |
|||
return OUTPUT_LOCATION_VUE + OUTPUT_DIR_VUE |
|||
+ "/views/" + pc.getModuleName() + "/" |
|||
+ tableInfo.getEntityPath() + "/" + "index.vue"; |
|||
} |
|||
}); |
|||
|
|||
// 生成components文件(/views/shop/goods/components/edit.vue)
|
|||
templatePath = TEMPLATES_DIR + "/components.edit.vue.btl"; |
|||
focList.add(new FileOutConfig(templatePath) { |
|||
@Override |
|||
public String outputFile(TableInfo tableInfo) { |
|||
return OUTPUT_LOCATION_VUE + OUTPUT_DIR_VUE |
|||
+ "/views/" + pc.getModuleName() + "/" |
|||
+ tableInfo.getEntityPath() + "/components/" + tableInfo.getEntityPath() + "Edit.vue"; |
|||
} |
|||
}); |
|||
|
|||
// 生成components文件(/views/shop/goods/components/search.vue)
|
|||
templatePath = TEMPLATES_DIR + "/components.search.vue.btl"; |
|||
focList.add(new FileOutConfig(templatePath) { |
|||
@Override |
|||
public String outputFile(TableInfo tableInfo) { |
|||
return OUTPUT_LOCATION_VUE + OUTPUT_DIR_VUE |
|||
+ "/views/" + pc.getModuleName() + "/" |
|||
+ tableInfo.getEntityPath() + "/components/" + "search.vue"; |
|||
} |
|||
}); |
|||
|
|||
cfg.setFileOutConfigList(focList); |
|||
mpg.setCfg(cfg); |
|||
|
|||
mpg.execute(); |
|||
} |
|||
|
|||
} |
|||
@ -1,244 +0,0 @@ |
|||
package com.gxwebsoft.generator; |
|||
|
|||
import com.baomidou.mybatisplus.annotation.IdType; |
|||
import com.baomidou.mybatisplus.core.toolkit.StringPool; |
|||
import com.baomidou.mybatisplus.generator.AutoGenerator; |
|||
import com.baomidou.mybatisplus.generator.InjectionConfig; |
|||
import com.baomidou.mybatisplus.generator.config.*; |
|||
import com.baomidou.mybatisplus.generator.config.po.TableInfo; |
|||
import com.baomidou.mybatisplus.generator.config.rules.NamingStrategy; |
|||
import com.gxwebsoft.generator.engine.BeetlTemplateEnginePlus; |
|||
|
|||
import java.util.ArrayList; |
|||
import java.util.HashMap; |
|||
import java.util.List; |
|||
import java.util.Map; |
|||
|
|||
/** |
|||
* 小程序模块-代码生成工具 |
|||
* |
|||
* @author WebSoft |
|||
* @since 2021-09-05 00:31:14 |
|||
*/ |
|||
public class PwlGenerator { |
|||
// 输出位置
|
|||
private static final String OUTPUT_LOCATION = System.getProperty("user.dir"); |
|||
//private static final String OUTPUT_LOCATION = "D:/codegen"; // 不想生成到项目中可以写磁盘路径
|
|||
// 输出目录
|
|||
private static final String OUTPUT_DIR = "/src/main/java"; |
|||
// Vue文件输出位置
|
|||
private static final String OUTPUT_LOCATION_VUE = "/Users/gxwebsoft/VUE/oa"; |
|||
// Vue文件输出目录
|
|||
private static final String OUTPUT_LOCATION_UNIAPP = "/Users/gxwebsoft/APP/oa"; |
|||
// Vue文件输出目录
|
|||
private static final String OUTPUT_DIR_VUE = "/src"; |
|||
// 作者名称
|
|||
private static final String AUTHOR = "科技小王子"; |
|||
// 是否在xml中添加二级缓存配置
|
|||
private static final boolean ENABLE_CACHE = false; |
|||
// 数据库连接配置
|
|||
private static final String DB_URL = "jdbc:mysql://47.119.165.234:3308/modules?useUnicode=true&characterEncoding=utf8&useSSL=false&serverTimezone=GMT%2B8"; |
|||
private static final String DB_DRIVER = "com.mysql.cj.jdbc.Driver"; |
|||
private static final String DB_USERNAME = "modules"; |
|||
private static final String DB_PASSWORD = "8YdLnk7KsPAyDXGA"; |
|||
// 包名
|
|||
private static final String PACKAGE_NAME = "com.gxwebsoft"; |
|||
// 模块名
|
|||
private static final String MODULE_NAME = "pwl"; |
|||
// 需要生成的表
|
|||
private static final String[] TABLE_NAMES = new String[]{ |
|||
"pwl_project" |
|||
}; |
|||
// 需要去除的表前缀
|
|||
private static final String[] TABLE_PREFIX = new String[]{ |
|||
"tb_" |
|||
}; |
|||
// 不需要作为查询参数的字段
|
|||
private static final String[] PARAM_EXCLUDE_FIELDS = new String[]{ |
|||
"tenant_id", |
|||
"create_time", |
|||
"update_time" |
|||
}; |
|||
// 查询参数使用String的类型
|
|||
private static final String[] PARAM_TO_STRING_TYPE = new String[]{ |
|||
"Date", |
|||
"LocalDate", |
|||
"LocalTime", |
|||
"LocalDateTime" |
|||
}; |
|||
// 查询参数使用EQ的类型
|
|||
private static final String[] PARAM_EQ_TYPE = new String[]{ |
|||
"Integer", |
|||
"Boolean", |
|||
"BigDecimal" |
|||
}; |
|||
// 是否添加权限注解
|
|||
private static final boolean AUTH_ANNOTATION = true; |
|||
// 是否添加日志注解
|
|||
private static final boolean LOG_ANNOTATION = true; |
|||
// controller的mapping前缀
|
|||
private static final String CONTROLLER_MAPPING_PREFIX = "/api"; |
|||
// 模板所在位置
|
|||
private static final String TEMPLATES_DIR = "/src/test/java/com/gxwebsoft/generator/templates"; |
|||
|
|||
public static void main(String[] args) { |
|||
// 代码生成器
|
|||
AutoGenerator mpg = new AutoGenerator(); |
|||
|
|||
// 全局配置
|
|||
GlobalConfig gc = new GlobalConfig(); |
|||
gc.setOutputDir(OUTPUT_LOCATION + OUTPUT_DIR); |
|||
gc.setAuthor(AUTHOR); |
|||
gc.setOpen(false); |
|||
gc.setFileOverride(true); |
|||
gc.setEnableCache(ENABLE_CACHE); |
|||
gc.setSwagger2(true); |
|||
gc.setIdType(IdType.AUTO); |
|||
gc.setServiceName("%sService"); |
|||
mpg.setGlobalConfig(gc); |
|||
|
|||
// 数据源配置
|
|||
DataSourceConfig dsc = new DataSourceConfig(); |
|||
dsc.setUrl(DB_URL); |
|||
// dsc.setSchemaName("public");
|
|||
dsc.setDriverName(DB_DRIVER); |
|||
dsc.setUsername(DB_USERNAME); |
|||
dsc.setPassword(DB_PASSWORD); |
|||
mpg.setDataSource(dsc); |
|||
|
|||
// 包配置
|
|||
PackageConfig pc = new PackageConfig(); |
|||
pc.setModuleName(MODULE_NAME); |
|||
pc.setParent(PACKAGE_NAME); |
|||
mpg.setPackageInfo(pc); |
|||
|
|||
// 策略配置
|
|||
StrategyConfig strategy = new StrategyConfig(); |
|||
strategy.setNaming(NamingStrategy.underline_to_camel); |
|||
strategy.setColumnNaming(NamingStrategy.underline_to_camel); |
|||
strategy.setInclude(TABLE_NAMES); |
|||
strategy.setTablePrefix(TABLE_PREFIX); |
|||
strategy.setSuperControllerClass(PACKAGE_NAME + ".common.core.web.BaseController"); |
|||
strategy.setEntityLombokModel(true); |
|||
strategy.setRestControllerStyle(true); |
|||
strategy.setControllerMappingHyphenStyle(true); |
|||
strategy.setLogicDeleteFieldName("deleted"); |
|||
mpg.setStrategy(strategy); |
|||
|
|||
// 模板配置
|
|||
TemplateConfig templateConfig = new TemplateConfig(); |
|||
templateConfig.setController(TEMPLATES_DIR + "/controller.java"); |
|||
templateConfig.setEntity(TEMPLATES_DIR + "/entity.java"); |
|||
templateConfig.setMapper(TEMPLATES_DIR + "/mapper.java"); |
|||
templateConfig.setXml(TEMPLATES_DIR + "/mapper.xml"); |
|||
templateConfig.setService(TEMPLATES_DIR + "/service.java"); |
|||
templateConfig.setServiceImpl(TEMPLATES_DIR + "/serviceImpl.java"); |
|||
mpg.setTemplate(templateConfig); |
|||
mpg.setTemplateEngine(new BeetlTemplateEnginePlus()); |
|||
|
|||
// 自定义模板配置
|
|||
InjectionConfig cfg = new InjectionConfig() { |
|||
@Override |
|||
public void initMap() { |
|||
Map<String, Object> map = new HashMap<>(); |
|||
map.put("packageName", PACKAGE_NAME); |
|||
map.put("paramExcludeFields", PARAM_EXCLUDE_FIELDS); |
|||
map.put("paramToStringType", PARAM_TO_STRING_TYPE); |
|||
map.put("paramEqType", PARAM_EQ_TYPE); |
|||
map.put("authAnnotation", AUTH_ANNOTATION); |
|||
map.put("logAnnotation", LOG_ANNOTATION); |
|||
map.put("controllerMappingPrefix", CONTROLLER_MAPPING_PREFIX); |
|||
this.setMap(map); |
|||
} |
|||
}; |
|||
String templatePath = TEMPLATES_DIR + "/param.java.btl"; |
|||
List<FileOutConfig> focList = new ArrayList<>(); |
|||
focList.add(new FileOutConfig(templatePath) { |
|||
@Override |
|||
public String outputFile(TableInfo tableInfo) { |
|||
return OUTPUT_LOCATION + OUTPUT_DIR + "/" |
|||
+ PACKAGE_NAME.replace(".", "/") |
|||
+ "/" + pc.getModuleName() + "/param/" |
|||
+ tableInfo.getEntityName() + "Param" + StringPool.DOT_JAVA; |
|||
} |
|||
}); |
|||
/** |
|||
* 以下是生成VUE项目代码 |
|||
* 生成文件的路径 /api/shop/goods/index.ts |
|||
*/ |
|||
templatePath = TEMPLATES_DIR + "/index.ts.btl"; |
|||
|
|||
focList.add(new FileOutConfig(templatePath) { |
|||
@Override |
|||
public String outputFile(TableInfo tableInfo) { |
|||
return OUTPUT_LOCATION_VUE + OUTPUT_DIR_VUE |
|||
+ "/api/" + pc.getModuleName() + "/" |
|||
+ tableInfo.getEntityPath() + "/" + "index.ts"; |
|||
} |
|||
}); |
|||
focList.add(new FileOutConfig() { |
|||
@Override |
|||
public String outputFile(TableInfo tableInfo) { |
|||
return OUTPUT_LOCATION_UNIAPP + OUTPUT_DIR_VUE |
|||
+ "/api/" + pc.getModuleName() + "/" |
|||
+ tableInfo.getEntityPath() + "/" + "index.ts"; |
|||
} |
|||
}); |
|||
// 生成TS文件 (/api/shop/goods/model/index.ts)
|
|||
templatePath = TEMPLATES_DIR + "/model.ts.btl"; |
|||
focList.add(new FileOutConfig(templatePath) { |
|||
@Override |
|||
public String outputFile(TableInfo tableInfo) { |
|||
return OUTPUT_LOCATION_VUE + OUTPUT_DIR_VUE |
|||
+ "/api/" + pc.getModuleName() + "/" |
|||
+ tableInfo.getEntityPath() + "/model/" + "index.ts"; |
|||
} |
|||
}); |
|||
focList.add(new FileOutConfig(templatePath) { |
|||
@Override |
|||
public String outputFile(TableInfo tableInfo) { |
|||
return OUTPUT_LOCATION_UNIAPP + OUTPUT_DIR_VUE |
|||
+ "/api/" + pc.getModuleName() + "/" |
|||
+ tableInfo.getEntityPath() + "/model/" + "index.ts"; |
|||
} |
|||
}); |
|||
// 生成Vue文件(/views/shop/goods/index.vue)
|
|||
templatePath = TEMPLATES_DIR + "/index.vue.btl"; |
|||
focList.add(new FileOutConfig(templatePath) { |
|||
@Override |
|||
public String outputFile(TableInfo tableInfo) { |
|||
return OUTPUT_LOCATION_VUE + OUTPUT_DIR_VUE |
|||
+ "/views/" + pc.getModuleName() + "/" |
|||
+ tableInfo.getEntityPath() + "/" + "index.vue"; |
|||
} |
|||
}); |
|||
|
|||
// 生成components文件(/views/shop/goods/components/edit.vue)
|
|||
templatePath = TEMPLATES_DIR + "/components.edit.vue.btl"; |
|||
focList.add(new FileOutConfig(templatePath) { |
|||
@Override |
|||
public String outputFile(TableInfo tableInfo) { |
|||
return OUTPUT_LOCATION_VUE + OUTPUT_DIR_VUE |
|||
+ "/views/" + pc.getModuleName() + "/" |
|||
+ tableInfo.getEntityPath() + "/components/" + tableInfo.getEntityPath() + "Edit.vue"; |
|||
} |
|||
}); |
|||
|
|||
// 生成components文件(/views/shop/goods/components/search.vue)
|
|||
templatePath = TEMPLATES_DIR + "/components.search.vue.btl"; |
|||
focList.add(new FileOutConfig(templatePath) { |
|||
@Override |
|||
public String outputFile(TableInfo tableInfo) { |
|||
return OUTPUT_LOCATION_VUE + OUTPUT_DIR_VUE |
|||
+ "/views/" + pc.getModuleName() + "/" |
|||
+ tableInfo.getEntityPath() + "/components/" + "search.vue"; |
|||
} |
|||
}); |
|||
|
|||
cfg.setFileOutConfigList(focList); |
|||
mpg.setCfg(cfg); |
|||
|
|||
mpg.execute(); |
|||
} |
|||
|
|||
} |
|||
@ -1,251 +0,0 @@ |
|||
package com.gxwebsoft.generator; |
|||
|
|||
import com.baomidou.mybatisplus.annotation.IdType; |
|||
import com.baomidou.mybatisplus.core.toolkit.StringPool; |
|||
import com.baomidou.mybatisplus.generator.AutoGenerator; |
|||
import com.baomidou.mybatisplus.generator.InjectionConfig; |
|||
import com.baomidou.mybatisplus.generator.config.*; |
|||
import com.baomidou.mybatisplus.generator.config.po.TableInfo; |
|||
import com.baomidou.mybatisplus.generator.config.rules.NamingStrategy; |
|||
import com.gxwebsoft.generator.engine.BeetlTemplateEnginePlus; |
|||
|
|||
import java.util.ArrayList; |
|||
import java.util.HashMap; |
|||
import java.util.List; |
|||
import java.util.Map; |
|||
|
|||
/** |
|||
* 核心模块-代码生成工具 |
|||
* |
|||
* @author WebSoft |
|||
* @since 2021-09-05 00:31:14 |
|||
*/ |
|||
public class SysGenerator { |
|||
// 输出位置
|
|||
private static final String OUTPUT_LOCATION = System.getProperty("user.dir"); |
|||
//private static final String OUTPUT_LOCATION = "D:/codegen"; // 不想生成到项目中可以写磁盘路径
|
|||
// 输出目录
|
|||
private static final String OUTPUT_DIR = "/src/main/java"; |
|||
// Vue文件输出位置
|
|||
private static final String OUTPUT_LOCATION_VUE = "/Users/gxwebsoft/VUE/mp"; |
|||
// Vue文件输出目录
|
|||
private static final String OUTPUT_LOCATION_UNIAPP = "/Users/gxwebsoft/VUE/mp"; |
|||
// Vue文件输出目录
|
|||
private static final String OUTPUT_DIR_VUE = "/src"; |
|||
// 作者名称
|
|||
private static final String AUTHOR = "科技小王子"; |
|||
// 是否在xml中添加二级缓存配置
|
|||
private static final boolean ENABLE_CACHE = false; |
|||
// 数据库连接配置
|
|||
private static final String DB_URL = "jdbc:mysql://47.119.165.234:3308/gxwebsoft_core?useUnicode=true&characterEncoding=utf8&useSSL=false&serverTimezone=GMT%2B8"; |
|||
private static final String DB_DRIVER = "com.mysql.cj.jdbc.Driver"; |
|||
private static final String DB_USERNAME = "gxwebsoft_core"; |
|||
private static final String DB_PASSWORD = "jdj7HYEdYHnYEFBy"; |
|||
// 包名
|
|||
private static final String PACKAGE_NAME = "com.gxwebsoft"; |
|||
// 模块名
|
|||
private static final String MODULE_NAME = "common.system"; |
|||
// 需要生成的表
|
|||
private static final String[] TABLE_NAMES = new String[]{ |
|||
// "sys_website",
|
|||
// "sys_website_field",
|
|||
// "sys_domain",
|
|||
// "sys_company",
|
|||
// "sys_user_verify"
|
|||
// "sys_user_role",
|
|||
// "sys_user_oauth"
|
|||
}; |
|||
// 需要去除的表前缀
|
|||
private static final String[] TABLE_PREFIX = new String[]{ |
|||
"sys_", |
|||
"tb_" |
|||
}; |
|||
// 不需要作为查询参数的字段
|
|||
private static final String[] PARAM_EXCLUDE_FIELDS = new String[]{ |
|||
"tenant_id", |
|||
"create_time", |
|||
"update_time" |
|||
}; |
|||
// 查询参数使用String的类型
|
|||
private static final String[] PARAM_TO_STRING_TYPE = new String[]{ |
|||
"Date", |
|||
"LocalDate", |
|||
"LocalTime", |
|||
"LocalDateTime" |
|||
}; |
|||
// 查询参数使用EQ的类型
|
|||
private static final String[] PARAM_EQ_TYPE = new String[]{ |
|||
"Integer", |
|||
"Boolean", |
|||
"BigDecimal" |
|||
}; |
|||
// 是否添加权限注解
|
|||
private static final boolean AUTH_ANNOTATION = true; |
|||
// 是否添加日志注解
|
|||
private static final boolean LOG_ANNOTATION = true; |
|||
// controller的mapping前缀
|
|||
private static final String CONTROLLER_MAPPING_PREFIX = "/api"; |
|||
// 模板所在位置
|
|||
private static final String TEMPLATES_DIR = "/src/test/java/com/gxwebsoft/generator/templates"; |
|||
|
|||
public static void main(String[] args) { |
|||
// 代码生成器
|
|||
AutoGenerator mpg = new AutoGenerator(); |
|||
|
|||
// 全局配置
|
|||
GlobalConfig gc = new GlobalConfig(); |
|||
gc.setOutputDir(OUTPUT_LOCATION + OUTPUT_DIR); |
|||
gc.setAuthor(AUTHOR); |
|||
gc.setOpen(false); |
|||
gc.setFileOverride(true); |
|||
gc.setEnableCache(ENABLE_CACHE); |
|||
gc.setSwagger2(true); |
|||
gc.setIdType(IdType.AUTO); |
|||
gc.setServiceName("%sService"); |
|||
mpg.setGlobalConfig(gc); |
|||
|
|||
// 数据源配置
|
|||
DataSourceConfig dsc = new DataSourceConfig(); |
|||
dsc.setUrl(DB_URL); |
|||
// dsc.setSchemaName("public");
|
|||
dsc.setDriverName(DB_DRIVER); |
|||
dsc.setUsername(DB_USERNAME); |
|||
dsc.setPassword(DB_PASSWORD); |
|||
mpg.setDataSource(dsc); |
|||
|
|||
// 包配置
|
|||
PackageConfig pc = new PackageConfig(); |
|||
pc.setModuleName(MODULE_NAME); |
|||
pc.setParent(PACKAGE_NAME); |
|||
mpg.setPackageInfo(pc); |
|||
|
|||
// 策略配置
|
|||
StrategyConfig strategy = new StrategyConfig(); |
|||
strategy.setNaming(NamingStrategy.underline_to_camel); |
|||
strategy.setColumnNaming(NamingStrategy.underline_to_camel); |
|||
strategy.setInclude(TABLE_NAMES); |
|||
strategy.setTablePrefix(TABLE_PREFIX); |
|||
strategy.setSuperControllerClass(PACKAGE_NAME + ".common.core.web.BaseController"); |
|||
strategy.setEntityLombokModel(true); |
|||
strategy.setRestControllerStyle(true); |
|||
strategy.setControllerMappingHyphenStyle(true); |
|||
strategy.setLogicDeleteFieldName("deleted"); |
|||
mpg.setStrategy(strategy); |
|||
|
|||
// 模板配置
|
|||
TemplateConfig templateConfig = new TemplateConfig(); |
|||
templateConfig.setController(TEMPLATES_DIR + "/controller.java"); |
|||
templateConfig.setEntity(TEMPLATES_DIR + "/entity.java"); |
|||
templateConfig.setMapper(TEMPLATES_DIR + "/mapper.java"); |
|||
templateConfig.setXml(TEMPLATES_DIR + "/mapper.xml"); |
|||
templateConfig.setService(TEMPLATES_DIR + "/service.java"); |
|||
templateConfig.setServiceImpl(TEMPLATES_DIR + "/serviceImpl.java"); |
|||
mpg.setTemplate(templateConfig); |
|||
mpg.setTemplateEngine(new BeetlTemplateEnginePlus()); |
|||
|
|||
// 自定义模板配置
|
|||
InjectionConfig cfg = new InjectionConfig() { |
|||
@Override |
|||
public void initMap() { |
|||
Map<String, Object> map = new HashMap<>(); |
|||
map.put("packageName", PACKAGE_NAME); |
|||
map.put("paramExcludeFields", PARAM_EXCLUDE_FIELDS); |
|||
map.put("paramToStringType", PARAM_TO_STRING_TYPE); |
|||
map.put("paramEqType", PARAM_EQ_TYPE); |
|||
map.put("authAnnotation", AUTH_ANNOTATION); |
|||
map.put("logAnnotation", LOG_ANNOTATION); |
|||
map.put("controllerMappingPrefix", CONTROLLER_MAPPING_PREFIX); |
|||
this.setMap(map); |
|||
} |
|||
}; |
|||
String templatePath = TEMPLATES_DIR + "/param.java.btl"; |
|||
List<FileOutConfig> focList = new ArrayList<>(); |
|||
focList.add(new FileOutConfig(templatePath) { |
|||
@Override |
|||
public String outputFile(TableInfo tableInfo) { |
|||
return OUTPUT_LOCATION + OUTPUT_DIR + "/" |
|||
+ PACKAGE_NAME.replace(".", "/") |
|||
+ "/" + pc.getModuleName() + "/param/" |
|||
+ tableInfo.getEntityName() + "Param" + StringPool.DOT_JAVA; |
|||
} |
|||
}); |
|||
/** |
|||
* 以下是生成VUE项目代码 |
|||
* 生成文件的路径 /api/shop/goods/index.ts |
|||
*/ |
|||
templatePath = TEMPLATES_DIR + "/index.ts.btl"; |
|||
|
|||
focList.add(new FileOutConfig(templatePath) { |
|||
@Override |
|||
public String outputFile(TableInfo tableInfo) { |
|||
return OUTPUT_LOCATION_VUE + OUTPUT_DIR_VUE |
|||
+ "/api/" + pc.getModuleName() + "/" |
|||
+ tableInfo.getEntityPath() + "/" + "index.ts"; |
|||
} |
|||
}); |
|||
focList.add(new FileOutConfig() { |
|||
@Override |
|||
public String outputFile(TableInfo tableInfo) { |
|||
return OUTPUT_LOCATION_UNIAPP + OUTPUT_DIR_VUE |
|||
+ "/api/" + pc.getModuleName() + "/" |
|||
+ tableInfo.getEntityPath() + "/" + "index.ts"; |
|||
} |
|||
}); |
|||
// 生成TS文件 (/api/shop/goods/model/index.ts)
|
|||
templatePath = TEMPLATES_DIR + "/model.ts.btl"; |
|||
focList.add(new FileOutConfig(templatePath) { |
|||
@Override |
|||
public String outputFile(TableInfo tableInfo) { |
|||
return OUTPUT_LOCATION_VUE + OUTPUT_DIR_VUE |
|||
+ "/api/" + pc.getModuleName() + "/" |
|||
+ tableInfo.getEntityPath() + "/model/" + "index.ts"; |
|||
} |
|||
}); |
|||
focList.add(new FileOutConfig(templatePath) { |
|||
@Override |
|||
public String outputFile(TableInfo tableInfo) { |
|||
return OUTPUT_LOCATION_UNIAPP + OUTPUT_DIR_VUE |
|||
+ "/api/" + pc.getModuleName() + "/" |
|||
+ tableInfo.getEntityPath() + "/model/" + "index.ts"; |
|||
} |
|||
}); |
|||
// 生成Vue文件(/views/shop/goods/index.vue)
|
|||
templatePath = TEMPLATES_DIR + "/index.vue.btl"; |
|||
focList.add(new FileOutConfig(templatePath) { |
|||
@Override |
|||
public String outputFile(TableInfo tableInfo) { |
|||
return OUTPUT_LOCATION_VUE + OUTPUT_DIR_VUE |
|||
+ "/views/" + pc.getModuleName() + "/" |
|||
+ tableInfo.getEntityPath() + "/" + "index.vue"; |
|||
} |
|||
}); |
|||
|
|||
// 生成components文件(/views/shop/goods/components/edit.vue)
|
|||
templatePath = TEMPLATES_DIR + "/components.edit.vue.btl"; |
|||
focList.add(new FileOutConfig(templatePath) { |
|||
@Override |
|||
public String outputFile(TableInfo tableInfo) { |
|||
return OUTPUT_LOCATION_VUE + OUTPUT_DIR_VUE |
|||
+ "/views/" + pc.getModuleName() + "/" |
|||
+ tableInfo.getEntityPath() + "/components/" + tableInfo.getEntityPath() + "Edit.vue"; |
|||
} |
|||
}); |
|||
|
|||
// 生成components文件(/views/shop/goods/components/search.vue)
|
|||
templatePath = TEMPLATES_DIR + "/components.search.vue.btl"; |
|||
focList.add(new FileOutConfig(templatePath) { |
|||
@Override |
|||
public String outputFile(TableInfo tableInfo) { |
|||
return OUTPUT_LOCATION_VUE + OUTPUT_DIR_VUE |
|||
+ "/views/" + pc.getModuleName() + "/" |
|||
+ tableInfo.getEntityPath() + "/components/" + "search.vue"; |
|||
} |
|||
}); |
|||
|
|||
cfg.setFileOutConfigList(focList); |
|||
mpg.setCfg(cfg); |
|||
|
|||
mpg.execute(); |
|||
} |
|||
|
|||
} |
|||
@ -1,50 +0,0 @@ |
|||
package com.gxwebsoft.generator.engine; |
|||
|
|||
import com.baomidou.mybatisplus.generator.config.builder.ConfigBuilder; |
|||
import com.baomidou.mybatisplus.generator.engine.AbstractTemplateEngine; |
|||
import org.beetl.core.Configuration; |
|||
import org.beetl.core.GroupTemplate; |
|||
import org.beetl.core.Template; |
|||
import org.beetl.core.resource.FileResourceLoader; |
|||
|
|||
import java.io.FileOutputStream; |
|||
import java.io.IOException; |
|||
import java.util.Map; |
|||
|
|||
/** |
|||
* Beetl模板引擎实现文件输出 |
|||
* |
|||
* @author WebSoft |
|||
* @since 2021-09-05 00:30:28 |
|||
*/ |
|||
public class BeetlTemplateEnginePlus extends AbstractTemplateEngine { |
|||
private GroupTemplate groupTemplate; |
|||
|
|||
@Override |
|||
public AbstractTemplateEngine init(ConfigBuilder configBuilder) { |
|||
super.init(configBuilder); |
|||
try { |
|||
Configuration cfg = Configuration.defaultConfiguration(); |
|||
groupTemplate = new GroupTemplate(new FileResourceLoader(), cfg); |
|||
} catch (IOException e) { |
|||
logger.error(e.getMessage(), e); |
|||
} |
|||
return this; |
|||
} |
|||
|
|||
@Override |
|||
public void writer(Map<String, Object> objectMap, String templatePath, String outputFile) throws Exception { |
|||
Template template = groupTemplate.getTemplate(templatePath); |
|||
try (FileOutputStream fileOutputStream = new FileOutputStream(outputFile)) { |
|||
template.binding(objectMap); |
|||
template.renderTo(fileOutputStream); |
|||
} |
|||
logger.debug("模板:" + templatePath + "; 文件:" + outputFile); |
|||
} |
|||
|
|||
@Override |
|||
public String templateFilePath(String filePath) { |
|||
return filePath + ".btl"; |
|||
} |
|||
|
|||
} |
|||
Loading…
Reference in new issue