MyBatis Generator 自定义生成注释的方法
作者：mizhoux 发布时间：2023-02-24 12:37:23
标签：MyBatis,Generator,注释
最近做项目，ORM 使用的是 MyBatis，为了偷懒，我自然而然的想到了使用 MyBatis Generator（MBG）来生成数据库表对应的实体代码和 Mapper 代码。于是做了如下的配置（对 MBG 配置不熟悉的同学可以参考 Mybatis Generator最完整配置详解）：
<?xml version="1.0" encoding="UTF-8"?>
<!DOCTYPE generatorConfiguration PUBLIC
"-//mybatis.org//DTD MyBatis Generator Configuration 1.0//EN"
"http://mybatis.org/dtd/mybatis-generator-config_1_0.dtd">
<generatorConfiguration>
<!-- 指定数据库驱动的jdbc驱动jar包的位置 -->
<classPathEntry location="./mysql-connector-java-5.1.40.jar" />
<context id="mysql" defaultModelType="hierarchical" targetRuntime="MyBatis3Simple" >
<!-- 生成的 Java 文件的编码 -->
<property name="javaFileEncoding" value="UTF-8"/>
<!-- 格式化 Java 代码 -->
<property name="javaFormatter" value="org.mybatis.generator.api.dom.DefaultJavaFormatter"/>
<!-- 格式化 XML 代码 -->
<property name="xmlFormatter" value="org.mybatis.generator.api.dom.DefaultXmlFormatter"/>
<!-- 配置数据库连接 -->
<jdbcConnection driverClass="com.mysql.jdbc.Driver" connectionURL="jdbc:mysql://localhost:3306/test?characterEncoding=utf-8" userId="root" password="123456">
</jdbcConnection>
<!-- 生成实体的位置 -->
<javaModelGenerator targetPackage="me.mizhoux.model" targetProject="src/main/java">
<property name="enableSubPackages" value="true"/>
</javaModelGenerator>
<!-- 生成 Mapper 接口的位置 -->
<sqlMapGenerator targetPackage="me.mizhoux.mapper" targetProject="src/main/java">
<property name="enableSubPackages" value="true"/>
</sqlMapGenerator>
<!-- 生成 Mapper XML 的位置 -->
<javaClientGenerator targetPackage="me.mizhoux.mapper" type="XMLMAPPER" targetProject="src/main/java">
<property name="enableSubPackages" value="true"/>
</javaClientGenerator>
<!-- 设置数据库的表名和实体类名 -->
<table tableName="t_user" domainObjectName="User">
<!-- generatedKey用于生成生成主键的方法 -->
<generatedKey column="id" sqlStatement="SELECT LAST_INSERT_ID()"/>
</table>
</context>
</generatorConfiguration>
数据库建库建表的代码：
CREATE SCHEMA `db_test` DEFAULT CHARACTER SET utf8 COLLATE utf8_unicode_ci ;
CREATE TABLE `db_test`.`t_user` (
`id` INT NOT NULL AUTO_INCREMENT COMMENT '用户 ID',
`username` VARCHAR(30) NULL COMMENT '用户名称',
`password` VARCHAR(20) NULL COMMENT '用户密码',
`birthday` DATE NULL COMMENT '用户生日',
PRIMARY KEY (`id`),
UNIQUE INDEX `username_UNIQUE` (`username` ASC)
) COMMENT = '用户';
开开心心，执行命令，开始生成代码：
java -jar mybatis-generator-core-1.3.7.jar -configfile generatorConfig.xml -overwrite
然后查看生成的 Java 实体类：
看着这个注释，让我有点纠结啊 —— 为什么不是数据库中每个字段对应的注释呢？查找相关资料，得知 MBG 生成的是由 org.mybatis.generator.api.CommentGenerator 来控制的。这是一个接口，MBG 的默认实现类是做 org.mybatis.generator.internal.DefaultCommentGenerator。当你在 generatorConfig.xml 中配置了 commentGenerator 标签，那么默认状态下，生成注释的工作，将由 DefaultCommentGenerator来完成。 所以我们来查看下这个 DefaultCommentGenerator 的源码：
public class DefaultCommentGenerator implements CommentGenerator {
// 属性，即配置在 commentGenerator 标签之内的 Property 标签
private Properties properties;
// 是否不生成日期
private boolean suppressDate;
// 是否不生成注释
private boolean suppressAllComments;
// 是否添加数据库内的注释
private boolean addRemarkComments;
// 日期格式化
private SimpleDateFormat dateFormat;
public DefaultCommentGenerator() {
super();
properties = new Properties();
suppressDate = false;
suppressAllComments = false;
addRemarkComments = false;
}
@Override
public void addConfigurationProperties(Properties properties) {
this.properties.putAll(properties);
suppressDate = isTrue(properties
.getProperty(PropertyRegistry.COMMENT_GENERATOR_SUPPRESS_DATE));
suppressAllComments = isTrue(properties
.getProperty(PropertyRegistry.COMMENT_GENERATOR_SUPPRESS_ALL_COMMENTS));
addRemarkComments = isTrue(properties
.getProperty(PropertyRegistry.COMMENT_GENERATOR_ADD_REMARK_COMMENTS));
String dateFormatString = properties.getProperty(PropertyRegistry.COMMENT_GENERATOR_DATE_FORMAT);
if (StringUtility.stringHasValue(dateFormatString)) {
dateFormat = new SimpleDateFormat(dateFormatString);
}
}
// 其他代码
...
}
addRemarkComments 这个属性，看来就是用来生成数据库注释用的 —— 好开心，那把它设置为 true 试试：
<generatorConfiguration>
<!-- 指定数据库驱动的jdbc驱动jar包的位置 -->
<classPathEntry location="./mysql-connector-java-5.1.40.jar" />
<context id="mysql" defaultModelType="hierarchical" targetRuntime="MyBatis3Simple" >
<property name="javaFileEncoding" value="UTF-8"/>
<!-- 其他 Property -->
<commentGenerator>
<property name="suppressDate" value="true"/>
<property name="addRemarkComments" value="true"/>
</commentGenerator>
...
</context>
</generatorConfiguration>
运行命令：
java -jar mybatis-generator-core-1.3.7.jar -configfile generatorConfig.xml -overwrite
数据库注释倒是拿到了，但是生成的一堆其他信息，看着实在是太扎眼了。查看源码，发现这些内容已经写死在 DefaultCommentGenerator 中了，没有办法自定义。
自己动手丰衣足食，我们为啥不自己写个类实现 CommentGenerator 接口，然后自定义自己想要的注释呢。查看 commentGenerator 的 DTD，发现正好 commentGenerator 有个 type 属性，可以用来指定自己的注释实现类：
查看 CommentGenerator 接口，发现里面的方法非常多，不仅包含了生成 Java 实体注释对应的方法，还包括了生成 XML 中注释的方法。所以我们先写一个默认的实现类，实现CommentGenerator 接口，但不做任何操作 —— 因为 DefaultCommentGenerator 本文已经存在了，为了避免混淆，就叫它SimpleCommentGenerator吧。然后定义我们自己的注释类，MySQLCommentGenerator，继承 SimpleCommentGenerator，重写我们需要的方法：
public class MySQLCommentGenerator extends SimpleCommentGenerator {
private Properties properties;
public MySQLCommentGenerator() {
properties = new Properties();
}
@Override
public void addConfigurationProperties(Properties properties) {
// 获取自定义的 properties
this.properties.putAll(properties);
}
@Override
public void addModelClassComment(TopLevelClass topLevelClass, IntrospectedTable introspectedTable) {
String author = properties.getProperty("author");
String dateFormat = properties.getProperty("dateFormat", "yyyy-MM-dd");
SimpleDateFormat dateFormatter = new SimpleDateFormat(dateFormat);
// 获取表注释
String remarks = introspectedTable.getRemarks();
topLevelClass.addJavaDocLine("/**");
topLevelClass.addJavaDocLine(" * " + remarks);
topLevelClass.addJavaDocLine(" *");
topLevelClass.addJavaDocLine(" * @author " + author);
topLevelClass.addJavaDocLine(" * @date " + dateFormatter.format(new Date()));
topLevelClass.addJavaDocLine(" */");
}
@Override
public void addFieldComment(Field field, IntrospectedTable introspectedTable, IntrospectedColumn introspectedColumn) {
// 获取列注释
String remarks = introspectedColumn.getRemarks();
field.addJavaDocLine("/**");
field.addJavaDocLine(" * " + remarks);
field.addJavaDocLine(" */");
}
}
因为我们现在要使用到我们自己自定义的 CommentGenerator ，所以我们 通过代码的方式来操作 MBG：
public class Generator {
public static void main( String[] args ) throws Exception {
List<String> warnings = new ArrayList<>();
File configFile = new File("generatorConfig.xml");
ConfigurationParser cp = new ConfigurationParser(warnings);
Configuration config = cp.parseConfiguration(configFile);
DefaultShellCallback callback = new DefaultShellCallback(true);
MyBatisGenerator myBatisGenerator = new MyBatisGenerator(config, callback, warnings);
myBatisGenerator.generate(null);
}
}
然后配置 generatorConfig.xml设置我们自己的注释生成器：
<?xml version="1.0" encoding="UTF-8"?>
<!DOCTYPE generatorConfiguration PUBLIC
"-//mybatis.org//DTD MyBatis Generator Configuration 1.0//EN"
"http://mybatis.org/dtd/mybatis-generator-config_1_0.dtd">
<generatorConfiguration>
<!-- 指定数据库驱动的jdbc驱动jar包的位置 -->
<!-- 不再需要，因为 jar 包已经在 classpath 中
<classPathEntry location="./mysql-connector-java-5.1.40.jar" />
-->
<context id="mysql" defaultModelType="hierarchical" targetRuntime="MyBatis3Simple" >
...
<!-- 自定义注释生成器 -->
<commentGenerator type="me.mizhoux.mbgcomment.MySQLCommentGenerator">
<property name="author" value="Michael Chow"/>
<property name="dateFormat" value="yyyy/MM/dd"/>
</commentGenerator>
...
</context>
</generatorConfiguration>
完整的 Maven 项目在 我的 GitHub。现在，我们运行主类 Generator，成功生成了数据库中的注释：
等等，好像有点不对劲！
类的注释怎么没有了！
想来应该是 JDBC 连接 MySQL 的时候需要添加什么属性才能获取表的注释，上网查询，发现是 useInformationSchema，需要将其设置为 true（看来是 MBG 给自己的 DefaultCommentGenerator 开了小灶）：
<?xml version="1.0" encoding="UTF-8"?>
<!DOCTYPE generatorConfiguration PUBLIC
"-//mybatis.org//DTD MyBatis Generator Configuration 1.0//EN"
"http://mybatis.org/dtd/mybatis-generator-config_1_0.dtd">
<generatorConfiguration>
<context id="mysql" defaultModelType="hierarchical" targetRuntime="MyBatis3Simple" >
...
<!-- 自定义注释生成器 -->
<commentGenerator type="me.mizhoux.mbgcomment.MySQLCommentGenerator">
<property name="author" value="Michael Chow"/>
<property name="dateFormat" value="yyyy/MM/dd"/>
</commentGenerator>
<!-- 配置数据库连接 -->
<jdbcConnection driverClass="com.mysql.jdbc.Driver" connectionURL="jdbc:mysql://localhost:3306/test?characterEncoding=utf-8" userId="root" password="123456">
<!-- 设置 useInformationSchema 属性为 true -->
<property name="useInformationSchema" value="true" />
</jdbcConnection>
...
</context>
</generatorConfiguration>
然后再次运行主类 Generator：
成功的生成了类主食和字段注释~
我这里并没有处理注释是多行文本的情况 —— 留给有兴趣的读者吧~
小项目地址：https://github.com/mizhoux/mbg-comment
来源：https://segmentfault.com/a/1190000016525887
0
投稿猜你喜欢
- 本文实例讲述了C#读取csv格式文件的方法。分享给大家供大家参考。具体实现方法如下：一、CSV文件规则 1 开头是不留空，以行为单
本文介绍了spring整合JMS实现同步收发消息(基于ActiveMQ的实现)，分享给大家，具体如下：1. 安装ActiveMQ注意：JDK- Sequence序列Sequence 是Kotlin标准库提供的一种容器类型。它和Iterable一样具备对集合进行多步骤操作能力，但是却是
- 安卓虽然已经成为了移动设备第一操作系统，且影响力也延伸到了汽车和tv端，不过对于谷歌来说，需要依靠Java来做安卓开发一直是一个心病，因为O
- MyBatis根据条件批量修改字段背景：给学生改作业，只要是对的都批量进行数据库的修改代码以及注释conttoller@RestContro
- 一、微服务简介 Ⅰ、我对微服务的理解微服务是软件开发的一种架构方式，由单一的应用小程序构成的小服务；一个软件系统由多个服务组成；在微服务中，
- https://www.jb51.net/article/191716.htm 此篇博文对flyway讲解的很清楚了，我在这只是稍
- 本文实例展示了WinForm实现为ComboBox绑定数据源并提供下拉提示功能，这是一个非常有实用价值的功能，具体实现方法如下：主要功能代码
- 本文实例讲述了Android通过json向MySQL中写入数据的方法。分享给大家供大家参考，具体如下：先说一下如何通过json将Androi
- 系统有很多光标类型 ：Cursors 类 (System.Windows.Input) | Microsoft Docs本章介绍如何自定义光
- 一.小伙伴们在做钱包支付中，相信会有个绕不过去的输入支付密码页面。下面小编给个效果图：898342572738938468.png实现的原理
- android跑马灯出现重复跳动、不滚动问题，本文给出解决方案，供大家参考。原因：页面有View被重新绘制了、焦点被抢占例如：1、TextV
- 一、Maven聚合开发_继承关系 Maven中
- 目录前言Binder的使用模糊进程间调用Binder原理ioctlbinder初始化总结前言Binder是安卓中实现IPC（进程间通信的）常
- 1.摘要Android手机间通过蓝牙方式进行通信，有两种常见的方式，一种是socket方式，另一种是通过Gatt Server（Androi
- Task执行任务,等待任务完成代码：//任务Func<int> Funcs = () =>{? ? Console.Wri
- 在日常工作中，我们有时会需要修改字体的颜色来突出文本重点，让读者更容易抓住文章要点。在今天这篇文章中，我将为大家介绍如何以编程方式，在Wor
- 大概理解查了一个小时的资料：async和await发现这个大神的解释一针见血，深得我心！以最简单的例子，解释了async和await。妙~~
- 开发环境win10Android Studio效果用于多级菜单展示，或选择。如 每个省，市，县；如 树木的病虫害；关键代码 @overrid
- 大家好，这是 [C#.NET 拾遗补漏] 系列的第 08 篇文章，今天讲 C# 强大的 LINQ 查询。LINQ 是我最喜欢的 C# 语言特
