位置：首页>> 软件编程>> java编程>> java开发ServiceLoader实现机制及SPI应用

java开发ServiceLoader实现机制及SPI应用

作者：梦想实现家_Z　　发布时间：2022-12-24 09:55:05　

标签：java,ServiceLoader,SPI

前言

做过java web开发的小伙伴大多数时候都需要链接数据库，这个时候就需要配置数据库引擎DriverClassName参数，这样我们的java应用才能通过数据库厂商给的Driver与指定的数据库建立通信。但是这里就有一个疑问：

java.sql.Driver是jdk自带的接口，它是由BoostrapClassLoader加载的，DriverClassName是外部厂商提供的具体实现，是由AppClassLoader加载的，要建立与数据库的通信，必然是通过java.sql.Driver接口方法发起的，那么在java.sql.Driver是如何拿到具体实现的呢？它是不是违背了ClassLoader的双亲委派模式呢？

如何绕过双亲委派模式

为了拿到AppClassLoader中加载的java.sql.Driver实现类，我们可以查看一下DriverManager是怎么处理的：

public static Driver getDriver(String url)
throws SQLException {
println("DriverManager.getDriver("" + url + "")");
ensureDriversInitialized();
......
}
private static void ensureDriversInitialized() {
......
AccessController.doPrivileged(new PrivilegedAction<Void>() {
public Void run() {
// 核心代码ServiceLoader
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;
}
});
......
}

我们最终可以发现，DriverManager通过ServiceLoader.load(Driver.class)就拿到了我们配置的DriverClassName实现类。这就实现在DriverManager中拿到了外部提供的Driver实现，绕过来双亲委派模式。

ServiceLoader实现机制

我们来看一下ServiceLoader是如何实现SPI机制的，先从ServiceLoader.load()方法入手：

public static <S> ServiceLoader<S> load(Class<S> service) {
// 从当前线程中获取ClassLoader
ClassLoader cl = Thread.currentThread().getContextClassLoader();
// 创建一个ServiceLoader对象
return new ServiceLoader<>(Reflection.getCallerClass(), service, cl);
}

1.从当前线程中获取ClassLoader；因为在创建AppClassLoader后，将AppClassLoader设置进当前线程的上下文中；

2.根据ClassLoader以及目标接口类创建一个ServiceLoader对象；

其实ServiceLoader核心代码在hasNext()方法中：

@Override
public boolean hasNext() {
if (acc == null) {
return hasNextService();
} else {
PrivilegedAction<Boolean> action = new PrivilegedAction<>() {
public Boolean run() { return hasNextService(); }
};
return AccessController.doPrivileged(action, acc);
}
}

最终都会调用到hasNextService()方法中：

private boolean hasNextService() {
// nextProvider默认为null，如果通过next()取出来了，nextProvider就会变成null
while (nextProvider == null && nextError == null) {
try {
// 找到目标实现类
Class<?> clazz = nextProviderClass();
if (clazz == null)
return false;
if (clazz.getModule().isNamed()) {
// ignore class if in named module
continue;
}
// 判断service接口是否和clazz有父子关系
if (service.isAssignableFrom(clazz)) {
Class<? extends S> type = (Class<? extends S>) clazz;
// 获取无参构造函数
Constructor<? extends S> ctor
= (Constructor<? extends S>)getConstructor(clazz);
// 包装成一个ProviderImpl对象
ProviderImpl<S> p = new ProviderImpl<S>(service, type, ctor, acc);
// 并赋值给nextProvider
nextProvider = (ProviderImpl<T>) p;
} else {
fail(service, clazz.getName() + " not a subtype");
}
} catch (ServiceConfigurationError e) {
nextError = e;
}
}
return true;
}

外部提供的实现类一定要有一个无参构造函数，否则会导致ServiceLoader加载失败；

我们下面再继续深入看看ServiceLoader是怎么找到实现类的：

private Class<?> nextProviderClass() {
if (configs == null) {
try {
// 拼接文件名："META-INF/services/接口名称"
// 比如接口名为：java.sql.Driver，
// 那么文件路径就是："META-INF/services/java.sql.Driver"
String fullName = PREFIX + service.getName();
// 没有指定ClassLoader，就通过getSystemClassLoader()加载目标文件
if (loader == null) {
configs = ClassLoader.getSystemResources(fullName);
} else if (loader == ClassLoaders.platformClassLoader()) {
// 如果是platformClassLoader，它没有class path，那么看看BootLoader有没有class path
if (BootLoader.hasClassPath()) {
configs = BootLoader.findResources(fullName);
} else {
configs = Collections.emptyEnumeration();
}
} else {
// 通过指定classLoader加载目标文件
configs = loader.getResources(fullName);
}
} catch (IOException x) {
fail(service, "Error locating configuration files", x);
}
}
// 上面代码只会执行一次，这样configs就不会为null，下次进来直接取下一个实现类
// 把configs内容解析成一个迭代器
while ((pending == null) || !pending.hasNext()) {
if (!configs.hasMoreElements()) {
return null;
}
pending = parse(configs.nextElement());
}
// 通过迭代器获取下一个实现类名称
String cn = pending.next();
try {
// 通过类名反射成Class对象
return Class.forName(cn, false, loader);
} catch (ClassNotFoundException x) {
fail(service, "Provider " + cn + " not found");
return null;
}
}