[KOGITO-9774] Registering kubernetes service loaders in native mode

apache · Sep 5, 2023 · fd29a5a · fd29a5a
1 parent 3a67eb1
commit fd29a5a
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/...rc/main/java/org/kie/kogito/addons/quarkus/kubernetes/KogitoAddOnKubernetesProcessor.java b/...rc/main/java/org/kie/kogito/addons/quarkus/kubernetes/KogitoAddOnKubernetesProcessor.java
@@ -16,6 +16,7 @@
 package org.kie.kogito.addons.quarkus.kubernetes;
 
 import org.jboss.logmanager.Level;
+import org.kie.kogito.addons.k8s.resource.catalog.KubernetesServiceCatalog;
 import org.kie.kogito.addons.quarkus.k8s.EndpointCallerProducer;
 import org.kie.kogito.addons.quarkus.k8s.EndpointDiscoveryProducer;
 import org.kie.kogito.addons.quarkus.k8s.config.ServiceDiscoveryConfigBuilder;
@@ -27,6 +28,7 @@
 import io.quarkus.deployment.builditem.FeatureBuildItem;
 import io.quarkus.deployment.builditem.LogCategoryBuildItem;
 import io.quarkus.deployment.builditem.RunTimeConfigBuilderBuildItem;
+import io.quarkus.deployment.builditem.nativeimage.ServiceProviderBuildItem;
 
 class KogitoAddOnKubernetesProcessor extends AnyEngineKogitoAddOnProcessor {
 
@@ -62,4 +64,9 @@ void runtimeInitConfigBuilderProducer(BuildProducer<RunTimeConfigBuilderBuildIte
     public void produceLoggingCategories(BuildProducer<LogCategoryBuildItem> categories) {
         categories.produce(new LogCategoryBuildItem("okhttp3.OkHttpClient", Level.WARN));
     }
+
+    @BuildStep
+    public ServiceProviderBuildItem nativeRegistration() {
+        return ServiceProviderBuildItem.allProvidersFromClassPath(KubernetesServiceCatalog.class.getCanonicalName());
+    }
 }