程序员

Java SPI 机制解析

2018-09-28  本文已影响3993人  匠丶

本文以JDBC为例深入讲解 java spi 机制,将帮助你理解:什么是SPI,SPI实现原理,SPI的使用和SPI的优劣。

什么是SPI

SPI,Service Provider Interface,主要用于框架设计。比如java.sql.Driver接口,不同厂商可以针对同一接口做出不同的实现,mysql和oracle都有不同的实现提供给用户,而Java的SPI机制可以为某个接口寻找服务实现。

当服务的提供者提供了一种接口的实现之后,需要在classpath下的META-INF/services/目录里创建一个以服务接口命名的文件,这个文件里的内容就是这个接口的具体的实现类。当其他的程序需要这个服务的时候,就可以通过查找这个jar包(一般都是以jar包做依赖)的META-INF/services/中的配置文件,配置文件中有接口的具体实现类名,可以根据这个类名进行加载实例化,就可以使用该服务了。例如mysql的jar包:


SPI实现原理

下面以JDBC为例深入讲解spi的实现原理:
在JDBC4.0之前,我们开发有连接数据库的时候,通常会用Class.forName("com.mysql.jdbc.Driver")这句先加载数据库相关的驱动,然后再进行获取连接等的操作。而JDBC4.0之后不需要用Class.forName("com.mysql.jdbc.Driver")来加载驱动,直接获取连接就可以了,现在这种方式就是使用了Java的SPI扩展机制来实现。

           //1、加载驱动
            //Class.forName("com.mysql.jdbc.Driver");
            //2、建立连接
            Connection connection = DriverManager.getConnection("url", "user", "passport");
            //3、写prepareStatement
            PreparedStatement preparedStatement = connection.prepareStatement("SELECT * from table where 1=1 limit 1 ");
            //4、执行sql
            ResultSet resultSet = preparedStatement.executeQuery();

1、进入到DriverManager
DriverManager中有一段静态代码块,用于初始化驱动。

    /**
     * Load the initial JDBC drivers by checking the System property
     * jdbc.properties and then use the {@code ServiceLoader} mechanism
     */
    static {
        loadInitialDrivers();
        println("JDBC DriverManager initialized");
    }

为了能够更清晰的看到代码的功能,我们截取部分关键代码分析。

 private static void loadInitialDrivers() {
//使用SPI的ServiceLoader来加载接口的实现
  ServiceLoader<Driver> loadedDrivers = ServiceLoader.load(Driver.class);
//获取迭代器
  Iterator driversIterator = loadedDrivers.iterator();
//遍历所有的驱动实现
   try{
                    while(driversIterator.hasNext()) {
                        driversIterator.next();
                    }
                } catch(Throwable t) {
                // Do nothing
                }

 println("DriverManager.initialize: jdbc.drivers = " + drivers);

  for (String aDriver : driversList) {
            try {
                println("DriverManager.Initialize: loading " + aDriver);
                Class.forName(aDriver, true,
                        ClassLoader.getSystemClassLoader());
            } catch (Exception ex) {
                println("DriverManager.Initialize: load failed: " + ex);
            }
        }

}

2、进入到ServiceLoader,ServiceLoader是查找服务的实现工具
ServiceLoaderd定义了查找实现类的目录位置,和实现了Iterator的内部类LazyIterator。LazyIterator重写了hasNext和next方法,将类的加载延迟到了DriverManager调用的时候(也就是所谓的Lazy)。

DriverManager在遍历的时候,首先调用driversIterator.hasNext()方法,这里会搜索classpath下以及jar包中所有的META-INF/services目录下的java.sql.Driver文件,并找到文件中的实现类的名字,此时并没有实例化具体的实现类。然后是调用driversIterator.next();方法,此时就会根据驱动名字具体实例化各个实现类了。现在驱动就被找到并实例化了。

     //查找配置文件的目录
    private static final String PREFIX = "META-INF/services/";

    // The class or interface representing the service being loaded
    //表示要被加载的服务的类或接口
    private Class<S> service;

    // The class loader used to locate, load, and instantiate providers
    private ClassLoader loader;

    // Cached providers, in instantiation order
   // 缓存已经被实例化的服务提供者,按照实例化的顺序存储
    private LinkedHashMap<String,S> providers = new LinkedHashMap<>();

    // The current lazy-lookup iterator
    private LazyIterator lookupIterator;

   //为指定的服务使用指定的类加载器来创建一个ServiceLoader,去加载实现了java.sql.Driver接口的类
    public static <S> ServiceLoader<S> load(Class<S> service) {
        ClassLoader cl = Thread.currentThread().getContextClassLoader();
        return ServiceLoader.load(service, cl);
    }

    private ServiceLoader(Class<S> svc, ClassLoader cl) {
        service = svc;
        loader = cl;
        reload();
    }

 //重新加载,就相当于重新创建ServiceLoader了,用于新的服务提供者安装到正在运行的Java虚拟机中的情况
    public void reload() {
 //清空缓存中所有已实例化的服务提供者
        providers.clear();
//新建一个迭代器,该迭代器会从头查找和实例化服务提供者
        lookupIterator = new LazyIterator(service, loader);
    }

ServiceLoader不是实例化以后,就去读取配置文件中的具体实现,并进行实例化。而是等到使用迭代器去遍历的时候,才会加载对应的配置文件去解析,调用hasNext方法的时候会去加载配置文件进行解析,调用next方法的时候进行实例化并缓存。所有的配置文件只会加载一次,服务提供者也只会被实例化一次,重新加载配置文件可使用reload方法。

SPI的使用

SPI的使用可以仿照DriverManager的方式。如下:

ServiceLoader<Driver> loadedDrivers = ServiceLoader.load(xxx.class);
//获取迭代器
Iterator<Driver> driversIterator = loadedDrivers.iterator();
//遍历
while(driversIterator.hasNext()) {
    driversIterator.next();
    //可以做具体的业务逻辑
}

SPI的缺点

通过上面的分析可以发现,SPI需要把所有的实现都实例化了,即便我们不需要,也都给实例化了。引用dubbo文档的说法是:

  • JDK 标准的 SPI 会一次性实例化扩展点所有实现,如果有扩展实现初始化很耗时,但如果没用上也加载,会很浪费资源。
  • 如果扩展点加载失败,连扩展点的名称都拿不到了。比如:JDK 标准的 ScriptEngine,通过 getName() 获取脚本类型的名称,但如果 RubyScriptEngine 因为所依赖的 jruby.jar 不存在,导致 RubyScriptEngine 类加载失败,这个失败原因被吃掉了,和 ruby 对应不起来,当用户执行 ruby 脚本时,会报不支持 ruby,而不是真正失败的原因。
上一篇下一篇

猜你喜欢

热点阅读