springboot使用mybatis插件動態修改sql

一、關於mybatis的定位

mybatis很大的一個功能就是解析mapper.xml文件，按照規則解析sql,並交由數據庫驅動最終執行sql,然后對結果集進行處理

二、先說一下我們要做的事情：在sql執行前對sql進行動態修改，接收到結果后，對結果再次進行修改

於是有了如上圖設計
那攔截器到底要如何設計才能做到修改sql呢，就要說到sqlSession的四大對象executor, statementHandler, parameterHandler，resultHandler。對於這四大對象有很多文章解析，這里我們只用到statementHandler和resultSetHandler

三、准備

實體類

四、自定義攔截

注解里聲明了要攔截的對象，分別是statementHandler和resultSetHandler

在sql執行前攔截到statementHandle

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;

這樣，就可以對結果集進行處理