Details
diff --git a/model/api/src/main/java/org/keycloak/provider/ProviderLoader.java b/model/api/src/main/java/org/keycloak/provider/ProviderLoader.java
new file mode 100644
index 0000000..79d78b0
--- /dev/null
+++ b/model/api/src/main/java/org/keycloak/provider/ProviderLoader.java
@@ -0,0 +1,12 @@
+package org.keycloak.provider;
+
+import java.util.List;
+
+/**
+ * @author <a href="mailto:sthorger@redhat.com">Stian Thorgersen</a>
+ */
+public interface ProviderLoader {
+
+ List<ProviderFactory> load(Spi spi);
+
+}
diff --git a/model/api/src/main/java/org/keycloak/provider/ProviderLoaderFactory.java b/model/api/src/main/java/org/keycloak/provider/ProviderLoaderFactory.java
new file mode 100644
index 0000000..a76955c
--- /dev/null
+++ b/model/api/src/main/java/org/keycloak/provider/ProviderLoaderFactory.java
@@ -0,0 +1,12 @@
+package org.keycloak.provider;
+
+/**
+ * @author <a href="mailto:sthorger@redhat.com">Stian Thorgersen</a>
+ */
+public interface ProviderLoaderFactory {
+
+ boolean supports(String type);
+
+ ProviderLoader create(ClassLoader baseClassLoader, String resource);
+
+}
diff --git a/server/src/main/resources/META-INF/keycloak-server.json b/server/src/main/resources/META-INF/keycloak-server.json
index 9f0d03e..1f92932 100755
--- a/server/src/main/resources/META-INF/keycloak-server.json
+++ b/server/src/main/resources/META-INF/keycloak-server.json
@@ -1,4 +1,8 @@
{
+ "providers": [
+ "classpath:${jboss.server.config.dir}/providers/*"
+ ],
+
"admin": {
"realm": "master"
},
diff --git a/services/src/main/java/org/keycloak/provider/DefaultProviderLoader.java b/services/src/main/java/org/keycloak/provider/DefaultProviderLoader.java
new file mode 100644
index 0000000..6359969
--- /dev/null
+++ b/services/src/main/java/org/keycloak/provider/DefaultProviderLoader.java
@@ -0,0 +1,27 @@
+package org.keycloak.provider;
+
+import java.util.LinkedList;
+import java.util.List;
+import java.util.ServiceLoader;
+
+/**
+ * @author <a href="mailto:sthorger@redhat.com">Stian Thorgersen</a>
+ */
+public class DefaultProviderLoader implements ProviderLoader {
+
+ private ClassLoader classLoader;
+
+ public DefaultProviderLoader(ClassLoader classLoader) {
+ this.classLoader = classLoader;
+ }
+
+ @Override
+ public List<ProviderFactory> load(Spi spi) {
+ LinkedList<ProviderFactory> list = new LinkedList<ProviderFactory>();
+ for (ProviderFactory f : ServiceLoader.load(spi.getProviderFactoryClass(), classLoader)) {
+ list.add(f);
+ }
+ return list;
+ }
+
+}
diff --git a/services/src/main/java/org/keycloak/provider/DefaultProviderLoaderFactory.java b/services/src/main/java/org/keycloak/provider/DefaultProviderLoaderFactory.java
new file mode 100644
index 0000000..4985da4
--- /dev/null
+++ b/services/src/main/java/org/keycloak/provider/DefaultProviderLoaderFactory.java
@@ -0,0 +1,18 @@
+package org.keycloak.provider;
+
+/**
+ * @author <a href="mailto:sthorger@redhat.com">Stian Thorgersen</a>
+ */
+public class DefaultProviderLoaderFactory implements ProviderLoaderFactory {
+
+ @Override
+ public boolean supports(String type) {
+ return false;
+ }
+
+ @Override
+ public ProviderLoader create(ClassLoader baseClassLoader, String resource) {
+ return new DefaultProviderLoader(baseClassLoader);
+ }
+
+}
diff --git a/services/src/main/java/org/keycloak/provider/FileSystemProviderLoaderFactory.java b/services/src/main/java/org/keycloak/provider/FileSystemProviderLoaderFactory.java
new file mode 100644
index 0000000..4c887eb
--- /dev/null
+++ b/services/src/main/java/org/keycloak/provider/FileSystemProviderLoaderFactory.java
@@ -0,0 +1,63 @@
+package org.keycloak.provider;
+
+import org.jboss.logging.Logger;
+
+import java.io.File;
+import java.io.FilenameFilter;
+import java.net.URL;
+import java.net.URLClassLoader;
+import java.util.LinkedList;
+import java.util.List;
+
+/**
+ * @author <a href="mailto:sthorger@redhat.com">Stian Thorgersen</a>
+ */
+public class FileSystemProviderLoaderFactory implements ProviderLoaderFactory {
+
+ private static final Logger log = Logger.getLogger(FileSystemProviderLoaderFactory.class);
+
+ @Override
+ public boolean supports(String type) {
+ return "classpath".equals(type);
+ }
+
+ @Override
+ public ProviderLoader create(ClassLoader baseClassLoader, String resource) {
+ return new DefaultProviderLoader(createClassLoader(baseClassLoader, resource.split(";")));
+ }
+
+ private static URLClassLoader createClassLoader(ClassLoader parent, String... files) {
+ try {
+ List<URL> urls = new LinkedList<URL>();
+
+ for (String f : files) {
+ if (f.endsWith("*")) {
+ File dir = new File(f.substring(0, f.length() - 1));
+ if (dir.isDirectory()) {
+ for (File file : dir.listFiles(new JarFilter())) {
+ urls.add(file.toURI().toURL());
+ }
+ }
+ } else {
+ urls.add(new File(f).toURI().toURL());
+ }
+ }
+
+ log.debug("Loading providers from " + urls.toString());
+
+ return new URLClassLoader(urls.toArray(new URL[urls.size()]), parent);
+ } catch (Exception e) {
+ throw new RuntimeException(e);
+ }
+ }
+
+ private static class JarFilter implements FilenameFilter {
+
+ @Override
+ public boolean accept(File dir, String name) {
+ return name.toLowerCase().endsWith(".jar");
+ }
+
+ }
+
+}
diff --git a/services/src/main/java/org/keycloak/provider/ProviderManager.java b/services/src/main/java/org/keycloak/provider/ProviderManager.java
new file mode 100644
index 0000000..6a301fd
--- /dev/null
+++ b/services/src/main/java/org/keycloak/provider/ProviderManager.java
@@ -0,0 +1,74 @@
+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;
+ }
+
+}
diff --git a/services/src/main/java/org/keycloak/services/DefaultKeycloakSessionFactory.java b/services/src/main/java/org/keycloak/services/DefaultKeycloakSessionFactory.java
index fbee2e9..ef3b176 100755
--- a/services/src/main/java/org/keycloak/services/DefaultKeycloakSessionFactory.java
+++ b/services/src/main/java/org/keycloak/services/DefaultKeycloakSessionFactory.java
@@ -6,6 +6,7 @@ import org.keycloak.models.KeycloakSession;
import org.keycloak.models.KeycloakSessionFactory;
import org.keycloak.provider.Provider;
import org.keycloak.provider.ProviderFactory;
+import org.keycloak.provider.ProviderManager;
import org.keycloak.provider.Spi;
import java.util.HashMap;
@@ -24,6 +25,8 @@ public class DefaultKeycloakSessionFactory implements KeycloakSessionFactory {
private Map<Class<? extends Provider>, Map<String, ProviderFactory>> factoriesMap = new HashMap<Class<? extends Provider>, Map<String, ProviderFactory>>();
public void init() {
+ ProviderManager pm = new ProviderManager(getClass().getClassLoader(), Config.scope().getArray("providers"));
+
for (Spi spi : ServiceLoader.load(Spi.class)) {
Map<String, ProviderFactory> factories = new HashMap<String, ProviderFactory>();
factoriesMap.put(spi.getProviderClass(), factories);
@@ -32,7 +35,11 @@ public class DefaultKeycloakSessionFactory implements KeycloakSessionFactory {
if (provider != null) {
this.provider.put(spi.getProviderClass(), provider);
- ProviderFactory factory = loadProviderFactory(spi, provider);
+ ProviderFactory factory = pm.load(spi, provider);
+ if (factory == null) {
+ throw new RuntimeException("Failed to find provider " + provider + " for " + spi.getName());
+ }
+
Config.Scope scope = Config.scope(spi.getName(), provider);
factory.init(scope);
@@ -40,7 +47,7 @@ public class DefaultKeycloakSessionFactory implements KeycloakSessionFactory {
log.debugv("Loaded SPI {0} (provider = {1})", spi.getName(), provider);
} else {
- for (ProviderFactory factory : ServiceLoader.load(spi.getProviderFactoryClass())) {
+ for (ProviderFactory factory : pm.load(spi)) {
Config.Scope scope = Config.scope(spi.getName(), factory.getId());
factory.init(scope);
@@ -59,15 +66,6 @@ public class DefaultKeycloakSessionFactory implements KeycloakSessionFactory {
}
}
- private ProviderFactory loadProviderFactory(Spi spi, String id) {
- for (ProviderFactory factory : ServiceLoader.load(spi.getProviderFactoryClass())) {
- if (factory.getId().equals(id)){
- return factory;
- }
- }
- throw new RuntimeException("Failed to find provider " + id + " for " + spi.getName());
- }
-
public KeycloakSession create() {
return new DefaultKeycloakSession(this);
}
diff --git a/services/src/main/resources/META-INF/services/org.keycloak.provider.ProviderLoaderFactory b/services/src/main/resources/META-INF/services/org.keycloak.provider.ProviderLoaderFactory
new file mode 100644
index 0000000..724fe5f
--- /dev/null
+++ b/services/src/main/resources/META-INF/services/org.keycloak.provider.ProviderLoaderFactory
@@ -0,0 +1,2 @@
+org.keycloak.provider.DefaultProviderLoaderFactory
+org.keycloak.provider.FileSystemProviderLoaderFactory
\ No newline at end of file
diff --git a/testsuite/integration/src/main/resources/log4j.properties b/testsuite/integration/src/main/resources/log4j.properties
index 778d39a..6a329fd 100755
--- a/testsuite/integration/src/main/resources/log4j.properties
+++ b/testsuite/integration/src/main/resources/log4j.properties
@@ -8,6 +8,8 @@ log4j.logger.org.keycloak=info
# Enable to view loaded SPI and Providers
# log4j.logger.org.keycloak.services.DefaultKeycloakSessionFactory=debug
+# log4j.logger.org.keycloak.provider.ProviderManager=debug
+# log4j.logger.org.keycloak.provider.FileSystemProviderLoaderFactory=debug
# Enable to view database updates
# log4j.logger.org.keycloak.connections.jpa.updater.liquibase.LiquibaseJpaUpdaterProvider=debug