SPI(Service Provider Interface),是JDK内置的一种服务提供发现机制,可以用来启用框架扩展和替换组件,主要是被框架的开发人员使用。

什么是SPI机制

SPI(Service Provider Interface),是JDK内置的一种服务提供发现机制,可以用来启用框架扩展和替换组件,主要是被框架的开发人员使用,比如java.sql.Driver接口,其他不同厂商可以针对同一接口做出不同的实现,MySQL和PostgreSQL都有不同的实现提供给用户,而Java的SPI机制可以为某个接口寻找服务实现。Java中SPI机制主要思想是将装配的控制权移到程序之外,在模块化设计中这个机制尤其重要,其核心思想就是 解耦

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

SPI机制的简单示例

网上找了个例子:Java SPI思想梳理

我们现在需要使用一个内容搜索接口,搜索的实现可能是基于文件系统的搜索,也可能是基于数据库的搜索。

  • 先定义好接口
  1. public interface Search {
  2. public List<String> searchDoc(String keyword);
  3. }
  • 文件搜索实现
  1. public class FileSearch implements Search{
  2. @Override
  3. public List<String> searchDoc(String keyword) {
  4. System.out.println("文件搜索 "+keyword);
  5. return null;
  6. }
  7. }
  • 数据库搜索实现
  1. public class DatabaseSearch implements Search{
  2. @Override
  3. public List<String> searchDoc(String keyword) {
  4. System.out.println("数据搜索 "+keyword);
  5. return null;
  6. }
  7. }
  • resources 接下来可以在resources下新建META-INF/services/目录,然后新建接口全限定名的文件:com.cainiao.ys.spi.learn.Search,里面加上我们需要用到的实现类
  1. com.cainiao.ys.spi.learn.FileSearch
  • 测试方法
  1. public class TestCase {
  2. public static void main(String[] args) {
  3. ServiceLoader<Search> s = ServiceLoader.load(Search.class);
  4. Iterator<Search> iterator = s.iterator();
  5. while (iterator.hasNext()) {
  6. Search search = iterator.next();
  7. search.searchDoc("hello world");
  8. }
  9. }
  10. }

可以看到输出结果:文件搜索 hello world
如果在com.cainiao.ys.spi.learn.Search文件里写上两个实现类,那最后的输出结果就是两行了。
这就是因为ServiceLoader.load(Search.class)在加载某接口时,会去META-INF/services下找接口的全限定名文件,再根据里面的内容加载相应的实现类。
这就是spi的思想,接口的实现由provider实现,provider只用在提交的jar包里的META-INF/services下根据平台定义的接口新建文件,并添加进相应的实现类内容就好。

SPI机制的广泛应用

SPI机制 - JDBC DriverManager

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

JDBC接口定义

首先在java中定义了接口java.sql.Driver,并没有具体的实现,具体的实现都是由不同厂商来提供的。

mysql实现

在mysql的jar包mysql-connector-java-6.0.6.jar中,可以找到META-INF/services目录,该目录下会有一个名字为java.sql.Driver的文件,文件内容是com.mysql.cj.jdbc.Driver,这里面的内容就是针对Java中定义的接口的实现。

postgresql实现

同样在postgresql的jar包postgresql-42.0.0.jar中,也可以找到同样的配置文件,文件内容是org.postgresql.Driver,这是postgresql对Java的java.sql.Driver的实现。

使用方法

上面说了,现在使用SPI扩展来加载具体的驱动,我们在Java中写连接数据库的代码的时候,不需要再使用Class.forName("com.mysql.jdbc.Driver")来加载驱动了,而是直接使用如下代码:

  1. String url = "jdbc:xxxx://xxxx:xxxx/xxxx";
  2. Connection conn = DriverManager.getConnection(url,username,password);
  3. .....

SPI机制深入理解

SPI机制通常怎么使用

看完上面的几个例子解析,应该都能知道大概的流程了:

  • 有关组织或者公司定义标准。
  • 具体厂商或者框架开发者实现。
  • 程序猿使用。

定义标准

定义标准,就是定义接口。比如接口java.sql.Driver

具体厂商或者框架开发者实现

厂商或者框架开发者开发具体的实现:
META-INF/services目录下定义一个名字为接口全限定名的文件,比如java.sql.Driver文件,文件内容是具体的实现名字,比如me.cxis.sql.MyDriver
写具体的实现me.cxis.sql.MyDriver,都是对接口Driver的实现。

程序猿使用

我们会引用具体厂商的jar包来实现我们的功能:

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

SPI和API的区别是什么

这里实际包含两个问题,第一个SPI和API的区别?第二个什么时候用API,什么时候用SPI?

SPI - “接口”位于“调用方”所在的“包”中

  • 概念上更依赖调用方。
  • 组织上位于调用方所在的包中。
  • 实现位于独立的包中。
  • 常见的例子是:插件模式的插件。

API - “接口”位于“实现方”所在的“包”中

通过上面的解析,可以发现,我们使用SPI机制的缺陷:

  • 不能按需加载,需要遍历所有的实现,并实例化,然后在循环中才能找到我们需要的实现。如果不想用某些实现类,或者某些类实例化很耗时,它也被载入并实例化了,这就造成了浪费。
  • 获取某个实现类的方式不够灵活,只能通过Iterator形式获取,不能根据某个参数来获取对应的实现类。
  • 多个并发多线程使用ServiceLoader类的实例是不安全的。