自定义代码生成器

已于 2023-06-16 13:09:33 修改
阅读量196 收藏
点赞数
文章标签： java spring boot mybatis
于 2023-06-13 13:30:19 首次发布
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/qq_43975645/article/details/131186314
版权

自定义代码生成器

前提

  1. 这是在开发自己写的数据采集项目中，根据实际需要生成期望模板的生成器，支持多表，支持任意模型的插拔（如：只生成Controller和Service，其他模型不要）
  2. 基于SpringBoot项目，配置文件为yml类型
  3. 使用MySQL数据源
  4. 使用Mybatis-Plus+Spring JPA，所以会有JPA的注解存在

结果展示

在这里插入图片描述
在这里插入图片描述
在这里插入图片描述

流程

文件结构
在这里插入图片描述

1. 项目上配置好数据源信息

在这里插入图片描述

spring:
  datasource:
	driver-class-name: com.mysql.cj.jdbc.Driver
	url: jdbc:mysql://localhost:3306/自己的数据库?useUnicode=true&useSSL=false&characterEncoding=utf8&serverTimezone=Asia/Shanghai&rewriteBatchedStatements=true&generateSimpleParameterMetadata=true&nullCatalogMeansCurrent=true
	username: 数据库账号
	password: 数据库密码

2. 基本信息properites中做好配置

[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-jjsARYCE-1686633840240)(:/7a455cb600b141b0ac69aac0e9d7f8ed)]

# 表名，逗号分隔，大小写不敏感
tableName=SINA_ASHARES_QUOTES
# 工程名
projectDirectory=betrice-data-acquisition
# 工程下生成的代码的父目录，程序会拼接在path之后
secondaryModule=finance
# 工程下生成的位置
path=src\\main\\java\\com\\devilvan\\betrice\\
# 作者
author=Evad.Wu
# 父目录下的单位目录
packageName=com.devilvan.betrice

3. BetriceCodeGenerator中选择好生成的模型

在这里插入图片描述

4. 在BetriceCodeGenerator中运行

在这里插入图片描述

代码

BetriceCodeGenerator

public class BetriceCodeGenerator {
	private static final String SUFFIX_JAVA = ".java";
	private static final String SUFFIX_XML = ".xml";
	private static Map<String, Object> basicMap;

	static {
		basicMap = loadBasic();
	}

	public static void main(String[] args) {
		List<String> tableList = Optional.of(Arrays.asList(String.valueOf(basicMap.get("tableName")).toUpperCase(Locale.ROOT).split(",")))
				.filter(f -> f.size() > 0)
				.orElseThrow(() -> new BetriceException("待生成的数据库表名不能为空！"));
		BetriceDataSourceTbDto betriceDataSourceTbDto = assemDataSourceTbDto(tableList);
		List<BetriceDbDto> dbDtoList = printTableStructure(betriceDataSourceTbDto);
		generate(dbDtoList);
	}

	/**
	 * 生成模型文件开关
	 * @param dbDtoList 待生成代码的数据表信息
	 */
	private static void generate(List<BetriceDbDto> dbDtoList) {
		for (BetriceDbDto dbDto : dbDtoList) {
			generateVo(dbDto);
			generatePo(dbDto);
			generateDto(dbDto);
			generateController(dbDto);
			generateService(dbDto);
			generateServiceImpl(dbDto);
			generateRepository(dbDto);
			generateMapper(dbDto);
			generateMapperXml(dbDto);
		}
	}

	/**
	 * 初始化方法，加载basic.properties 获取如表名、包名、作者等基础信息
	 * @return 解析properties的key-value后封装的Map
	 */
	private static Map<String, Object> loadBasic() {
		HashMap<String, Object> map = new HashMap<>();
		Properties pro = new Properties();
		try {
			pro.load(new FileReader("betrice-common\\betrice-common-codegen\\src\\main\\resources\\basic.properties"));
		} catch (IOException e) {
			e.printStackTrace();
		}
		String tableName = String.valueOf(pro.get("tableName"));
		String projectDirectory = String.valueOf(pro.get("projectDirectory"));
		String secondaryModule = String.valueOf(pro.get("secondaryModule"));
		String path = pro.get("path") + secondaryModule + "\\";
		String author = String.valueOf(pro.get("author"));
		String packageName = pro.get("packageName") + "." + secondaryModule;
		map.put("tableName", tableName);
		map.put("projectDirectory", projectDirectory);
		map.put("secondaryModule", secondaryModule);
		map.put("path", path);
		map.put("author", author);
		map.put("packageName", packageName);
		map.put("date", new SimpleDateFormat("yyyy-MM-dd").format(new Date()));
		return map;
	}

	/**
	 * springboot项目，获取application.yml中数据源信息
	 * @param tableList 需要生成的表名
	 * @return 数据源信息
	 */
	private static BetriceDataSourceTbDto assemDataSourceTbDto(List<String> tableList) {
		ClassPathResource resource = new ClassPathResource("application-parent.yml");
		YamlPropertiesFactoryBean ymlFactory = new YamlPropertiesFactoryBean();
		ymlFactory.setResources(resource);
		Properties pro = Optional.ofNullable(ymlFactory.getObject()).orElseThrow(() -> new BetriceException("获取配置失败！"));
		String driver = String.valueOf(pro.get("spring.datasource.driver-class-name"));
		String url = String.valueOf(pro.get("spring.datasource.url"));
		String uname = String.valueOf(pro.get("spring.datasource.username"));
		String pwd = String.valueOf(pro.get("spring.datasource.password"));
		BetriceDataSourceTbDto tbDto = new BetriceDataSourceTbDto(driver, url, uname, pwd);
		tbDto.setTableList(tableList);
		return tbDto;
	}

	/**
	 * 根据数据源信息获取到对应数据表的字段
	 * @param tbDto 数据源信息
	 * @return 数据表、字段信息
	 */
	private static List<BetriceDbDto> printTableStructure(BetriceDataSourceTbDto tbDto) {
		List<BetriceDbDto> dbDtoList = new ArrayList<>();
		// 每张表默认的审计字段，在生成的Model中只显示业务字段，而通过继承一个BasePo管理这些审计字段
		List<String> auditList = List.of("ID", "CREATOR", "CREATE_TIME", "MODIFIER", "MODIFY_TIME", "LOGIC_DEL", "REC_VER");
		try {
			// JDBC的过程了，加载MySQL驱动-连接-执行-结果集
			Class.forName(tbDto.getDriver());
			Connection connection = DriverManager.getConnection(tbDto.getUrl(), tbDto.getUname(), tbDto.getPwd());
			DatabaseMetaData metaData = connection.getMetaData();
			// 获取所有表
			List<String> tableList = tbDto.getTableList();
			for (String table : tableList) {
				// 根据表名获取数据表的元数据
				ResultSet tableResultSet = metaData.getTables(null, null, table, new String[]{"TABLE"});
				while (tableResultSet.next()) {
					String tableName = tableResultSet.getString("TABLE_NAME").toUpperCase(Locale.ROOT);
					// 将表名转化成类名
					String className = tableName2className(table);
					System.out.println("tableName: " + tableName);
					// 获取数据表的描述-comment
					String comment = tableResultSet.getString("REMARKS");
					if (StrUtil.isNotEmpty(comment)) {
						throw new BetriceException("表描述不能为空！");
					}
					System.out.println("comment: " + comment);
					BetriceDbDto dbDto = new BetriceDbDto(table, className, comment);
					// 获取表字段结构
					ResultSet columnResultSet = metaData.getColumns(null, "%", tableName, "%");
					List<BetriceCgField> betriceCgFieldList = new ArrayList<>();
					// 遍历表中每个字段
					while (columnResultSet.next()) {
						String columnName = columnResultSet.getString("COLUMN_NAME");
						if (auditList.contains(columnName)) {
							continue;
						}
						String fieldName = columnName2Field(columnName);
						String columnType = columnResultSet.getString("TYPE_NAME");
						String javaType = columnType2JavaType(columnType);
						int datasize = columnResultSet.getInt("COLUMN_SIZE");
						int digits = columnResultSet.getInt("DECIMAL_DIGITS");
						// false-不可为空
						boolean nullable = columnResultSet.getInt("NULLABLE") == 1;
						String remarks = columnResultSet.getString("REMARKS");
						BetriceCgField betriceCgField = new BetriceCgField(columnName, fieldName, columnType, javaType, datasize, digits, nullable, remarks);
						betriceCgFieldList.add(betriceCgField);
					}
					dbDto.setFieldList(betriceCgFieldList);
					dbDtoList.add(dbDto);
					System.out.println("=================================");
				}
			}
		} catch (ClassNotFoundException | SQLException e) {
			e.printStackTrace();
		}
		return dbDtoList;
	}

	/**
	 * 表名转类名
	 * @param table 表名
	 * @return 类名
	 */
	private static String tableName2className(String table) {
		StringBuilder sb = new StringBuilder();
		String[] split = table.toLowerCase(Locale.ROOT).split("_");
		for (String s : split) {
			if (s.length() == 1) {
				sb.append(s.toUpperCase(Locale.ROOT));
			} else {
				sb.append(s.substring(0, 1).toUpperCase(Locale.ROOT)).append(s.substring(1));
			}
		}
		return sb.toString();
	}

	/**
	 * DB字段类型转Java类型
	 * @param columnType 字段类型
	 * @return Java类型
	 */
	private static String columnType2JavaType(String columnType) {
		if ("VARCHAR".equals(columnType) || "CHAR".equals(columnType) || "TINYTEXT".equals(columnType) ||
				"MEDIUNTEXT".equals(columnType) || "TEXT".equals(columnType) || "LONGTEXT".equals(columnType)) {
			return "String";
		} else if ("DATE".equals(columnType) || "TIME".equals(columnType) || "DATETIME".equals(columnType) ||
				"TIMESTAMP".equals(columnType) || "YEAR".equals(columnType)) {
			return "Date";
		} else if ("TINYINT".equals(columnType) || "SMALLINT".equals(columnType) || "MEDIUMINT".equals(columnType) ||
				"INT".equals(columnType)) {
			return "Integer";
		} else if ("BIGINT".equals(columnType)) {
			return "Long";
		} else if ("FLOAT".equals(columnType)) {
			return "Float";
		} else if ("DOUBLE".equals(columnType)) {
			return "Double";
		} else if ("DECIMAL".equals(columnType)) {
			return "BigDecimal";
		}
		return "String";
	}

	/**
	 * DB字段名转Java属性名
	 * @param columnName DB字段名
	 * @return Java属性名
	 */
	private static String columnName2Field(String columnName) {
		StringBuilder sb = new StringBuilder();
		String[] split = columnName.toLowerCase(Locale.ROOT).split("_");
		for (int i = 0; i < split.length; i++) {
			if (i == 0) {
				sb.append(split[0]);
			} else if (split[i].length() == 1) {
				sb.append(split[i].toUpperCase(Locale.ROOT));
			} else {
				sb.append(split[i].substring(0, 1).toUpperCase(Locale.ROOT)).append(split[i].substring(1));
			}
		}
		return sb.toString();
	}

	/**
	 * 根据freemarker模板，生成文件到对应目录下
	 * @param templateName 模板名称
	 * @param file 待生成的文件名
	 * @param basicMap 待生成的文件中的基本信息
	 */
	private static void generateFileByTemplate(String templateName, File file, Map<String, Object> basicMap) {
		try {
			Template template = FreemarkerTemplateUtil.getTemplate(templateName);
			FileOutputStream fos = new FileOutputStream(file);
			Writer out = new BufferedWriter(new OutputStreamWriter(fos, StandardCharsets.UTF_8), 10240);
			template.process(basicMap, out);
			FreemarkerTemplateUtil.clearCache();
		} catch (IOException | TemplateException e) {
			e.printStackTrace();
		}
	}

	private static void generateVo(BetriceDbDto dbDto) {
		String parentPath = basicMap.get("projectDirectory") + "\\" + basicMap.get("path") + "\\vo\\";
		String templateName = "vo.ftl";
		File mapperFile = new File(parentPath);
		if (!mapperFile.exists()) {
			mapperFile.mkdirs();
		}
		File path = new File(parentPath + dbDto.getClassName() + "Vo" + SUFFIX_JAVA);
		Map<String, Object> map = new HashMap<>(basicMap);
		map.put("dbDto", dbDto);
		generateFileByTemplate(templateName, path, map);
	}

	private static void generatePo(BetriceDbDto dbDto) {
		String parentPath = basicMap.get("projectDirectory") + "\\" + basicMap.get("path") + "\\po\\";
		String templateName = "po.ftl";
		File mapperFile = new File(parentPath);
		if (!mapperFile.exists()) {
			mapperFile.mkdirs();
		}
		File path = new File(parentPath + dbDto.getClassName() + "Po" + SUFFIX_JAVA);
		Map<String, Object> map = new HashMap<>(basicMap);
		map.put("dbDto", dbDto);
		generateFileByTemplate(templateName, path, map);
	}

	private static void generateController(BetriceDbDto dbDto) {
		String parentPath = basicMap.get("projectDirectory") + "\\" + basicMap.get("path") + "\\controller\\";
		String templateName = "controller.ftl";
		File mapperFile = new File(parentPath);
		if (!mapperFile.exists()) {
			mapperFile.mkdirs();
		}
		File path = new File(parentPath + dbDto.getClassName() + "Controller" + SUFFIX_JAVA);
		Map<String, Object> map = new HashMap<>(basicMap);
		map.put("dbDto", dbDto);
		generateFileByTemplate(templateName, path, map);
	}

	private static void generateDto(BetriceDbDto dbDto) {
		String parentPath = basicMap.get("projectDirectory") + "\\" + basicMap.get("path") + "\\dto\\";
		String templateName = "dto.ftl";
		File mapperFile = new File(parentPath);
		if (!mapperFile.exists()) {
			mapperFile.mkdirs();
		}
		File path = new File(parentPath + dbDto.getClassName() + "Dto" + SUFFIX_JAVA);
		Map<String, Object> map = new HashMap<>(basicMap);
		map.put("dbDto", dbDto);
		generateFileByTemplate(templateName, path, map);
	}

	private static void generateService(BetriceDbDto dbDto) {
		String parentPath = basicMap.get("projectDirectory") + "\\" + basicMap.get("path") + "\\service\\";
		String templateName = "service.ftl";
		File mapperFile = new File(parentPath);
		if (!mapperFile.exists()) {
			mapperFile.mkdirs();
		}
		File path = new File(parentPath + dbDto.getClassName() + "Service" + SUFFIX_JAVA);
		Map<String, Object> map = new HashMap<>(basicMap);
		map.put("dbDto", dbDto);
		generateFileByTemplate(templateName, path, map);
	}

	private static void generateServiceImpl(BetriceDbDto dbDto) {
		String parentpPath = basicMap.get("projectDirectory") + "\\" + basicMap.get("path") + "\\service\\impl\\";
		String templateName = "serviceImpl.ftl";
		File mapperFile = new File(parentpPath);
		if (!mapperFile.exists()) {
			mapperFile.mkdirs();
		}
		File path = new File(parentpPath + dbDto.getClassName() + "ServiceImpl" + SUFFIX_JAVA);
		Map<String, Object> map = new HashMap<>(basicMap);
		map.put("dbDto", dbDto);
		generateFileByTemplate(templateName, path, map);
	}

	private static void generateMapper(BetriceDbDto dbDto) {
		String parentPath = basicMap.get("projectDirectory") + "\\" + basicMap.get("path") + "\\mapper\\";
		String templateName = "mapper.ftl";
		File mapperFile = new File(parentPath);
		if (!mapperFile.exists()) {
			mapperFile.mkdirs();
		}
		File path = new File(parentPath + dbDto.getClassName() + "Mapper" + SUFFIX_JAVA);
		Map<String, Object> map = new HashMap<>(basicMap);
		map.put("dbDto", dbDto);
		generateFileByTemplate(templateName, path, map);
	}

	private static void generateMapperXml(BetriceDbDto dbDto) {
		String parentPath = basicMap.get("projectDirectory") + "\\src\\main\\resources\\mapper\\";
		String templateName = "mapperXml.ftl";
		File mapperFile = new File(parentPath);
		if (!mapperFile.exists()) {
			mapperFile.mkdirs();
		}
		File path = new File(parentPath + dbDto.getClassName() + "Mapper" + SUFFIX_XML);
		Map<String, Object> map = new HashMap<>(basicMap);
		map.put("dbDto", dbDto);
		generateFileByTemplate(templateName, path, map);
	}

	private static void generateRepository(BetriceDbDto dbDto) {
		String parentPath = basicMap.get("projectDirectory") + "\\" + basicMap.get("path") + "\\repository\\";
		String templateName = "repository.ftl";
		File mapperFile = new File(parentPath);
		if (!mapperFile.exists()) {
			mapperFile.mkdirs();
		}
		File path = new File(parentPath + dbDto.getClassName() + "Repository" + SUFFIX_JAVA);
		Map<String, Object> map = new HashMap<>(basicMap);
		map.put("dbDto", dbDto);
		generateFileByTemplate(templateName, path, map);
	}
}

BetriceDbDto

@Data
@NoArgsConstructor
@AllArgsConstructor
@ApiModel(value = "数据库表信息 数据传输对象")
public class BetriceDbDto implements Serializable {
	@Serial
	private static final long serialVersionUID = -179434144799166939L;
	@ApiModelProperty(value = "表名")
	private String tableName;
	@ApiModelProperty(value = "类名")
	private String className;
	@ApiModelProperty(value = "表描述")
	private String comment;
	@ApiModelProperty(value = "字段值对象")
	private List<BetriceCgField> fieldList;

	public BetriceDbDto(String tableName, String className, String comment) {
		this.tableName = tableName;
		this.className = className;
		this.comment = comment;
	}
}

BetriceDataSourceTbDto

@Data
@NoArgsConstructor
@AllArgsConstructor
@EqualsAndHashCode(callSuper = true)
@ApiModel(value = "数据源信息 数据传输对象")
public class BetriceDataSourceTbDto extends BetriceDataSourceDto{
	@ApiModelProperty(value = "需要生成的表名")
	private List<String> tableList;

	public BetriceDataSourceTbDto(String driver, String url, String uname, String pwd) {
		super(driver, url, uname, pwd);
	}
}

BetriceDataSourceDto

@Data
@NoArgsConstructor
@AllArgsConstructor
@ApiModel(value = "数据源信息 数据传输对象")
public class BetriceDataSourceDto{
	@ApiModelProperty(value = "数据库驱动")
	private String driver;
	@ApiModelProperty(value = "数据库URL")
	private String url;
	@ApiModelProperty(value = "账号")
	private String uname;
	@ApiModelProperty(value = "密码")
	private String pwd;
}

BetriceCgField

@Data
@NoArgsConstructor
@AllArgsConstructor
@ApiModel(value = "表字段及转换相关字段")
public class BetriceCgField {
	@ApiModelProperty(value = "字段名称")
	private String columnName;
	@ApiModelProperty(value = "字段名称转换的属性名称")
	private String fieldName;
	@ApiModelProperty(value = "字段类型")
	private String columnType;
	@ApiModelProperty(value = "Java类型")
	private String javaType;
	@ApiModelProperty(value = "字段长度")
	private Integer dataSize;
	@ApiModelProperty(value = "精度")
	private Integer digits;
	@ApiModelProperty(value = "是否为空")
	private Boolean nullable;
	@ApiModelProperty(value = "字段描述")
	private String remark;
}

FreemarkerTemplateUtil-Freemarker生成模板工具类

public class FreemarkerTemplateUtil {
	private static final Configuration CONFIGURATION = new Configuration(Configuration.VERSION_2_3_31);

	static {
		//这里比较重要，用来指定加载模板所在的路径
		CONFIGURATION.setTemplateLoader(new ClassTemplateLoader(FreemarkerTemplateUtil.class,
				"/static/ftl"));
		CONFIGURATION.setDefaultEncoding("UTF-8");
		CONFIGURATION.setTemplateExceptionHandler(TemplateExceptionHandler.RETHROW_HANDLER);
		CONFIGURATION.setCacheStorage(NullCacheStorage.INSTANCE);
	}

	public static Template getTemplate(String templateName) throws IOException {
		return CONFIGURATION.getTemplate(templateName);
	}

	public static void clearCache() {
		CONFIGURATION.clearTemplateCache();
	}
}

模型模板ftl文件

controller.ftl

package ${packageName}.controller;

import ${packageName}.po.${dbDto.className}Po;
import ${packageName}.service.${dbDto.className}Service;
import com.devilvan.betrice.common.BetriceResultVo;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiOperation;
import org.springframework.web.bind.annotation.*;

import javax.annotation.Resource;

/**
 * @author ${author}
 * @Description ${dbDto.comment} 控制器
 * @date ${date}
 */
@RestController
@RequestMapping(value = "${dbDto.className?uncap_first}Controller")
@Api(tags = "${dbDto.comment} 控制器")
public class ${dbDto.className}Controller {
    @Resource(name = "${dbDto.className?uncap_first}ServiceImpl")
    private ${dbDto.className}Service ${dbDto.className?uncap_first}Service;

    /**
     * 总请求控制器
     *
     * @return 形式返回
     */
    @ApiOperation(value = "总请求控制器")
    @RequestMapping(value = "do${dbDto.className}")
    public BetriceResultVo<${dbDto.className}Po> do${dbDto.className}() {
        BetriceResultVo<${dbDto.className}Po> betriceResultVo = new BetriceResultVo<>();
        return betriceResultVo;
    }

    /**
     * 采集报文控制器
     *
     * @return 形式返回
     */
    @ApiOperation(value = "采集报文控制器")
    @RequestMapping(value = "do${dbDto.className}Crawler")
    public BetriceResultVo<${dbDto.className}Po> do${dbDto.className}Crawler() {
        BetriceResultVo<${dbDto.className}Po> betriceResultVo = new BetriceResultVo<>();
        return betriceResultVo;
    }

    /**
     * 解析报文控制器
     *
     * @param request 报文
     * @return 形式返回
     */
    @ApiOperation(value = "解析报文控制器")
    @RequestMapping(value = "do${dbDto.className}Request")
    public BetriceResultVo<${dbDto.className}Po> do${dbDto.className}Request(@RequestBody String request) {
        BetriceResultVo<${dbDto.className}Po> betriceResultVo = new BetriceResultVo<>();
        return betriceResultVo;
    }
}

dto.ftl

package ${packageName}.dto;

import io.swagger.annotations.ApiModel;
import io.swagger.annotations.ApiModelProperty;
import lombok.*;

import java.util.Date;

/**
* @author ${author}
* @Description ${dbDto.comment} 数据传输对象
* @date ${date}
*/
@Data
@NoArgsConstructor
@AllArgsConstructor
@ApiModel(value = "${dbDto.comment} 数据传输对象")
public class ${dbDto.className}Dto {
<#if dbDto.fieldList ??>
    <#list dbDto.fieldList as field>
    @ApiModelProperty(value = "${field.remark ! ''}")
    private <#if field.columnType == 'BIT'>Boolean<#else>${field.javaType}</#if> ${field.fieldName};
    </#list>
</#if>
}

mapper.ftl

package ${packageName}.mapper;


import com.baomidou.mybatisplus.core.mapper.BaseMapper;
import ${packageName}.po.${dbDto.className}Po;
import org.apache.ibatis.annotations.Mapper;

/**
* @author ${author}
* @Description ${dbDto.comment} 持久层
* @date ${date}
*/
@Mapper
public interface ${dbDto.className}Mapper extends BaseMapper<${dbDto.className}Po> {

}

mapperXml.ftl

<?xml version="1.0" encoding="UTF-8" ?>
<!DOCTYPE mapper PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN" "http://mybatis.org/dtd/mybatis-3-mapper.dtd" >
<mapper namespace="${packageName}.mapper.${dbDto.className}Mapper">
</mapper>

po.ftl

package ${packageName}.po;

import com.baomidou.mybatisplus.annotation.TableField;
import com.baomidou.mybatisplus.annotation.TableName;
import com.devilvan.betrice.common.BasePo;
import io.swagger.annotations.ApiModel;
import io.swagger.annotations.ApiModelProperty;
import lombok.Getter;
import lombok.NoArgsConstructor;
import lombok.Setter;
import lombok.ToString;
import lombok.experimental.Accessors;
import org.hibernate.annotations.DynamicInsert;
import org.springframework.data.jpa.domain.support.AuditingEntityListener;

import javax.persistence.*;
import java.io.Serial;
import java.io.Serializable;
import java.util.Date;

/**
* @author ${author}
* @Description ${dbDto.comment} 映射
* @date ${date}
*/
@Setter
@Getter
@ToString
@NoArgsConstructor
@Accessors(chain = true)
@TableName(value = "${dbDto.tableName}")

@Entity
@Table(name = "${dbDto.tableName}")
@DynamicInsert
@EntityListeners(AuditingEntityListener.class)
@ApiModel(value = "${dbDto.comment} 映射")
public class ${dbDto.className}Po extends BasePo implements Serializable {
<#if dbDto.fieldList ??>
    <#list dbDto.fieldList as field>
    @ApiModelProperty(value = "${field.remark ! ''}")
    @TableField(value = "${field.columnName}")
    @Column(name = "${field.columnName}", columnDefinition = "<#if field.javaType != 'Date'>${field.columnType}(${field.dataSize}) <#else>${field.columnType} </#if><#if field.nullable = false>NOT NULL </#if>COMMENT '${field.remark}'")
    private <#if field.columnType == 'BIT'>Boolean<#else>${field.javaType}</#if> ${field.fieldName};
    </#list>
</#if>
}

repository.ftl

package ${packageName}.repository;


import ${packageName}.po.${dbDto.className}Po;
import org.springframework.data.jpa.repository.JpaRepository;
import org.springframework.stereotype.Repository;

import java.util.List;

/**
* @author ${author}
* @Description ${dbDto.comment} 持久层
* @date ${date}
*/
@Repository
public interface ${dbDto.className}Repository extends JpaRepository<${dbDto.className}Po, Long> {
}

service.ftl

package ${packageName}.service;

import ${packageName}.dto.${dbDto.className}Dto;
import ${packageName}.po.${dbDto.className}Po;

import java.util.List;

/**
* @author ${author}
* @Description ${dbDto.comment} 业务逻辑接口
* @date ${date}
*/
public interface ${dbDto.className}Service {
    /**
    * 保存
    *
    * @param ${dbDto.className?uncap_first}Dto 控制层处理后的数据传输对象
    * @return 保存回调对象
    */
    ${dbDto.className}Po saveUpdate(${dbDto.className}Dto ${dbDto.className?uncap_first}Dto);
    /**
    * 批量保存
    *
    * @param list 控制层处理后的数据传输对象集合
    * @return 保存回调对象
    */
    List<${dbDto.className}Po> saveUpdateBatch(List<${dbDto.className}Dto> list);
}

serviceImpl.ftl

package ${packageName}.service.impl;

import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import ${packageName}.dto.${dbDto.className}Dto;
import ${packageName}.po.${dbDto.className}Po;
import ${packageName}.service.${dbDto.className}Service;
import ${packageName}.mapper.${dbDto.className}Mapper;
import ${packageName}.repository.${dbDto.className}Repository;
import org.springframework.stereotype.Service;
import com.devilvan.betrice.util.CopyUtil;

import javax.annotation.Resource;
import java.util.List;

/**
* @author ${author}
* @Description ${dbDto.comment} 业务逻辑类
* @date ${date}
*/
@Service
public class ${dbDto.className}ServiceImpl extends ServiceImpl<${dbDto.className}Mapper, ${dbDto.className}Po> implements ${dbDto.className}Service {
    @Resource(name = "${dbDto.className?uncap_first}Repository")
    private ${dbDto.className}Repository ${dbDto.className?uncap_first}Repository;

    @Override
    public ${dbDto.className}Po saveUpdate(${dbDto.className}Dto ${dbDto.className?uncap_first}Dto) {
        ${dbDto.className}Po ${dbDto.className?uncap_first}Po = new ${dbDto.className}Po();
        CopyUtil.fullCopy(${dbDto.className?uncap_first}Dto, ${dbDto.className?uncap_first}Po);
        return ${dbDto.className?uncap_first}Repository.saveAndFlush(${dbDto.className?uncap_first}Po);
    }

    @Override
    public List<${dbDto.className}Po> saveUpdateBatch(List<${dbDto.className}Dto> list) {
        List<${dbDto.className}Po> ${dbDto.className?uncap_first}PoList = CopyUtil.fullCopyCollections(list, ${dbDto.className}Po.class);
        return ${dbDto.className?uncap_first}Repository.saveAllAndFlush(${dbDto.className?uncap_first}PoList);
    }
}

vo.ftl

package ${packageName}.vo;

import com.alibaba.fastjson.annotation.JSONField;
import io.swagger.annotations.ApiModel;
import io.swagger.annotations.ApiModelProperty;
import lombok.AllArgsConstructor;
import lombok.Data;
import lombok.NoArgsConstructor;

import java.util.Date;

/**
* @author ${author}
* @Description ${dbDto.comment} 值对象
* @date ${date}
*/
@Data
@NoArgsConstructor
@AllArgsConstructor
@ApiModel(value = "${dbDto.comment} 值对象")
public class ${dbDto.className}Vo {
<#if dbDto.fieldList ??>
    <#list dbDto.fieldList as field>
    @ApiModelProperty(value = "${field.remark ! ''}")
    @JSONField(alternateNames = {"${field.fieldName}"})
    private <#if field.columnType == 'BIT'>Boolean<#else>${field.javaType}</#if> ${field.fieldName};
    </#list>
</#if>
}
MyBatis-Plus 自定义代码生成器
奇妙的Bug之旅
01-07 591
MyBatis-Plus 是一个开源的 MyBatis 增强工具，为我们提供了更简洁、高效的开发方式。其中的代码生成器是其重要的功能之一，可以帮助开发者快速生成 MyBatis 的实体类、Mapper 接口以及对应的 XML 文件。然而，有时我们需要根据自己的需求对代码生成器进行定制化，本文将介绍如何自定义 MyBatis-Plus 的代码生成器。接下来，我们创建一个自定义的代码生成器类，继承自 MyBatis-Plus 的 AutoGenerator 类。// TODO: 添加自定义的代码生成逻辑。
Java培训：自定义代码生成器
Blue92120的博客
11-15 561
需求：遍历学生集合，显示集合总条数，id不要逗号隔开，显示学生的生日(只显示年月日)，钱包显示整数并显示单位`元`，用户姓名做脱敏处理(如果是两个字第二个字显示为星号，例如张三显示为`张*`，如果大于两个字，中间字显示为星号，例如成吉思汗显示为`成*汗`，诸葛亮显示为`诸*亮`)在代码中可以由JDBC的Connection对象通过getMetaData方法获取而来，主要封装了是对数据库本身的一些整体综合信息，例如数据库的产品名称，数据库的版本号，数据库的URL，是否支持事务等等。存着其他数据库的信息。
尝试着写自己的代码生成器
weixin_33737134的博客
03-27 66
         我主要以三层模式为基础，安装昨天的代码加以修改，来完成自己的通用的代码生成器，由于是初学而且专业课占去了大部分的时间，只是尝试的写点功能，还需要来熟悉下IO下的一些类，所以这里写的只是为了作为练习作为以后的复习方便而已，灰常不完善，慢慢改进，通过学习代码生成器和三层模式的结构，如果我要在这两者学习间找到着力点，重点是在三层模式，不过基本还是要依托于对ADO.NET实现机制的熟悉程...
关于自定义代码生成器
cxm52013144的博客
03-30 145
在2010年以前一直都喜欢用动软的代码生成器来生成三层结构，借此缩短开发时间，但是在做完ERP项目之后发现，有时代码本身就存在着相同或者类似的地方，只要我们用心去找还是可以找到很多的，在遇到这个问题的时候要么抽象成泛型接口，要么就用制定为项目而生的代码生成器来用，具体就是看项目本身和业务本身的需求了。最近也是在从事一款HR产品的编程，在整个产品的过程中因为周期短，项目量重，所以我把整个项目的服务层...
自定義代碼生成器
xchnba2107的博客
11-23 200
public class GenerationVlEntity { //老值 private String originalValue; //新值 private String newValue; public String getNewValue() { return newValue; } public void setNewValue(String newValue) { this.newValue ...
mybatis-puls自定义代码生成器
01-04
MyBatis-Puls是一款基于MyBatis框架的增强工具，它在MyBatis的基础上提供了许多便利的功能，其中包括自定义代码生成器。这个功能允许开发者根据数据库表结构快速生成对应的Java代码，大大提高了开发效率，降低了出错...
扩展MyBatisPlus代码生成器实现自定义源码生成，可生成前端页面、vo对象、dto对象等代码
最新发布
03-07
通过深入研究源码、理解代码生成的流程，我们可以灵活地定制代码生成器，实现前端页面、VO对象、DTO对象等自定义代码的生成，从而提高开发效率，降低维护成本。在实践中，不断尝试和调试，结合官方文档和社区资源，...
springboot-mybatis-generator-cache-demo:数据源使用阿里巴巴Druid，通过MyBatis-Generator实现自动生产模型，Mapper以及Mapper.xml，以及扩展MyBatis Generator，通过继承DefaultCommentGenerator的方式来设置MyBatis自定义代码生成器中生成实体类的注释
03-23
本项目"springboot-mybatis-generator-cache-demo"旨在展示如何在SpringBoot应用中整合MyBatis-Generator和阿里巴巴的Druid数据源，并利用MyBatis-Generator自定义代码生成器来提高开发效率。 首先，Druid是阿里...
QT自定义界面生成器
10-12
QT自定义界面生成器的一大亮点就是能自动生成QSS代码，这意味着开发者无需手动编写复杂的样式规则，只需在界面上进行直观的设置，生成器就会生成对应的QSS代码，这大大降低了设计复杂界面的难度。 使用QT自定义界面...
自己编写的简易的代码生成器
09-13
自己编写的简易的代码生成器，c#代码编辑，纯手工打造
generator生成工具（自定义）
04-11
可以按照自己所需的代码风格实现自定义生成实体类的注释、属性注释，实现序列化接口等功能
自己写的 代码生成工具
07-31
自己写的一个代码成功工具 将pdm 生成java代码 需要自己 制作模板 用velocity。
代码生成器-可自定义模版-guns
10-26
系统地讲解了如何构建一个日常生产环境实用的基于Spring Boot并且集成springmvc + shiro + mybatis-plus + beetl的后台管理系统,可管理代码生成模版,管理连接生成代码的数据库. Guns框架自带的功能：1.用户管理 2.角色管理 3.部门管理 4.菜单管理 5.字典管理 6.业务日志 7.登录日志 8.监控管理 9.通知管理 10.代码生成 Guns特点： 1. 基于SpringBoot,简化了大量项目配置和maven依赖,让您更专注于业务开发,独特的分包方式,代码多而不乱。 2. 完善的日志记录体系，可记录登录日志，业务操作日志(可记录操作前和操作后的数据)，异常日志到数据库，通过@BussinessLog注解和LogObjectHolder.me().set()方法，业务操作日志可具体记录哪个用户，执行了哪些业务，修改了哪些数据，并且日志记录为异步执行，详情请见@BussinessLog注解和LogObjectHolder,LogManager,LogAop类。 3. 利用beetl模板引擎对前台页面进行封装和拆分，使臃肿的html代码变得简洁，更加易维护。 4. 对常用js插件进行二次封装，使js代码变得简洁，更加易维护，具体请见webapp/static/js/common文件夹内js代码。 5. 利用ehcache框架对经常调用的查询进行缓存，提升运行速度，具体请见ConstantFactory类中@Cacheable标记的方法。 6. controller层采用map + warpper方式的返回结果，返回给前端更为灵活的数据，具体参见com.stylefeng.guns.modular.system.warpper包中具体类。 7. 防止XSS攻击,通过XssFilter类对所有的输入的非法字符串进行过滤以及替换。 8. 简单可用的代码生成体系，通过SimpleTemplateEngine可生成带有主页跳转和增删改查的通用控制器、html页面以及相关的js，还可以生成Service和Dao，并且这些生成项都为可选的，通过ContextConfig下的一些列xxxSwitch开关,可灵活控制生成模板代码，让您把时间放在真正的业务上。 9. 控制器层统一的异常拦截机制,利用@ControllerAdvice统一对异常拦截,具体见com.stylefeng.guns.core.aop.GlobalExceptionHandler类。 10. 页面统一的js key-value单例模式写法，每个页面生成一个唯一的全局变量，提高js的利用效率，并且有效防止多个人员开发引起的函数名/类名冲突，并且可以更好地去维护代码
.NET代码生成器 C# 版（可自定义模板函，模板类，及视图实体）
12-05
列出主要功能：  访问三种不同数据库  产生数据库的树形列表  查看数据库属性（表，视图，存储过程，字段）  通过表名自动产生SQL基本语句（select,insert,update,delete）  执行sql语句并返回结果  获得视图，存储过程的sql定义  用户自定义代码生成（可选择层，可选择生成的DAL函数）  用户批量生成代码文件（解决方案文件，项目文件，各个层的代码文件）  通过sql语句，生成视图字段实体。  可定义函数模板和类模板，使代码生成更加通用化。 说明：这需要事先安装.net framework 3.5 。这个完全出自本人之手。有什么建议望指正。 邮箱：hp_co@qq.com 。 类似于李天平的，但那个太死了，这个可以自定义函数模板和类模板，及视图实体。 文档请发邮件找我索取。 还有这个完全免费，不需要任何注册。希望大家提意见，我及时改正。 (奉献源代码)由于控件问题，自己调整主窗体，控件在dll文件夹里
自定义代码生成器使用volocity模板+genUtil进行渲染生成
J的博客
05-29 268
使用volocity模板+genUtil进行渲染生成
自己写代码生成器
ASDTP的博客
06-10 1458
1、先看界面效果根据默认的连接字符串查询出系统有多少表，在下拉框中加载出来，然后点击生成代码，生成数据模型类2、后端代码实现        private void button1_Click(object sender, EventArgs e)        {            comTable.Items.Clear();            DataTable table = Ge...
Myabtis-Plus自定义代码生成器与自定义代码生成模板
a1091758887的博客
08-29 1471
在编写代码时我们会编写大量与业务无关的代码，比如Contoller，Service，Entity以及一些增删改查接口的定义，这些代码基本都是固定的模板，手动编写浪费大量时间，今天介绍一下Mybatis-Plus代码生成器来简化代码的开发，减少重复代码的编写。 ......
如何自定义代码生成器（上）
CZ-001的博客
11-16 3169
​ 在项目开发过程中，有很多业务模块的代码是具有一定规律性的，例如controller控制器、service接口、service实现类、mapper接口、model实体类等等，这部分代码可以使用代码生成器生成，我们就可以将更多的时间放在业务逻辑上。传统的开发步骤：​ 创建数据库和表 ​ 根据表设计实体类 ​ 编写mapper接口 ​ 编写service接口和实现类 ​ 编写controller控制器 ​ 编写前端页面 ​ 前后端联调基于代码生成器开发步骤：​ 创建数据库和表 ​ 使用代码生成器生成实体类、m
Node.js实现自定义代码生成器：步骤与实战
除了基本的Curd代码生成，代码生成器还具备扩展性，支持模板定制开发，允许用户安装和使用自定义模板。例如，它能够处理前端项目的模板，将生成的代码无缝地插入到前端项目的适当位置，利用webpack的热更新功能，...
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

加把劲骑士RideOn

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付：¥1
获取中
扫码支付

您的余额不足，请更换扫码支付或充值

打赏作者

实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值