分享一个自己写的关于mybatis的实用工具方法

今天和大家分享的是关于自己写的一个mybatis的实用小工具。

一、先来看一下问题

1、测试问题前的准备工作:

1）、准备数据。在数据库中创建一个person表，插入需要的数据，如下图：

2）、根据上面表，通过mybatis自动生成java代码，如下图：

2、需求：查询person表中离职的人员。

通过执行SQL语句查询应该是这样的语句（select * from person where enabled=0 or enabled is null），如下图：

1）、通过mybatis自动生成的Java代码去查询。

先看下自动生成的文件中，对enabled字段可以添加的条件。如下图：

从图中的方法中可以看到，只有一个andEnabledNotEqualTo(Boolean value)方法，接近需求的条件。添加这个条件，运行java代码，看下结果。如下图：

但从这个结果中可以看出，并没有实现需求。这个时候就会发现mybatis自动生成的Example.Criteria不能达到自己的需求。这里能实现的方法有很多，需要考虑下怎么实现这个需求比较简单了。

这里先看一个简单的实现方案，如下图：

从上图可以看出，通过ClassUtils.invokeCriteriaMethod(Object criteria,String[] str,String method,Integer paramQuantity),添加了自定义的条件（nabled=0 or enabled is null），就满足了需求。而且使用非常方便。下面就具体介绍这个公共方法。

二、工具方法（ClassUtils.invokeCriteriaMethod）

1、先看Example.Criteria 的具体添加条件的方法的代码。如下图：

从图中可以看出Criteria是继承了GeneratedCriteria,且所有添加条件的方式都是调用GeneratedCriteria的addCriterion方法。如果能直接执行addCriterion方法将需要的条件添加到对象中，那就能达到我们的需求了，但有一个问题，这个addCriterion方法的访问表示符是protected，不能直接访问，这个时候就需要用到java的反射了。通过反射就能执行addCriterion了。

2、ClassUtils.invokeCriteriaMethod详细介绍，代码如下图：

主要步骤如下：

1）、Example.Criteria的实例对象通过getClass()方法获取Criteria的类对象clazz。

2）、因为addCiteria方法在父类GeneratedCriteria中，通过clazz.getSuperclass() 获取GeneratedCriteria的类对象clazz。

3）、通过类对象的getDeclaredMethods()获取GeneratedCriteria的所有方法。

4）、对方法进行遍历，根据方法名和参数的个数来找到需要执行的方法。

5）、通过Methods.setAccessible(true)方法取消Java语言访问检查。

6）、通过Methods.invoke(criteria, str)，执行对应的方法。

这样就能很方便的添加自定义查询条件了。