Mybatis工具—数据库操作器(SqlRunner)

  MyBatis提供了一个用于操作数据库的SqlRunner工具类,对JDBC做了很好的封装。结合工具类SQL,能够很方便地通过Java代码执行SQL语句并检索SQL执行结果。如下示例代码所示:

String sql = new SQL()
    .SELECT("*")
	.FROM("user_info")
	.WHERE("id = #{id}", "name = #{name}")
	.toString();
Connection connection = DriverManager.getConnection("数据库连接");
SqlRunner sqlRunner = new SqlRunner(connection);
Map<String, Object> result = sqlRunner.selectOne(sql, 1, "Jason");

  SqlRunner不仅提供了上述selectOne()方法,还提供了:

  • SqlRunner#closeConnection():用于关闭Connection对象。
  • SqlRunner#selectOne(String sql, Object… args):执行SELECT语句,SQL语句中可以使用占位符,如果SQL中包含占位符,则可变参数用于为参数占位符赋值,该方法只返回一条记录。若查询结果行数不等于一,则会抛出SQLException异常。
  • SqlRunner#selectAll(String sql, Object… args):该方法和selectOne()方法的作用相同,只不过该方法可以返回多条记录,方法返回值是一个List对象,List中包含多个Map对象,每个Map对象对应数据库中的一行记录。
  • SqlRunner#insert(String sql, Object… args):执行一条INSERT语句,插入一条记录。
  • SqlRunner#update(String sql, Object… args):执行更新SQL语句。
  • SqlRunner#delete(String sql, Object… args):执行删除SQL语句。
  • SqlRunner#run(String sql):执行任意一条SQL语句,最好为DDL语句。

selectAll

  这里以最具代表性的selectAll(String sql, Object… args)方法为例,来进一步剖析SqlRunner类源代码逻辑。

public List<Map<String, Object>> selectAll(String sql, Object... args) throws SQLException {
    try (PreparedStatement ps = connection.prepareStatement(sql)) {
      setParameters(ps, args);
      try (ResultSet rs = ps.executeQuery()) {
        return getResults(rs);
      }
    }
}

  如源代码所示,selectAll详细处理逻辑如下,

  1. 调用Connection对象的prepareStatement()方法获取PreparedStatement对象后;
  2. 调用setParameters()方法为SQL中的参数占位符赋值;
  3. 调用PreparedStatement的executeQuery()方法执行查询操作,并获取查询结果;
  4. 调用getResults()方法,将ResultSet对象转换为List对象,其中List对象中的每个Map对象对应数据库中的一条记录。

setParameters占位符赋值

  在通过Connection对象获取到PreparedStatement对象后,首先调用setParameters方法来对SQL语句中的占位符进行赋值,源代码详情如下,

private void setParameters(PreparedStatement ps, Object... args) throws SQLException {
    for (int i = 0, n = args.length; i < n; i++) {
      if (args[i] == null) {
        throw new SQLException("SqlRunner requires an instance of Null to represent typed null values for JDBC compatibility");
      } else if (args[i] instanceof Null) {
        ((Null) args[i]).getTypeHandler().setParameter(ps, i + 1, null, ((Null) args[i]).getJdbcType());
      } else {
        TypeHandler typeHandler = typeHandlerRegistry.getTypeHandler(args[i].getClass());
        if (typeHandler == null) {
          throw new SQLException("SqlRunner could not find a TypeHandler instance for " + args[i].getClass());
        } else {
          typeHandler.setParameter(ps, i + 1, args[i], null);
        }
      }
    }
}

  如源代码所示,setParameters方法首先会对可变参数args进行遍历,并将得到的值进行如下处理:

  1. 如果该值为null,直接抛出SQLException;
  2. 如果该值为枚举类Null,则寻找Null对应的TypeHandler对象进行参数赋值;
  3. 如果该值不为null,也不为枚举类Null,则寻找该值对应的TypeHandler对象进行参数赋值。

getResults处理结果集

  在组装好SQL语句,并通过PreparedStatement的executeQuery()方法获得查询结果后,此时就该getResults大显身手处理结果集。

 private List<Map<String, Object>> getResults(ResultSet rs) throws SQLException {
    List<Map<String, Object>> list = new ArrayList<>();
    List<String> columns = new ArrayList<>();
    List<TypeHandler<?>> typeHandlers = new ArrayList<>();
    // 获取ResultSetMetaData对象,用于遍历结果
    ResultSetMetaData rsmd = rs.getMetaData();
    for (int i = 0, n = rsmd.getColumnCount(); i < n; i++) {
      columns.add(rsmd.getColumnLabel(i + 1));
      try {
        Class<?> type = Resources.classForName(rsmd.getColumnClassName(i + 1));
        TypeHandler<?> typeHandler = typeHandlerRegistry.getTypeHandler(type);
        if (typeHandler == null) {
          typeHandler = typeHandlerRegistry.getTypeHandler(Object.class);
        }
        typeHandlers.add(typeHandler);
      } catch (Exception e) {
        typeHandlers.add(typeHandlerRegistry.getTypeHandler(Object.class));
      }
    }
    while (rs.next()) {
      Map<String, Object> row = new HashMap<>();
      for (int i = 0, n = columns.size(); i < n; i++) {
        String name = columns.get(i);
        TypeHandler<?> handler = typeHandlers.get(i);
        row.put(name.toUpperCase(Locale.ENGLISH), handler.getResult(rs, name));
      }
      list.add(row);
    }
    return list;
}

  如上述源代码所示,

  1. 通过ResultSet类的getMetaData()方法获取ResultSetMetaData对象;
  2. 通过ResultSetMetaData对象遍历获取所有列的名称和对应的TypeHandler对象,并将其所有遍历来的结果放入集合中待后续使用;
  3. 遍历ResultSet对象,根据步骤2获取的列名和TypeHandler对象,调用TypeHandler对象的getSesult()方法,将JDBC类型转换为Java类型,然后将ResultSet对象中的记录行转换为Map对象,再将结果放入结果集中以待遍历完后返回。

你可能感兴趣的:(mybatis,mybatis,数据库,java)