SPI 机制-插件化扩展功能

andyxh

发布于 2019-09-05 16:53:31

9130

发布于 2019-09-05 16:53:31

SPI(Service Provider Interfaces),中文直译服务提供者接口，一种服务发现机制。可能很多人都不太熟悉这个机制，但是平常或多或少都用到了这个机制，比如我们使用 JDBC 连接操作数据库的时候。

SPI 主要适用于功能扩展的场景，如一些框架提供某一部分功能可以由第三方开发人员扩展，满足其自身业务需求。

假设我们在公司内实现了一个统一登陆框架，框架内部仅仅提供用户名/密码登陆方式。后来 A 部门想使用该框架，但是他们想增加微信登陆授权。正常情况下，我们可以改动登陆框架代码，增加微信登陆实现方式。如果后面又增加 QQ 登陆，淘宝登陆那？也只能不断相应的实现。

SPI 实现方式

这种情况如果使用 SPI，可以在不用改动框架代码前提下，增加新的登陆实现方式。下面用代码演示如何使用 SPI。

定义接口

首先我们新建一个 maven 项目 oauth-api，在这个项目创建一个公共接口。

public interface OauthLoginService {
    void login();
}

第三方实现该接口

再新建一个 maven 项目 wechat-oauth ，引入上面 oauth-api 依赖

public class WechatLoginService implements OauthLoginService {
    @Override
    public void login() {
        System.out.println("使用微信登陆授权");
    }
}

定义配置文件

SPI 需要将接口实现定义在配置文件中，文件名为接口全名称，如 com.andyxh.OauthLoginService，配置文件需放在 resources\META-INF\services 文件夹下。文件内容如下：

com.another.WechatLoginService

加载接口实现类

新建 maven 项目 oauth-login,在这个项目中引入 wechat-oauth 与 oauth-api 依赖。SPI 核心将会使用 java.util.ServiceLoader读取上面上面定义配置文件，加载所有服务实现类。使用代码如下：

ServiceLoader<OauthLoginService> serviceLoader=ServiceLoader.load(OauthLoginService.class);
serviceLoader.forEach(OauthLoginService::login);

打印结果：

使用微信登陆授权

SPI 实际应用

上面说过 JDBC 中使用到 SPI 进制。 JDK 定义标准数据库接口，相应的数据库厂商实现这类接口。以 mysql-connector-javal 为例。

        <dependency>
            <groupId>mysql</groupId>
            <artifactId>mysql-connector-java</artifactId>
            <version>8.0.16</version>
        </dependency>

mysql jar 包 META-INF/services 中存在java.sql.Driver 文件，这个文件定义了实现类。

com.mysql.cj.jdbc.Driver

可以看到 java.sql.Driver 是标准 SPI 接口，而 com.mysql.cj.jdbc.Driver 是 mysql 标准实现接口。

何时加载 java.sql.Driver？

我们将会使用 DriverManager.getConnection 获取相应数据库连接。这个类内部存在一个静态代码块，将会使用 ServiceLoader 加载实现类。

    static {
        loadInitialDrivers();
        println("JDBC DriverManager initialized");
    }

 private static void loadInitialDrivers() {
        ....
     ServiceLoader<Driver> loadedDrivers = ServiceLoader.load(Driver.class);  
     Iterator<Driver> driversIterator = loadedDrivers.iterator();  
     try{  
         while(driversIterator.hasNext()) {  
         driversIterator.next();  
     }  
     } catch(Throwable t) {  
         // Do nothing  
     }  
         return null;  
     }
    ....
    }