介绍

TgDao是一款基于Mybatis的编译期SQL生成器,利用注解来表达SQL,能根据你的方法签名生成对应的Mapper.xml文件。 它能减少你日常开发中大量简单SQL的编写,由于它只是生成Mapper.xml文件,因此对于复杂的查询场景, 你同样可以自己编写来完成一些工具所无法生成的SQL。

@Table(name = "T_User")
public class User {
    @Id("id")
    private int id;
    private String username;
    private int age;
}

复制代码

上面的model定义了模型和数据库表的关系,看到下面这些方法的签名,聪明的你肯定能猜出每个方法的sql吧,这就是这个库要做的工作。

@DaoGen(model = User.class)
public interface UserDao {
    @Select
    @OrderBy("id desc")
    List<User> queryUser(@Condition(criterion = Criterions.EQUAL, column = "username") String name,
                         @Condition(criterion = Criterions.GREATER, attach = Attach.OR) int age,
                         @Limit int limit, @OffSet int offset);

    @Select
    List<User> queryUser2(@Condition(criterion = Criterions.GREATER, column = "age") int min,
                          @Condition(criterion = Criterions.LESS, column = "age") int max);

    @Select
    List<User> queryUser3(@Condition(criterion = Criterions.EQUAL, column = "username") String name,
                          @Condition(attach = Attach.OR, column = "id", criterion = Criterions.IN) String[] ids);

    @Insert(useGeneratedKeys = true, keyProperty = "id")
    int insert(User user);

    @BatchInsert(columns = "username,age")
    int batchInsert(List<User> users);

    @Update
    @ModelConditions({
            @ModelCondition(field = "id")
    })
    int update(User user);

    @Delete
    int delete(@Condition(criterion = Criterions.GREATER, column = "age") int min,
               @Condition(criterion = Criterions.LESS, column = "age") int max);
}
复制代码

文档

引入如下依赖:

<dependency>
  <groupId>com.github.twogoods</groupId>
  <artifactId>tgdao-core</artifactId>
  <version>0.1.3</version>
</dependency>
复制代码

Table与Model关联

@Table记录数据表的名字 @Id记录主键信息 @Column映射了表字段和属性的关系,如果表字段和类属性同名,那么可以省略这个注解 @Ingore忽略这个类属性,没有哪个表字段与它关联

@Table(name = "T_User")
public class User {
    @Id("id")
    private int id;

    private String username;
    private String password;
    private int age;

    @Column("old_address")
    private String oldAddress;
    @Column("now_address")
    private String nowAddress;

    private int state;

    @Column("created_at")
    private Timestamp createdAt;
    @Column("updated_at")
    private Timestamp updatedAt;

    @Ignore
    private String remrk;
复制代码

查询

@Select
@OrderBy("id desc")
List<User> queryUser(@Condition(criterion = Criterions.EQUAL, column = "username") String name,
                    @Condition(criterion = Criterions.GREATER, attach = Attach.OR) int age,
                    @Condition(column = "id", criterion = Criterions.IN) String[] ids,
                    @Limit int limit, @OffSet int offset);
复制代码
@Select
  • columns:默认 select *可以配置columns("username,age")选择部分字段;
  • SqlMode:有两个选择,SqlMode.SELECTIVE 和 SqlMode.COMMON,区别是selective会检查查询条件的字段是否为null来实现动态的查询, 即<if test="name != null">username = #{name}</if>
@Condition
  • criterion:查询条件,=,<,>,in等,具体见Criterions
  • column:与表字段的对应,若与字段名相同可不配置
  • attach:连接 and,or, 默认是and
  • test:selective下的判断表达式,即<if test="username != null">里的test属性

@Limit@OffSet为分页字段。 方法的参数不加任何注解一样会被当做查询条件,如下面两个函数效果是一样的:

@Select()
List<User> queryUser(Integer age);

@Select()
List<User> queryUser(@Condition(criterion = Criterions.EQUAL, column = "age") Integer age);
复制代码
查询Model

上面的例子在查询条件比较多时方法参数会比较多,我们可以把查询条件封装到一个类里,使用@ModelConditions来注解查询条件,注意被@ModelConditions只能有一个参数。

@Select
@Page
@ModelConditions({
       @ModelCondition(field = "username", criterion = Criterions.EQUAL),
       @ModelCondition(field = "minAge", column = "age", criterion = Criterions.GREATER),
       @ModelCondition(field = "maxAge", column = "age", criterion = Criterions.LESS),
       @ModelCondition(field = "ids", column = "id", criterion = Criterions.IN),
       @ModelCondition(field = "idArr", column = "id", criterion = Criterions.IN, paramType = InType.ARRAY)
})
List<User> queryUser5(UserSearch userSearch);
复制代码
@ModelCondition
  • field:必填,查询条件中类对应的属性
  • column:对应的表字段
  • paramType:in 查询下才需要配置,数组为array,List为collection类型
  • test:selective下的判断表达式,即<if test="username != null">里的test属性

@Page只能用在ModelConditions下的查询,并且方法参数的那个类应该有offsetlimit这两个属性。

注:

@Select(columns = "username,age")
List<User> queryUser(Integer age);

@Select(columns = "username,age")
List<User> queryUser2param(Integer age, String username);

<select id="queryUser" resultMap="XXX">select username,age from T_User
    <where>
      <if test="age != null">AND age = #{age}</if>
    </where>
</select>

<select id="queryUser2param" resultMap="XXX">select username,age from T_User
    <where>
      <if test="age != null">AND age = #{age}</if>
      <if test="username != null">AND username = #{username}</if>
    </where>
</select>
复制代码

两个函数生成的sql如上,@Select的属性SqlMode默认是Selective,所以两个都有条件判断,但是这里第一个函数的sql, Mybatis不支持,执行会报错,类似no age getter in java.lang.Interger,Mybatis会把这唯一的一个参数当做对象来取里面的值。 解决方法:函数签名里强加@Param()注解,或者@Select里使用sqlMode = SqlMode.COMMON去掉生成sql里的if判断。 这个问题只会在方法只有一个参数的情况下发生,第二个函数生成的sql是ok的。

分页

查询参数里@Limit@OffSet或查询model里@Page的分页功能都比较原始,TgDao只是一款SQL生成器而已,因此你可以使用各种插件, 或者与其他框架集成。对于分页,可以无缝与PageHelper整合。

@Select
List<User> queryUser2(@Condition(criterion = Criterions.GREATER, column = "age") int min,
                @Condition(criterion = Criterions.LESS, column = "age") int max);


@Test
public void testQueryUser2() throws Exception {
   PageHelper.offsetPage(1, 10);
   List<User> users = mapper.queryUser2(12, 30);
   PageInfo page = new PageInfo<>(users);
   System.out.println(page.getTotal());
   Assert.assertTrue(page.getList().size() > 0);
}
复制代码

插入

@Insert(useGeneratedKeys = true, keyProperty = "id")//获取自增id
int insert(User user);

@BatchInsert(columns = "username,age")//插入的列
int batchInsert(List<User> users);
复制代码

BatchInsert强烈建议写columns,因为生成的语句并不会过滤null字段,数据库中插入null易报错。


更新

@Update(columns = "username,age")//选择更新某几个列
@ModelConditions({
       @ModelCondition(field = "id")
})
int update(User user);
复制代码

删除

@Delete
int delete(@Condition(criterion = Criterions.GREATER, column = "age") int min,
          @Condition(criterion = Criterions.LESS, column = "age") int max);

@Delete
@ModelConditions({
       @ModelCondition(attach = Attach.AND, field = "minAge", column = "age", criterion = Criterions.GREATER),
       @ModelCondition(attach = Attach.AND, field = "maxAge", column = "age", criterion = Criterions.LESS)
})
int delete2(UserSearch userSearch);
复制代码

selective

@Select@Count@Update@Delete都有selective这个属性,这个属性有两个值,分别是SqlMode.COMMONSqlMode.SELECTIVE。 它们的区别在下面这段生成的xml里显示的很清楚,SqlMode.SELECTIVE引入了Mybatis的动态SQL能力。

<!-- SELECTIVE -->
  <select id="queryUser" resultMap="BaseResultMap">select username,age from T_User 
    <where>
      <if test="name!=null and name!=''">AND username = #{name}</if>
      <if test="age != null">OR age = #{age}</if>
    </where>
  </select>
  
  <!-- COMMON -->
  <select id="queryUser" resultMap="BaseResultMap">select username,age from T_User 
    <where>
      AND username = #{name} OR age = #{age}
    </where>
  </select>
复制代码

@Select@Count默认的selective属性是SqlMode.SELECTIVE,这样查询语句可以充分利用Mybatis的动态SQL能力。 而@Update@Delete默认是SqlMode.COMMON,这样做的原因是:selective模式下如果参数全是null会使得where语句里没有任何条件, 最终变成全表的更新和删除,这是一个极其危险的动作。所以@Update@Delete慎用SqlMode.SELECTIVE模式。

@Params

在介绍这个注解时要先介绍一下Mybatis自己的@Param注解,@Param注解在方法的参数上,给参数定义了一个名字, 这样可以在xml的sql里使用这个名字来取得参数所对应的值。如下:

List<User> queryUser(@Param("name") String name);
    
    <select id="queryUser">select * from T_User where username=#{name} </select>
复制代码

明明参数就叫name,为什么还要@Param注解一个名字name呢?这是因为Java编译完,会丢掉参数名,以至于运行期mybatis不知道这个参数叫什么,所以需要注解一个名字。 在运行时看到mybatis报错如:Parameter 'XXX' not found. Available parameters are... 这就是没有这个注解导致的问题。 但是在Java8里我们已经可以通过给javac 添加-parameters参数来保留参数名字信息,这样mybatis会利用这个信息,这样就不需要加@Param注解了。 maven可以通过如下方式设置:

<plugin>
      <groupId>org.apache.maven.plugins</groupId>
      <artifactId>maven-compiler-plugin</artifactId>
      <version>3.1</version>
      <configuration>
          <compilerArgs>
              <arg>-parameters</arg>
          </compilerArgs>
      </configuration>
  </plugin>
复制代码

然而有一种情况-parameters也无能为力,List<User> queryUser4(List ids);当参数是collection或者数组类型时,mybatis依旧无法认出ids这个参数,只认collectionarray。 而@Params注解是Mybatis自身注解@Param-parameters外的另外一种解决方案。@Params可以注解在类和方法上, 被它注解的类和方法会在编译期自动给所有方法参数加上@Param注解,它借鉴了lombok的方式在编译期修改抽象语法树从而改变编译生成的字节码文件。

@Select(columns = "username,age")
    @Params
    List<User> queryUser(Integer age, String username);
    
    //编译后
    List<User> queryUser(@Param("age") Integer var1, @Param("username") String var2);
复制代码

说明

  • 编译生成的XML文件与Mapper接口在同一个包下
  • 只支持Java8和MySql
  • 修改了源代码中方法的定义或者model里和数据表的映射关系,发现编译出来的xml却没有改变,这是增量编译的原因。生成一个xml同时需要model和mapper interface两个部分, 如果你只修改了其中一个的代码,那么另一个未修改的代码编译器就不做处理,这样这一次编译就无法得到全部的信息,所以TgDao无法生成最新版本的xml。 解决方法是每次mvn clean compile先清除一下编译目录,更好的方案正在寻找...