Mybatis源码分析(五)探究SQL语句的执行过程

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/w372426096/article/details/89174195

一、重温JDBC

Java Database Connectivity,简称JDBC。是Java语言中用来规范客户端程序如何来访问数据库的应用程序接口,提供了诸如查询和更新数据库中数据的方法。 随着Java ORM框架的发展,已经很少有机会再在生产系统中写JDBC的代码来访问数据库了,但是基本流程我们还是要熟悉。下面以一个简单的查询为例,温故一下JDBC。

public static void main(String[] args) throws Exception {
   Connection conn = getConnection();  
   String sql = "select * from user where 1=1 and id = ?";
   PreparedStatement stmt = conn.prepareStatement(sql);
   stmt.setString(1, "501440165655347200");
   ResultSet rs = stmt.executeQuery();
   while(rs.next()){
       String username = rs.getString("username");
       System.out.print("姓名: " + username);
   }
}

从上面的代码来看,一次简单的数据库查询操作,可以分为几个步骤。

  • 创建Connection连接

  • 传入参数化查询SQL语句构建预编译对象PreparedStatement

  • 设置参数

  • 执行SQL

  • 从结果集中获取数据

那么,咱们的主角Mybatis是怎样完成这一过程的呢?不着急,咱们一个一个来看。

二、sqlSession

在上一章节的内容中,我们已经看到了在Service层通过@Autowired注入的userMapper是个代理类,在执行方法的时候实际上调用的是代理类的invoke通知方法。

public class MapperProxy<T> implements InvocationHandler{
   public Object invoke(Object proxy, Method method, Object[] args)
       final MapperMethod mapperMethod = cachedMapperMethod(method);
       return mapperMethod.execute(sqlSession, args);
   }
}

1 、创建MapperMethod对象

MapperMethod对象里面就两个属性,SqlCommand和MethodSignature。

SqlCommand包含了执行方法的名称和方法的类型,比如 UNKNOWN,INSERT,UPDATE,DELETE,SELECT,FLUSH。 MethodSignature可以简单理解为方法的签名信息。里面包含:返回值类型、是否void、是否为集合类型、是否为Cursor等,主要还获取到了方法参数上的@Param注解的名称,方便下一步获取参数值。 比如,如果方法上加了@Param的参数: UsergetUserById(@Param(value="id")Stringid,@Param(value="password")Stringpassword);,参数会被解析成 {0=id,1=password}

2、执行

判断方法的SQL类型和返回值类型 ,调用相应的方法。以方法 UsergetUserById(Stringid,Stringpassword)为例,会调用到selectOne()方法。

public class MapperMethod {

   public Object execute(SqlSession sqlSession, Object[] args) {
       Object result;
       switch (command.getType()) {
       case INSERT: {}
       case UPDATE: {}
       case DELETE: {}
       case SELECT:
       if (method.returnsVoid() && method.hasResultHandler()) {
           //无返回值
       } else if (method.returnsMany()) {
           //返回集合类型
           result = executeForMany(sqlSession, args);
       } else if (method.returnsMap()) {
           //返回Map类型
           result = executeForMap(sqlSession, args);
       } else if (method.returnsCursor()) {
           //返回Cursor
           result = executeForCursor(sqlSession, args);
       } else {
           //将参数args转换为SQL命令的参数
           //默认会添加一个《param+参数索引》的参数名
           //{password=123456, id=501441819331002368, param1=501441819331002368, param2=123456}
           Object param = method.convertArgsToSqlCommandParam(args);
           result = sqlSession.selectOne(command.getName(), param);
       }
       ...
       return result;
   }
}

可以看到,sqlSession.selectOne就可以获取到数据库中的值并完成转换工作。这里的sqlSession就是 SqlSessionTemplate实例的对象,所以它会调用到

public class SqlSessionTemplate{
   public <T> T selectOne(String statement, Object parameter) {
       return this.sqlSessionProxy.<T> selectOne(statement, parameter);
   }
}

sqlSessionProxy也是个代理对象。关于它的创建咱们上节课也很认真的分析了,总之它实际会调用到 SqlSessionInterceptor.invoke()

3、创建sqlSession对象

sqlSession我们熟悉呀,它作为MyBatis工作的主要顶层API,表示和数据库交互的会话,完成必要数据库增删改查功能。关于它的创建、执行、提交和资源清理都是在SqlSessionInterceptor的通知方法中完成的。

private class SqlSessionInterceptor implements InvocationHandler {
   public Object invoke(Object proxy, Method method, Object[] args) throws Throwable {
       //创建SqlSession对象
       SqlSession sqlSession = getSqlSession(
       SqlSessionTemplate.this.sqlSessionFactory,
       SqlSessionTemplate.this.executorType,
       SqlSessionTemplate.this.exceptionTranslator);
       try {
           //调用sqlSession实际方法
           Object result = method.invoke(sqlSession, args);
           return result;
       } catch (Throwable t) {
           ....
       } finally {
           if (sqlSession != null) {
               closeSqlSession(sqlSession, SqlSessionTemplate.this.sqlSessionFactory);
           }
       }
   }
}

上面的重点就是创建了SqlSession并执行它的方法,它是一个DefaultSqlSession实例的对象,里面主要有一个通过configuration创建的执行器,在这里它是SimpleExecutor。

那么,invoke方法实际调用的就是 DefaultSqlSession.selectOne()

三、获取BoundSql对象

DefaultSqlSession中的 selectOne()方法最终也会调用到 selectList()方法。它先从数据大管家configuration中根据请求方法的全名称拿到对应的MappedStatement对象,然后调用执行器的查询方法。

1、获取MappedStatement对象

//statement是调用方法的全名称,parameter为参数的Map
public <E> List<E> selectList(String statement, Object parameter, RowBounds rowBounds) {
   //在mapper.xml中每一个SQL节点都会封装为MappedStatement对象
   //在configuration中就可以通过请求方法的全名称获取对应的MappedStatement对象
   MappedStatement ms = configuration.getMappedStatement(statement);
   return executor.query(ms, wrapCollection(parameter), rowBounds, Executor.NO_RESULT_HANDLER);
}

其中有个方法 wrapCollection(parameter)我们可以了解下,如果参数为集合类型或者数组类型,它会将参数名称设置为相应类型的名称。

private Object wrapCollection(final Object object) {
   if (object instanceof Collection) {
       StrictMap<Object> map = new StrictMap<Object>();
       map.put("collection", object);
       if (object instanceof List) {
       map.put("list", object);
       }
       return map;
   } else if (object != null && object.getClass().isArray()) {
       StrictMap<Object> map = new StrictMap<Object>();
       map.put("array", object);
       return map;
   }
   return object;
}

2、获取BoundSql对象

在configuration这个大管家对象中,保存着mapper.xml里面所有的SQL节点。每一个节点对应一个MappedStatement对象,而动态生成的各种sqlNode保存在SqlSource对象,SqlSource对象有一个方法就是getBoundSql()。 我们先来看一下BoundSql类哪有哪些属性。

public class BoundSql {    
   //动态生成的SQL,解析完毕带有占位性的SQL
   private final String sql;
   //每个参数的信息。比如参数名称、输入/输出类型、对应的JDBC类型等
   private final List<ParameterMapping> parameterMappings;
   //参数
   private final Object parameterObject;
   private final Map<String, Object> additionalParameters;
   private final MetaObject metaParameters;
}

看到这几个属性,也就解释了BoundSql 的含义。即表示动态生成的SQL语句和相应的参数信息。

不知大家是否还有印象,不同类型的SQL会生成不同类型的SqlSource对象。比如静态SQL会生成StaticSqlSource对象,动态SQL会生成DynamicSqlSource对象。

  • 静态SQL

静态SQL比较简单,直接就创建了BoundSql对象并返回。

public class StaticSqlSource implements SqlSource {
   return new BoundSql(configuration, sql, parameterMappings, parameterObject);
}
  • 动态SQL

动态SQL要根据不同的sqlNode节点,调用对应的apply方法,有的还要通过Ognl表达式来判断是否需要添加当前节点,比如IfSqlNode。

public class DynamicSqlSource implements SqlSource {
   public BoundSql getBoundSql(Object parameterObject) {
       DynamicContext context = new DynamicContext(configuration, parameterObject);
       //rootSqlNode为sqlNode节点的最外层封装,即MixedSqlNode。
       //解析完所有的sqlNode,将sql内容设置到context
       rootSqlNode.apply(context);
       SqlSourceBuilder sqlSourceParser = new SqlSourceBuilder(configuration);
       Class<?> parameterType = parameterObject == null ? Object.class : parameterObject.getClass();
       //设置参数信息 将SQL#{}替换为占位符
       SqlSource sqlSource = sqlSourceParser.parse(context.getSql(), parameterType, context.getBindings());
       //创建BoundSql对象
       BoundSql boundSql = sqlSource.getBoundSql(parameterObject);
       for (Map.Entry<String, Object> entry : context.getBindings().entrySet()) {
           boundSql.setAdditionalParameter(entry.getKey(), entry.getValue());
       }
       return boundSql;
   }
}

rootSqlNode.apply(context)是一个迭代调用的过程。最后生成的内容保存在DynamicContext对象,比如 select*fromuser WHERE uid=#{uid}

然后调用SqlSourceBuilder.parse()方法。它主要做了两件事:

1、将SQL语句中的#{}替换为占位符 2、将#{}里面的字段封装成ParameterMapping对象,添加到parameterMappings。

ParameterMapping对象保存的就是参数的类型信息,如果没有配置则为null。 ParameterMapping{property='uid',mode=IN,javaType=classjava.lang.Object,jdbcType=null,numericScale=null,resultMapId='null',jdbcTypeName='null',expression='null'}

最后返回的BoundSql对象就包含一个带有占位符的SQL和参数的具体信息。

四、执行SQL

创建完BoundSql对象,调用query方法,来到CachingExecutor.query()。这个方法的前面是二级缓存的判断,如果开启了二级缓存且缓存中有数据,就返回。

1、缓存

public class CachingExecutor implements Executor {
   public <E> List<E> query(MappedStatement ms, Object parameterObject,
       RowBounds rowBounds, ResultHandler resultHandler,
       CacheKey key, BoundSql boundSql)throws SQLException
       //二级缓存的应用
       //如果配置</cache>则走入这个流程
       Cache cache = ms.getCache();
       if (cache != null) {
           flushCacheIfRequired(ms);
           if (ms.isUseCache() && resultHandler == null) {
               //从缓存中获取数据
               List<E> list = (List<E>) tcm.getObject(cache, key);
               if (list == null) {
                   list = delegate.<E> query(ms, parameterObject, rowBounds, resultHandler, key, boundSql);
                   tcm.putObject(cache, key, list); // issue #578 and #116
               }
               return list;
           }
       }
       return delegate.<E> query(ms, parameterObject, rowBounds, resultHandler, key, boundSql);
   }
}

接着看query方法,创建PreparedStatement预编译对象,执行SQL并获取返回集合。

public class SimpleExecutor extends BaseExecutor {
   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();
           //获取Statement的类型,即默认的PreparedStatementHandler
           //需要注意,在这里如果配置了插件,则StatementHandler可能返回的是一个代理
           StatementHandler handler = configuration.newStatementHandler(wrapper, ms, parameter, rowBounds, resultHandler, boundSql);
           //创建PreparedStatement对象,并设置参数值
           stmt = prepareStatement(handler, ms.getStatementLog());
           //执行execute 并返回结果集
           return handler.<E>query(stmt, resultHandler);
       } finally {
           closeStatement(stmt);
       }
   }
}

prepareStatement方法获取数据库连接并构建Statement对象设置SQL参数。

1、创建PreparedStatement

public class SimpleExecutor extends BaseExecutor {
   private Statement prepareStatement(StatementHandler handler, Log statementLog) {
       Statement stmt;
       Connection connection = getConnection(statementLog);
       stmt = handler.prepare(connection, transaction.getTimeout());
       handler.parameterize(stmt);
       return stmt;
   }
}
  • 获取Connection连接

我们看到getConnection方法就是获取Connection连接的地方。但这个Connection也是一个代理对象,它的调用程序处理器为 ConnectionLogger。显然,它是为了更方便的打印日志。

public abstract class BaseExecutor implements Executor {
   protected Connection getConnection(Log statementLog) throws SQLException {
       //从c3p0连接池中获取一个连接
       Connection connection = transaction.getConnection();
       //如果日志级别为Debug,则为这个连接生成代理对象返回
       //它的处理类为ConnectionLogger
       if (statementLog.isDebugEnabled()) {
           return ConnectionLogger.newInstance(connection, statementLog, queryStack);
       } else {
           return connection;
       }
   }
}
  • 执行预编译

这个跟我们的JDBC代码是一样的,拿到SQL,调用Connection连接的prepareStatement(sql)。但由于connection是一个代理对象,似乎又没那么简单。

public class PreparedStatementHandler
   protected Statement instantiateStatement(Connection connection) throws SQLException {
       String sql = boundSql.getSql();
       return connection.prepareStatement(sql);
   }
}

所以,在执行的 onnection.prepareStatement(sql)的时候,实际调用的是ConnectionLogger类的invoke()。

public final class ConnectionLogger extends BaseJdbcLogger implements InvocationHandler {
   public Object invoke(Object proxy, Method method, Object[] params)throws Throwable {
       try {
           if ("prepareStatement".equals(method.getName())) {
               if (isDebugEnabled()) {
                   debug(" Preparing: " + removeBreakingWhitespace((String) params[0]), true);
               }        
               //调用connection.prepareStatement
               PreparedStatement stmt = (PreparedStatement) method.invoke(connection, params);
               //又为stmt创建了代理对象,通知类为PreparedStatementLogger
               stmt = PreparedStatementLogger.newInstance(stmt, statementLog, queryStack);
               return stmt;
           }
       }
   }
}
public static PreparedStatement newInstance(PreparedStatement stmt, Log statementLog, int queryStack) {
   InvocationHandler handler = new PreparedStatementLogger(stmt, statementLog, queryStack);
   ClassLoader cl = PreparedStatement.class.getClassLoader();
   return (PreparedStatement) Proxy.newProxyInstance(cl,
           new Class[]{PreparedStatement.class, CallableStatement.class}, handler);
}

果然没那么简单,最后返回的PreparedStatement又是个代理对象。

  • 设置参数

我们知道,在设置参数的时候,你有很多可选项,比如 stmt.setString()、stmt.setInt()、stmt.setFloat()等,或者粗暴一点就 stmt.setObject()

当然了,Mybatis作为一个优秀的ORM框架,不可能这么粗暴。它先是根据参数的Java类型获取所有JDBC类型的处理器,再根据JDBC的类型获取对应的处理器。在这里我们没有配置JDBC类型,所以就是它的类型为NULL,最后返回的就是StringTypeHandler。

关于类型处理器的匹配和查询规则,咱们在Mybatis源码分析(三)通过实例来看typeHandlers已经详细分析过,就不再细看。

public class StringTypeHandler extends BaseTypeHandler<String> {
   public void setNonNullParameter(PreparedStatement ps, int i,
           String parameter, JdbcType jdbcType)throws SQLException {
       ps.setString(i, parameter);
   }
}

2、执行

在SQL预编译完成之后,调用execute()执行。

public class PreparedStatementHandler{
   public <E> List<E> query(Statement statement, ResultHandler resultHandler) {
       PreparedStatement ps = (PreparedStatement) statement;
       ps.execute();
       return resultSetHandler.<E> handleResultSets(ps);
   }
}

这里的PreparedStatement对象也是个代理类,在调用通知类PreparedStatementLogger,执行execute的时候,只是打印了参数的值。即 Parameters:501868995461251072(String)

五、处理返回值

上面的方法我们看到SQL已经提交给数据库执行,那么最后一步就是获取返回值。

public class DefaultResultSetHandler implements ResultSetHandler {
   public List<Object> handleResultSets(Statement stmt) throws SQLException {
       final List<Object> multipleResults = new ArrayList<Object>();
       int resultSetCount = 0;
       //将ResultSet封装成ResultSetWrapper对象
       ResultSetWrapper rsw = getFirstResultSet(stmt);
       //返回mapper.xml中配置的rsultMap 实际上我们没有配置,但会有默认的一个
       List<ResultMap> resultMaps = mappedStatement.getResultMaps();
       int resultMapCount = resultMaps.size();
       //处理数据库的返回值,最后加入到multipleResults
       while (rsw != null && resultMapCount > resultSetCount) {
           ResultMap resultMap = resultMaps.get(resultSetCount);
           handleResultSet(rsw, resultMap, multipleResults, null);
           resultSetCount++;
       }
       //返回
       return collapseSingleResultList(multipleResults);
   }
}

1、ResultSetWrapper对象

上面的代码我们看到,第一步就把ResultSet对象封装成了ResultSetWrapper对象,关于它还需要具体来看。

public class ResultSetWrapper {
   public ResultSetWrapper(ResultSet rs, Configuration configuration) throws SQLException {
       //所有已注册的类型处理器
       this.typeHandlerRegistry = configuration.getTypeHandlerRegistry()
       //ResultSet对象
       this.resultSet = rs;
       //元数据 列名、列类型等信息
       final ResultSetMetaData metaData = rs.getMetaData();
       final int columnCount = metaData.getColumnCount();
       //循环列,将列名、列对应的JDBC类型和列对应的Java类型都获取到
       for (int i = 1; i <= columnCount; i++) {
           columnNames.add(metaData.getColumnName(i));
           jdbcTypes.add(JdbcType.forCode(metaData.getColumnType(i)));
           classNames.add(metaData.getColumnClassName(i));
       }
   }
}

上面的重点是拿到数据库列上的信息,在解析的时候会用到。

2、处理返回值

handleResultSet方法最后调用到 DefaultResultSetHandler.handleRowValuesForSimpleResultMap()

public class DefaultResultSetHandler implements ResultSetHandler {
   private void handleRowValuesForSimpleResultMap(ResultSetWrapper rsw,
           ResultMap resultMap, ResultHandler<?> resultHandler,
           RowBounds rowBounds, ResultMapping parentMapping)throws SQLException {
       DefaultResultContext<Object> resultContext = new DefaultResultContext<Object>();
       //跳过行 Mybatis的RowBounds分页功能
       skipRows(rsw.getResultSet(), rowBounds);
       while (shouldProcessMoreRows(resultContext, rowBounds) && rsw.getResultSet().next()) {
           ResultMap discriminatedResultMap = resolveDiscriminatedResultMap(rsw.getResultSet(), resultMap, null);
           Object rowValue = getRowValue(rsw, discriminatedResultMap);
           storeObject(resultHandler, resultContext, rowValue, parentMapping, rsw.getResultSet());
       }
   }
}

在这个地方涉及到Mybatis中分页的一个对象RowBounds。但实际上,我们基本不会用到它。因为它是一个逻辑分页,而非物理分页。

  • RowBounds

RowBounds对象中有两个属性控制着分页: offset、limit。offset是说分页从第几条数据开始,limit是说一共取多少条数据。因为我们没有配置它,所以它默认是offset从0开始,limit取Int的最大值。

public class RowBounds {
   public static final int NO_ROW_OFFSET = 0;
   public static final int NO_ROW_LIMIT = Integer.MAX_VALUE;
   public RowBounds() {
       this.offset = NO_ROW_OFFSET;
       this.limit = NO_ROW_LIMIT;
   }
}

skipRows方法就是来跳过offset,它的实现也比较简单。

private void skipRows(ResultSet rs, RowBounds rowBounds) throws SQLException {
   for (int i = 0; i < rowBounds.getOffset(); i++) {
       rs.next();
   }
}

offset跳过之后,怎么控制Limit的呢?这就要看上面的while循环了。

while (shouldProcessMoreRows(resultContext, rowBounds) && rsw.getResultSet().next()) {
   //处理数据
}

关键在于shouldProcessMoreRows()方法,它其实是个简单的判断。

private boolean shouldProcessMoreRows(ResultContext<?> context,
                           RowBounds rowBounds) throws SQLException {
   //就是看已经取到的数据是否小与Limit
   return context.getResultCount() < rowBounds.getLimit();
}
  • 获取

while循环获取ResultSet的每一行数据,然后通过rs.getxxx()获取数据。

public class DefaultResultSetHandler implements ResultSetHandler {
   private Object getRowValue(ResultSetWrapper rsw, ResultMap resultMap) throws SQLException {
       final ResultLoaderMap lazyLoader = new ResultLoaderMap();
       //创建返回值类型,比如我们返回的是User实体类
       Object rowValue = createResultObject(rsw, resultMap, lazyLoader, null);
       if (rowValue != null && !hasTypeHandlerForResultObject(rsw, resultMap.getType())) {
           final MetaObject metaObject = configuration.newMetaObject(rowValue);
           boolean foundValues = this.useConstructorMappings;
           if (shouldApplyAutomaticMappings(resultMap, false)) {
           //自动映射
           foundValues = applyAutomaticMappings(rsw, resultMap, metaObject, null) || foundValues;
       }
       //这个处理配置的ResultMap,就是手动配置数据库列名与Java实体类字段的映射
       foundValues = applyPropertyMappings(rsw, resultMap, metaObject, lazyLoader, null) || foundValues;
       foundValues = lazyLoader.size() > 0 || foundValues;
       rowValue = foundValues || configuration.isReturnInstanceForEmptyRow() ? rowValue : null;
       }
       return rowValue;
   }
}

1、获取返回值类型

第一步是获取返回值类型,过程就是拿到Class对象,然后获取构造器,设置可访问并返回实例。

private  <T> T instantiateClass(Class<T> type,
           List<Class<?>> constructorArgTypes, List<Object> constructorArgs) {
   Constructor<T> constructor;
   if (constructorArgTypes == null || constructorArgs == null) {
       //获取构造器
       constructor = type.getDeclaredConstructor();
       if (!constructor.isAccessible()) {
           //设置可访问
           constructor.setAccessible(true);
       }
       //返回实例
       return constructor.newInstance();
   }
}

返回后,又把它包装成了MetaObject对象。Mybatis会根据返回值类型的不同,包装成不同的Wrapper对象。本例中,由于是一个实体类,会返回BeanWrapper。

private MetaObject(Object object, ObjectFactory objectFactory,

           ObjectWrapperFactory objectWrapperFactory, ReflectorFactory reflectorFactory) {
   this.originalObject = object;
   this.objectFactory = objectFactory;
   this.objectWrapperFactory = objectWrapperFactory;
   this.reflectorFactory = reflectorFactory;
   if (object instanceof ObjectWrapper) {
       this.objectWrapper = (ObjectWrapper) object;
   } else if (objectWrapperFactory.hasWrapperFor(object)) {
       this.objectWrapper = objectWrapperFactory.getWrapperFor(this, object);
   } else if (object instanceof Map) {
       this.objectWrapper = new MapWrapper(this, (Map) object);
   } else if (object instanceof Collection) {
       this.objectWrapper = new CollectionWrapper(this, (Collection) object);
   } else {
       this.objectWrapper = new BeanWrapper(this, object);
   }
}

2、applyAutomaticMappings

在mapper.xml中我们可以声明一个resultMap节点,将数据库中列的名称和Java中字段名称对应起来,应用到SQL节点的resultMap中。也可以不配置它,直接利用resultType返回一个Bean即可。但是这两种方式会对应两种解析方法。

private boolean applyAutomaticMappings(ResultSetWrapper rsw,
       ResultMap resultMap, MetaObject metaObject,
       String columnPrefix) throws SQLException {  
   //获取相应字段的类型处理器
   List<UnMappedColumnAutoMapping> autoMapping = createAutomaticMappings(rsw, resultMap, metaObject, columnPrefix);
   boolean foundValues = false;
   if (!autoMapping.isEmpty()) {
       for (UnMappedColumnAutoMapping mapping : autoMapping) {
           final Object value = mapping.typeHandler.getResult(rsw.getResultSet(), mapping.column);
           if (value != null) {
               foundValues = true;
           }
           if (value != null || (configuration.isCallSettersOnNulls() && !mapping.primitive)) {
               //因为返回值类型是一个BeanWapper,通过反射把值设置到JavaBean中。
               metaObject.setValue(mapping.property, value);
           }
       }
   }
   return foundValues;
}
  • 获取

上面代码的重点是获取对应字段的类型处理器,调用对应类型处理器的getResult方法从ResultSet中拿到数据的值。

//type是Java字段的类型 jdbcType是数据库列的JDBC类型
private <T> TypeHandler<T> getTypeHandler(Type type, JdbcType jdbcType) {
   //先从所有的处理器中获取Java类型的处理器
   Map<JdbcType, TypeHandler<?>> jdbcHandlerMap = getJdbcHandlerMap(type);
   TypeHandler<?> handler = null;
   if (jdbcHandlerMap != null) {
       //再根据JDBC的类型获取实际的处理器
       handler = jdbcHandlerMap.get(jdbcType);
   }
   return (TypeHandler<T>) handler;
}

以ID为例,在Java中是String类型,在数据库中是VARCHAR,最后返回的类型处理器是StringTypeHandler。调用的时候,就很简单了。

return rs.getString(columnName);
  • 设置

通过rs.getString()拿到值之后,然后向返回值类型中设置。因为我们返回的是一个JavaBean,对应的是BeanWapper对象,方法中其实就是反射调用。

public class BeanWrapper extends BaseWrapper {
   private void setBeanProperty(PropertyTokenizer prop, Object object, Object value) {
       try {
           Invoker method = metaClass.getSetInvoker(prop.getName());
           Object[] params = {value};
           try {
               method.invoke(object, params);
           } catch (Throwable t) {
           throw ExceptionUtil.unwrapThrowable(t);
           }
       } catch (Throwable t) {
           throw new ReflectionException("Could not set property '" + prop.getName() + "' of '" + object.getClass() + "' with value '" + value + "' Cause: " + t.toString(), t);
       }
   }
}

把所有的列都解析完,返回指定的Bean。最后加入到list,整个方法返回。在selectOne方法中,取List的第一条数据。如果数据记录大于1,就是出错了。

public class DefaultSqlSession implements SqlSession {
   public <T> T selectOne(String statement, Object parameter) {
       List<T> list = this.<T>selectList(statement, parameter);
       if (list.size() == 1) {
           return list.get(0);
       } else if (list.size() > 1) {
           throw new TooManyResultsException("Expected one result (or null)
                         to be returned by selectOne(), but found: " + list.size());
       } else {
           return null;
       }
   }
}

六、总结

关于Mybatis执行方法的整个过程,我们简单归纳一下。

  • 获取SqlSession,根据方法的返回值类型调用不同的方法。比如 selectOne

  • 获取BoundSql对象,根据传递的参数生成SQL语句

  • 从数据库连接池获取Connection对象,并为它创建代理,以便打印日志

  • 从Connection中获取PreparedStatement预编译对象,并为它创建代理

  • 预编译SQL,并设置参数

  • 执行、返回数据集合

  • 将数据集转换为Java对象

看到这里,再回忆下我们开头的JDBC实例的步骤,可以看到它们两者之间的主流程都是一样的。Mybatis只是在此基础上做了一些封装,更好的服务于我们的应用。

猜你喜欢

转载自blog.csdn.net/w372426096/article/details/89174195