feat: introduce multiple clients

quarkiverse · Sep 23, 2024 · f5de351 · f5de351
1 parent 9abc7f3
commit f5de351
Show file tree

Hide file tree

Showing 151 changed files with 45,187 additions and 10,703 deletions.
diff --git a/build-parent/pom.xml b/build-parent/pom.xml
@@ -30,5 +30,20 @@
             </dependency>
         </dependencies>
     </dependencyManagement>
-
+    <build>
+        <pluginManagement>
+        <plugins>
+            <plugin>
+                <groupId>io.quarkus</groupId>
+                <artifactId>quarkus-maven-plugin</artifactId>
+                <version>${quarkus.version}</version>
+            </plugin>
+            <plugin>
+                <groupId>io.quarkus</groupId>
+                <artifactId>quarkus-config-doc-maven-plugin</artifactId>
+                <version>${quarkus.version}</version>
+            </plugin>
+        </plugins>
+        </pluginManagement>
+    </build>
 </project>
diff --git a/...deployment/src/main/java/io/quarkus/amazon/cloudwatch/deployment/CloudWatchProcessor.java b/...deployment/src/main/java/io/quarkus/amazon/cloudwatch/deployment/CloudWatchProcessor.java
@@ -65,10 +65,18 @@ void setup(
     }
 
     @BuildStep
-    void discover(BeanRegistrationPhaseBuildItem beanRegistrationPhase,
+    void discoverClientInjectionPoints(BeanRegistrationPhaseBuildItem beanRegistrationPhase,
+            BuildProducer<RequireAmazonClientInjectionBuildItem> requireClientInjectionProducer) {
+
+        discoverClientInjectionPointsInternal(beanRegistrationPhase, requireClientInjectionProducer);
+    }
+
+    @BuildStep
+    void discover(
+            List<RequireAmazonClientInjectionBuildItem> amazonClientInjectionPoints,
             BuildProducer<RequireAmazonClientBuildItem> requireClientProducer) {
 
-        discoverClient(beanRegistrationPhase, requireClientProducer);
+        discoverClient(amazonClientInjectionPoints, requireClientProducer);
     }
 
     @BuildStep
@@ -79,15 +87,15 @@ void discoverTelemetry(BuildProducer<RequireAmazonTelemetryBuildItem> telemetryP
 
     @BuildStep
     void setupClient(List<RequireAmazonClientBuildItem> clientRequirements,
-            BuildProducer<AmazonClientBuildItem> clientProducer) {
+            BuildProducer<RequireAmazonClientTransportBuilderBuildItem> clientProducer) {
 
         setupClient(clientRequirements, clientProducer, buildTimeConfig.sdk(), buildTimeConfig.syncClient(),
                 buildTimeConfig.asyncClient());
     }
 
     @BuildStep(onlyIf = AmazonHttpClients.IsAmazonApacheHttpServicePresent.class)
     @Record(ExecutionTime.RUNTIME_INIT)
-    void setupApacheSyncTransport(List<AmazonClientBuildItem> amazonClients, CloudWatchRecorder recorder,
+    void setupApacheSyncTransport(List<RequireAmazonClientTransportBuilderBuildItem> amazonClients, CloudWatchRecorder recorder,
             AmazonClientApacheTransportRecorder transportRecorder,
             BuildProducer<AmazonClientSyncTransportBuildItem> syncTransports) {
 
@@ -100,7 +108,7 @@ void setupApacheSyncTransport(List<AmazonClientBuildItem> amazonClients, CloudWa
 
     @BuildStep(onlyIf = AmazonHttpClients.IsAmazonAwsCrtHttpServicePresent.class)
     @Record(ExecutionTime.RUNTIME_INIT)
-    void setupAwsCrtSyncTransport(List<AmazonClientBuildItem> amazonClients, CloudWatchRecorder recorder,
+    void setupAwsCrtSyncTransport(List<RequireAmazonClientTransportBuilderBuildItem> amazonClients, CloudWatchRecorder recorder,
             AmazonClientAwsCrtTransportRecorder transportRecorder,
             BuildProducer<AmazonClientSyncTransportBuildItem> syncTransports) {
 
@@ -113,7 +121,8 @@ void setupAwsCrtSyncTransport(List<AmazonClientBuildItem> amazonClients, CloudWa
 
     @BuildStep(onlyIf = AmazonHttpClients.IsAmazonUrlConnectionHttpServicePresent.class)
     @Record(ExecutionTime.RUNTIME_INIT)
-    void setupUrlConnectionSyncTransport(List<AmazonClientBuildItem> amazonClients, CloudWatchRecorder recorder,
+    void setupUrlConnectionSyncTransport(List<RequireAmazonClientTransportBuilderBuildItem> amazonClients,
+            CloudWatchRecorder recorder,
             AmazonClientUrlConnectionTransportRecorder transportRecorder,
             BuildProducer<AmazonClientSyncTransportBuildItem> syncTransports) {
 
@@ -126,7 +135,7 @@ void setupUrlConnectionSyncTransport(List<AmazonClientBuildItem> amazonClients,
 
     @BuildStep(onlyIf = AmazonHttpClients.IsAmazonNettyHttpServicePresent.class)
     @Record(ExecutionTime.RUNTIME_INIT)
-    void setupNettyAsyncTransport(List<AmazonClientBuildItem> amazonClients, CloudWatchRecorder recorder,
+    void setupNettyAsyncTransport(List<RequireAmazonClientTransportBuilderBuildItem> amazonClients, CloudWatchRecorder recorder,
             AmazonClientNettyTransportRecorder transportRecorder,
             BuildProducer<AmazonClientAsyncTransportBuildItem> asyncTransports,
             EventLoopGroupBuildItem eventLoopSupplier) {
@@ -140,7 +149,8 @@ void setupNettyAsyncTransport(List<AmazonClientBuildItem> amazonClients, CloudWa
 
     @BuildStep(onlyIf = AmazonHttpClients.IsAmazonAwsCrtHttpServicePresent.class)
     @Record(ExecutionTime.RUNTIME_INIT)
-    void setupAwsCrtAsyncTransport(List<AmazonClientBuildItem> amazonClients, CloudWatchRecorder recorder,
+    void setupAwsCrtAsyncTransport(List<RequireAmazonClientTransportBuilderBuildItem> amazonClients,
+            CloudWatchRecorder recorder,
             AmazonClientAwsCrtTransportRecorder transportRecorder,
             BuildProducer<AmazonClientAsyncTransportBuildItem> asyncTransports) {
 
@@ -156,6 +166,7 @@ void setupAwsCrtAsyncTransport(List<AmazonClientBuildItem> amazonClients, CloudW
     void createClientBuilders(CloudWatchRecorder recorder,
             AmazonClientCommonRecorder commonRecorder,
             AmazonClientOpenTelemetryRecorder otelRecorder,
+            List<RequireAmazonClientInjectionBuildItem> amazonClientInjections,
             List<RequireAmazonTelemetryBuildItem> amazonRequireTelemtryClients,
             List<AmazonClientSyncTransportBuildItem> syncTransports,
             List<AmazonClientAsyncTransportBuildItem> asyncTransports,
@@ -170,6 +181,7 @@ void createClientBuilders(CloudWatchRecorder recorder,
                 commonRecorder,
                 otelRecorder,
                 buildTimeConfig,
+                amazonClientInjections,
                 amazonRequireTelemtryClients,
                 syncTransports,
                 asyncTransports,

diff --git a/cloudwatch/runtime/src/main/java/io/quarkus/amazon/cloudwatch/runtime/CloudWatchConfig.java b/cloudwatch/runtime/src/main/java/io/quarkus/amazon/cloudwatch/runtime/CloudWatchConfig.java
@@ -1,31 +1,16 @@
 package io.quarkus.amazon.cloudwatch.runtime;
 
 import io.quarkus.amazon.common.runtime.AsyncHttpClientConfig;
-import io.quarkus.amazon.common.runtime.AwsConfig;
-import io.quarkus.amazon.common.runtime.SdkConfig;
+import io.quarkus.amazon.common.runtime.HasAmazonClientRuntimeConfig;
 import io.quarkus.amazon.common.runtime.SyncHttpClientConfig;
 import io.quarkus.runtime.annotations.ConfigDocSection;
 import io.quarkus.runtime.annotations.ConfigPhase;
 import io.quarkus.runtime.annotations.ConfigRoot;
 import io.smallrye.config.ConfigMapping;
-import io.smallrye.config.WithParentName;
 
 @ConfigMapping(prefix = "quarkus.cloudwatch")
 @ConfigRoot(phase = ConfigPhase.RUN_TIME)
-public interface CloudWatchConfig {
-    /**
-     * AWS SDK client configurations
-     */
-    @WithParentName
-    @ConfigDocSection
-    SdkConfig sdk();
-
-    /**
-     * AWS services configurations
-     */
-    @ConfigDocSection
-    AwsConfig aws();
-
+public interface CloudWatchConfig extends HasAmazonClientRuntimeConfig {
     /**
      * Sync HTTP transport configurations
      */

diff --git a/...dwatch/runtime/src/main/java/io/quarkus/amazon/cloudwatch/runtime/CloudWatchRecorder.java b/...dwatch/runtime/src/main/java/io/quarkus/amazon/cloudwatch/runtime/CloudWatchRecorder.java
@@ -18,13 +18,8 @@ public CloudWatchRecorder(CloudWatchConfig config) {
     }
 
     @Override
-    public RuntimeValue<AwsConfig> getAwsConfig() {
-        return new RuntimeValue<>(config.aws());
-    }
-
-    @Override
-    public RuntimeValue<SdkConfig> getSdkConfig() {
-        return new RuntimeValue<>(config.sdk());
+    public RuntimeValue<HasAmazonClientRuntimeConfig> getAmazonClientsConfig() {
+        return new RuntimeValue<>(config);
     }
 
     @Override

diff --git a/...oyment/src/main/java/io/quarkus/amazon/cloudwatch/deployment/CloudWatchLogsProcessor.java b/...oyment/src/main/java/io/quarkus/amazon/cloudwatch/deployment/CloudWatchLogsProcessor.java
@@ -65,10 +65,18 @@ void setup(
     }
 
     @BuildStep
-    void discover(BeanRegistrationPhaseBuildItem beanRegistrationPhase,
+    void discoverClientInjectionPoints(BeanRegistrationPhaseBuildItem beanRegistrationPhase,
+            BuildProducer<RequireAmazonClientInjectionBuildItem> requireClientInjectionProducer) {
+
+        discoverClientInjectionPointsInternal(beanRegistrationPhase, requireClientInjectionProducer);
+    }
+
+    @BuildStep
+    void discover(
+            List<RequireAmazonClientInjectionBuildItem> amazonClientInjectionPoints,
             BuildProducer<RequireAmazonClientBuildItem> requireClientProducer) {
 
-        discoverClient(beanRegistrationPhase, requireClientProducer);
+        discoverClient(amazonClientInjectionPoints, requireClientProducer);
     }
 
     @BuildStep
@@ -79,15 +87,16 @@ void discoverTelemetry(BuildProducer<RequireAmazonTelemetryBuildItem> telemetryP
 
     @BuildStep
     void setupClient(List<RequireAmazonClientBuildItem> clientRequirements,
-            BuildProducer<AmazonClientBuildItem> clientProducer) {
+            BuildProducer<RequireAmazonClientTransportBuilderBuildItem> clientProducer) {
 
         setupClient(clientRequirements, clientProducer, buildTimeConfig.sdk(), buildTimeConfig.syncClient(),
                 buildTimeConfig.asyncClient());
     }
 
     @BuildStep(onlyIf = AmazonHttpClients.IsAmazonApacheHttpServicePresent.class)
     @Record(ExecutionTime.RUNTIME_INIT)
-    void setupApacheSyncTransport(List<AmazonClientBuildItem> amazonClients, CloudWatchLogsRecorder recorder,
+    void setupApacheSyncTransport(List<RequireAmazonClientTransportBuilderBuildItem> amazonClients,
+            CloudWatchLogsRecorder recorder,
             AmazonClientApacheTransportRecorder transportRecorder,
             BuildProducer<AmazonClientSyncTransportBuildItem> syncTransports) {
 
@@ -100,7 +109,8 @@ void setupApacheSyncTransport(List<AmazonClientBuildItem> amazonClients, CloudWa
 
     @BuildStep(onlyIf = AmazonHttpClients.IsAmazonAwsCrtHttpServicePresent.class)
     @Record(ExecutionTime.RUNTIME_INIT)
-    void setupAwsCrtSyncTransport(List<AmazonClientBuildItem> amazonClients, CloudWatchLogsRecorder recorder,
+    void setupAwsCrtSyncTransport(List<RequireAmazonClientTransportBuilderBuildItem> amazonClients,
+            CloudWatchLogsRecorder recorder,
             AmazonClientAwsCrtTransportRecorder transportRecorder,
             BuildProducer<AmazonClientSyncTransportBuildItem> syncTransports) {
 
@@ -113,7 +123,8 @@ void setupAwsCrtSyncTransport(List<AmazonClientBuildItem> amazonClients, CloudWa
 
     @BuildStep(onlyIf = AmazonHttpClients.IsAmazonUrlConnectionHttpServicePresent.class)
     @Record(ExecutionTime.RUNTIME_INIT)
-    void setupUrlConnectionSyncTransport(List<AmazonClientBuildItem> amazonClients, CloudWatchLogsRecorder recorder,
+    void setupUrlConnectionSyncTransport(List<RequireAmazonClientTransportBuilderBuildItem> amazonClients,
+            CloudWatchLogsRecorder recorder,
             AmazonClientUrlConnectionTransportRecorder transportRecorder,
             BuildProducer<AmazonClientSyncTransportBuildItem> syncTransports) {
 
@@ -126,7 +137,8 @@ void setupUrlConnectionSyncTransport(List<AmazonClientBuildItem> amazonClients,
 
     @BuildStep(onlyIf = AmazonHttpClients.IsAmazonNettyHttpServicePresent.class)
     @Record(ExecutionTime.RUNTIME_INIT)
-    void setupNettyAsyncTransport(List<AmazonClientBuildItem> amazonClients, CloudWatchLogsRecorder recorder,
+    void setupNettyAsyncTransport(List<RequireAmazonClientTransportBuilderBuildItem> amazonClients,
+            CloudWatchLogsRecorder recorder,
             AmazonClientNettyTransportRecorder transportRecorder,
             BuildProducer<AmazonClientAsyncTransportBuildItem> asyncTransports,
             EventLoopGroupBuildItem eventLoopSupplier) {
@@ -140,7 +152,8 @@ void setupNettyAsyncTransport(List<AmazonClientBuildItem> amazonClients, CloudWa
 
     @BuildStep(onlyIf = AmazonHttpClients.IsAmazonAwsCrtHttpServicePresent.class)
     @Record(ExecutionTime.RUNTIME_INIT)
-    void setupAwsCrtAsyncTransport(List<AmazonClientBuildItem> amazonClients, CloudWatchLogsRecorder recorder,
+    void setupAwsCrtAsyncTransport(List<RequireAmazonClientTransportBuilderBuildItem> amazonClients,
+            CloudWatchLogsRecorder recorder,
             AmazonClientAwsCrtTransportRecorder transportRecorder,
             BuildProducer<AmazonClientAsyncTransportBuildItem> asyncTransports) {
 
@@ -156,6 +169,7 @@ void setupAwsCrtAsyncTransport(List<AmazonClientBuildItem> amazonClients, CloudW
     void createClientBuilders(CloudWatchLogsRecorder recorder,
             AmazonClientCommonRecorder commonRecorder,
             AmazonClientOpenTelemetryRecorder otelRecorder,
+            List<RequireAmazonClientInjectionBuildItem> amazonClientInjections,
             List<RequireAmazonTelemetryBuildItem> amazonRequireTelemtryClients,
             List<AmazonClientSyncTransportBuildItem> syncTransports,
             List<AmazonClientAsyncTransportBuildItem> asyncTransports,
@@ -170,6 +184,7 @@ void createClientBuilders(CloudWatchLogsRecorder recorder,
                 commonRecorder,
                 otelRecorder,
                 buildTimeConfig,
+                amazonClientInjections,
                 amazonRequireTelemtryClients,
                 syncTransports,
                 asyncTransports,

diff --git a/...logs/runtime/src/main/java/io/quarkus/amazon/cloudwatch/runtime/CloudWatchLogsConfig.java b/...logs/runtime/src/main/java/io/quarkus/amazon/cloudwatch/runtime/CloudWatchLogsConfig.java
@@ -1,31 +1,16 @@
 package io.quarkus.amazon.cloudwatch.runtime;
 
 import io.quarkus.amazon.common.runtime.AsyncHttpClientConfig;
-import io.quarkus.amazon.common.runtime.AwsConfig;
-import io.quarkus.amazon.common.runtime.SdkConfig;
+import io.quarkus.amazon.common.runtime.HasAmazonClientRuntimeConfig;
 import io.quarkus.amazon.common.runtime.SyncHttpClientConfig;
 import io.quarkus.runtime.annotations.ConfigDocSection;
 import io.quarkus.runtime.annotations.ConfigPhase;
 import io.quarkus.runtime.annotations.ConfigRoot;
 import io.smallrye.config.ConfigMapping;
-import io.smallrye.config.WithParentName;
 
 @ConfigMapping(prefix = "quarkus.cloudwatchlogs")
 @ConfigRoot(phase = ConfigPhase.RUN_TIME)
-public interface CloudWatchLogsConfig {
-    /**
-     * AWS SDK client configurations
-     */
-    @WithParentName
-    @ConfigDocSection
-    SdkConfig sdk();
-
-    /**
-     * AWS services configurations
-     */
-    @ConfigDocSection
-    AwsConfig aws();
-
+public interface CloudWatchLogsConfig extends HasAmazonClientRuntimeConfig {
     /**
      * Sync HTTP transport configurations
      */

diff --git a/...gs/runtime/src/main/java/io/quarkus/amazon/cloudwatch/runtime/CloudWatchLogsRecorder.java b/...gs/runtime/src/main/java/io/quarkus/amazon/cloudwatch/runtime/CloudWatchLogsRecorder.java
@@ -18,13 +18,8 @@ public CloudWatchLogsRecorder(CloudWatchLogsConfig config) {
     }
 
     @Override
-    public RuntimeValue<AwsConfig> getAwsConfig() {
-        return new RuntimeValue<>(config.aws());
-    }
-
-    @Override
-    public RuntimeValue<SdkConfig> getSdkConfig() {
-        return new RuntimeValue<>(config.sdk());
+    public RuntimeValue<HasAmazonClientRuntimeConfig> getAmazonClientsConfig() {
+        return new RuntimeValue<>(config);
     }
 
     @Override