Druid源码学习（二）-init方法的驱动加载

今天继续看DruidDataSource的init 方法。这个方法从字面意思上看是初始化的含义。也就是说在真正获取连接之前数据源还需要做一些工作。这里我会分几篇文章来总结自己的学习，因为init很复杂。

从头梳理一个问题： Druid连接池如何加载驱动

在介绍Druid之前，先回顾一下JDBC如果获取连接的。JDBC获取连接2步骤：

//1.加载驱动(开发推荐的方式)
Class.forName("com.mysql.jdbc.Driver");

String url = "jdbc:mysql://localhost:3306/shen";
String username = "shen";
String password = "Anxin062039";
Connection conn = null;
            
//2.获取与数据库的链接
conn = DriverManager.getConnection(url, username, password);

这里有个逻辑需要搞清楚：Class.forName 和 DriverManager.getConnection是什么关系？

Class.forName的目的是加载驱动类，加载驱动类时会执行驱动类的静态代码块。静态代码块中就是把本驱动注册到DriverManager中。

public class Driver extends NonRegisteringDriver implements java.sql.Driver {
    public Driver() throws SQLException {
    }

    static {
        try {
            DriverManager.registerDriver(new Driver());
        } catch (SQLException var1) {
            throw new RuntimeException("Can't register driver!");
        }
    }
}

DriverManager类也有静态代码块，这个代码块会调用loadInitialDrivers方法，这个方法可以通过SPI方式加载工程中所有的数据库驱动类。

AccessController.doPrivileged(new PrivilegedAction<Void>() {
            public Void run() {

                ServiceLoader<Driver> loadedDrivers = ServiceLoader.load(Driver.class);
                Iterator<Driver> driversIterator = loadedDrivers.iterator();

                /* Load these drivers, so that they can be instantiated.
                 * It may be the case that the driver class may not be there
                 * i.e. there may be a packaged driver with the service class
                 * as implementation of java.sql.Driver but the actual class
                 * may be missing. In that case a java.util.ServiceConfigurationError
                 * will be thrown at runtime by the VM trying to locate
                 * and load the service.
                 *
                 * Adding a try catch block to catch those runtime errors
                 * if driver not available in classpath but it's
                 * packaged as service and that service is there in classpath.
                 */
                try{
                    while(driversIterator.hasNext()) {
                        driversIterator.next();
                    }
                } catch(Throwable t) {
                // Do nothing
                }
                return null;
            }
        });

那么最后一步就是getConnection了。在DriverManager中的getConnection方法是调用的Driver的connect方法来获取连接的。核心代码如下：

 Connection con = aDriver.driver.connect(url, info);

至此，JDBC获取连接的逻辑梳理完毕了。那么再来看Druid的init方法。

init方法在开始的时候会调用DruidDriver.getInstance(); 这就是上一篇文章中提到的为了修复一个BUG。之前版本（1.1.13）在多线程的情况下，init方法容易造成死锁的情况出现。这里既然是DruidDriver自然联想到是驱动。那么这里和其他的启动逻辑应该是一样的。内部会调用DriverManager的方式，那么也就会使用SPI方式加载所有的驱动。代码如下：

 static {
        AccessController.doPrivileged(new PrivilegedAction<Object>() {
            @Override
            public Object run() {
                registerDriver(instance);
                return null;
            }
        });
    }


。。。

public static boolean registerDriver(Driver driver) {
        try {
            DriverManager.registerDriver(driver);

            try {
                MBeanServer mbeanServer = ManagementFactory.getPlatformMBeanServer();

                ObjectName objectName = new ObjectName(MBEAN_NAME);
                if (!mbeanServer.isRegistered(objectName)) {
                    mbeanServer.registerMBean(instance, objectName);
                }
            } catch (Throwable ex) {
                if (LOG == null) {
                    LOG = LogFactory.getLog(DruidDriver.class);
                }
                LOG.warn("register druid-driver mbean error", ex);
            }

            return true;
        } catch (Exception e) {
            if (LOG == null) {
                LOG = LogFactory.getLog(DruidDriver.class);
            }
            
            LOG.error("registerDriver error", e);
        }

        return false;
    }

这个为了解决BUG的方法同时也完成了工程中数据库驱动的加载。那么往后看init逻辑，在流程中有一个叫resolveDriver方法。这个方法就是解析驱动的方法。这个方法使用了JdbcUtil来创建Driver代码如下：

driver = JdbcUtils.createDriver(driverClassLoader, driverClass);

工具类主要就是通过classload来加载类，最后通过Instance来实例化驱动类。

 public static Driver createDriver(ClassLoader classLoader, String driverClassName) throws SQLException {
        Class<?> clazz = null;
        if (classLoader != null) {
            try {
                clazz = classLoader.loadClass(driverClassName);
            } catch (ClassNotFoundException e) {
                // skip
            }
        }

        if (clazz == null) {
            try {
                ClassLoader contextLoader = Thread.currentThread().getContextClassLoader();
                if (contextLoader != null) {
                    clazz = contextLoader.loadClass(driverClassName);
                }
            } catch (ClassNotFoundException e) {
                // skip
            }
        }

        if (clazz == null) {
            try {
                clazz = Class.forName(driverClassName);
            } catch (ClassNotFoundException e) {
                throw new SQLException(e.getMessage(), e);
            }
        }

        try {
            return (Driver) clazz.newInstance();
        } catch (IllegalAccessException e) {
            throw new SQLException(e.getMessage(), e);
        } catch (InstantiationException e) {
            throw new SQLException(e.getMessage(), e);
        }
    }

在上述逻辑中clazz = Class.forName(driverClassName);不会被执行，因为之前已经加载了驱动类。至此，在Init 方法中数据库驱动被实例化并进行赋值。

总结：

1.复习了JDBC创建连接的逻辑。梳理清楚了class.forName方法与DriverManager方法的关系

2.介绍了Init方法中加载驱动，以及实例化驱动的逻辑。

3.复习了SPI扩展方式，这种方式可以应用在很多场景中。我之前的CAT与RocketMQ集成就用到了SPI扩展。