Skip to content

4.2.codegenerator

abel533 edited this page Mar 8, 2018 · 1 revision

4.2 代码生成器文档

代码生成器是基于 MBG 插件的,所以需要配合 MBG 使用。

一个简单的 MBG 配置如下:

<?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" targetRuntime="MyBatis3Simple" defaultModelType="flat">
        <property name="javaFileEncoding" value="UTF-8"/>
        <!--配置是否使用通用 Mapper 自带的注释扩展,默认 true-->
        <!--<property name="useMapperCommentGenerator" value="false"/>-->

        <!--通用 Mapper 插件,可以生成带注解的实体类-->
        <plugin type="tk.mybatis.mapper.generator.MapperPlugin">
            <property name="mappers" value="tk.mybatis.mapper.common.Mapper,tk.mybatis.mapper.hsqldb.HsqldbMapper"/>
            <property name="caseSensitive" value="true"/>
            <property name="forceAnnotation" value="true"/>
            <property name="beginningDelimiter" value="`"/>
            <property name="endingDelimiter" value="`"/>
        </plugin>

        <!--通用代码生成器插件-->
        <plugin type="tk.mybatis.mapper.generator.TemplateFilePlugin">
            <property name="targetProject" value="src/test/java"/>
            <property name="targetPackage" value="test.mapper"/>
            <property name="templatePath" value="generator/mapper.ftl"/>
            <property name="mapperSuffix" value="Dao"/>
            <property name="fileName" value="${tableClass.shortClassName}${mapperSuffix}.java"/>
        </plugin>

        <jdbcConnection driverClass="com.mysql.jdbc.Driver"
                        connectionURL="jdbc:mysql://localhost:3306/test"
                        userId="root"
                        password="">
        </jdbcConnection>

        <!--MyBatis 生成器只需要生成 Model-->
        <javaModelGenerator targetPackage="test.model" targetProject="./src/test/java"/>

        <table tableName="user%">
            <generatedKey column="id" sqlStatement="JDBC"/>
        </table>
    </context>
</generatorConfiguration>

在这个配置中,我们只关注 tk.mybatis.mapper.generator.TemplateFilePlugin

4.2.1 基于模板的插件 TemplateFilePlugin

这个插件中除了几个必备的属性外,还可以增加任意的属性,属性完全是为了给模板提供数据。

先看一个基本完整的配置:

<!--测试输出单个文件,每个表都会生成一个对应的文件-->
<plugin type="tk.mybatis.mapper.generator.TemplateFilePlugin">
    <property name="singleMode" value="false"/>
    <property name="targetProject" value="src/test/resources"/>
    <property name="targetPackage" value=""/>
    <property name="templatePath" value="generator/test-one.ftl"/>
    <property name="fileName" value="${tableClass.shortClassName}Test.txt"/>
    <!--默认值是下面这个,可以不配置-->
    <property name="templateFormatter" value="tk.mybatis.mapper.generator.formatter.FreemarkerTemplateFormatter"/>
</plugin>

下面介绍必备的属性。

1. targetProject

用于指定目标项目,一般是 src/main/java 或者 src/main/resource 这样的目录。 还可以是 src/test/java 或者 src/test/resource 这样的目录。

在多模块项目中,还能通过相对路径指定为其他的目录,例如:

<property name="targetProject" value="../myproject-api/src/main/java"/>

这个属性值有一个要求,就是目录必须存在,否则不会生成代码!

2. targetPackage

用于指定包的部分,虽然是这个名字,实际上就是路径。

这个属性指定的路径如果不存在,就会自动创建。

这个属性的值可以为空。

例如 mapper/admin 用于生成 mapper/admin/ 目录,或者 tk.mybatis.mapper 生成包(本质上还是目录)。

这个属性还有一个特殊的地方,它还支持使用模板,就和下面的 fileName 一样,举个简单的使用场景。

你可能在生成前端代码的时候,希望将表对应的 JSP 生成在自己的一个目录中,此时可以配置为:

<property name="targetPackage" value="WEB-INF/jsp/${tableClass.lowerCaseName}/"/>

模板中可以用到的属性,这里都能用,其他属性后面会介绍。

通过这个路径也能看出来,配置一个插件只能根据模板在一个指定位置(targetProject 和 targetPackage 决定的目录)生成一个文件。

3. templatePath

指定模板路径,可以是任意能够通过 ClassLoader 能够获取的位置,文件类型没有限制。

例如示例中的 generator/test-one.ftl

这个属性必须指定,否则不会生成代码!

4. fileName

这个属性用于指定生成文件的名字,这个值支持使用模板,例如上面的 ${tableClass.shortClassName}Test.txt,具体可用的属性会在后面介绍。

这个属性必须指定,否则不会生成代码!

5. templateFormatter

这个属性可选,默认使用基于 FreeMarker 的实现!

默认情况下,你需要添加下面的依赖:

<dependency>
    <groupId>org.freemarker</groupId>
    <artifactId>freemarker</artifactId>
    <version>2.3.23</version>
</dependency>

默认的实现类为:tk.mybatis.mapper.generator.formatter.FreemarkerTemplateFormatter

这个类实现了两个接口 TemplateFormatter, ListTemplateFormatter

这俩接口分别对应下面 singleMode 参数值的 truefalse

也就是一个表生成一个文件,或者多个表生成一个文件。

对于一般情况下,都是第一种情况。但是在配置文件中,可能会用到多个表的信息。

如果你想使用其他模板引擎,可以自己实现上面的接口。

6. singleMode

上面已经提过,默认为 true

一个表生成一个文件时,可用属性可以参考 generator/test-one.ftl,表的属性在 tableClass 中。

多个表生成一个文件时,可用属性可以参考 generator/test-all.ftl,所有表的属性在 tableClassSet 中,通过遍历可以获取单个的信息。

7. 其他你需要的属性

模板中需要的特殊信息都可以通过 <property> 方法设置,在模板中直接使用这里定义的属性名来使用,后面例子的中的 mapperSuffix 就是这种属性。

4.2.2 TemplateFilePlugin 配置示例

因为模板需要根据业务进行设计,所以这里只提供了两个简单的 mapper 目标和两个完整属性的示例模板。

因为一个模板只能生成一类的文件,所以如果要生成多个不同的文件,就需要配置多个插件。

这种设计很灵活,因为自由度很高,所以代价就是配置的多。

但是正常情况下,根据业务设计的一套模板基本是固定的,不会有太多变化,所以用起来并不麻烦。

例如下面的示例:

<!--通用代码生成器插件-->
<!--mapper接口-->
<plugin type="tk.mybatis.mapper.generator.TemplateFilePlugin">
    <property name="targetProject" value="src/test/java"/>
    <property name="targetPackage" value="test.mapper"/>
    <property name="templatePath" value="generator/mapper.ftl"/>
    <property name="mapperSuffix" value="Dao"/>
    <property name="fileName" value="${tableClass.shortClassName}${mapperSuffix}.java"/>
</plugin>
<!--mapper.xml-->
<plugin type="tk.mybatis.mapper.generator.TemplateFilePlugin">
    <property name="targetProject" value="src/test/resources"/>
    <property name="targetPackage" value="mappers"/>
    <property name="mapperPackage" value="test.mapper"/>
    <property name="templatePath" value="generator/mapperXml.ftl"/>
    <property name="mapperSuffix" value="Dao"/>
    <property name="fileName" value="${tableClass.shortClassName}${mapperSuffix}.xml"/>
</plugin>
<!--测试输出单个文件,每个表都会生成一个对应的文件-->
<plugin type="tk.mybatis.mapper.generator.TemplateFilePlugin">
    <property name="targetProject" value="src/test/resources"/>
    <property name="targetPackage" value=""/>
    <property name="templatePath" value="generator/test-one.ftl"/>
    <property name="fileName" value="${tableClass.shortClassName}Test.txt"/>
    <!--默认值是下面这个,可以不配置-->
    <property name="templateFormatter"
              value="tk.mybatis.mapper.generator.formatter.FreemarkerTemplateFormatter"/>
</plugin>
<!--测试输出整个文件,所有表都可用,一次只生成一个文件,用于聚合所有表使用-->
<plugin type="tk.mybatis.mapper.generator.TemplateFilePlugin">
    <property name="singleMode" value="false"/>
    <property name="targetProject" value="src/test/resources"/>
    <property name="targetPackage" value=""/>
    <property name="templatePath" value="generator/test-all.ftl"/>
    <property name="fileName" value="TestAll.txt"/>
</plugin>

前两个会生成 Dao 后缀的 Mapper 接口和 XML,其中有个针对性的参数 mapperSuffix 用于配置后缀, 还有个 mapperPackage 在生成 XML 时获取接口的包名(因为和这里的 targetPackage 可以不同)。

后两个插件用于演示所有可用的属性,而且是两种不同的模式。

在表和实体上可用的所有属性如下:

特殊:targetPackage值在 ${package} 中。

<!-- 详细日期用法参考:http://freemarker.apache.org/docs/ref_builtins_date.html -->
当前时间:
<#assign dateTime = .now>
日期:${dateTime?date}
时间:${dateTime?time}
格式化:${dateTime?string["yyyy-MM-dd HH:mm:ss"]}


所有配置的属性信息:
<#list props?keys as key>
${key} - ${props[key]}
</#list>

实体和表的信息:
表名:${tableClass.tableName}
变量名:${tableClass.variableName}
小写名:${tableClass.lowerCaseName}
类名:${tableClass.shortClassName}
全名:${tableClass.fullClassName}
包名:${tableClass.packageName}

列的信息:
=====================================
<#if tableClass.pkFields??>
主键:
    <#list tableClass.pkFields as field>
    -------------------------------------
    列名:${field.columnName}
    列类型:${field.jdbcType}
    字段名:${field.fieldName}
    注释:${field.remarks}
    类型包名:${field.typePackage}
    类型短名:${field.shortTypeName}
    类型全名:${field.fullTypeName}
    是否主键:${field.identity?c}
    是否可空:${field.nullable?c}
    是否为BLOB列:${field.blobColumn?c}
    是否为String列:${field.stringColumn?c}
    是否为字符串列:${field.jdbcCharacterColumn?c}
    是否为日期列:${field.jdbcDateColumn?c}
    是否为时间列:${field.jdbcTimeColumn?c}
    是否为序列列:${field.sequenceColumn?c}
    列长度:${field.length?c}
    列精度:${field.scale}
    </#list>
</#if>

<#if tableClass.baseFields??>
基础列:
    <#list tableClass.baseFields as field>
    -------------------------------------
    列名:${field.columnName}
    列类型:${field.jdbcType}
    字段名:${field.fieldName}
    注释:${field.remarks}
    类型包名:${field.typePackage}
    类型短名:${field.shortTypeName}
    类型全名:${field.fullTypeName}
    是否主键:${field.identity?c}
    是否可空:${field.nullable?c}
    是否为BLOB列:${field.blobColumn?c}
    是否为String列:${field.stringColumn?c}
    是否为字符串列:${field.jdbcCharacterColumn?c}
    是否为日期列:${field.jdbcDateColumn?c}
    是否为时间列:${field.jdbcTimeColumn?c}
    是否为序列列:${field.sequenceColumn?c}
    列长度:${field.length?c}
    列精度:${field.scale}
    </#list>
</#if>

<#if tableClass.blobFields??>
Blob列:
    <#list tableClass.blobFields as field>
    -------------------------------------
    列名:${field.columnName}
    列类型:${field.jdbcType}
    字段名:${field.fieldName}
    注释:${field.remarks}
    类型包名:${field.typePackage}
    类型短名:${field.shortTypeName}
    类型全名:${field.fullTypeName}
    是否主键:${field.identity?c}
    是否可空:${field.nullable?c}
    是否为BLOB列:${field.blobColumn?c}
    是否为String列:${field.stringColumn?c}
    是否为字符串列:${field.jdbcCharacterColumn?c}
    是否为日期列:${field.jdbcDateColumn?c}
    是否为时间列:${field.jdbcTimeColumn?c}
    是否为序列列:${field.sequenceColumn?c}
    列长度:${field.length?c}
    列精度:${field.scale}
    </#list>
</#if>

=====================================
全部列(包含了pk,base,blob 字段,可用的属性和上面的一样):
<#if tableClass.allFields??>
列名 - 字段名
    <#list tableClass.allFields as field>
    ${field.columnName} - ${field.fieldName}
    </#list>
</#if>

4.2.3 测试执行

上面示例就是本项目的测试代码,在 src/test/resources/generator/generatorConfig.xml 中。

还提供了一种 Java 编码方式运行的类,src/test/java/ 中的 tk.mybatis.mapper.generator.Generator,配置上面 xml 中的数据库信息就可以生成。

测试生成的部分结果如下。

实体:

@Table(name = "`user_info`")
public class UserInfo {
    @Id
    @Column(name = "`Id`")
    @GeneratedValue(generator = "JDBC")
    private Integer id;

Dao:

package test.mapper;

import test.model.UserInfo;

/**
* 通用 Mapper 代码生成器
*
* @author mapper-generator
*/
public interface UserInfoDao extends tk.mybatis.mapper.common.Mapper<UserInfo> {

}

XML:

<?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="test.mapper.UserInfoDao">

</mapper>

test-one.ftl 生成的信息如下:

目标package: 

当前时间2017-11-6
22:00:45
2017-11-06 22:00:45

所有配置的属性信息:
targetPackage - 
templateFormatter - tk.mybatis.mapper.generator.formatter.FreemarkerTemplateFormatter
templatePath - generator/test-one.ftl
targetProject - src/test/resources
fileName - ${tableClass.shortClassName}Test.txt

实体和表的信息表名user_info
变量名userInfo
小写名userinfo
类名UserInfo
全名test.model.UserInfo
包名test.model

列的信息:
=====================================
主键:
    -------------------------------------
    列名Id
    列类型INTEGER
    字段名id
    注释类型包名java.lang
    类型短名Integer
    类型全名java.lang.Integer
    是否主键true
    是否可空false
    是否为BLOB列false
    是否为String列false
    是否为字符串列false
    是否为日期列false
    是否为时间列false
    是否为序列列false
    列长度10
    列精度0

基础列:
    -------------------------------------
    列名username
    列类型VARCHAR
    字段名username
    注释用户名
    类型包名java.lang
    类型短名String
    类型全名java.lang.String
    是否主键false
    是否可空false
    是否为BLOB列false
    是否为String列true
    是否为字符串列true
    是否为日期列false
    是否为时间列false
    是否为序列列false
    列长度32
    列精度0
    -------------------------------------
    列名password
    列类型VARCHAR
    字段名password
    注释密码
    类型包名java.lang
    类型短名String
    类型全名java.lang.String
    是否主键false
    是否可空true
    是否为BLOB列false
    是否为String列true
    是否为字符串列true
    是否为日期列false
    是否为时间列false
    是否为序列列false
    列长度32
    列精度0
    -------------------------------------
    列名usertype
    列类型VARCHAR
    字段名usertype
    注释用户类型
    类型包名java.lang
    类型短名String
    类型全名java.lang.String
    是否主键false
    是否可空true
    是否为BLOB列false
    是否为String列true
    是否为字符串列true
    是否为日期列false
    是否为时间列false
    是否为序列列false
    列长度2
    列精度0
    -------------------------------------
    列名enabled
    列类型INTEGER
    字段名enabled
    注释是否可用
    类型包名java.lang
    类型短名Integer
    类型全名java.lang.Integer
    是否主键false
    是否可空true
    是否为BLOB列false
    是否为String列false
    是否为字符串列false
    是否为日期列false
    是否为时间列false
    是否为序列列false
    列长度10
    列精度0
    -------------------------------------
    列名realname
    列类型VARCHAR
    字段名realname
    注释真实姓名
    类型包名java.lang
    类型短名String
    类型全名java.lang.String
    是否主键false
    是否可空true
    是否为BLOB列false
    是否为String列true
    是否为字符串列true
    是否为日期列false
    是否为时间列false
    是否为序列列false
    列长度32
    列精度0
    -------------------------------------
    列名qq
    列类型VARCHAR
    字段名qq
    注释QQ
    类型包名java.lang
    类型短名String
    类型全名java.lang.String
    是否主键false
    是否可空true
    是否为BLOB列false
    是否为String列true
    是否为字符串列true
    是否为日期列false
    是否为时间列false
    是否为序列列false
    列长度14
    列精度0
    -------------------------------------
    列名email
    列类型VARCHAR
    字段名email
    注释类型包名java.lang
    类型短名String
    类型全名java.lang.String
    是否主键false
    是否可空true
    是否为BLOB列false
    是否为String列true
    是否为字符串列true
    是否为日期列false
    是否为时间列false
    是否为序列列false
    列长度100
    列精度0
    -------------------------------------
    列名tel
    列类型VARCHAR
    字段名tel
    注释联系电话
    类型包名java.lang
    类型短名String
    类型全名java.lang.String
    是否主键false
    是否可空true
    是否为BLOB列false
    是否为String列true
    是否为字符串列true
    是否为日期列false
    是否为时间列false
    是否为序列列false
    列长度255
    列精度0

Blob列:

=====================================
全部列列名 - 字段名
    Id - id
    username - username
    password - password
    usertype - usertype
    enabled - enabled
    realname - realname
    qq - qq
    email - email
    tel - tel

4.2.4 最后

基础的代码生成器是很简单的,和 Java 拼字符串输出很像,这里只是使用了模板。

几乎所有人都在 JSP 中用过的 EL 就是一种模板,可能你会 <c:forEach 这种,但是联想不到这里的代码生成器而已。

后续会在 https://github.com/abel533/Mybatis-Spring 项目中提供一套模板做为示例。