場景：

系統模組查詢資料庫需要根據使用者的id去篩選資料。那麼如果在 每個sql加user_id的過濾顯然不明確。所以要在查詢前將sql拼接上條件，做統一管理。

開發環境：

spring boot + mybatis

只需一個攔截類即可搞定（在看程式碼前需要了解註解@Intercepts()）：

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

  private Logger logger = LoggerFactory.getLogger(SqlInterceptor.class);

  @Override
  public Object intercept(Invocation invocation) throws Throwable {
 logger.info("Interceptor......");
//    獲取sql
 MappedStatement mappedStatement = (MappedStatement) invocation.getArgs()[0];
 Object parameter = invocation.getArgs()[1];
 BoundSql boundSql = mappedStatement.getBoundSql(parameter);
 String oldsql = boundSql.getSql();
 logger.info("old:"+oldsql);

//    判斷sql是否有where條件。改變sql。
 boolean status = oldsql.toLowerCase().contains("where");

 if (status) {
   BoundSql newBoundSql = new BoundSql(mappedStatement.getConfiguration(),oldsql + " and 1=1",boundSql.getParameterMappings(),boundSql.getParameterObject());
   MappedStatement newMs = copyFromMappedStatement(mappedStatement,new BoundSqlSqlSource(newBoundSql));
   invocation.getArgs()[0] = newMs;
 }
// 繼續執行
 Object result = invocation.proceed();
 return result;
  }

  @Override
  public Object plugin(Object target) {
 return Plugin.wrap(target,this);
  }

  @Override
  public void setProperties(Properties properties) {

  }

  // 複製原始MappedStatement
  private MappedStatement copyFromMappedStatement(MappedStatement ms,SqlSource newSqlSource) {
 MappedStatement.Builder builder = new MappedStatement.Builder(ms.getConfiguration(),ms.getId(),newSqlSource,ms.getSqlCommandType());
 builder.resource(ms.getResource());
 builder.fetchSize(ms.getFetchSize());
 builder.statementType(ms.getStatementType());
 builder.keyGenerator(ms.getKeyGenerator());
 if (ms.getKeyProperties() != null) {
   for (String keyProperty : ms.getKeyProperties()) {
     builder.keyProperty(keyProperty);
   }
 }
 builder.timeout(ms.getTimeout());
 builder.parameterMap(ms.getParameterMap());
 builder.resultMaps(ms.getResultMaps());
 builder.cache(ms.getCache());
 builder.useCache(ms.isUseCache());
 return builder.build();
  }

  public static class BoundSqlSqlSource implements SqlSource {
 BoundSql boundSql;

 public BoundSqlSqlSource(BoundSql boundSql) {
   this.boundSql = boundSql;
 }

 public BoundSql getBoundSql(Object parameterObject) {
   return boundSql;
 }
  }

}

這樣重啟訪問即可發現每次查詢都會走這邊！

到此這篇關於SpringBoot整合MybatisSQL過濾@Intercepts的實現的文章就介紹到這了,更多相關SpringBoot MybatisSQL過濾@Intercepts內容請搜尋我們以前的文章或繼續瀏覽下面的相關文章希望大家以後多多支援我們！