首頁 > 軟體

mybatis-plus查詢原始碼詳解

2022-03-02 16:01:23

設定詳情

pom.xml

dependency>
     <groupId>com.baomidou</groupId>
     <artifactId>mybatis-plus-boot-starter</artifactId>
     <version>3.4.1</version>
 </dependency>

mapper

public interface GenTableMapper extends BaseMapper<GenTable> {
}

測試類

@RunWith(SpringRunner.class)
@SpringBootTest(classes = GendemoApplication.class)
public class BlockqueueTestDemo {
    @Autowired
    GenTableMapper genTableMapper;
    @Test
    public void test(){
        List<GenTable> genTables = 
        genTableMapper.selectList(new QueryWrapper<>());
    }
}

debug流程

1.發現 genTableMapper 是一個代理物件型別。

2.進入代理物件MybatisMapperProxy , 呼叫其invoke 方法,方法的Class型別為BaseMapper.selectList()

3.其中cachedInvoker()方法會返回一個PlainMethodInvoker ,它重寫了MapperMethodInvoker 介面的invoke()方法

4.最終會呼叫MybatisMapperMethodexecute()方法

public class MybatisMapperMethod {
    public Object execute(SqlSession sqlSession, Object[] args) {
        Object result;
        switch (command.getType()) {
            case INSERT: {
                Object param = method.convertArgsToSqlCommandParam(args);
                result = rowCountResult(sqlSession.insert(command.getName(), param));
                break;
            }
            case UPDATE: {
                Object param = method.convertArgsToSqlCommandParam(args);
                result = rowCountResult(sqlSession.update(command.getName(), param));
                break;
            }
            case DELETE: {
                Object param = method.convertArgsToSqlCommandParam(args);
                result = rowCountResult(sqlSession.delete(command.getName(), param));
                break;
            }
            case SELECT:
                if (method.returnsVoid() && method.hasResultHandler()) {
                    executeWithResultHandler(sqlSession, args);
                    result = null;
                } else if (method.returnsMany()) {
                    result = executeForMany(sqlSession, args);
                } else if (method.returnsMap()) {
                    result = executeForMap(sqlSession, args);
                } else if (method.returnsCursor()) {
                    result = executeForCursor(sqlSession, args);
                } else {
                    Object param = method.convertArgsToSqlCommandParam(args);
                    // TODO 這裡下面改了
                    if (IPage.class.isAssignableFrom(method.getReturnType())) {
                        result = executeForIPage(sqlSession, args);
                        // TODO 這裡上面改了
                    } else {
                        result = sqlSession.selectOne(command.getName(), param);
                        if (method.returnsOptional()
                            && (result == null || !method.getReturnType().equals(result.getClass()))) {
                            result = Optional.ofNullable(result);
                        }
                    }
                }
                break;
            case FLUSH:
                result = sqlSession.flushStatements();
                break;
            default:
                throw new BindingException("Unknown execution method for: " + command.getName());
        }
        if (result == null && method.getReturnType().isPrimitive() && !method.returnsVoid()) {
            throw new BindingException("Mapper method '" + command.getName()
                + " attempted to return null from a method with a primitive return type (" + method.getReturnType() + ").");
        }
        return result;
    }
}

5.這是經過判斷會進入executeForMany(sqlSession, args)方法,此時方法和引數都顯示出來了。sqlSession的型別是SqlSessionTemplate , 為什麼要注意這個 sqlSession的型別?因為SqlSession是一個介面,有很多實現類,有時候我們並不知道到底呼叫了哪個實現類的selectList()方法,這個時候我們看型別就知道了,就可以進入SqlSessionTemplate類,找到selectList()打上斷點,debug就過來了。

6.利用同樣的方法,又呼叫了DefaultSqlSessionselectList()方法。


7.來到DefaultSqlSessionselectList() 方法中,此時已經進入到mybatis的原始碼範圍了。executor的型別是MybatisCachingExecutor

8.此時要注意MybatisCachingExecutor 代理類的handler是一個Plugin

9.因為我使用到了分頁外掛,所以會來到com.github.pagehelperPageInterceptor

10.由MybatisCachingExecutor來執行查詢

11.MybatisCachingExecutor 委派 BaseExecutor 執行查詢

12.最終委派到PreparedStatementHandler來處理

13.最後由DefaultResultSetHandler來封裝結果集

@Override
  public List<Object> handleResultSets(Statement stmt) throws SQLException {
    ErrorContext.instance().activity("handling results").object(mappedStatement.getId());

    final List<Object> multipleResults = new ArrayList<>();
    int resultSetCount = 0;
    ResultSetWrapper rsw = getFirstResultSet(stmt);
    List<ResultMap> resultMaps = mappedStatement.getResultMaps();
    int resultMapCount = resultMaps.size();
    validateResultMapsCount(rsw, resultMapCount);
    while (rsw != null && resultMapCount > resultSetCount) {
      ResultMap resultMap = resultMaps.get(resultSetCount);
      handleResultSet(rsw, resultMap, multipleResults, null);
      rsw = getNextResultSet(stmt);
      cleanUpAfterHandlingResultSet();
      resultSetCount++;
    }
    String[] resultSets = mappedStatement.getResultSets();
    if (resultSets != null) {
      while (rsw != null && resultSetCount < resultSets.length) {
        ResultMapping parentMapping = nextResultMaps.get(resultSets[resultSetCount]);
        if (parentMapping != null) {
          String nestedResultMapId = parentMapping.getNestedResultMapId();
          ResultMap resultMap = configuration.getResultMap(nestedResultMapId);
          handleResultSet(rsw, resultMap, null, parentMapping);
        }
        rsw = getNextResultSet(stmt);
        cleanUpAfterHandlingResultSet();
        resultSetCount++;
      }
    return collapseSingleResultList(multipleResults);
  }

總結

為什麼要一層一層的封裝?一層一層的委派?
這裡面和快取有關,有興趣的可以自己瞭解一下。

下一篇打算記錄一下 mybatis-plus 的自動設定過程,有興趣的可以持續關注一下。
文中有錯誤的地方不吝賜教謝謝。

到此這篇關於mybatis-plus查詢原始碼走讀的文章就介紹到這了,更多相關mybatis-plus查詢內容請搜尋it145.com以前的文章或繼續瀏覽下面的相關文章希望大家以後多多支援it145.com!


IT145.com E-mail:sddin#qq.com