动态加载jdbc驱动(可测试jdbc不同版本兼容性)

缘由

接到一个新任务，需要测试应用对mysql jdbc不同版本驱动的兼容性。这里希望能够以代码的形式，手动加载不同版本的mysql jdbc驱动，比如有下列的驱动版本，我要一个一个进行测试：

• mysql-connector-java-5.1.6.jar
• mysql-connector-java-5.1.31.jar
• mysql-connector-java-5.1.35.jar
• mysql-connector-java-5.1.39.jar
• mysql-connector-java-6.0.3.jar

实现

参考stackoverflow上做法:

http://stackoverflow.com/questions/14478870/dynamically-load-the-jdbc-driver

觉得挺有意思的，写了如下代码，可以实现动态加载不同版本的jdbc驱动进行一些测试:

import java.net.URL;
import java.net.URLClassLoader;
import java.sql.Connection;
import java.sql.Driver;
import java.sql.DriverManager;
import java.sql.DriverPropertyInfo;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.sql.SQLFeatureNotSupportedException;
import java.sql.Statement;
import java.util.HashMap;
import java.util.Map;
import java.util.Properties;
import java.util.logging.Logger;

public class MycatMulitJdbcVersionTest {

    private static final String JDBC_URL = "jdbc:mysql://localhost:3306/testdb?serverTimezone=UTC&useSSL=false";
    private static final String USER = "root";
    private static final String PASSWORD = "mysql";
    private static final Map<String, String> jdbcVersionMap = new HashMap<String, String>();
    private static final Map<String, Driver> tmpDriverMap = new HashMap<String, Driver>();

    // 动态加载jdbc驱动
    private static void dynamicLoadJdbc(String mysqlJdbcFile) throws Exception {
        URL u = new URL("jar:file:lib/" + mysqlJdbcFile + "!/");
        String classname = jdbcVersionMap.get(mysqlJdbcFile);
        URLClassLoader ucl = new URLClassLoader(new URL[] { u });
        Driver d = (Driver)Class.forName(classname, true, ucl).newInstance();
        DriverShim driver = new DriverShim(d);
        DriverManager.registerDriver(driver);
        tmpDriverMap.put(mysqlJdbcFile, driver);
    }

    // 每一次测试完卸载对应版本的jdbc驱动
    private static void dynamicUnLoadJdbc(String mysqlJdbcFile) throws SQLException {
        DriverManager.deregisterDriver(tmpDriverMap.get(mysqlJdbcFile));
    }

    // 进行一次测试
    private static void testOneVersion(String mysqlJdbcFile) {

        System.out.println("start test mysql jdbc version : " + mysqlJdbcFile);

        try {
            dynamicLoadJdbc(mysqlJdbcFile);
        } catch (Exception e1) {
            e1.printStackTrace();
        }

        Connection conn = null;
        try {
            conn = DriverManager.getConnection(JDBC_URL, USER, PASSWORD);
            Statement stmt = conn.createStatement();
            ResultSet rs = stmt.executeQuery("select user()");
            System.out.println("select user() output : ");
            while(rs.next()) {
                System.out.println(rs.getObject(1));
            }
            rs = stmt.executeQuery("show tables");
            System.out.println("show tables output : ");
            while(rs.next()) {
                System.out.println(rs.getObject(1));
            }
        } catch (SQLException e) {
            e.printStackTrace();
        } finally {
            if(conn != null) {
                try {
                    conn.close();
                } catch (SQLException e) {
                    e.printStackTrace();
                }
            }
        }

        try {
            dynamicUnLoadJdbc(mysqlJdbcFile);
        } catch (SQLException e) {
            e.printStackTrace();
        }

        System.out.println("end !!!");
        System.out.println();
    }

    public static void main(String[] args) {

        jdbcVersionMap.put("mysql-connector-java-6.0.3.jar", "com.mysql.cj.jdbc.Driver");
        jdbcVersionMap.put("mysql-connector-java-5.1.6.jar", "com.mysql.jdbc.Driver");
        jdbcVersionMap.put("mysql-connector-java-5.1.31.jar", "com.mysql.jdbc.Driver");
        jdbcVersionMap.put("mysql-connector-java-5.1.35.jar", "com.mysql.jdbc.Driver");
        jdbcVersionMap.put("mysql-connector-java-5.1.39.jar", "com.mysql.jdbc.Driver");

        for(String mysqlJdbcFile : jdbcVersionMap.keySet()) {
            testOneVersion(mysqlJdbcFile);
        }

    }

}

class DriverShim implements Driver {
    private Driver driver;
    DriverShim(Driver d) { this.driver = d; }
    public boolean acceptsURL(String u) throws SQLException {
        return this.driver.acceptsURL(u);
    }
    public Connection connect(String u, Properties p) throws SQLException {
        return this.driver.connect(u, p);
    }
    @Override
    public DriverPropertyInfo[] getPropertyInfo(String url, Properties info) throws SQLException {
        return this.driver.getPropertyInfo(url, info);
    }
    @Override
    public int getMajorVersion() {
        return this.driver.getMajorVersion();
    }
    @Override
    public int getMinorVersion() {
        return this.driver.getMinorVersion();
    }
    @Override
    public boolean jdbcCompliant() {
        return this.driver.jdbcCompliant();
    }
    @Override
    public Logger getParentLogger() throws SQLFeatureNotSupportedException {
        return this.driver.getParentLogger();
    }
}