关于Mybatis自定义拦截器如何动态修改sql及执行结果(上)

本文分享了在MyBatis中使用拦截器动态修改SQL及处理结果集的方法。通过自定义拦截器,作者展示了如何在SQL执行前后对SQL语句进行动态修改,以及如何在SQL执行后对结果集进行二次处理。

最近一直在研究mybatis的拦截器,也算小有收获,和大家一起分享。

一、关于mybatis的定位

mybatis很大的一个功能就是解析mapper.xml文件,按照规则解析sql,并交由数据库驱动最终执行sql,然后对结果集进行处理

二、先说一下我们要做的事情:在sql执行前对sql进行动态修改,接收到结果后,对结果再次进行修改

于是有了如上图设计

三、那拦截器到底要如何设计才能做到修改sql呢,就要说到sqlSession的四大对象executor, statementHandler, parameterHandler,resultHandler。对于这四大对象有很多文章解析,这里我们只用到statementHandler和resultSetHandler

四、准备(为了测试,也为了过程中能够看到数据变化)

数据库表:

 实体类:

Mapper以及Mapper.xml 这里就不一一列举了

五、自定义拦截器

 注解里声明了要拦截的对象,分别是statementHandler和resultSetHandler

 在sql执行前拦截到statementHandler

public Object process(Invocation invocation) throws Throwable {

	StatementHandler statementHandler = (StatementHandler) invocation.getTarget();
	BaseStatementHandler delegate = (BaseStatementHandler) ReflectUtil.getFieldValue(statementHandler, "delegate");
	MappedStatement mappedStatement = (MappedStatement) ReflectUtil.getFieldValue(delegate, "mappedStatement");
	Object parameterObject = boundSql.getParameterObject();
	
	MetaObject metaMappedStatement = MetaObject.forObject(mappedStatement
			, new DefaultObjectFactory(), new DefaultObjectWrapperFactory(), new DefaultReflectorFactory());
	SqlNode sqlNode = (SqlNode) metaMappedStatement.getValue("sqlSource.rootSqlNode");
	//动态生成sql
	DynamicContext context = new DynamicContext(mappedStatement.getConfiguration(), boundSql.getParameterObject());
	sqlNode.apply(context);
	String contextSql = context.getSql();
	// 将实体类中的方法加载到
	Map<String, Method> methodMap = getMethodMap(parameterObject);

	modifySql(contextSql, parameterObject, key, methodMap);
	for (Map.Entry<String, Object> entry : context.getBindings().entrySet()) {
		boundSql.setAdditionalParameter(entry.getKey(), entry.getValue());
	}

	return invocation.proceed();
}

 在sql返回结果后拦截resultSetHandler

public Object process(Invocation invocation) throws Throwable {

	Object result = invocation.proceed();

	//如果结果集是一个List
	if (result instanceof ArrayList) {
		ArrayList resultList = (ArrayList) result;
		for (int i = 0; i < resultList.size(); i++) {
			//依次获取其中的对象
			Object object = resultList.get(i);
				//利用反射对结果集中的每个对象进行处理,
				handleObject(object);
				System.out.println(object);
			}
			resultList.set(i,object);
		}
		return resultList;
	}

	return result;
}

 这样,就可以对结果集进行处理

 

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值