先po张效果图
如图
- 输入entity名字
-
选择项目, 具体可以用项目中的模块来做
这个工具目前生成的内容包含:DTO,Controller, Service, Mapper, sqlmap 另外还会根据yaml配置文件在dev环境下生成基本的数据表结构,其中包含了几个常见接口,比如,添加、更新、删除、根据id查询,列表查询这些接口。
使用方法:
1. 独立启动,浏览器访问项目根目录/
2. 添入实体名,属性,并选择对应的项目
3. (可选)在高级选项添加入注释的作者或者表前缀等信息
4. 点击一键生成
关键代码
@Override
public void generate(GenerateConfig cfg) throws Exception {
long startTime = System.currentTimeMillis();
if (StringUtils.isEmpty(cfg.getModelName())) {
ExceptionHolder.throwParaErrorException();
}
Map<String, String> bindMap = getBindMap(cfg);
//生成ResultDto ExceptionHolder etc..
createCommonClass(bindMap, cfg);
//生成model
createModel(bindMap, cfg);
//生成控制器
createController(bindMap, cfg);
//生成service
//createService(bindMap, cfg);
//生成service 实现类
createServiceImpl(bindMap, cfg);
//生成dao
createMapper(bindMap, cfg);
//生成sqlmap
if (cfg.getIsCreateSqlMap()) {
createSqlMapper(bindMap, cfg);
}
//建表
if (cfg.getIsCreateTable()) {
createTable(cfg);
}
double s = (System.currentTimeMillis() - startTime) / 1000D;
System.out.println("Generate use " + s + " seconds");
}
private void createModel(Map bindMap, GenerateConfig cfg) throws FileNotFoundException {
String model = cfg.getModelName();
Template template = groupTemplate.getTemplate("/Model.tpl");
String modelPackage = getModelPackageToFileSeparator(cfg);
String packagePath = cfg.getBasePackage() + "entity" + modelPackage;
String modelPath = CHAR_SEPARATOR + model + ".java";
String path = getCreatePath(cfg, packagePath);
//fields
String fieldDisplay = getFieldDisplay(cfg);
bindMap.put("fieldDefinition", fieldDisplay);
//setter and getter
String[] split = cfg.getFieldSplit();
StringBuilder setget = new StringBuilder();
for (String e : split) {
if (e.length() > 3) {
String[] field = e.trim().replaceAll(LINE_RN, "").split(LINE_SS);
String methodName = StringUtil.toFirstUpCase(field[1]);
setget.append("\t\tpublic void set" + methodName + "(" + field[0] + " " + field[1] + ")");
setget.append("{\n\t\tthis." + field[1] + " = " + field[1] + ";\n\t}\n");
setget.append("\t\tpublic " + field[0] + " get" + methodName + "()");
setget.append("{\n\t\treturn this." + field[1] + ";\n\t}\n");
}
}
bindMap.put("setget", setget);
rend(template, path, modelPath, bindMap);
}
private void createController(Map<String, String> bindMap, GenerateConfig cfg) throws Exception {
try {
Template template = groupTemplate.getTemplate("/Controller.tpl");
String model = cfg.getModelName();
String modelPackage = getModelPackageToFileSeparator(cfg);
String packagePath = cfg.getBasePackage() + "controller" + modelPackage;
String modelPath = CHAR_SEPARATOR + model + "Controller.java";
String path = getCreatePath(cfg, packagePath);
rend(template, path, modelPath, bindMap);
} catch (Exception e) {
throw e;
}
}
/**
* 生成service
*
* @param bindMap
* @param cfg
* @throws IOException
*/
@Deprecated
private void createService(Map<String, String> bindMap, GenerateConfig cfg) throws IOException {
try {
Template template = groupTemplate.getTemplate("/Service.tpl");
String model = cfg.getModelName();
String modelPackage = getModelPackageToFileSeparator(cfg);
String packagePath = cfg.getBasePackage() + "service" + modelPackage;
String modelPath = CHAR_SEPARATOR + model + "Service.java";
String path = getCreatePath(cfg, packagePath);
rend(template, path, modelPath, bindMap);
} catch (Exception e) {
throw e;
}
}
/**
* 生成service实现类
*
* @param bindMap
* @param cfg
* @throws IOException
*/
private void createServiceImpl(Map<String, String> bindMap, GenerateConfig cfg) throws IOException {
try {
Template template = groupTemplate.getTemplate("/ServiceImpl.tpl");
String model = cfg.getModelName();
String modelPackage = getModelPackageToFileSeparator(cfg);
String packagePath = cfg.getBasePackage() + "service" + modelPackage;
String modelPath = CHAR_SEPARATOR + model + "Service.java";
String path = getCreatePath(cfg, packagePath);
rend(template, path, modelPath, bindMap);
} catch (Exception e) {
throw e;
}
}
private void createMapper(Map bindMap, GenerateConfig cfg) throws Exception {
try {
Template template = groupTemplate.getTemplate("/Mapper.tpl");
String model = cfg.getModelName();
String modelPackage = getModelPackageToFileSeparator(cfg);
String packagePath = cfg.getBasePackage() + "mapper" + modelPackage;
String modelPath = CHAR_SEPARATOR + model + "Mapper.java";
String path = getCreatePath(cfg, packagePath);
rend(template, path, modelPath, bindMap);
} catch (Exception e) {
throw e;
}
}
转载于:https://blog.51cto.com/viphyy/2370537