75 lines
2.4 KiB
Java
75 lines
2.4 KiB
Java
|
package org.keycloak.provider;
|
||
|
|
|
||
|
|
import org.jboss.logging.Logger;
|
||
|
|
|
||
|
|
import java.util.HashMap;
|
||
|
|
import java.util.LinkedList;
|
||
|
|
import java.util.List;
|
||
|
|
import java.util.Map;
|
||
|
|
import java.util.ServiceLoader;
|
||
|
|
|
||
|
|
/**
|
||
|
|
* @author <a href="mailto:sthorger@redhat.com">Stian Thorgersen</a>
|
||
|
|
*/
|
||
|
|
public class ProviderManager {
|
||
|
|
|
||
|
|
private static final Logger log = Logger.getLogger(ProviderManager.class);
|
||
|
|
|
||
|
|
private List<ProviderLoader> loaders = new LinkedList<ProviderLoader>();
|
||
|
|
private Map<String, List<ProviderFactory>> cache = new HashMap<String, List<ProviderFactory>>();
|
||
|
|
|
||
|
|
public ProviderManager(ClassLoader baseClassLoader, String... resources) {
|
||
|
|
List<ProviderLoaderFactory> factories = new LinkedList<ProviderLoaderFactory>();
|
||
|
|
for (ProviderLoaderFactory f : ServiceLoader.load(ProviderLoaderFactory.class)) {
|
||
|
|
factories.add(f);
|
||
|
|
}
|
||
|
|
|
||
|
|
log.debugv("Provider loaders {0}", factories);
|
||
|
|
|
||
|
|
loaders.add(new DefaultProviderLoader(baseClassLoader));
|
||
|
|
|
||
|
|
if (resources != null) {
|
||
|
|
for (String r : resources) {
|
||
|
|
String type = r.substring(0, r.indexOf(':'));
|
||
|
|
String resource = r.substring(r.indexOf(':') + 1, r.length());
|
||
|
|
|
||
|
|
boolean found = false;
|
||
|
|
for (ProviderLoaderFactory f : factories) {
|
||
|
|
if (f.supports(type)) {
|
||
|
|
loaders.add(f.create(baseClassLoader, resource));
|
||
|
|
found = true;
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
if (!found) {
|
||
|
|
throw new RuntimeException("Provider loader for " + r + " not found");
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
public synchronized List<ProviderFactory> load(Spi spi) {
|
||
|
|
List<ProviderFactory> factories = cache.get(spi.getName());
|
||
|
|
if (factories == null) {
|
||
|
|
factories = new LinkedList<ProviderFactory>();
|
||
|
|
for (ProviderLoader loader : loaders) {
|
||
|
|
List<ProviderFactory> f = loader.load(spi);
|
||
|
|
if (f != null) {
|
||
|
|
factories.addAll(f);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
return factories;
|
||
|
|
}
|
||
|
|
|
||
|
|
public synchronized ProviderFactory load(Spi spi, String providerId) {
|
||
|
|
for (ProviderFactory f : load(spi)) {
|
||
|
|
if (f.getId().equals(providerId)) {
|
||
|
|
return f;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
return null;
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|