springboot使用mybatis外掛動態修改sql
阿新 • • 發佈:2020-10-07
一、關於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;
這樣,就可以對結果集進行處理