了解一下Java SPI的原理

2019-09-30 07:12:12来源：博客园阅读 ()

了解一下Java SPI的原理

1 为什么写这篇文章？

近期，本人在学习dubbo相关的知识，但是在dubbo官网中有提到Java的 SPI，这个名词之前未接触过，所以就去看了看，感觉还是有很多地方有使用的，比如jdbc、log相关的技术上均有使用，还是很有用处的，就在这里总结一下自己的学习内容！（本文有参考相关资料：比如dubbo官网、相关blog等）

2 SPI是什么？

Java SPI(Service Provider Interface)是JDK内置的一种动态加载扩展点的实现。在ClassPath的META-INF/services目录下放置一个与接口同名的文本文件，文件的内容为接口的实现类，多个实现类用换行符分隔。JDK中使用java.util.ServiceLoader来加载具体的实现。

Java SPI 实际上是“基于接口的编程＋策略模式＋配置文件”组合实现的动态加载机制。

3 自定义一个SPI

3.1 创建工程

创建dubbo-spi的工程，这里展示一下完整的spi示例程序结构：

3.2 创建接口

在包top.flygrk.ishare.spi.service下创建接口： SPIService

package top.flygrk.ishare.spi.service;

/**
 * @Package top.flygrk.ishare.spi.service
 * @Version V1.0
 * @Description: SPIService 接口
 */
public interface SPIService {
    /**
     * 接口方法： say()
     */
    String say();
}

3.3 创建实现类： ASPIServiceImpl和BSPIServiceImpl

在包top.flygrk.ishare.spi.service.impl下创建ASPIServiceImpl和BSPIServiceImpl类，均实现SPIservice接口：

ASPIServiceImpl

package top.flygrk.ishare.spi.service.impl;

import top.flygrk.ishare.spi.service.SPIService;

/**
 * @Package top.flygrk.ishare.spi.service.impl
 * @Version V1.0
 * @Description: SPIService 实现类 ASPIServiceImpl
 */
public class ASPIServiceImpl implements SPIService {
    @Override
    public String say() {
        return "ASPIServiceImpl";
    }
}

BSPIServiceImpl

package top.flygrk.ishare.spi.service.impl;

import top.flygrk.ishare.spi.service.SPIService;

/**
 * @Package top.flygrk.ishare.spi.service.impl
 * @Version V1.0
 * @Description: SPIService 实现类 BSPIServiceImpl
 */
public class BSPIServiceImpl implements SPIService {
    @Override
    public String say() {
        return "BSPIServiceImpl";
    }
}

3.4 创建文件top.flygrk.ishare.spi.service.SPIService

在resource目录下，创建META-INF/services目录，并在该目录下创建top.flygrk.ishare.spi.service.SPIService文件（该文件名为接口的全路径，需保持一致），并在该文件中配置两个实现类的全路径：

top.flygrk.ishare.spi.service.impl.ASPIServiceImpl
top.flygrk.ishare.spi.service.impl.BSPIServiceImpl

3.5 创建测试类TestSPIService

在包top.flygrk.ishare.demo下创建TestSPIService类，用于测试该SPI服务

package top.flygrk.ishare.demo;

import top.flygrk.ishare.spi.service.SPIService;

import java.util.Iterator;
import java.util.ServiceLoader;

/**
 * @Package top.flygrk.ishare.demo
 * @Version V1.0
 * @Description: 测试 SPIService
 */
public class TestSPIService {

    public static void main(String[] args) {
        // ServiceLoader实现了Iterable接口，可以遍历出所有的服务实现者
        ServiceLoader<SPIService> serviceLoaders = ServiceLoader.load(SPIService.class);

        /*
         * 方法1： 迭代器
         */
        Iterator<SPIService> spiServiceIterator = serviceLoaders.iterator();
        while (spiServiceIterator != null && spiServiceIterator.hasNext()) {
            SPIService spiService = spiServiceIterator.next();
            System.out.println(spiService.getClass().getName() + " : " + spiService.say());
        }


        /*
         * 迭代方法2： foreach
         */
//        for (SPIService spiService : serviceLoaders) {
//            System.out.println(spiService.getClass().getName() + " : " + spiService.say());
//        }

    }

}

3.6 测试类运行结果

top.flygrk.ishare.spi.service.impl.ASPIServiceImpl : ASPIServiceImpl
top.flygrk.ishare.spi.service.impl.BSPIServiceImpl : BSPIServiceImpl

4 SPI原理分析

在我们阅读源码前，我们先提出以下几个问题，然后我们再去带着问题去源码中找答案：

1) META-INF/services目录下的文件有什么用？为什么要用接口的全路径命名？是否可以更改接口名称？里面的内容为什么要用实现类的全路径？
2） ServiceLoader 是如何获取到SPIService的全部实现的？
3）如果我们只想取ASPIServiceImpl，并不想去操作BSPIServiceImpl，如何去操作？

4.1 ServiceLoader结构

我们先看一下ServiceLoader类的结构：

进入ServiceLoader类的源码，我们可以看到以下定义的一些常量：

各位肯定注意到了一点： private static final String PREFIX = "META-INF/services/";，这个PREFIX后面的路径不正是我们在上述示例中创建和接口保持一致的文件的目录吗？还有services、loader、acc、lookupIterator和providers表达的意思在源码上方的注释中也进行了描述，下面我将各个属性的释义标注一下：

// 配置文件的目录
private static final String PREFIX = "META-INF/services/";
// 要加载服务的类或者接口
// The class or interface representing the service being loaded
private final Class<S> service;
// 服务加载器
// The class loader used to locate, load, and instantiate providers
private final ClassLoader loader;
// 访问控制上下文
// The access control context taken when the ServiceLoader is created
private final AccessControlContext acc;
// 服务实例的缓存
// Cached providers, in instantiation order
private LinkedHashMap<String,S> providers = new LinkedHashMap<>();
// 懒加载的迭代器
// The current lazy-lookup iterator
private LazyIterator lookupIterator;

4.2 ServiceLoader的加载过程

看完了上面ServiceLoader的结构，下面我们再来看看ServiceLoader是如何一步步加载的。我们在TestSPIService类上的main方法第一行打上断点：

然后使用debug的方式调试，进入ServiceLoader的源码，会依次进入以下几个函数：

经过这些步骤之后，serviceLoader内部包含有一个Iterator迭代器，下面我们来仔细看一下这个迭代器的作用！

4.3 迭代器lookupIterator的操作

在上述4.2步骤加载完成之后，serviceLoader内的lookupIterator的内容如下：

然后使用iterator()方法获取Iterator迭代器时，执行如下的程序：

file

在经过上述过程之后，我们拿到了Iterator迭代器，这时我们看下spiServiceIterator的内容：

是不是很奇怪，还是只有SPIService，不要忘记了，他内部的迭代器可是懒加载的！我们继续跟进代码，进入到hasNext()方法。

从上面可以知道，acc一直为null的，所以这时候，他进入了hasNextService()方法：

重头戏来了，我们可以看到其中的 PREFIX，这个内容就是我们配置的文件。再仔细的跟进代码，我们会进入到parse()方法，该方法用于按照行读取出文件中的内容，并保存到Iterator<String> 中。

故而，再通过 nextName = pending.next(); 执行后，获取到top.flygrk.ishare.spi.service.impl.ASPIServiceImpl，继而进行后续的next()方法操作。

然后进入到nextService()方法：

再nextService()方法里，使用了反射的技术，根据前面从文件中读取到的实现类全路径top.flygrk.ishare.spi.service.impl.ASPIServiceImpl获取到该实现类的对象！走到这里，也就基本上了解了SPI，但是我们能只获取ASPIServiceImpl，而不去获取BSPIServiceImpl吗？对不起，这里不允许这样，只能通过迭代器遍历出所有的内容！除非人为干预（外层循环比对完成之后退出循环）。接下来的步骤就和前面几乎一致了，这里不再细述～