文档章节

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

zgw06629
 zgw06629
发布于 2015/04/27 14:25
字数 825
阅读 537
收藏 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
mybatis-plus sql注入原理

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

暗中观察
2018/07/29
0
0
疑难杂症-MyBatis一级缓存引起的分页插件失效

症状:使用自定义MyBatis分页插件,只有分页参数不同的方法在短时间内使用不同分页参数查询出来的结果相同。 病因:自定义MyBatis插件拦截目标为StatementHandler,而在同一个SqlSession中,...

SawyerZhou
2018/12/06
0
0
mybatis/ibatis直接执行sql语句

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

lee123lee
2013/09/21
0
1
MyBatis如何防止SQL注入

MyBatis如何防止SQL注入 SQL注入是一种代码注入技术,用于攻击数据驱动的应用,恶意的SQL语句被插入到执行的实体字段中(例如,为了转储数据库内容给攻击者)。[摘自] SQL injection - Wikip...

优惠券发放
2018/05/02
0
0

没有更多内容

加载失败,请刷新页面

加载更多

大数据反欺诈技术架构

一年多以前,有朋友让我聊一下你们的大数据反欺诈架构是怎么实现的,以及我们途中踩了哪些坑,怎么做到从30min延迟优化到1s内完成实时反欺诈。当时呢第一是觉得不合适,第二也是觉得场景比较...

微笑向暖wx
14分钟前
0
0
flink-系统内部消息传递的exactly once语义

At Most once,At Least once和Exactly once 在分布式系统中,组成系统的各个计算机是独立的。这些计算机有可能fail。 一个sender发送一条message到receiver。根据receiver出现fail时sender如...

xtof
22分钟前
0
0
iOS程序执行顺序和UIViewController 的生命周期(整理)

说明:此文是自己的总结笔记,主要参考: iOS程序的启动执行顺序 AppDelegate 及 UIViewController 的生命周期 UIView的生命周期 言叶之庭.jpeg 一. iOS程序的启动执行顺序 程序启动顺序图 iO...

壹峰
23分钟前
0
0
配置网络、远程登录、Linux秘钥认证

配置网络 一台服务器安装完系统之后不管是为了方便管理还是业务需要,我们都要给它配置ip地址。让机器能够联网。在现实的生产环境的当中,往往我们给服务器配置的ip都是提前规划好的,但是在...

李超小牛子
27分钟前
0
0
dotConnect for Oracle入门指南(五):检索和修改数据

【下载dotConnect for Oracle最新版本】 dotConnect for Oracle(原名OraDirect.NET)建立在ADO.NET技术上,为基于Oracle数据库的应用程序提供完整的解决方案。它为设计应用程序结构带来了新的...

电池盒
27分钟前
0
0

没有更多内容

加载失败,请刷新页面

加载更多

返回顶部
顶部