Auto-discover extensions (#187)

imx-org · Mar 15, 2024 · 0755ac2 · 0755ac2
1 parent b04a609
commit 0755ac2
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 2 deletions.
diff --git a/...spatial/src/main/java/nl/geostandaarden/imx/orchestrate/ext/spatial/SpatialExtension.java b/...spatial/src/main/java/nl/geostandaarden/imx/orchestrate/ext/spatial/SpatialExtension.java
@@ -1,10 +1,12 @@
 package nl.geostandaarden.imx.orchestrate.ext.spatial;
 
+import com.google.auto.service.AutoService;
 import java.util.Set;
 import nl.geostandaarden.imx.orchestrate.ext.spatial.geometry.GeometryTypeFactory;
 import nl.geostandaarden.imx.orchestrate.model.OrchestrateExtension;
 import nl.geostandaarden.imx.orchestrate.model.types.ValueTypeFactory;
 
+@AutoService(OrchestrateExtension.class)
 public final class SpatialExtension implements OrchestrateExtension {
 
   @Override

diff --git a/gateway/src/main/java/nl/geostandaarden/imx/orchestrate/gateway/GatewayConfiguration.java b/gateway/src/main/java/nl/geostandaarden/imx/orchestrate/gateway/GatewayConfiguration.java
@@ -16,10 +16,10 @@
 import nl.geostandaarden.imx.orchestrate.engine.OrchestrateEngine;
 import nl.geostandaarden.imx.orchestrate.engine.source.Source;
 import nl.geostandaarden.imx.orchestrate.engine.source.SourceType;
-import nl.geostandaarden.imx.orchestrate.ext.spatial.SpatialExtension;
 import nl.geostandaarden.imx.orchestrate.gateway.schema.SchemaFactory;
 import nl.geostandaarden.imx.orchestrate.model.ComponentRegistry;
 import nl.geostandaarden.imx.orchestrate.model.Model;
+import nl.geostandaarden.imx.orchestrate.model.OrchestrateExtension;
 import nl.geostandaarden.imx.orchestrate.model.loader.ModelLoader;
 import nl.geostandaarden.imx.orchestrate.model.loader.ModelLoaderRegistry;
 import nl.geostandaarden.imx.orchestrate.model.types.ValueTypeFactory;
@@ -41,7 +41,7 @@ public class GatewayConfiguration {
 
   @Bean
   public GraphQlSource graphQlSource() throws IOException {
-    var extensions = Set.of(new SpatialExtension());
+    var extensions = resolveExtensions();
 
     var componentRegistry = new ComponentRegistry();
     extensions.forEach(extension -> extension.registerComponents(componentRegistry));
@@ -84,6 +84,13 @@ public GraphQLSchema schema() {
     };
   }
 
+  private Set<OrchestrateExtension> resolveExtensions() {
+    return ServiceLoader.load(OrchestrateExtension.class)
+        .stream()
+        .map(ServiceLoader.Provider::get)
+        .collect(toUnmodifiableSet());
+  }
+
   private Set<ModelLoader> resolveModelLoaders() {
     return ServiceLoader.load(ModelLoader.class)
         .stream()