拒绝一次性买卖:MyBatis的mapper和repository可重复生成工具

背景

MyBatis的历史可谓久远了,码农们也在用着各式各样的代码生成工具。然而这些工具大部分都有一个缺点,那就是只能一次性生成文件。若是咱们期间在生成的文件里作了修改,再次生成时,不少工具会覆盖咱们的修改。git

为何会在生成文件后进行修改呢? 由于工具只会帮咱们生成通用的数据库访问方法(好比只生成基本的CURD操做),咱们不可避免的要根据实际的业务须要,添加其余的操做方法。github

同时,数据库也不是设计完以后就一成不变的了,咱们也可能在开发的过程当中,调整已经建好的表结构。这个时候问题就来了,利用工具再生成一次?那就要人肉合并修改;手动添加更改后的字段进去? 太多了怕遗漏。sql

因此,咱们须要找到一个方法,解决这个痛点。数据库

原理

熟悉.NET的同窗可能知道,大名鼎鼎Visual Studio也会帮开发人员生成不少代码,好比asp.net中的aspx的后台代码,它是如何保证被工具所生成的代码片断和开发人员本身写的代码片断不冲突的呢?它实际上用到了C#的分部类(partial)特性。mybatis

简单来讲,分部类,就是把一个类的代码,放到多个文件中去写,C#编译器负责把他们编译到一个类中。有了这个特性,代码生成器就只专一他负责的partial文件就能够了,开发人员的代码写到另一个partial文件中,当年用partial + T4,写了不少代码生成模板,屡试不爽。app

但咱们的JAVA不支持这个神器啊(这里说句题外话,几年前我从C#转到JAVA的时候,就感受C#在语言层面比JAVA好太多了,如今好几年没碰C#了,不知道它又先进到什么程度了),怎么办呢?asp.net

只有用不是办法的办法了,那就是继承。实体类、Repository接口,用继承的方式,把工具生成的代码和预留给开发人员人肉的代码,分割到两个文件中。工具

但mapper.xml怎么办?这个MyBatis帮咱们想好了(赞一个),利用namespace便可作到。只要namespace指向同一个Repository接口,不管是不是在同一个xml文件里,MyBatis均可以正确找到。ui

例如咱们有一个Repository是这么定义的:spa

public interface UserRepository{
    //aaa
    User selectByPrimaryKey(@Param("id") Long id);

    User selectByAccount(@Param("account") String account);
}

那么如下的两个mapper.xml结合起来是彻底可用的

UserMapper1.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="com.abc.demo.repository.UserRepository">
    <sql id="TableName">
        jm_user
    </sql>
   <!--asdfdsfsdf-->
    <sql id="BaseColumnList">
     `id`, `account`, `email`, `is_active`
    </sql>

    <resultMap id="BaseResultMap" type="com.abc.demo.entity.User" autoMapping="false">
        <result column="id" property="id" jdbcType="BIGINT"/>
        <result column="account" property="account" jdbcType="VARCHAR"/>
        <result column="email" property="email" jdbcType="VARCHAR"/>
        <result column="is_active" property="isActive" jdbcType="BIT"/>
    </resultMap>
    <select id="selectByPrimaryKey" resultMap="BaseResultMap">
        select
        <include refid="BaseColumnList"/>
        from
        <include refid="TableName"/>
        where
            `id` = #{id}
    </select>
</mapper>

UserMapper2.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="com.abc.demo.repository.UserRepository">
    <select id="selectByAccount" resultMap="BaseResultMap">
        select
        <include refid="BaseColumnList"/>
        from
        <include refid="TableName"/>
        where
            `account` = #{account}
    </select>
</mapper>

你看,UserMapper2.xml中只定义了selectByAccount方法,BaseColumnList、TableName、BaseResultMap都没有从新定义,能够直接用UserMapper1.xml的。

jasmine —— 基于数据库模型和velocity模板的代码生成工具

利用以上原理,我写了一个代码生成工具,读取数据库模型,并基于velocity模板,生成代码。
能够命令行形式运行,也能够做为IDEA的插件运行。

项目地址:https://github.com/kongxiangxin/jasmine
工具下载地址:https://github.com/kongxiangxin/jasmine/releases

源码里提供了一个示例DEMO,里面包含了MyBatis的代码生成模板,支持以下特性:

  1. 一键生成实体类、MyBatis Repository、MyBatis Mapper
  2. 实体类、MyBatis Repository和MyBatis Mapper均利用继承策略,划分出XXXX和XXXXBase两个文件,其中XXXX若是文件存在则不覆盖,XXXXBase每次生成都会覆盖。若是在生成后你须要作一些代码上的调整,请在XXXX文件中修改,而不要在XXXXBase中修改。这样作的好处是一旦咱们的表结构发生变化须要从新生成时,不会覆盖您手动改过的代码。
  3. 若是表存在is_deleted字段,生成的delete方法是逻辑删除而不是物理删除。
  4. 若是表存在record_version字段,update语句带有乐观锁,即update .... set record_version=record_version + 1 where .... and record_version=#{record_version}
  5. 若是表存在create_time,insert语句这一列的值是now()
  6. 若是表存在update_time, insert和update语句这一列的值是now()

Quick Start

  1. 去releases页面,下载最新的jasmine-[version].zip,解压。
  2. clone源码至本地,根据实际状况,修改demo/jasmine.properties中jdbc相关的配置(主要是数据库链接配置)
  3. 执行如下命令:
/path/to/jasmine-[version]/bin/jasmine /path/to/jasmine-src/demo/jasmine.properties

若是一切正常,会在demo下看到生成出来的文件

相关文章
相关标签/搜索