chore: springboot example

Author: jackie-linz

examples/SpringBootHikariExample/build.gradle.kts

@@ -23,6 +23,8 @@ dependencies {
 	implementation("org.springframework.boot:spring-boot-starter-data-jdbc")
 	implementation("org.springframework.boot:spring-boot-starter-web")
 	implementation("org.postgresql:postgresql:42.7.3")
+    implementation("software.amazon.awssdk:sts:2.25.60")
+    implementation("software.amazon.awssdk:rds:2.25.70")
     implementation(project(":aws-advanced-jdbc-wrapper"))
 
 }

examples/SpringBootHikariExample/src/main/java/software/amazon/SpringBootHikariExample/SpringBootHikariExampleApplication.java

@@ -18,10 +18,25 @@
 
 import org.springframework.boot.SpringApplication;
 import org.springframework.boot.autoconfigure.SpringBootApplication;
+import software.amazon.awssdk.services.sts.StsClient;
+import software.amazon.awssdk.services.sts.auth.StsAssumeRoleCredentialsProvider;
+import software.amazon.awssdk.services.sts.model.AssumeRoleRequest;
+import software.amazon.jdbc.authentication.AwsCredentialsManager;
 
 @SpringBootApplication
 public class SpringBootHikariExampleApplication {
   public static void main(String[] args) {
+    AwsCredentialsManager.setCustomHandler((hostSpec, props) -> {
+      StsClient stsClient = StsClient.builder().build();
+      AssumeRoleRequest req = AssumeRoleRequest.builder()
+          .roleArn("arn:aws:iam::005965230347:role/iam-poc-testuserb")
+          .roleSessionName("test123")
+          .build();
+      return StsAssumeRoleCredentialsProvider.builder()
+          .stsClient(stsClient)
+          .refreshRequest(req)
+          .build();
+    });
     SpringApplication.run(SpringBootHikariExampleApplication.class, args);
   }
 }

examples/SpringBootHikariExample/src/main/resources/application.yml

@@ -1,12 +1,12 @@
 spring:
   datasource:
-    url: jdbc:aws-wrapper:postgresql://db-identifier.cluster-XYZ.us-east-2.rds.amazonaws.com:5432/postgres
-    username: username
-    password: password
+    url: jdbc:aws-wrapper:postgresql://rds-iam-poc-nonprod.cluster-cjbyytyiqj35.ap-southeast-2.rds.amazonaws.com:5432/postgres
+    username: testuserb
+#    password: password
     driver-class-name: software.amazon.jdbc.Driver
     hikari:
       data-source-properties:
-        wrapperPlugins: failover,efm2
+        wrapperPlugins: failover,efm2,iam
         wrapperDialect: aurora-pg
       max-lifetime: 840000
       minimum-idle: 20