Java中SPI机制原理解析

news2024/12/28 22:18:39

使用SPI机制前后的代码变化

加载MySQL对JDBC的Driver接口实现
在未使用SPI机制之前，使用JDBC操作数据库的时候，一般会写如下的代码：
```
// 通过这行代码手动加载MySql对Driver接口的实现类
Class.forName("com.mysql.jdbc.Driver")
DriverManager.getConnection("jdbc:mysql://127.0.0.1:3306/study", "root", "root");
```
在使用了SPI机制之后，就直接可以调用DriverManager.getConnection()方法获取连接了。
SLF4J查找日志门面实现
在未使用SPI机制之前，SLF4J日志门面在查找具体的日志实现时，需要每个实现提供一个叫org.slf4j.com.impl.StaticLoggerBinder.class类，这里其实就跟SPI机制有点类似了，这个类实现了和SPI机制中的配置文件的功能，通过这个类来实现和具体实现的绑定关系。

使用了SPI机制之后则是通过SLF4JServiceProvider类来实现和具体实现的绑定关系。

SPI代码实现案例

假设现在有一个Plugin的接口，调用方通过PluginFactory中的installPlugins方法来加载实现实现类，并进行调用，代码如下所示：

// Plugin接口
public interface Plugin {
    boolean install(Map<Object, Object> context);
}

public class PluginFactory {
    public void installPlugins() {
        Map<Object, Object> context = new HashMap<>();
        context.put("_beans", new ArrayList<>());
        context.put("_version", "1.0.0");
        context.put("_aspects", new HashMap<>());

        // 这里通过ServiceLoader加载Plugin的实现类
        ServiceLoader<Plugin> loader = ServiceLoader.load(Plugin.class);
        for (Plugin plugin : loader) {
            plugin.install(context);
        }
    }

    public static void main(String[] args) {
        PluginFactory factory = new PluginFactory();
        factory.installPlugins();
    }
}

Plugin接口实现者的代码：

public class LogPlugin implements Plugin {
    private static final Logger LOGGER = LogManager.getLogger();

    @Override
    public boolean install(Map<Object, Object> context) {
        LOGGER.info("Login plugin is initiating...");
        return true;
    }
}