fix(kafka): add native support for aws-msk-iam-auth (#268)

streamthoughts · Sep 7, 2023 · 41e5140 · davidlanouette · Sep 7, 2023 · 41e5140
1 parent 28d3374
commit 41e5140
Show file tree

Hide file tree

Showing 5 changed files with 68 additions and 15 deletions.
diff --git a/jikkou-cli/pom.xml b/jikkou-cli/pom.xml
@@ -111,10 +111,6 @@ limitations under the License.
             <groupId>ch.qos.logback</groupId>
             <artifactId>logback-classic</artifactId>
         </dependency>
-        <dependency>
-            <groupId>software.amazon.msk</groupId>
-            <artifactId>aws-msk-iam-auth</artifactId>
-        </dependency>
         <!-- START dependencies for Micronaut/Picocli -->
         <dependency>
             <groupId>io.micronaut</groupId>

diff --git a/jikkou-extension-kafka/pom.xml b/jikkou-extension-kafka/pom.xml
@@ -47,6 +47,12 @@ limitations under the License.
             <artifactId>jikkou-extension-core</artifactId>
             <version>${project.version}</version>
         </dependency>
+        <dependency>
+            <groupId>software.amazon.msk</groupId>
+            <artifactId>aws-msk-iam-auth</artifactId>
+            <version>1.1.9</version>
+            <scope>compile</scope>
+        </dependency>
         <dependency>
             <groupId>org.jetbrains</groupId>
             <artifactId>annotations</artifactId>

diff --git a/...ources/META-INF/native-image/software.amazon.msk/aws-msk-iam-auth/native-image.properties b/...ources/META-INF/native-image/software.amazon.msk/aws-msk-iam-auth/native-image.properties
@@ -0,0 +1 @@
+Args = -H:AdditionalSecurityProviders=software.amazon.msk.auth.iam.internals.ClassLoaderAwareIAMSaslClientProvider,software.amazon.msk.auth.iam.internals.ClassLoaderAwareIAMSaslClientProvider
diff --git a/.../resources/META-INF/native-image/software.amazon.msk/aws-msk-iam-auth/reflect-config.json b/.../resources/META-INF/native-image/software.amazon.msk/aws-msk-iam-auth/reflect-config.json
@@ -0,0 +1,37 @@
+[
+  {
+    "name" : "software.amazon.msk.auth.iam.IAMClientCallbackHandler",
+    "allDeclaredConstructors" : true,
+    "allPublicConstructors" : true,
+    "allDeclaredMethods" : true,
+    "allPublicMethods" : true
+  },
+  {
+    "name":"software.amazon.msk.auth.iam.IAMLoginModule",
+    "allDeclaredConstructors" : true,
+    "allPublicConstructors" : true,
+    "allDeclaredMethods" : true,
+    "allPublicMethods" : true
+  },
+  {
+    "name":"software.amazon.msk.auth.iam.internals.IAMSaslClient.ClassLoaderAwareIAMSaslClientFactory",
+    "allDeclaredConstructors" : true,
+    "allPublicConstructors" : true,
+    "allDeclaredMethods" : true,
+    "allPublicMethods" : true
+  },
+  {
+    "name":"software.amazon.msk.auth.iam.internals.IAMSaslClient.IAMSaslClientFactory",
+    "allDeclaredConstructors" : true,
+    "allPublicConstructors" : true,
+    "allDeclaredMethods" : true,
+    "allPublicMethods" : true
+  },
+  {
+    "name":"software.amazon.msk.auth.iam.internals.AuthenticationResponse",
+    "allDeclaredConstructors" : true,
+    "allPublicConstructors" : true,
+    "allDeclaredMethods" : true,
+    "allPublicMethods" : true
+  }
+]
diff --git a/pom.xml b/pom.xml
@@ -38,7 +38,8 @@ limitations under the License.
     <name>Jikkou Project</name>
 
     <description>
-        Jikkou :: A tool designed to provide an efficient and easy way to manage, automate, and provision resource configurations for Kafka
+        Jikkou :: A tool designed to provide an efficient and easy way to manage, automate, and provision resource
+        configurations for Kafka
     </description>
 
     <url>https://streamthoughts.github.io/jikkou/</url>
@@ -234,8 +235,8 @@ limitations under the License.
                 <groupId>org.jsonschema2pojo</groupId>
                 <artifactId>jsonschema2pojo-maven-plugin</artifactId>
                 <version>1.2.1</version>
-	    </dependency>
-	    <dependency>
+            </dependency>
+            <dependency>
                 <groupId>software.amazon.msk</groupId>
                 <artifactId>aws-msk-iam-auth</artifactId>
                 <version>1.1.9</version>
@@ -437,8 +438,8 @@ limitations under the License.
                             <include>src/integration-test/java/**/*.java</include>
                         </includes>
 
-                        <importOrder /> <!-- standard import order -->
-                        <removeUnusedImports /> <!-- self-explanatory -->
+                        <importOrder/> <!-- standard import order -->
+                        <removeUnusedImports/> <!-- self-explanatory -->
                         <!-- no need to specify files, inferred automatically, but you can if you want -->
 
                         <!-- make sure every file has the following copyright header.
@@ -677,18 +678,30 @@ limitations under the License.
                                     <jikkou>
                                         <artifacts>
                                             <artifact>
-                                                <path>{{artifactsDir}}/{{distributionName}}-{{projectVersion}}-linux-x86_64.tar.gz</path>
-                                                <transform>artifacts/{{distributionName}}-{{projectEffectiveVersion}}-linux-x86_64.tar.gz</transform>
+                                                <path>
+                                                    {{artifactsDir}}/{{distributionName}}-{{projectVersion}}-linux-x86_64.tar.gz
+                                                </path>
+                                                <transform>
+                                                    artifacts/{{distributionName}}-{{projectEffectiveVersion}}-linux-x86_64.tar.gz
+                                                </transform>
                                                 <platform>linux-x86_64</platform>
                                             </artifact>
                                             <artifact>
-                                                <path>{{artifactsDir}}/{{distributionName}}-{{projectVersion}}-linux-x86_64.zip</path>
-                                                <transform>artifacts/{{distributionName}}-{{projectEffectiveVersion}}-linux-x86_64.zip</transform>
+                                                <path>
+                                                    {{artifactsDir}}/{{distributionName}}-{{projectVersion}}-linux-x86_64.zip
+                                                </path>
+                                                <transform>
+                                                    artifacts/{{distributionName}}-{{projectEffectiveVersion}}-linux-x86_64.zip
+                                                </transform>
                                                 <platform>linux-x86_64</platform>
                                             </artifact>
                                             <artifact>
-                                                <path>{{artifactsDir}}/{{distributionName}}-{{projectVersion}}-osx-x86_64.zip</path>
-                                                <transform>artifacts/{{distributionName}}-{{projectEffectiveVersion}}-osx-x86_64.zip</transform>
+                                                <path>
+                                                    {{artifactsDir}}/{{distributionName}}-{{projectVersion}}-osx-x86_64.zip
+                                                </path>
+                                                <transform>
+                                                    artifacts/{{distributionName}}-{{projectEffectiveVersion}}-osx-x86_64.zip
+                                                </transform>
                                                 <platform>osx-x86_64</platform>
                                             </artifact>
                                         </artifacts>