代码生成器，生成controller、service、service-impl、dao、mapper、xml文件、vo、实体
代码生成器，生成controller、service、service-impl、dao、mapper、xml文件、vo、实体
·
正文
这个代码生成器不是用模板，所以如果需要配合公司的工具类等等，则需要修改代码。
该生成器会生成controller、service、service-impl、dao、mapper、xml文件、vo、实体
代码
package com.example.jpacodegenerator;
/**
* @author JunHao Huang
* @since 2023/6/7 13:41
*/
import java.io.File;
import java.io.FileWriter;
import java.io.IOException;
import java.util.Arrays;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
public class JpaCodeGenerator {
private static final String BASE_PACKAGE = "cn.com.infox.attendance";
private static final String ENTITY_PACKAGE = BASE_PACKAGE + ".entity";
private static final String REPOSITORY_PACKAGE = BASE_PACKAGE + ".repository";
private static final String SERVICE_PACKAGE = BASE_PACKAGE + ".service";
private static final String SERVICE_IMPL_PACKAGE = SERVICE_PACKAGE + ".impl";
private static final String CONTROLLER_PACKAGE = BASE_PACKAGE + ".controller";
private static final String VO_PACKAGE = BASE_PACKAGE + ".vo";
private static final String MAPPER_PACKAGE = BASE_PACKAGE + ".mapper";
private static final List<String> RESERVED_KEYWORDS = Arrays.asList("abstract", "assert", "boolean", "break", "byte", "case", "catch", "char", "class", "const", "continue", "default", "do", "double", "else", "enum", "extends", "false", "final", "finally", "float", "for", "goto", "if", "implements", "import", "instanceof", "int", "interface", "long", "native", "new", "null", "package", "private", "protected", "public", "return", "short", "static", "strictfp", "super", "switch", "synchronized", "this", "throw", "throws", "transient", "true", "try", "void", "volatile", "while");
public static void main(String[] args) throws IOException {
String tableName = "我的测试表";
String author = "huangjunhao";
// 实体类名和字段名可以根据需求自行修改
generateEntity("myTest", tableName + "实体类", author, "id", "name");
generateVO("myTest", tableName + "实体类VO", author, "id", "name");
generateMapper("myTest", tableName + "数据库访问层", author);
Map<String, String> cls = new HashMap<>();
cls.put("id", "id");
cls.put("name", "name");
cls.put("is_deleted", "isDeleted");
cls.put("create_user", "createUser");
cls.put("create_time", "createTime");
cls.put("update_user", "updateUser");
cls.put("update_time", "updateTime");
cls.put("status", "status");
generateXML("myTest", cls, "my_test");
generateService("myTest", tableName + "服务接口", author);
generateServiceImpl("myTest", tableName + "服务实现类", author);
generateController("myTest", tableName + "控制器", author, tableName, tableName + "接口");
}
private static void generateXML(String entityName, Map<String, String> cls, String tableName) throws IOException {
StringBuilder sb = new StringBuilder();
sb.append("<?xml version=\"1.0\" encoding=\"UTF-8\"?>\n");
sb.append("<!DOCTYPE mapper PUBLIC \"-//mybatis.org//DTD Mapper 3.0//EN\" \"http://mybatis.org/dtd/mybatis-3-mapper.dtd\">\n");
sb.append("<mapper namespace=\"").append(MAPPER_PACKAGE).append(".").append(capitalize(entityName)).append("Mapper\">\n");
sb.append("\t<!-- 通用查询映射结果 -->\n");
sb.append("\t<resultMap id=\"").append(entityName).append("ResultMap\" type=\"").append(ENTITY_PACKAGE).append(".").append(capitalize(entityName))
.append("\">\n");
cls.forEach((key, value) -> {
sb.append("\t\t<result column=\"").append(key).append("\" property=\"").append(value).append("\"/>\n");
});
sb.append("\t</resultMap>\n\n");
sb.append("\t<select id=\"select").append(capitalize(entityName)).append("Page\" resultMap=\"").append(toCamelCase(entityName))
.append("ResultMap\">\n");
sb.append("\t\tselect * from ").append(tableName).append(" where is_deleted = 0\n");
sb.append("\t</select>\n\n");
sb.append("</mapper>");
writeFile("src/main/resources/mapper/", capitalize(entityName)+"Mapper", sb.toString(), ".xml", capitalize(entityName));
}
/**
* 生成实体类
*/
private static void generateEntity(String entityName, String todo, String author, String... fieldNames) throws IOException {
String className = capitalize(entityName);
StringBuilder sb = new StringBuilder();
sb.append("package ").append(ENTITY_PACKAGE).append(";\n\n");
sb.append("import cn.com.infox.core.mp.base.BaseEntity;\n");
sb.append("import com.baomidou.mybatisplus.annotation.TableName;\n");
sb.append("import lombok.Data;\n");
sb.append("import lombok.EqualsAndHashCode;\n");
sb.append("import com.baomidou.mybatisplus.annotation.TableName;\n");
sb.append("import io.swagger.annotations.ApiModelProperty;\n\n");
sb.append("/**\n");
classAddAuthor(todo, author, sb);
sb.append("@Data\n");
sb.append("@EqualsAndHashCode(callSuper = true)\n");
sb.append("@TableName(\"").append(toSnakeCase(entityName)).append("\")\n");
sb.append("public class ").append(className).append(" extends BaseEntity {\n\n");
// 生成字段
for (String fieldName : fieldNames) {
if (RESERVED_KEYWORDS.contains(fieldName)) {
fieldName = "_" + fieldName;
}
sb.append("\t@ApiModelProperty()\n");
sb.append("\tprivate String ").append(fieldName).append(";\n\n");
}
sb.append("}");
writeFile("src/main/java/" + ENTITY_PACKAGE.replaceAll("\\.", "/"), capitalize(entityName), sb.toString(), ".java", capitalize(entityName));
writeFile("src/main/java/" + VO_PACKAGE.replaceAll("\\.", "/"), capitalize(entityName) + "VO", sb.toString(), ".java", capitalize(entityName));
}
/**
* 生成实体类VO
*/
private static void generateVO(String entityName, String todo, String author, String... fieldNames) throws IOException {
String className = capitalize(entityName);
StringBuilder sb = new StringBuilder();
sb.append("package ").append(ENTITY_PACKAGE).append("vo;\n\n");
sb.append("import ").append(ENTITY_PACKAGE).append(".").append(className).append(";\n");
sb.append("import com.baomidou.mybatisplus.annotation.TableField;\n");
sb.append("import lombok.Data;\n");
sb.append("import lombok.EqualsAndHashCode;\n\n");
sb.append("/**\n");
classAddAuthor(todo, author, sb);
sb.append("@Data\n");
sb.append("@EqualsAndHashCode(callSuper = true)\n");
sb.append("public class ").append(className).append("VO extends ").append(className).append("{\n\n");
// 生成字段
sb.append("\tprivate static final long serialVersionUID = 323961866895098324L;").append(";\n\n");
sb.append("}");
writeFile("src/main/java/" + VO_PACKAGE.replaceAll("\\.", "/"), capitalize(entityName) + "VO", sb.toString(), ".java", capitalize(entityName));
}
private static void classAddAuthor(String todo, String author, StringBuilder sb) {
sb.append("* " + todo + "\n");
sb.append("* \n");
sb.append("* @author " + author + "\n");
sb.append("*/\n");
}
/**
* 生成 Repository
*/
private static void generateRepository(String entityName) throws IOException {
String className = capitalize(entityName) + "Repository";
StringBuilder sb = new StringBuilder();
sb.append("package ").append(REPOSITORY_PACKAGE).append(";\n\n");
sb.append("import org.springframework.data.jpa.repository.JpaRepository;\n");
sb.append("import ").append(ENTITY_PACKAGE).append(".").append(capitalize(entityName)).append(";\n\n");
sb.append("public interface ").append(className).append(" extends JpaRepository<")
.append(capitalize(entityName)).append(", Long> {\n\n").append("}");
writeFile("src/main/java/", REPOSITORY_PACKAGE, sb.toString(), ".java", capitalize(entityName));
}
private static void generateMapper(String entityName, String todo, String author) throws IOException {
String className = capitalize(entityName) + "Mapper";
StringBuilder sb = new StringBuilder();
sb.append("package ").append(MAPPER_PACKAGE).append(";\n\n");
sb.append("import java.util.List;\n\n");
sb.append("import ").append(VO_PACKAGE).append(".").append(capitalize(entityName)).append("VO;\n");
sb.append("import com.baomidou.mybatisplus.core.mapper.BaseMapper;\n");
sb.append("import com.baomidou.mybatisplus.core.metadata.IPage;\n");
sb.append("import ").append(ENTITY_PACKAGE).append(".").append(capitalize(entityName) + ";\n\n");
sbAddAuthor(todo, author, sb);
sb.append("public interface ").append(className).append(" extends BaseMapper<")
.append(capitalize(entityName)).append("> {\n\n");
sb.append("\t/**\n");
sb.append("\t* 自定义分页\n");
sb.append("\t* \n");
sb.append("\t* @param page 分页数据\n");
sb.append("\t* @param ").append(toCamelCase(entityName)).append(" 查询条件\n");
sb.append("\t* @return List<").append(capitalize(entityName)).append("VO>\n");
sb.append("\t*/\n");
sb.append("\tList<").append(capitalize(entityName)).append("VO> select").append(capitalize(entityName)).append("Page")
.append("(IPage page, ").append(capitalize(capitalize(entityName))).append("VO ").append(toCamelCase(entityName))
.append(");\n");
sb.append("}");
writeFile("src/main/java/" + MAPPER_PACKAGE.replaceAll("\\.", "/"), capitalize(entityName) + "Mapper", sb.toString(), ".java", capitalize(entityName));
}
/**
* 生成 Service 接口
*/
private static void generateService(String entityName, String todo, String author) throws IOException {
String className = capitalize(entityName) + "Service";
StringBuilder sb = new StringBuilder();
sb.append("package ").append(SERVICE_PACKAGE).append(";\n\n");
sb.append("import ").append(VO_PACKAGE).append(".").append(capitalize(entityName)).append("VO;\n");
sb.append("import cn.com.infox.core.mp.base.BaseService;\n");
sb.append("import ").append(ENTITY_PACKAGE).append(".").append(capitalize(entityName)).append(";\n");
sb.append("import com.baomidou.mybatisplus.core.metadata.IPage;\n\n");
sbAddAuthor(todo, author, sb);
sb.append("public interface ").append(className).append(" extends BaseService<")
.append(capitalize(entityName)).append("> {\n\n");
sb.append("\t/**\n");
sb.append("\t* 自定义分页\n");
sb.append("\t* \n");
sb.append("\t* @param page 分页参数\n");
sb.append("\t* @param ").append(toCamelCase(entityName)).append(" 查询条件\n");
sb.append("\t* @return ").append("分页数据IPage<").append(capitalize(entityName)).append(">\n");
sb.append("\t*/\n");
sb.append("\tIPage<").append(capitalize(entityName)).append("VO> select" + capitalize(entityName) + "Page")
.append("(IPage<").append(capitalize(entityName)).append("VO> page, ").append(capitalize(entityName))
.append("VO ").append(toCamelCase(entityName)).append(");\n");
sb.append("}");
writeFile("src/main/java/" + SERVICE_PACKAGE.replaceAll("\\.", "/"), capitalize(entityName) + "Service", sb.toString(), ".java", capitalize(entityName));
}
private static void sbAddAuthor(String todo, String author, StringBuilder sb) {
sb.append("/**\n");
sb.append("* " + todo + "\n");
sb.append("*\n");
sb.append("* @author " + author + "\n");
sb.append("*/\n");
}
/**
* 生成 Service 实现类
*/
private static void generateServiceImpl(String entityName, String todo, String author) throws IOException {
String className = capitalize(entityName) + "ServiceImpl";
StringBuilder sb = new StringBuilder();
sb.append("package ").append(SERVICE_PACKAGE).append(".impl;\n\n");
sb.append("import ").append(ENTITY_PACKAGE).append(".").append(capitalize(entityName)).append(";\n");
sb.append("import ").append(MAPPER_PACKAGE).append(".").append(capitalize(entityName)).append("Mapper;\n");
sb.append("import ").append(SERVICE_PACKAGE).append(".").append(capitalize(entityName)).append("Service;\n");
sb.append("import ").append(VO_PACKAGE).append(".").append(capitalize(entityName)).append("VO;\n");
sb.append("import cn.com.infox.core.mp.base.BaseServiceImpl;\n");
sb.append("import com.baomidou.mybatisplus.core.metadata.IPage;\n");
sb.append("import org.springframework.stereotype.Service;\n\n");
sbAddAuthor(todo, author, sb);
sb.append("@Service\n");
sb.append("public class ").append(className).append(" extends BaseServiceImpl<")
.append(capitalize(entityName)).append("Mapper, ").append(capitalize(entityName))
.append(">").append(" implements ").append(capitalize(entityName))
.append("Service {\n\n");
sb.append("\t@Override\n");
sb.append("\tpublic IPage<").append(capitalize(entityName)).append("VO> select")
.append(capitalize(entityName)).append("Page(IPage<").append(capitalize(entityName))
.append("VO> page, ").append(capitalize(entityName)).append("VO ").append(toCamelCase(entityName))
.append(") {\n");
sb.append("\t\treturn page.setRecords(baseMapper.select").append(capitalize(entityName)).append("Page(page, ")
.append(toCamelCase(entityName)).append("));\n");
sb.append("\t}\n");
sb.append("}");
writeFile("src/main/java/" + SERVICE_IMPL_PACKAGE.replaceAll("\\.", "/"), capitalize(entityName) + "ServiceImpl", sb.toString(), ".java", capitalize(entityName));
}
/**
* 生成 Controller
*/
private static void generateController(String entityName, String todo, String author, String apiValue, String apiTags) throws IOException {
String className = capitalize(entityName) + "Controller";
StringBuilder sb = new StringBuilder();
sb.append("package ").append(CONTROLLER_PACKAGE).append(";\n\n");
sb.append("import ").append(ENTITY_PACKAGE).append(".").append(capitalize(entityName)).append(";\n");
sb.append("import ").append(SERVICE_PACKAGE).append(".").append(capitalize(entityName)).append("Service;\n");
sb.append("import ").append(VO_PACKAGE).append(".").append(capitalize(entityName)).append("VO;\n");
sb.append("import cn.com.infox.core.boot.ctrl.BasicController;\n");
sb.append("import cn.com.infox.core.mp.support.Condition;\n");
sb.append("import cn.com.infox.core.mp.support.Query;\n");
sb.append("import cn.com.infox.core.tool.api.R;\n");
sb.append("import cn.com.infox.core.tool.utils.BeanUtil;\n");
sb.append("import cn.com.infox.core.tool.utils.Func;\n");
sb.append("import com.baomidou.mybatisplus.core.metadata.IPage;\n");
sb.append("import com.baomidou.mybatisplus.extension.plugins.pagination.Page;\n");
sb.append("import com.github.xiaoymin.knife4j.annotations.ApiOperationSupport;\n");
sb.append("import io.swagger.annotations.Api;\n");
sb.append("import io.swagger.annotations.ApiOperation;\n");
sb.append("import io.swagger.annotations.ApiParam;\n");
sb.append("import lombok.AllArgsConstructor;\n");
sb.append("import org.springframework.web.bind.annotation.*;\n\n");
sb.append("import javax.validation.Valid;\n");
sb.append("import java.util.stream.Collectors;\n\n");
sbAddAuthor(todo, author, sb);
sb.append("@RestController\n");
sb.append("@AllArgsConstructor\n");
sb.append("@RequestMapping(\"/").append(toCamelCase(entityName)).append("\")\n");
sb.append("@Api(value = \"/").append(apiValue).append("\", tags = \"").append(apiTags).append("\")\n");
sb.append("public class ").append(className).append(" extends BasicController {\n\n");
sb.append("\tprivate ").append(capitalize(entityName)).append("Service ")
.append(toCamelCase(entityName)).append("Service;\n\n");
//详情
controllerDetail(entityName, sb);
//分页
controllerList(entityName, apiValue, sb);
//自定义分页
controllerPage(entityName, apiValue, sb);
//新增
controllerSave(entityName, sb);
//修改
controllerUpdate(entityName, sb);
//新增或修改
controllerSubmit(entityName, sb);
controllerDelete(entityName, sb);
sb.append("}");
writeFile("src/main/java/" + CONTROLLER_PACKAGE.replaceAll("\\.", "/"), capitalize(entityName) + "Controller", sb.toString(), ".java", capitalize(entityName));
}
private static void controllerDelete(String entityName, StringBuilder sb) {
sb.append("\t/**\n");
sb.append("\t* 逻辑删除\n");
sb.append("\t*/\n");
sb.append("\t@PostMapping(\"/remove\")\n");
sb.append("\t@ApiOperationSupport(order = 7)\n");
sb.append("\t@ApiOperation(value = \"逻辑删除\", notes = \"传入ids\")\n");
sb.append("\tpublic R remove(@ApiParam(value = \"主键集合\", required = true) @RequestBody String ids) {\n");
sb.append("\t\treturn R.status(").append(toCamelCase(entityName)).append("Service.deleteLogic(Func.toStrList(ids)));\n");
sb.append("\t}\n\n");
}
private static void controllerSubmit(String entityName, StringBuilder sb) {
sb.append("\t/**\n");
sb.append("\t* 新增或修改\n");
sb.append("\t*/\n");
sb.append("\t@PostMapping(\"/submit\")\n");
sb.append("\t@ApiOperationSupport(order = 6)\n");
sb.append("\t@ApiOperation(value = \"新增或修改\", notes = \"传入" + entityName + "\")\n");
sb.append("\tpublic R submit(@Valid @RequestBody ")
.append(capitalize(entityName)).append(" ").append(toCamelCase(entityName)).append(") {\n");
sb.append("\t\treturn R.status(").append(toCamelCase(entityName)).append("Service.saveOrUpdate(")
.append(toCamelCase(entityName)).append("));\n");
sb.append("\t}\n\n");
}
private static void controllerUpdate(String entityName, StringBuilder sb) {
sb.append("\t/**\n");
sb.append("\t* 修改\n");
sb.append("\t*/\n");
sb.append("\t@PostMapping(\"/update\")\n");
sb.append("\t@ApiOperationSupport(order = 5)\n");
sb.append("\t@ApiOperation(value = \"修改\", notes = \"传入" + entityName + "\")\n");
sb.append("\tpublic R update(@Valid @RequestBody ")
.append(capitalize(entityName)).append(" ").append(toCamelCase(entityName)).append(") {\n");
sb.append("\t\treturn R.status(").append(toCamelCase(entityName)).append("Service.updateById(")
.append(toCamelCase(entityName)).append("));\n");
sb.append("\t}\n\n");
}
private static void controllerSave(String entityName, StringBuilder sb) {
sb.append("\t/**\n");
sb.append("\t* 新增\n");
sb.append("\t*/\n");
sb.append("\t@PostMapping(\"/save\")\n");
sb.append("\t@ApiOperationSupport(order = 4)\n");
sb.append("\t@ApiOperation(value = \"新增\", notes = \"传入" + entityName + "\")\n");
sb.append("\tpublic R save(@Valid @RequestBody ")
.append(capitalize(entityName)).append(" ").append(toCamelCase(entityName)).append(") {\n");
sb.append("\t\treturn R.status(").append(toCamelCase(entityName)).append("Service.save(")
.append(toCamelCase(entityName)).append("));\n");
sb.append("\t}\n\n");
}
private static void controllerPage(String entityName, String apiValue, StringBuilder sb) {
sb.append("\t/**\n");
sb.append("\t* 自定义分页 " + apiValue + "\n");
sb.append("\t*/\n");
sb.append("\t@GetMapping(\"/page\")\n");
sb.append("\t@ApiOperationSupport(order = 3)\n");
sb.append("\t@ApiOperation(value = \"分页\", notes = \"传入" + entityName + "\")\n");
sb.append("\tpublic R<IPage<").append(capitalize(entityName)).append("VO>> page(")
.append(capitalize(entityName)).append("VO ").append(toCamelCase(entityName)).append(", Query query").append("){\n");
sb.append("\t\tIPage<").append(capitalize(entityName)).append("VO> ").append("pages = ").
append(toCamelCase(entityName)).append("Service.select").append(capitalize(entityName))
.append("Page(Condition.getPage(query), ").append(toCamelCase(entityName)).append(");\n");
sb.append("\t\treturn R.data(pages);\n");
sb.append("\t}\n\n");
}
private static void controllerList(String entityName, String apiValue, StringBuilder sb) {
sb.append("\t/**\n");
sb.append("\t* 分页 " + apiValue + "\n");
sb.append("\t*/\n");
sb.append("\t@GetMapping(\"/list\")\n");
sb.append("\t@ApiOperationSupport(order = 2)\n");
sb.append("\t@ApiOperation(value = \"分页\", notes = \"传入" + entityName + "\")\n");
sb.append("\tpublic R<IPage<").append(capitalize(entityName)).append("VO>> list(")
.append(capitalize(entityName)).append(" ").append(toCamelCase(entityName)).append(", Query query").append("){\n");
sb.append("\t\tIPage<").append(capitalize(entityName)).append("> ").append("pages = ")
.append(toCamelCase(entityName)).append("Service.page(Condition.getPage(query), Condition.getQueryWrapper(")
.append(toCamelCase(entityName)).append("));\n");
sb.append("\t\tIPage<").append(capitalize(entityName)).append("VO> page = new Page<>();\n");
sb.append("\t\tBeanUtil.copy(pages, page);\n");
sb.append("\t\tpage.setRecords(pages.getRecords().stream().map(p -> BeanUtil.copy(p, ")
.append(capitalize(entityName)).append("VO.class)).collect(Collectors.toList()));\n");
sb.append("\t\treturn R.data(page);\n");
sb.append("\t}\n\n");
}
private static void controllerDetail(String entityName, StringBuilder sb) {
sb.append("\t/**\n");
sb.append("\t* 详情\n");
sb.append("\t*/\n");
sb.append("\t@GetMapping(\"/detail\")\n");
sb.append("\t@ApiOperationSupport(order = 1)\n");
sb.append("\t@ApiOperation(value = \"详情\", notes = \"传入" + entityName + "\")\n");
sb.append("\tpublic R<").append(capitalize(entityName)).append("VO> detail(")
.append(capitalize(entityName)).append(" ").append(toCamelCase(entityName)).append("){\n");
sb.append("\t\t").append(capitalize(entityName)).append(" detail").append(" = ")
.append(toCamelCase(entityName)).append("Service.getOne(Condition.getQueryWrapper(").append(toCamelCase(entityName))
.append("));\n");
sb.append("\t\treturn R.data(detail == null ? null : BeanUtil.copy(detail, ").append(capitalize(entityName)).append("VO.class));\n");
sb.append("\t}\n\n");
}
private static String capitalize(String str) {
return str.substring(0, 1).toUpperCase() + str.substring(1);
}
private static String toSnakeCase(String str) {
return str.replaceAll("([a-z])([A-Z]+)", "$1_$2").toLowerCase();
}
private static String toCamelCase(String str) {
return str.substring(0, 1).toLowerCase() + str.substring(1);
}
private static void writeFile(String fileUrl, String className, String content, String fileType, String fileName) throws IOException {
File dir = new File(fileUrl);
if (!dir.exists()) {
dir.mkdirs();
}
FileWriter writer = new FileWriter(new File(dir, capitalize(className.substring(className.lastIndexOf(".") + 1)) + fileType));
writer.write(content);
writer.flush();
writer.close();
}
}
华为开发者空间，是为全球开发者打造的专属开发空间，汇聚了华为优质开发资源及工具，致力于让每一位开发者拥有一台云主机，基于华为根生态开发、创新。
更多推荐
0
0- 0
已为社区贡献1条内容
所有评论(0)