日韩无码专区无码一级三级片|91人人爱网站中日韩无码电影|厨房大战丰满熟妇|AV高清无码在线免费观看|另类AV日韩少妇熟女|中文日本大黄一级黄色片|色情在线视频免费|亚洲成人特黄a片|黄片wwwav色图欧美|欧亚乱色一区二区三区

RELATEED CONSULTING
相關(guān)咨詢
選擇下列產(chǎn)品馬上在線溝通
服務(wù)時間：8:30-17:00
你可能遇到了下面的問題
關(guān)閉右側(cè)工具欄

新聞中心

這里有您想知道的互聯(lián)網(wǎng)營銷解決方案
手把手教你開發(fā)MyBatis插件

 小伙伴們元宵節(jié)快樂，記得吃元宵哦～

在日常開發(fā)中，小伙伴們多多少少都有用過 MyBatis 插件，松哥猜測大家用的最多的就是 MyBatis 的分頁插件!不知道小伙伴們有沒有想過有一天自己也來開發(fā)一個 MyBatis 插件?

其實自己動手擼一個 MyBatis 插件并不難，今天松哥就把手帶大家擼一個 MyBatis 插件!

1.MyBatis 插件接口

即使你沒開發(fā)過 MyBatis 插件，估計也能猜出來，MyBatis 插件是通過攔截器來起作用的，MyBatis 框架在設(shè)計的時候，就已經(jīng)為插件的開發(fā)預(yù)留了相關(guān)接口，如下：

 
 
 
 
    
  
  
  
  
  1. public interface Interceptor {
    2. 
  
  
  
  
  2. 
  
  
  
  
  3.   Object intercept(Invocation invocation) throws Throwable;
    4. 
  
  
  
  
  4. 
  
  
  
  
  5.   default Object plugin(Object target) {
    6. 
  
  
  
  
  6.     return Plugin.wrap(target, this);
    7. 
  
  
  
  
  7.   }
    8. 
  
  
  
  
  8. 
  
  
  
  
  9.   default void setProperties(Properties properties) {
    10. 
  
  
  
  
  10.     // NOP
    11. 
  
  
  
  
  11.   }
    12. 
  
  
  
  
  12. 
  
  
  
  
  13. }
    14.

這個接口中就三個方法，第一個方法必須實現(xiàn)，后面兩個方法都是可選的。三個方法作用分別如下：

  1. intercept：這個就是具體的攔截方法，我們自定義 MyBatis 插件時，一般都需要重寫該方法，我們插件所完成的工作也都是在該方法中完成的。
  2. plugin：這個方法的參數(shù) target 就是攔截器要攔截的對象，一般來說我們不需要重寫該方法。Plugin.wrap 方法會自動判斷攔截器的簽名和被攔截對象的接口是否匹配，如果匹配，才會通過動態(tài)代理攔截目標對象。
  3. setProperties：這個方法用來傳遞插件的參數(shù)，可以通過參數(shù)來改變插件的行為。我們定義好插件之后，需要對插件進行配置，在配置的時候，可以給插件設(shè)置相關(guān)屬性，設(shè)置的屬性可以通過該方法獲取到。插件屬性設(shè)置像下面這樣：
 
 
 
 
    
  
  
  
  
    2. 
  
  
  
  
  2.     
    3. 
  
  
  
  
  3.         
    4. 
  
  
  
  
  4.     
    5. 
  
  
  
  
  5.

2.MyBatis 攔截器簽名

攔截器定義好了后，攔截誰?

這個就需要攔截器簽名來完成了!

攔截器簽名是一個名為 @Intercepts 的注解，該注解中可以通過 @Signature 配置多個簽名。@Signature 注解中則包含三個屬性：

  • type: 攔截器需要攔截的接口，有 4 個可選項，分別是：Executor、ParameterHandler、ResultSetHandler 以及 StatementHandler。
  • method: 攔截器所攔截接口中的方法名，也就是前面四個接口中的方法名，接口和方法要對應(yīng)上。
  • args: 攔截器所攔截方法的參數(shù)類型，通過方法名和參數(shù)類型可以鎖定唯一一個方法。

一個簡單的簽名可能像下面這樣：

 
 
 
 
    
  
  
  
  
  1. @Intercepts(@Signature(
    2. 
  
  
  
  
  2.         type = ResultSetHandler.class,
    3. 
  
  
  
  
  3.         method = "handleResultSets",
    4. 
  
  
  
  
  4.         args = {Statement.class}
    5. 
  
  
  
  
  5. ))
    6. 
  
  
  
  
  6. public class CamelInterceptor implements Interceptor {
    7. 
  
  
  
  
  7.     //...
    8. 
  
  
  
  
  8. }
    9.

3.被攔截的對象

根據(jù)前面的介紹，被攔截的對象主要有如下四個：

Executor

 
 
 
 
    
  
  
  
  
  1. public interface Executor {
    2. 
  
  
  
  
  2. 
  
  
  
  
  3.   ResultHandler NO_RESULT_HANDLER = null;
    4. 
  
  
  
  
  4. 
  
  
  
  
  5.   int update(MappedStatement ms, Object parameter) throws SQLException;
    6. 
  
  
  
  
  6. 
  
  
  
  
  7.    List query(MappedStatement ms, Object parameter, RowBounds rowBounds, ResultHandler resultHandler, CacheKey cacheKey, BoundSql boundSql) throws SQLException;
    8. 
  
  
  
  
  8. 
  
  
  
  
  9.    List query(MappedStatement ms, Object parameter, RowBounds rowBounds, ResultHandler resultHandler) throws SQLException;
    10. 
  
  
  
  
  10. 
  
  
  
  
  11.    Cursor queryCursor(MappedStatement ms, Object parameter, RowBounds rowBounds) throws SQLException;
    12. 
  
  
  
  
  12. 
  
  
  
  
  13.   List flushStatements() throws SQLException;
    14. 
  
  
  
  
  14. 
  
  
  
  
  15.   void commit(boolean required) throws SQLException;
    16. 
  
  
  
  
  16. 
  
  
  
  
  17.   void rollback(boolean required) throws SQLException;
    18. 
  
  
  
  
  18. 
  
  
  
  
  19.   CacheKey createCacheKey(MappedStatement ms, Object parameterObject, RowBounds rowBounds, BoundSql boundSql);
    20. 
  
  
  
  
  20. 
  
  
  
  
  21.   boolean isCached(MappedStatement ms, CacheKey key);
    22. 
  
  
  
  
  22. 
  
  
  
  
  23.   void clearLocalCache();
    24. 
  
  
  
  
  24. 
  
  
  
  
  25.   void deferLoad(MappedStatement ms, MetaObject resultObject, String property, CacheKey key, Class targetType);
    26. 
  
  
  
  
  26. 
  
  
  
  
  27.   Transaction getTransaction();
    28. 
  
  
  
  
  28. 
  
  
  
  
  29.   void close(boolean forceRollback);
    30. 
  
  
  
  
  30. 
  
  
  
  
  31.   boolean isClosed();
    32. 
  
  
  
  
  32. 
  
  
  
  
  33.   void setExecutorWrapper(Executor executor);
    34. 
  
  
  
  
  34. 
  
  
  
  
  35. }
    36.

各方法含義分別如下：

  • update：該方法會在所有的 INSERT、 UPDATE、 DELETE 執(zhí)行時被調(diào)用，如果想要攔截這些操作，可以通過該方法實現(xiàn)。
  • query：該方法會在 SELECT 查詢方法執(zhí)行時被調(diào)用，方法參數(shù)攜帶了很多有用的信息，如果需要獲取，可以通過該方法實現(xiàn)。
  • queryCursor：當 SELECT 的返回類型是 Cursor 時，該方法會被調(diào)用。
  • flushStatements：當 SqlSession 方法調(diào)用 flushStatements 方法或執(zhí)行的接口方法中帶有 @Flush 注解時該方法會被觸發(fā)。
  • commit：當 SqlSession 方法調(diào)用 commit 方法時該方法會被觸發(fā)。
  • rollback：當 SqlSession 方法調(diào)用 rollback 方法時該方法會被觸發(fā)。
  • getTransaction：當 SqlSession 方法獲取數(shù)據(jù)庫連接時該方法會被觸發(fā)。
  • close：該方法在懶加載獲取新的 Executor 后會被觸發(fā)。
  • isClosed：該方法在懶加載執(zhí)行查詢前會被觸發(fā)。

ParameterHandler

 
 
 
 
    
  
  
  
  
  1. public interface ParameterHandler {
    2. 
  
  
  
  
  2. 
  
  
  
  
  3.   Object getParameterObject();
    4. 
  
  
  
  
  4. 
  
  
  
  
  5.   void setParameters(PreparedStatement ps) throws SQLException;
    6. 
  
  
  
  
  6. 
  
  
  
  
  7. }
    8.

各方法含義分別如下：

  • getParameterObject：在執(zhí)行存儲過程處理出參的時候該方法會被觸發(fā)。
  • setParameters：設(shè)置 SQL 參數(shù)時該方法會被觸發(fā)。

ResultSetHandler

 
 
 
 
    
  
  
  
  
  1. public interface ResultSetHandler {
    2. 
  
  
  
  
  2. 
  
  
  
  
  3.    List handleResultSets(Statement stmt) throws SQLException;
    4. 
  
  
  
  
  4. 
  
  
  
  
  5.    Cursor handleCursorResultSets(Statement stmt) throws SQLException;
    6. 
  
  
  
  
  6. 
  
  
  
  
  7.   void handleOutputParameters(CallableStatement cs) throws SQLException;
    8. 
  
  
  
  
  8. 
  
  
  
  
  9. }
    10.

各方法含義分別如下：

  • handleResultSets：該方法會在所有的查詢方法中被觸發(fā)(除去返回值類型為 Cursor的查詢方法)，一般來說，如果我們想對查詢結(jié)果進行二次處理，可以通過攔截該方法實現(xiàn)。
  • handleCursorResultSets：當查詢方法的返回值類型為 Cursor時，該方法會被觸發(fā)。
  • handleOutputParameters：使用存儲過程處理出參的時候該方法會被調(diào)用。

StatementHandler

 
 
 
 
    
  
  
  
  
  1. public interface StatementHandler {
    2. 
  
  
  
  
  2. 
  
  
  
  
  3.   Statement prepare(Connection connection, Integer transactionTimeout)
    4. 
  
  
  
  
  4.       throws SQLException;
    5. 
  
  
  
  
  5. 
  
  
  
  
  6.   void parameterize(Statement statement)
    7. 
  
  
  
  
  7.       throws SQLException;
    8. 
  
  
  
  
  8. 
  
  
  
  
  9.   void batch(Statement statement)
    10. 
  
  
  
  
  10.       throws SQLException;
    11. 
  
  
  
  
  11. 
  
  
  
  
  12.   int update(Statement statement)
    13. 
  
  
  
  
  13.       throws SQLException;
    14. 
  
  
  
  
  14. 
  
  
  
  
  15.    List query(Statement statement, ResultHandler resultHandler)
    16. 
  
  
  
  
  16.       throws SQLException;
    17. 
  
  
  
  
  17. 
  
  
  
  
  18.    Cursor queryCursor(Statement statement)
    19. 
  
  
  
  
  19.       throws SQLException;
    20. 
  
  
  
  
  20. 
  
  
  
  
  21.   BoundSql getBoundSql();
    22. 
  
  
  
  
  22. 
  
  
  
  
  23.   ParameterHandler getParameterHandler();
    24. 
  
  
  
  
  24. 
  
  
  
  
  25. }
    26.

各方法含義分別如下：

  • prepare：該方法在數(shù)據(jù)庫執(zhí)行前被觸發(fā)。
  • parameterize：該方法在 prepare 方法之后執(zhí)行，用來處理參數(shù)信息。
  • batch：如果 MyBatis 的全劇配置中配置了 defaultExecutorType=”BATCH”，執(zhí)行數(shù)據(jù)操作時該方法會被調(diào)用。
  • update：更新操作時該方法會被觸發(fā)。
  • query：該方法在 SELECT 方法執(zhí)行時會被觸發(fā)。
  • queryCursor：該方法在 SELECT 方法執(zhí)行時，并且返回值為 Cursor 時會被觸發(fā)。

在開發(fā)一個具體的插件時，我們應(yīng)當根據(jù)自己的需求來決定到底攔截哪個方法。

4.開發(fā)分頁插件

4.1 內(nèi)存分頁

MyBatis 中提供了一個不太好用的內(nèi)存分頁功能，就是一次性把所有數(shù)據(jù)都查詢出來，然后在內(nèi)存中進行分頁處理，這種分頁方式效率很低，基本上沒啥用，但是如果我們想要自定義分頁插件，就需要對這種分頁方式有一個簡單了解。

內(nèi)存分頁的使用方式如下，首先在 Mapper 中添加 RowBounds 參數(shù)，如下：

 
 
 
 
    
  
  
  
  
  1. public interface UserMapper {
    2. 
  
  
  
  
  2.     List getAllUsersByPage(RowBounds rowBounds);
    3. 
  
  
  
  
  3. }
    4.

然后在 XML 文件中定義相關(guān) SQL：

 
 
 
 
    
  
  
  
  
    2. 
  
  
  
  
  2.     select * from user
    3. 
  
  
  
  
  3.

可以看到，在 SQL 定義時，壓根不用管分頁的事情，MyBatis 會查詢到所有的數(shù)據(jù)，然后在內(nèi)存中進行分頁處理。

Mapper 中方法的調(diào)用方式如下：

 
 
 
 
    
  
  
  
  
  1. @Test
    2. 
  
  
  
  
  2. public void test3() {
    3. 
  
  
  
  
  3.     UserMapper userMapper = sqlSessionFactory.openSession().getMapper(UserMapper.class);
    4. 
  
  
  
  
  4.     RowBounds rowBounds = new RowBounds(1,2);
    5. 
  
  
  
  
  5.     List list = userMapper.getAllUsersByPage(rowBounds);
    6. 
  
  
  
  
  6.     for (User user : list) {
    7. 
  
  
  
  
  7.         System.out.println("user = " + user);
    8. 
  
  
  
  
  8.     }
    9. 
  
  
  
  
  9. }
    10.

構(gòu)建 RowBounds 時傳入兩個參數(shù)，分別是 offset 和 limit，對應(yīng)分頁 SQL 中的兩個參數(shù)。也可以通過 RowBounds.DEFAULT 的方式構(gòu)建一個 RowBounds 實例，這種方式構(gòu)建出來的 RowBounds 實例，offset 為 0，limit 則為 Integer.MAX_VALUE，也就相當于不分頁。

這就是 MyBatis 中提供的一個很不實用的內(nèi)存分頁功能。

了解了 MyBatis 自帶的內(nèi)存分頁之后，接下來我們就可以來看看如何自定義分頁插件了。

4.2 自定義分頁插件

首先要聲明一下，這里松哥帶大家自定義 MyBatis 分頁插件，主要是想通過這個東西讓小伙伴們了解自定義 MyBatis 插件的一些條條框框，了解整個自定義插件的流程，分頁插件并不是我們的目的，自定義分頁插件只是為了讓大家的學習過程變得有趣一些而已。

接下來我們就來開啟自定義分頁插件之旅。

首先我們需要自定義一個 RowBounds，因為 MyBatis 原生的 RowBounds 是內(nèi)存分頁，并且沒有辦法獲取到總記錄數(shù)(一般分頁查詢的時候我們還需要獲取到總記錄數(shù))，所以我們自定義 PageRowBounds，對原生的 RowBounds 功能進行增強，如下：

 
 
 
 
    
  
  
  
  
  1. public class PageRowBounds extends RowBounds {
    2. 
  
  
  
  
  2.     private Long total;
    3. 
  
  
  
  
  3. 
  
  
  
  
  4.     public PageRowBounds(int offset, int limit) {
    5. 
  
  
  
  
  5.         super(offset, limit);
    6. 
  
  
  
  
  6.     }
    7. 
  
  
  
  
  7. 
  
  
  
  
  8.     public PageRowBounds() {
    9. 
  
  
  
  
  9.     }
    10. 
  
  
  
  
  10. 
  
  
  
  
  11.     public Long getTotal() {
    12. 
  
  
  
  
  12.         return total;
    13. 
  
  
  
  
  13.     }
    14. 
  
  
  
  
  14. 
  
  
  
  
  15.     public void setTotal(Long total) {
    16. 
  
  
  
  
  16.         this.total = total;
    17. 
  
  
  
  
  17.     }
    18. 
  
  
  
  
  18. }
    19.

可以看到，我們自定義的 PageRowBounds 中增加了 total 字段，用來保存查詢的總記錄數(shù)。

接下來我們自定義攔截器 PageInterceptor，如下：

 
 
 
 
    
  
  
  
  
  1. @Intercepts(@Signature(
    2. 
  
  
  
  
  2.         type = Executor.class,
    3. 
  
  
  
  
  3.         method = "query",
    4. 
  
  
  
  
  4.         args = {MappedStatement.class, Object.class, RowBounds.class, ResultHandler.class}
    5. 
  
  
  
  
  5. ))
    6. 
  
  
  
  
  6. public class PageInterceptor implements Interceptor {
    7. 
  
  
  
  
  7.     @Override
    8. 
  
  
  
  
  8.     public Object intercept(Invocation invocation) throws Throwable {
    9. 
  
  
  
  
  9.         Object[] args = invocation.getArgs();
    10. 
  
  
  
  
  10.         MappedStatement ms = (MappedStatement) args[0];
    11. 
  
  
  
  
  11.         Object parameterObject = args[1];
    12. 
  
  
  
  
  12.         RowBounds rowBounds = (RowBounds) args[2];
    13. 
  
  
  
  
  13.         if (rowBounds != RowBounds.DEFAULT) {
    14. 
  
  
  
  
  14.             Executor executor = (Executor) invocation.getTarget();
    15. 
  
  
  
  
  15.             BoundSql boundSql = ms.getBoundSql(parameterObject);
    16. 
  
  
  
  
  16.             Field additionalParametersField = BoundSql.class.getDeclaredField("additionalParameters");
    17. 
  
  
  
  
  17.             additionalParametersField.setAccessible(true);
    18. 
  
  
  
  
  18.             Map additionalParameters = (Map) additionalParametersField.get(boundSql);
    19. 
  
  
  
  
  19.             if (rowBounds instanceof PageRowBounds) {
    20. 
  
  
  
  
  20.                 MappedStatement countMs = newMappedStatement(ms, Long.class);
    21. 
  
  
  
  
  21.                 CacheKey countKey = executor.createCacheKey(countMs, parameterObject, RowBounds.DEFAULT, boundSql);
    22. 
  
  
  
  
  22.                 String countSql = "select count(*) from (" + boundSql.getSql() + ") temp";
    23. 
  
  
  
  
  23.                 BoundSql countBoundSql = new BoundSql(ms.getConfiguration(), countSql, boundSql.getParameterMappings(), parameterObject);
    24. 
  
  
  
  
  24.                 Set keySet = additionalParameters.keySet();
    25. 
  
  
  
  
  25.                 for (String key : keySet) {
    26. 
  
  
  
  
  26.                     countBoundSql.setAdditionalParameter(key, additionalParameters.get(key));
    27. 
  
  
  
  
  27.                 }
    28. 
  
  
  
  
  28.                 List countQueryResult = executor.query(countMs, parameterObject, RowBounds.DEFAULT, (ResultHandler) args[3], countKey, countBoundSql);
  
  
  
  
  29.                 Long count = (Long) countQueryResult.get(0);
    30. 
  
  
  
  
  30.                 ((PageRowBounds) rowBounds).setTotal(count);
    31. 
  
  
  
  
  31.             }
    32. 
  
  
  
  
  32.             CacheKey pageKey = executor.createCacheKey(ms, parameterObject, rowBounds, boundSql);
    33. 
  
  
  
  
  33.             pageKey.update("RowBounds");
    34. 
  
  
  
  
  34.             String pageSql = boundSql.getSql() + " limit " + rowBounds.getOffset() + "," + rowBounds.getLimit();
    35. 
  
  
  
  
  35.             BoundSql pageBoundSql = new BoundSql(ms.getConfiguration(), pageSql, boundSql.getParameterMappings(), parameterObject);
    36. 
  
  
  
  
  36.             Set keySet = additionalParameters.keySet();
    37. 
  
  
  
  
  37.             for (String key : keySet) {
    38. 
  
  
  
  
  38.                 pageBoundSql.setAdditionalParameter(key, additionalParameters.get(key));
    39. 
  
  
  
  
  39.             }
    40. 
  
  
  
  
  40.             List list = executor.query(ms, parameterObject, RowBounds.DEFAULT, (ResultHandler) args[3], pageKey, pageBoundSql);
    41. 
  
  
  
  
  41.             return list;
    42. 
  
  
  
  
  42.         }
    43. 
  
  
  
  
  43.         //不需要分頁，直接返回結(jié)果
    44. 
  
  
  
  
  44.         return invocation.proceed();
    45. 
  
  
  
  
  45.     }
    46. 
  
  
  
  
  46. 
  
  
  
  
  47.     private MappedStatement newMappedStatement(MappedStatement ms, Class longClass) {
    48. 
  
  
  
  
  48.         MappedStatement.Builder builder = new MappedStatement.Builder(
    49. 
  
  
  
  
  49.                 ms.getConfiguration(), ms.getId() + "_count", ms.getSqlSource(), ms.getSqlCommandType()
    50. 
  
  
  
  
  50.         );
    51. 
  
  
  
  
  51.         ResultMap resultMap = new ResultMap.Builder(ms.getConfiguration(), ms.getId(), longClass, new ArrayList<>(0)).build();
    52. 
  
  
  
  
  52.         builder.resource(ms.getResource())
    53. 
  
  
  
  
  53.                 .fetchSize(ms.getFetchSize())
    54. 
  
  
  
  
  54.                 .statementType(ms.getStatementType())
    55. 
  
  
  
  
  55.                 .timeout(ms.getTimeout())
    56. 
  
  
  
  
  56.                 .parameterMap(ms.getParameterMap())
    57. 
  
  
  
  
  57.                 .resultSetType(ms.getResultSetType())
    58. 
  
  
  
  
  58.                 .cache(ms.getCache())
    59. 
  
  
  
  
  59.                 .flushCacheRequired(ms.isFlushCacheRequired())
    60. 
  
  
  
  
  60.                 .useCache(ms.isUseCache())
    61. 
  
  
  
  
  61.                 .resultMaps(Arrays.asList(resultMap));
    62. 
  
  
  
  
  62.         if (ms.getKeyProperties() != null && ms.getKeyProperties().length > 0) {
    63. 
  
  
  
  
  63.             StringBuilder keyProperties = new StringBuilder();
    64. 
  
  
  
  
  64.             for (String keyProperty : ms.getKeyProperties()) {
    65. 
  
  
  
  
  65.                 keyProperties.append(keyProperty).append(",");
    66. 
  
  
  
  
  66.             }
    67. 
  
  
  
  
  67.             keyProperties.delete(keyProperties.length() - 1, keyProperties.length());
    68. 
  
  
  
  
  68.             builder.keyProperty(keyProperties.toString());
    69. 
  
  
  
  
  69.         }
    70. 
  
  
  
  
  70.         return builder.build();
    71. 
  
  
  
  
  71.     }
    72. 
  
  
  
  
  72. }
    73. 
 
 
 
 
    這是我們今天定義的核心代碼，涉及到的知識點松哥來給大家一個一個剖析。
    

      

    1. 首先通過 @Intercepts 注解配置攔截器簽名，從 @Signature 的定義中我們可以看到，攔截的是 Executor#query 方法，該方法有一個重載方法，通過 args 指定了方法參數(shù)，進而鎖定了重載方法(實際上該方法的另一個重載方法我們沒法攔截，那個是 MyBatis 內(nèi)部調(diào)用的，這里不做討論)。
      2. 

    2. 將查詢操作攔截下來之后，接下來我們的操作主要在 PageInterceptor#intercept 方法中完成，該方法的參數(shù)重包含了攔截對象的諸多信息。
      3. 

    3. 通過 invocation.getArgs() 獲取攔截方法的參數(shù)，獲取到的是一個數(shù)組，正常來說這個數(shù)組的長度為 4。數(shù)組第一項是一個 MappedStatement，我們在 Mapper.xml 中定義的各種操作節(jié)點和 SQL，都被封裝成一個個的 MappedStatement 對象了;數(shù)組第二項就是所攔截方法的具體參數(shù)，也就是你在 Mapper 接口中定義的方法參數(shù);數(shù)組的第三項是一個 RowBounds 對象，我們在 Mapper 接口中定義方法時不一定使用了 RowBounds 對象，如果我們沒有定義 RowBounds 對象，系統(tǒng)會給我們提供一個默認的 RowBounds.DEFAULT;數(shù)組第四項則是一個處理返回值的 ResultHandler。
      4. 

    4. 接下來判斷上一步提取到的 rowBounds 對象是否不為 RowBounds.DEFAULT，如果為 RowBounds.DEFAULT，說明用戶不想分頁;如果不為 RowBounds.DEFAULT，則說明用戶想要分頁，如果用戶不想分頁，則直接執(zhí)行最后的 return invocation.proceed();，讓方法繼續(xù)往下走就行了。
      5. 

    5. 如果需要進行分頁，則先從 invocation 對象中取出執(zhí)行器 Executor、BoundSql 以及通過反射拿出來 BoundSql 中保存的額外參數(shù)(如果我們使用了動態(tài) SQL，可能會存在該參數(shù))。BoundSql 中封裝了我們執(zhí)行的 Sql 以及相關(guān)的參數(shù)。
      6. 

    6. 接下來判斷 rowBounds 是否是 PageRowBounds 的實例，如果是，說明除了分頁查詢，還想要查詢總記錄數(shù)，如果不是，則說明 rowBounds 可能是 RowBounds 實例，此時只要分頁即可，不用查詢總記錄數(shù)。
      7. 

    7. 如果需要查詢總記錄數(shù)，則首先調(diào)用 newMappedStatement 方法構(gòu)造出一個新的 MappedStatement 對象出來，這個新的 MappedStatement 對象的返回值是 Long 類型的。然后分別創(chuàng)建查詢的 CacheKey、拼接查詢的 countSql，再根據(jù) countSql 構(gòu)建出 countBoundSql，再將額外參數(shù)添加進 countBoundSql 中。最后通過 executor.query 方法完成查詢操作，并將查詢結(jié)果賦值給 PageRowBounds 中的 total 屬性。
      8. 

    8. 接下來進行分頁查詢，有了第七步的介紹之后，分頁查詢就很簡單了，這里就不細說了，唯一需要強調(diào)的是，當我們啟動了這個分頁插件之后，MyBatis 原生的 RowBounds 內(nèi)存分頁會變成物理分頁，原因就在這里我們修改了查詢 SQL。
      9. 

    9. 最后將查詢結(jié)果返回。
      10. 

    

    在前面的代碼中，我們一共在兩個地方重新組織了 SQL，一個是查詢總記錄數(shù)的時候，另一個則是分頁的時候，都是通過 boundSql.getSql() 獲取到 Mapper.xml 中的 SQL 然后進行改裝，有的小伙伴在 Mapper.xml 中寫 SQL 的時候不注意，結(jié)尾可能加上了 ;，這會導致分頁插件重新組裝的 SQL 運行出錯，這點需要注意。松哥在 GitHub 上看到的其他 MyBatis 分頁插件也是一樣的，Mapper.xml 中 SQL 結(jié)尾不能有 ;。
    如此之后，我們的分頁插件就算是定義成功了。
    5.測試
    接下來我們對我們的分頁插件進行一個簡單測試。
    首先我們需要在全局配置中配置分頁插件，配置方式如下：
     
 
 
 
      
  
  
  
  
      2. 
  
  
  
  
    2.     
      3. 
  
  
  
  
    3. 
 
 
 
 
    接下來我們在 Mapper 中定義查詢接口：
     
 
 
 
      
  
  
  
  
    1. public interface UserMapper {
      2. 
  
  
  
  
    2.     List getAllUsersByPage(RowBounds rowBounds);
      3. 
  
  
  
  
    3. }
      4. 
 
 
 
 
    接下來定義 UserMapper.xml，如下：
     
 
 
 
      
  
  
  
  
      2. 
  
  
  
  
    2.     select * from user
      3. 
  
  
  
  
    3. 
 
 
 
 
    最后我們進行測試：
     
 
 
 
      
  
  
  
  
    1. @Test
      2. 
  
  
  
  
    2. public void test3() {
      3. 
  
  
  
  
    3.     UserMapper userMapper = sqlSessionFactory.openSession().getMapper(UserMapper.class);
      4. 
  
  
  
  
    4.     List list = userMapper.getAllUsersByPage(new RowBounds(1,2));
      5. 
  
  
  
  
    5.     for (User user : list) {
      6. 
  
  
  
  
    6.         System.out.println("user = " + user);
      7. 
  
  
  
  
    7.     }
      8. 
  
  
  
  
    8. }
      9. 
 
 
 
 
    這里在查詢時，我們使用了 RowBounds 對象，就只會進行分頁，而不會統(tǒng)計總記錄數(shù)。需要注意的時，此時的分頁已經(jīng)不是內(nèi)存分頁，而是物理分頁了，這點我們從打印出來的 SQL 中也能看到，如下：
    可以看到，查詢的時候就已經(jīng)進行了分頁了。
    當然，我們也可以使用 PageRowBounds 進行測試，如下：
     
 
 
 
      
  
  
  
  
    1. @Test
      2. 
  
  
  
  
    2. public void test4() {
      3. 
  
  
  
  
    3.     UserMapper userMapper = sqlSessionFactory.openSession().getMapper(UserMapper.class);
      4. 
  
  
  
  
    4.     PageRowBounds pageRowBounds = new PageRowBounds(1, 2);
      5. 
  
  
  
  
    5.     List list = userMapper.getAllUsersByPage(pageRowBounds);
      6. 
  
  
  
  
    6.     for (User user : list) {
      7. 
  
  
  
  
    7.         System.out.println("user = " + user);
      8. 
  
  
  
  
    8.     }
      9. 
  
  
  
  
    9.     System.out.println("pageRowBounds.getTotal() = " + pageRowBounds.getTotal());
      10. 
  
  
  
  
    10. }
      11. 
 
 
 
 
    此時通過 pageRowBounds.getTotal() 方法我們就可以獲取到總記錄數(shù)。
    6.小結(jié)
    好啦，今天主要和小伙伴們分享了我們?nèi)绾巫约洪_發(fā)一個 MyBatis 插件，插件功能其實都是次要的，最主要是希望小伙伴們能夠理解 MyBatis 的工作流程。
    本文轉(zhuǎn)載自微信公眾號「江南一點雨」，可以通過以下二維碼關(guān)注。轉(zhuǎn)載本文請聯(lián)系江南一點雨公眾號。
                                                    
    
                                                新聞名稱：手把手教你開發(fā)MyBatis插件                                                
    
                                                本文地址：http://www.5511xx.com/article/ccioshc.html