文档章节

Mybatis工具类--在执行(批量)sql前 先判断参数是否为空

zgw06629
 zgw06629
发布于 2015/04/27 14:25
字数 825
阅读 514
收藏 0

如有这样的dao方法:

List<User> getUsersByIds(List<Long> idList) ;
void deleteUsersByIds(List<Long> idList) ;
void batchAddUsers(List<User> userList);

若参数为空的话,执行相关sql时,会报错,因是不完整的sql,如下所示:

select * from user where id in 
delete from user where id in 
insert into user(name,idcard,...) values

若想避免此一情况,可以在调用dao方法前进行判断。

if(idList!=null && !idList.isEmpty()){
    //call dao here
}

能否省去这种判断呢?若参数为空,不去执行sql不就行了吗。

刚开始想到用spring aop, 但是发觉不易实现,想拦截dao方法,但报错:

Could not generate CGLIB subclass of class [class com.sun.proxy.$Proxy27]: Common causes of this problem include using a final class or a non-visible class;

因dao均是接口,而其实现类本身是一个由mybatis实现的动态代理类,现在spring又要对该代理类进行代理,但mybatis生成的代理类又不能再被代理(理由见上面的错误提示)。

而对servcie层进行代理处理起来也不方便,因参数不如dao方法直观,可能List参数是封装在一个业务对象中,如BussinessDTO.并且使用Spring Aop 还不通用(即每个项目都要有自己的一套)。

最后想到了用Mybatis的拦截器,拦截其最后的执行sql类(org.apache.ibatis.executor.Executor的某个具体实现子类),判断输入参数是否为空,若为空,直接返回,不往下执行了。

@Intercepts({
@Signature(type = Executor.class, method = "query", args = {
MappedStatement.class, Object.class, RowBounds.class,
ResultHandler.class }),
@Signature(type = Executor.class, method = "update", args = {
MappedStatement.class, Object.class }) })
public class MyBatisCheckEmptyBeforeExecuteInterceptor implements Interceptor {
    //...
}

注解定义需要拦截的方法。mybatis最终的执行方法只有两个,查询调用query方法,增、删、改均调用update方法。

具体拦截方法:

Object parameter = invocation.getArgs()[1];
if (parameter == null) { // 参数值为null
    MappedStatement ms = (MappedStatement) invocation.getArgs()[0];
    Class parameterType = ms.getParameterMap().getType();
    if (parameterType != null)// 实际存在输入参数 即并不是无参方法 如getAll()
        return getDefaultReturnValue(invocation); //直接返回一个默认值(根据调用方法的返回类型) 如new ArrayList()
}
//若dao 方法的输入参数为List的话,mybatis会自动将其封转到一个map中 key为list(数组参数一样 但key为array)
if (parameter instanceof Map) {
    Map map = (Map) parameter;
    //list参数是否为空或数组是否为空 
    if ((map.containsKey("list") && CollectionUtils.isEmpty((List) map.get("list")))
        || (map.containsKey("array") && ArrayUtils.isEmpty((Object[]) map.get("array"))))
        return getDefaultReturnValue(invocation);
}
//参数非空 继续往下执行
return invocation.proceed();

完整代码:

import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import java.util.Properties;

import org.apache.commons.collections.CollectionUtils;
import org.apache.commons.lang3.ArrayUtils;
import org.apache.ibatis.executor.Executor;
import org.apache.ibatis.mapping.MappedStatement;
import org.apache.ibatis.plugin.Interceptor;
import org.apache.ibatis.plugin.Intercepts;
import org.apache.ibatis.plugin.Invocation;
import org.apache.ibatis.plugin.Plugin;
import org.apache.ibatis.plugin.Signature;
import org.apache.ibatis.session.ResultHandler;
import org.apache.ibatis.session.RowBounds;

/**
 * 执行sql前统一判空 如getByIds(List<Long> idList) 若输入参数为空 直接返回 不再执行sql
 * 
 * @author zhuguowei
 *
 */
@Intercepts({
		@Signature(type = Executor.class, method = "query", args = {
				MappedStatement.class, Object.class, RowBounds.class,
				ResultHandler.class }),
		@Signature(type = Executor.class, method = "update", args = {
				MappedStatement.class, Object.class }) })
public class MyBatisCheckEmptyBeforeExecuteInterceptor implements Interceptor {

	@SuppressWarnings("rawtypes")
	@Override
	public Object intercept(Invocation invocation) throws Throwable {
		Object parameter = invocation.getArgs()[1];

		if (parameter == null) { // 参数值为空
			MappedStatement ms = (MappedStatement) invocation.getArgs()[0];
			Class parameterType = ms.getParameterMap().getType();
			if (parameterType != null)// 存在输入参数
				return getDefaultReturnValue(invocation);
		}
		if (parameter instanceof Map) {
			Map map = (Map) parameter;
			if ((map.containsKey("list") && CollectionUtils.isEmpty((List) map
					.get("list")))
					|| (map.containsKey("array") && ArrayUtils
							.isEmpty((Object[]) map.get("array"))))
				return getDefaultReturnValue(invocation);
		}
		return invocation.proceed();
	}

	/**
	 * 得到默认返回值
	 * 
	 * @param invocation
	 * @return
	 */
	@SuppressWarnings("rawtypes")
	private Object getDefaultReturnValue(Invocation invocation) {
		Class returnType = invocation.getMethod().getReturnType();
		if (returnType.equals(List.class))
			return new ArrayList();
		else if (returnType.equals(Integer.TYPE))
			return 0;
		return null;
	}

	/**
	 * 只拦截Executor
	 */
	@Override
	public Object plugin(Object target) {
		if (target instanceof Executor) {
			return Plugin.wrap(target, this);
		} else {
			return target;
		}
	}

	@Override
	public void setProperties(Properties properties) {

	}

}


© 著作权归作者所有

共有 人打赏支持
zgw06629
粉丝 16
博文 54
码字总数 30471
作品 0
海淀
程序员
私信 提问
加载中

评论(1)

i
itxx2016
推荐国内最流行的iBatis、MyBatis代码生成网站: fwjava.com
网站在线生成,操作极其简单,生成的代码十分规范好用,经历过实战的主流代码啊.
现在,大多数知名的互联网公司都在用它.
史上最简单的 MyBatis 教程(四)

1 前言 在史上最简单的 MyBatis 教程(一、二、三)中,咱们已经初步体验了 MyBatis 框架的特性,尤其是其支持普通的 SQL 语句,但如果仔细阅读前三篇博文的示例,大家会发现一个问题,那就是...

qq_35246620
2017/02/03
0
0
BeetlSQL,简单和强大数据库访问工具(更新)

beetlsql 特点 BeetSql是一个全功能DAO工具, 同时具有Hibernate 优点 & Mybatis优点功能,适用于承认以SQL为中心,同时又需求工具能自动能生成大量常用的SQL的应用。 无需注解,自动生成大量...

闲大赋
2015/08/17
0
61
mybatis-plus sql注入原理

MP版本为2.3 注入主要靠com.baomidou.mybatisplus.mapper.AutoSqlInjector 来完成 ①,mybatis先进行sql语句的映射,然后mybatis-plus再添加 ②,我们在AutoSqlInjector .injectDeleteByIdSq...

暗中观察
07/29
0
0
Mybatis3.3.x技术内幕(四):五鼠闹东京之执行器Executor设计原本

在上一篇博文中,已经分析了Mybatis事务相关的内容,而今天的这篇博文,很多内容都是在方法method内部,与事务无关,所以,建议暂时忘记事务概念,避免搅扰。 Mybatis对数据库的操作,都将委...

祖大俊
2016/04/27
1K
2
mybatis/ibatis直接执行sql语句

在项目中,使用的是mybatis3.0.5,但没有采用其提供的DAO层接口映射的策略,而且在进行多种属性联合查找时,需要底层提供通用的解决方案,所以需要mybatis直接执行sql语句,各个daoImpl均可调...

lee123lee
2013/09/21
0
1

没有更多内容

加载失败,请刷新页面

加载更多

Accept和Content-type的意思

Accept意思是我希望接收到的数据类型 Content-type意思是我发出去的数据类型

大灰狼wow
9分钟前
0
0
Java每天10道面试题,跟我走,offer有!(五)

41.Iterator、ListIterator 和 Enumeration的区别?   迭代器是一种设计模式, 它是一个对象, 它可以遍历并选择序列中的对象, 而开发人员不需要了解 该序列的底层结构。 迭代器通常被称为...

Java干货分享
9分钟前
0
0
meta 解决页面浏览器兼容性

使用最高级的ie内核,如果支持谷歌内核,使用谷歌内核 <meta http-equiv="X-UA-Compatible" content="IE=edge,chrome=1" /> 这 样写可以达到的效果是如果安装了GCF,则使用GCF来渲染页面,如...

之渊
11分钟前
0
0
极验验证demo(django+vue)

在使用之前,曾经试过用阿里云的人机验证,不过在签名部分比较复杂,下载sdk后需要自己写很多,折腾了一下,还是放弃。而腾讯云的人机验证python版本有demo,直接填写keyhe1secret就可以使用...

xiaoge2016
12分钟前
0
0
浅谈js回调

js回调极为简洁,无需声明,直接通过参数传入方法实体,调用方法实体的时候,可以直接调用方法名或者方法名加参数即可,以下看例子 socket.initWebSocket(this, userName, userId, (isSucce...

Carbenson
17分钟前
1
0

没有更多内容

加载失败,请刷新页面

加载更多

返回顶部
顶部