今天和大家分享的是关于自己写的一个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),执行对应的方法。
这样就能很方便的添加自定义查询条件了。
领取专属 10元无门槛券
私享最新 技术干货