mybatis查詢語句之封裝數據的示例分析

這篇文章主要介紹了mybatis查詢語句之封裝數據的示例分析，具有一定借鑒價值，感興趣的朋友可以參考下，希望大家閱讀完這篇文章之后大有收獲，下面讓小編帶著大家一起了解一下。

分析

mybatis在最后的查詢最終會走SimpleExecutor類的doQuery方法，

@Override
 public <E> List<E> doQuery(MappedStatement ms, Object parameter, RowBounds rowBounds, ResultHandler resultHandler, BoundSql boundSql) throws SQLException {
 Statement stmt = null;
 try {
 Configuration configuration = ms.getConfiguration();
 // 這里也就是采用了策略模式(個人感覺有點像)，實際的statementHandler為routingStatementHandler
 StatementHandler handler = configuration.newStatementHandler(wrapper, ms, parameter, rowBounds, resultHandler, boundSql);
 stmt = prepareStatement(handler, ms.getStatementLog());
 // 雖然是執行的routingStatementHandler.query，但返回結果的還是PreparedStatementHandler處理
 return handler.query(stmt, resultHandler);
 } finally {
 closeStatement(stmt);
 }
 }

private Statement prepareStatement(StatementHandler handler, Log statementLog) throws SQLException {
 Statement stmt;
 // 使用了代理模式，也可以理解為對connection進行了一層包裝，這里的作用就是加了log處理
 Connection connection = getConnection(statementLog);
 //進行預編譯,即類似jdbc的 sql,如 select * from user where id=?
 stmt = handler.prepare(connection, transaction.getTimeout());
 // 對執行查詢的sql進行參數設置
 handler.parameterize(stmt);
 return stmt;
 }

關于 handler.prepare的作用這里簡單介紹下，不做代碼分析。

會設置fetchSize，作用就是一次性從數據庫抓取數據，好像默認值是10條，如果每次只抓取一條，則進行rs.next的時候，會再次查庫。

如果是insert操作，并且數據庫主鍵自增且還設置了可以返回主鍵，則會還做獲取主鍵的操作。

先從設置參數說起，也就是handler.parameterize。先看下源碼，具體位置在DefaultParameterHandler類里面

@Override
 public void setParameters(PreparedStatement ps) {
 ErrorContext.instance().activity("setting parameters").object(mappedStatement.getParameterMap().getId());
 // 獲取配置文件里面的sql參數信息，如sql為select * from user where id=#{userId,jdbcType=INTEGER}
 // ParameterMapping 記錄了參數名也就是userId,還有記錄了對應的jdbc類型，還有對應的javaType等等，具體可以debug看下
 List<ParameterMapping> parameterMappings = boundSql.getParameterMappings();
 if (parameterMappings != null) {
 for (int i = 0; i < parameterMappings.size(); i++) {
 ParameterMapping parameterMapping = parameterMappings.get(i);
 if (parameterMapping.getMode() != ParameterMode.OUT) {
  Object value;
  String propertyName = parameterMapping.getProperty();
  // 如果為true，那么sql參數中有類似 user.name 格式
  if (boundSql.hasAdditionalParameter(propertyName)) { // issue #448 ask first for additional params
  value = boundSql.getAdditionalParameter(propertyName);
  } else if (parameterObject == null) {
  value = null;
  } else if (typeHandlerRegistry.hasTypeHandler(parameterObject.getClass())) {
  value = parameterObject;
  } else {
  // metaObject 類似一個工具類，它里面有一個反射工廠，可以專門解析一個類的信息，如字段的setter/getter/屬性信息，這里不做多余介紹
  // 1、下面詳細介紹 
  MetaObject metaObject = configuration.newMetaObject(parameterObject);
  value = metaObject.getValue(propertyName);// 取值
  }
  // 獲取對應的typeHandler，一般情況不設置的話，基本都是ObjectTypeHandler
  TypeHandler typeHandler = parameterMapping.getTypeHandler();
  JdbcType jdbcType = parameterMapping.getJdbcType();
  if (value == null && jdbcType == null) {
  jdbcType = configuration.getJdbcTypeForNull();
  }
  try {
  // 進行設值
  typeHandler.setParameter(ps, i + 1, value, jdbcType);
  } catch (TypeException e) {
  throw new TypeException("Could not set parameters for mapping: " + parameterMapping + ". Cause: " + e, e);
  } catch (SQLException e) {
  throw new TypeException("Could not set parameters for mapping: " + parameterMapping + ". Cause: " + e, e);
  }
 }
 }
 }
 }

對于上述代碼中的一部分這里負責將parameterObject的里面的值整出來(也就是傳入的參數)，如果參數是map結構，就從map里面取值，如果不是，如單個非javabean參數，則直接取值，如果是單個javabean，則通過metaObject類轉換成一個BeanWrapper，進行取值

這段代碼也就負責對預編譯后的sql設置參數，這里邏輯主要是圍繞以下步驟進行得，

獲取參數名，獲取參數值，獲取參數類型，然后做進行設值操作

/**
 * mybatis數據處理有單結果集和多結果集處理，一般多結果集出現存儲過程中，如果存儲過程中寫了兩條select語句，如
 * select * from user , select * from classes 這種情況這里不做介紹，因為本人用的不多，理解的也不是很透徹。
 * 這里不多做介紹，這里只針對簡單映射做一個大概介紹
 *
 */
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);
 // 如果不是多結果集映射，一般resultMaps的大小為1
 // resultMap中存儲的有類的字段屬性，數據庫字段名稱等信息
 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++;
 }
 // 處理slect 標簽的resultSets屬性，多個用逗號隔開，個人幾乎沒用過，略過
 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);
 }

以上代碼就是為結果映射做一個鋪墊，重點是在hanleResultSet方法里，

private void handleResultSet(ResultSetWrapper rsw, ResultMap resultMap, List<Object> multipleResults, ResultMapping parentMapping) throws SQLException {
 try {// 針對簡單映射，parentMapping是為Null的
 if (parentMapping != null) {
 handleRowValues(rsw, resultMap, null, RowBounds.DEFAULT, parentMapping);
 } else {
 // 默認使用defaultResultHandler，如需使用自定義的，則可在傳參加入resultHandler接口實現類
 if (resultHandler == null) {
  DefaultResultHandler defaultResultHandler = new DefaultResultHandler(objectFactory);
  // 處理結果，結果存在resultHandler里
  handleRowValues(rsw, resultMap, defaultResultHandler, rowBounds, null);
  multipleResults.add(defaultResultHandler.getResultList());
 } else {
  handleRowValues(rsw, resultMap, resultHandler, rowBounds, null);
 }
 }
 } finally {
 // issue #228 (close resultsets)
 closeResultSet(rsw.getResultSet());
 }
 }

public void handleRowValues(ResultSetWrapper rsw, ResultMap resultMap, ResultHandler<?> resultHandler, RowBounds rowBounds, ResultMapping parentMapping) throws SQLException {
 // 處理有嵌套映射的情況
 if (resultMap.hasNestedResultMaps()) {
 ensureNoRowBounds();
 checkResultHandler();
 handleRowValuesForNestedResultMap(rsw, resultMap, resultHandler, rowBounds, parentMapping);
 } else {//沒有嵌套映射
 handleRowValuesForSimpleResultMap(rsw, resultMap, resultHandler, rowBounds, parentMapping);
 }
 }

private void handleRowValuesForSimpleResultMap(ResultSetWrapper rsw, ResultMap resultMap, ResultHandler<?> resultHandler, RowBounds rowBounds, ResultMapping parentMapping)
 throws SQLException {
 DefaultResultContext<Object> resultContext = new DefaultResultContext<>();
 ResultSet resultSet = rsw.getResultSet();
 // 跳過多少行，到達指定記錄位置，如在傳參的時候傳入了rowBounds，則會根據該類的offset值跳到指定記錄位置
 skipRows(resultSet, rowBounds);
 // shouldProcessMoreRows 用來檢測是否能繼續對后續的結果進行映射
 while (shouldProcessMoreRows(resultContext, rowBounds) && !resultSet.isClosed() && resultSet.next()) {
 //用來處理resultMap節點中配置了discriminator節點，這里忽略掉
 ResultMap discriminatedResultMap = resolveDiscriminatedResultMap(resultSet, resultMap, null);
 // 得到的結果就是sql執行后的一行記錄，如返回User對象信息，則rowValue就代表一個user實例，里面已經有值了
 Object rowValue = getRowValue(rsw, discriminatedResultMap, null);
 //保存數據
 storeObject(resultHandler, resultContext, rowValue, parentMapping, resultSet);
 }
 }

private Object getRowValue(ResultSetWrapper rsw, ResultMap resultMap, String columnPrefix) throws SQLException {
 final ResultLoaderMap lazyLoader = new ResultLoaderMap();
 // 創建對象，可以理解為對resultMap節點的type屬性值，進行了反射處理，得到了一個對象，但屬性值都是默認值。
 Object rowValue = createResultObject(rsw, resultMap, lazyLoader, columnPrefix);
 if (rowValue != null && !hasTypeHandlerForResultObject(rsw, resultMap.getType())) {
 final MetaObject metaObject = configuration.newMetaObject(rowValue);
 boolean foundValues = this.useConstructorMappings;
 //是否需要自動映射，有三種映射，分別為None,partial,full，默認第二種，處理非嵌套映射，可通過autoMappingBehavior 配置
 if (shouldApplyAutomaticMappings(resultMap, false)) {
 // 映射resultMap中未明確指定的列，如類中含有username屬性，但是resultMap中沒配置，則通過這個進行數據映射，還是可以查詢到結果
 foundValues = applyAutomaticMappings(rsw, resultMap, metaObject, columnPrefix) || foundValues;
 }
 // 處理resultMap中指定的列
 foundValues = applyPropertyMappings(rsw, resultMap, metaObject, lazyLoader, columnPrefix) || foundValues;
 foundValues = lazyLoader.size() > 0 || foundValues;
 // 如果沒查詢到結果，但配置可返回空對象(指的是沒有設置屬性值得對象)，則返回空對象，否則返回null
 rowValue = foundValues || configuration.isReturnInstanceForEmptyRow() ? rowValue : null;
 }
 return rowValue;
 }

這里只介紹resultMap中有明確指定的列

private boolean applyPropertyMappings(ResultSetWrapper rsw, ResultMap resultMap, MetaObject metaObject, ResultLoaderMap lazyLoader, String columnPrefix)
 throws SQLException {
 // 獲取數據字段名
 final List<String> mappedColumnNames = rsw.getMappedColumnNames(resultMap, columnPrefix);
 boolean foundValues = false;
 // 獲取的數據就是resultMap節點中配置的result節點，有多個result節點，這個集合大小就是多少
 // 里面存儲的是屬性名/字段名等信息
 final List<ResultMapping> propertyMappings = resultMap.getPropertyResultMappings();
 for (ResultMapping propertyMapping : propertyMappings) {
 String column = prependPrefix(propertyMapping.getColumn(), columnPrefix);
 // 是否有嵌套映射
 if (propertyMapping.getNestedResultMapId() != null) {
 // the user added a column attribute to a nested result map, ignore it
 column = null;
 }
 // 針對1來說一般常與嵌套查詢配合使用
 // 2 判斷屬性基本映射
 // 3 多結果集的一個處理
 if (propertyMapping.isCompositeResult()// 1
  || (column != null && mappedColumnNames.contains(column.toUpperCase(Locale.ENGLISH)))// 2
  || propertyMapping.getResultSet() != null) {// 3
 // 獲取當前column字段對于的值，有用到typeHandler來進行參數的一個轉換
 Object value = getPropertyMappingValue(rsw.getResultSet(), metaObject, propertyMapping, lazyLoader, columnPrefix);
 
 //獲取類的屬性字段名
 final String property = propertyMapping.getProperty();
 if (property == null) {
  continue;
 } else if (value == DEFERRED) {// 類似占位符。處理懶加載數據
  foundValues = true;
  continue;
 }
 if (value != null) {
  foundValues = true;
 }
 if (value != null || (configuration.isCallSettersOnNulls() && !metaObject.getSetterType(property).isPrimitive())) {
  // 進行設置屬性值
  metaObject.setValue(property, value);
 }
 }
 }
 return foundValues;
 }

或許有人奇怪為啥沒看到查詢的對象有set操作，值就到了對象里面去了，這里全是metaObject給你操作了，具體的，大家可以自行了解這個類，只能說這個類的功能很強大。

感謝你能夠認真閱讀完這篇文章，希望小編分享的“mybatis查詢語句之封裝數據的示例分析”這篇文章對大家有幫助，同時也希望大家多多支持億速云，關注億速云行業資訊頻道，更多相關知識等著你來學習!