1. 前言
前几天写了篇关于Mybatis Plus代码生成器的文章，不少同学私下问我这个代码生成器是如何运作的，为什么要用到一些模板引擎，所以今天来说明下代码生成器的流程。
2. 代码生成器的使用场景
我们在编码中存在很多样板代码，格式较为固定，结构随着项目的迭代也比较稳定，而且数量巨大，这种代码写多了也没有什么技术含量，在这种情况下代码生成器可以有效提高我们的效率，其它情况并不适于使用代码生成器。
3. 代码生成器的制作流程
首先我们要制作模板，把样板代码的固定格式抽出来。然后把动态属性绑定到模板中，就像做填空题一样。所以在这个流程中模板引擎是最合适的。我们通过使用模板引擎的语法将数据动态地解析到静态模板中去，然后导出为编程中对应的文件就行了。
另外模板引擎有着丰富的绑定数据的指令集，可以让我们根据条件动态的绑定数据到模板中去。以Freemarker为例：
三元表达式：
?
|
1
|
${true ? 'checked': ''}
|
还有我们等下要用的遍历列表：
?
|
1
2
3
|
<#list fields as field>
private ${field.fieldType} ${field.fieldName};
</#list>
|
在Java开发中我们常用的模板引擎有Freemarker、Velocity、Thymeleaf ，随着Web开发中前后端分离的流行模板引擎的使用场景正在被压缩，但是它依然是一门有用的技术。
4. 代码生成器演示
接下来，我们以Freemarker为例写一个简单的代码生成器，来生成POJO类。需要引入Freemarker的依赖。
?
|
1
2
3
4
5
|
<dependency>
<groupId>org.freemarker</groupId>
<artifactId>freemarker</artifactId>
<version>2.3.28</version>
</dependency>
|
4.1 模板制作
POJO的结构可以分为以下几部分：
java.lang 包无需导入。
所以将这些规则封装到配置类中：
?
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
|
public class JavaProperties {
// 包名
private final String pkg;
// 类名
private final String entityName;
// 属性集合 需要改写 equals hash 保证名字可不重复 类型可重复
private final Set<Field> fields = new LinkedHashSet<>();
// 导入类的不重复集合
private final Set<String> imports = new LinkedHashSet<>();
public JavaProperties(String entityName, String pkg) {
this.entityName = entityName;
this.pkg = pkg;
}
public void addField(Class<?> type, String fieldName) {
// 处理 java.lang
final String pattern = "java.lang";
String fieldType = type.getName();
if (!fieldType.startsWith(pattern)) {
// 处理导包
imports.add(fieldType);
}
Field field = new Field();
// 处理成员属性的格式
int i = fieldType.lastIndexOf(".");
field.setFieldType(fieldType.substring(i + 1));
field.setFieldName(fieldName);
fields.add(field);
}
public String getPkg() {
return pkg;
}
public String getEntityName() {
return entityName;
}
public Set<Field> getFields() {
return fields;
}
public Set<String> getImports() {
return imports;
}
/**
* 成员属性封装对象.
*/
public static class Field {
// 成员属性类型
private String fieldType;
// 成员属性名称
private String fieldName;
public String getFieldType() {
return fieldType;
}
public void setFieldType(String fieldType) {
this.fieldType = fieldType;
}
public String getFieldName() {
return fieldName;
}
public void setFieldName(String fieldName) {
this.fieldName = fieldName;
}
/**
* 一个类的成员属性 一个名称只能出现一次
* 我们可以通过覆写equals hash 方法 然后放入Set
*
* @param o 另一个成员属性
* @return 比较结果
*/
@Override
public boolean equals(Object o) {
if (this == o) return true;
if (o == null || getClass() != o.getClass()) return false;
Field field = (Field) o;
return Objects.equals(fieldName, field.fieldName);
}
@Override
public int hashCode() {
return Objects.hash(fieldType, fieldName);
}
}
}
|
接着就是静态模板entity.ftl
?
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
|
package ${pkg};
<#list imports as impt>
import ${impt};
</#list>
/**
* the ${entityName} type
* @author felord.cn
*/
public class ${entityName} {
<#list fields as field>
private ${field.fieldType} ${field.fieldName};
</#list>
}
|
这里用到了Freemarker绑定数据的语法，比如List迭代渲染。
4.2 生成器编写
Freemarker通过声明配置并获取模板对象freemarker.template，该对象的process方法可以将动态数据绑定到模板中并导出为文件，最终实现了代码生成器，核心代码如下：
?
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
|
/**
* 简单的代码生成器.
*
* @param rootPath maven 的 java 目录
* @param templatePath 模板存放的文件夹
* @param templateName 模板的名称
* @param javaProperties 需要渲染对象的封装
* @throws IOException the io exception
* @throws TemplateException the template exception
*/
public static void autoCodingJavaEntity(String rootPath,
String templatePath,
String templateName,
JavaProperties javaProperties) throws IOException, TemplateException {
// freemarker 配置
Configuration configuration = new Configuration(Configuration.DEFAULT_INCOMPATIBLE_IMPROVEMENTS);
configuration.setDefaultEncoding("UTF-8");
// 指定模板的路径
configuration.setDirectoryForTemplateLoading(new File(templatePath));
// 根据模板名称获取路径下的模板
Template template = configuration.getTemplate(templateName);
// 处理路径问题
final String ext = ".java";
String javaName = javaProperties.getEntityName().concat(ext);
String packageName = javaProperties.getPkg();
String out = rootPath.concat(Stream.of(packageName.split("\\\\."))
.collect(Collectors.joining("/", "/", "/" + javaName)));
// 定义一个输出流来导出代码文件
OutputStreamWriter outputStreamWriter = new OutputStreamWriter(new FileOutputStream(out));
// freemarker 引擎将动态数据绑定的模板并导出为文件
template.process(javaProperties, outputStreamWriter);
}
|
通过执行以下代码即可生成一个UserEntity的POJO：
?
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
|
// 路径根据自己项目的特点调整
String rootPath = "C:\\\\Users\\\\felord\\\\IdeaProjects\\\\codegenerator\\\\src\\\\main\\\\java";
String packageName = "cn.felord.code";
String templatePath = "C:\\\\Users\\\\felord\\\\IdeaProjects\\\\codegenerator\\\\src\\\\main\\\\resources\\\\templates";
String templateName = "entity.ftl";
JavaProperties userEntity = new JavaProperties("UserEntity", packageName);
userEntity.addField(String.class, "username");
userEntity.addField(LocalDate.class, "birthday");
userEntity.addField(LocalDateTime.class, "addTime");
userEntity.addField(Integer.class, "gender");
userEntity.addField(Integer.class, "age");
autoCodingJavaEntity(rootPath, templatePath, templateName, userEntity);
|
生成的效果是不是跟手写的差不多：
5. 总结
这就是大部分代码生成器的机制，希望可以解答一些网友的疑问。多多关注：码农小胖哥 获取更多干货，相关的DEMO可通过公众号回复codegen获取。如果你有疑问可以通过微信MSW_623进行沟通。
到此这篇关于Java代码生成器的制作流程详解的文章就介绍到这了,更多相关Java代码生成器 制作内容请搜索快网idc以前的文章或继续浏览下面的相关文章希望大家以后多多支持快网idc！
原文链接：https://www.cnblogs.com/felordcn/p/13304856.html
相关文章
猜你喜欢
- 个人网站服务器域名解析设置指南：从购买到绑定全流程 2025-06-10
- 个人网站搭建：如何挑选具有弹性扩展能力的服务器？ 2025-06-10
- 个人服务器网站搭建：如何选择适合自己的建站程序或框架？ 2025-06-10
- 64M VPS建站：能否支持高流量网站运行？ 2025-06-10
- 64M VPS建站：怎样选择合适的域名和SSL证书？ 2025-06-10
