0
点赞
收藏
分享

微信扫一扫

DBUtils源码分析

肉肉七七 2022-08-19 阅读 68


其实,在这篇文章里,我只是分析了dbutis的query的运作流程。

至于类为什么要这样设计,蕴含的设计模式等等高级知识点咱们在下节再探讨。

先看看最简单的DBUtils是如何工作的。

 数据库里有一张表,student,里面就三个属性 姓名,学号,出生日期( xm,xh,birth)其中前两个是vchar,birth是date;



package dbutils;

import model.Student;

import org.apache.commons.dbutils.QueryRunner;
import org.apache.commons.dbutils.handlers.BeanHandler;
import org.apache.commons.dbutils.ResultSetHandler;
import org.junit.Test;

import java.sql.SQLException;

public class BeanExample {

@Test
public void testBean() {

QueryRunner qr = new QueryRunner(new MyDBSource());

String sql = "select * from student where xh=?";
Object params[] = { "02" };
Student s=null;
try {
ResultSetHandler<Student> rsh=new BeanHandler<>(Student.class);
s = qr.query(sql,rsh,params);
} catch (SQLException e) {
// TODO Auto-generated catch block
e.printStackTrace();
}
System.out.println(s.getXm());

}

}



其中,MyDBSource就是个提供数据源的工具而已。


public class MyDBSource implements DataSource {

private static String driverClassName = "com.mysql.jdbc.Driver";
private static String url = "jdbc:mysql://localhost:3306/webexample";

private static String userName = "root";
private static String passWord = "root";

@Override
public Connection getConnection() throws SQLException {
try {
Class.forName(driverClassName);
} catch (ClassNotFoundException e) {
// TODO Auto-generated catch block
e.printStackTrace();
}
return DriverManager.getConnection(url, userName, passWord);

}
//....
}

我们再看看BeanHandler的内部。


//BeanHandler.java
public BeanHandler(Class<T> type) {
this(type, ArrayHandler.ROW_PROCESSOR);
}
public BeanHandler(Class<T> type, RowProcessor convert) {
this.type = type;
this.convert = convert;
}

//ArrayHandler.java
public class ArrayHandler implements ResultSetHandler<Object[]> {
static final RowProcessor ROW_PROCESSOR = new BasicRowProcessor();
//....
}

//BasicRowProcessor.java
public class BasicRowProcessor implements RowProcessor {
private static final BeanProcessor defaultConvert = new BeanProcessor();

public BasicRowProcessor() {
this(defaultConvert);
}
//...
}



    ok,我们可以看到在BeanHandler中的convert,最终是BeanProcessor类型。



再下来就是我们的重头戏了。


s =  qr.query(sql,rsh,params);

我们看时序图

DBUtils源码分析_数据库

 


   1.1 prepareStatement(conn, sql);


protected PreparedStatement prepareStatement(Connection conn, String sql)
throws SQLException {
return conn.prepareStatement(sql);
}

    就是生成prepareStatement


    1.2 fillStatement(stmt, params)


    大家就是看名字也该知道,fillStatement就是填参数


    其核心代码如下:


for (int i = 0; i < params.length; i++) {
if (params[i] != null) {
stmt.setObject(i + 1, params[i]);
}
//...
}

     当然fillStatement在核心代码上面还有一块,就是检查sql中的问号数量与params的长度是否相等。


     1.3 handle(rs)


BeanHandler.java
public T handle(ResultSet rs) throws SQLException {
return rs.next() ? this.convert.toBean(rs, this.type) : null;
}

      在最开始分析BeanHandler的构造函数时,我们就已经知道了convert是BeanProcessor。


     1.3.1 toBean(rs, this.type)


public <T> T toBean(ResultSet rs, Class<T> type) throws SQLException {
PropertyDescriptor[] props = this.propertyDescriptors(type);
ResultSetMetaData rsmd = rs.getMetaData();
int[] columnToProperty = this.mapColumnsToProperties(rsmd, props);

return this.createBean(rs, type, props, columnToProperty);
}

      1.3.1.1 propertyDescriptors(type)


      这是运用内省(Introspector)获得类型(就是代码里的Student.class)的属性(property)


      1.3.1.2 mapColumnsToProperties(rsmd, props)


      这一步比较麻烦,为什么说麻烦呢。


      数据库里,一张表上有字段,现在这些字段存放在ResultSetMetaData里面


      在我们的bean里面,有属性(property),现在存放在props这个数组里。



      columnToProperty这里面就放的是字段与属性的对应关系。


      例如 columnToProperty[3]=4 就是说ResultSetMetaData里的第三个字段对应于bean的PropertyDescriptor里面的第四个属性。



      1.3.1.3 createBean(rs, type, props, columnToProperty)


 

//对源码略微有删改
//但绝对不影响核心思想
private <T> T createBean(ResultSet rs, Class<T> type,
PropertyDescriptor[] props, int[] columnToProperty)
throws SQLException {

T bean = this.newInstance(type);

for (int i = 1; i < columnToProperty.length; i++) {

PropertyDescriptor prop = props[columnToProperty[i]];
Class<?> propType = prop.getPropertyType();

Object value = null;
if(propType != null)
value = this.processColumn(rs, i, propType);

this.callSetter(bean, prop, value);
}

return bean;
}

    在createBean里面


    this.processColumn(rs, i, propType)


    就是获得value,那么我们已经知道resultset,还有这个value在resultset中的序列号还有value类型,那么该如何获取呢?


    代码我就不贴了,大家自己看源码吧,看上5秒钟就能知道内部逻辑了。



    1.3.1.3.1 callSetter(bean, prop, value)


    这里面的核心就是下面的代码


// Don't call setter if the value object isn't the right type
if (this.isCompatibleType(value, params[0])) {
setter.invoke(target, new Object[]{value});
} else {
throw new SQLException(
"Cannot set " + prop.getName() + ": incompatible types, cannot convert "
+ value.getClass().getName() + " to " + params[0].getName());
// value cannot be null here because isCompatibleType allows null
}



    如果value是个String,params却是个double那就得抛出异常了。



举报

相关推荐

0 条评论