Merge pull request #8 from stoconnor/ssm-and-sfn

Support for SSM and Step Functions

Author: Fabiano Oliveira

pom.xml

@@ -97,6 +97,12 @@
             <artifactId>cglib</artifactId>
             <version>3.2.6</version>
         </dependency>
+        <dependency>
+            <groupId>org.projectlombok</groupId>
+            <artifactId>lombok</artifactId>
+            <version>1.18.6</version>
+            <scope>provided</scope>
+        </dependency>
 
         <!-- test dependencies -->
         <dependency>
@@ -109,7 +115,7 @@
         <dependency>
             <groupId>cloud.localstack</groupId>
             <artifactId>localstack-utils</artifactId>
-            <version>0.1.13</version>
+            <version>0.1.19</version>
         </dependency>
         <dependency>
             <groupId>org.springframework</groupId>

src/main/java/xyz/fabiano/spring/localstack/LocalstackService.java

@@ -20,6 +20,7 @@ public enum LocalstackService {
     CLOUDFORMATION("cloudformation"),
     CLOUDWATCH("cloudwatch"),
     SSM("ssm"),
+    STEP_FUNCTION("stepfunctions"),
 
     ELASTICSEARCH("elasticsearch", false),
     ELASTICSEARCH_SERVICE("es", false);

src/main/java/xyz/fabiano/spring/localstack/autoconfigure/EveryAwsClientAutoConfiguration.java

@@ -12,8 +12,11 @@
 import com.amazonaws.services.route53.AmazonRoute53;
 import com.amazonaws.services.s3.AmazonS3;
 import com.amazonaws.services.simpleemail.AmazonSimpleEmailService;
+import com.amazonaws.services.simplesystemsmanagement.AWSSimpleSystemsManagement;
 import com.amazonaws.services.sns.AmazonSNS;
 import com.amazonaws.services.sqs.AmazonSQS;
+import com.amazonaws.services.stepfunctions.AWSStepFunctions;
+import com.amazonaws.services.stepfunctions.AWSStepFunctionsClient;
 import org.springframework.boot.autoconfigure.condition.ConditionalOnProperty;
 import org.springframework.context.annotation.Bean;
 import org.springframework.context.annotation.Configuration;
@@ -100,4 +103,10 @@ public AWSLambda awsLambda() {
     public AmazonCloudFormation amazonCloudFormation() {
         return amazonClientsHolder.amazonCloudFormation();
     }
+
+    @Bean
+    public AWSSimpleSystemsManagement awsSimpleSystemsManagement() { return  amazonClientsHolder.awsSimpleSystemsManagement(); }
+
+    @Bean
+    public AWSStepFunctions awsStepFunctions() { return amazonClientsHolder.awsStepFunctions(); }
 }

src/main/java/xyz/fabiano/spring/localstack/legacy/LocalstackDocker.java

@@ -186,6 +186,8 @@ public String getEndpointSSM() {
         return endpointForService(ServiceName.SSM);
     }
 
+    public String getEndpointStepFunction() { return endpointForService(ServiceName.STEPFUNCTIONS); }
+
 
     public String endpointForService(String serviceName) {
         if (serviceToPortMap == null) {

src/main/java/xyz/fabiano/spring/localstack/support/AmazonAsyncDockerClientsHolder.java

@@ -23,10 +23,14 @@
 import com.amazonaws.services.route53.AmazonRoute53AsyncClientBuilder;
 import com.amazonaws.services.simpleemail.AmazonSimpleEmailServiceAsync;
 import com.amazonaws.services.simpleemail.AmazonSimpleEmailServiceAsyncClientBuilder;
+import com.amazonaws.services.simplesystemsmanagement.AWSSimpleSystemsManagement;
+import com.amazonaws.services.simplesystemsmanagement.AWSSimpleSystemsManagementClientBuilder;
 import com.amazonaws.services.sns.AmazonSNSAsync;
 import com.amazonaws.services.sns.AmazonSNSAsyncClientBuilder;
 import com.amazonaws.services.sqs.AmazonSQSAsync;
 import com.amazonaws.services.sqs.AmazonSQSAsyncClientBuilder;
+import com.amazonaws.services.stepfunctions.AWSStepFunctions;
+import com.amazonaws.services.stepfunctions.AWSStepFunctionsClientBuilder;
 import xyz.fabiano.spring.localstack.legacy.LocalstackDocker;
 
 public final class AmazonAsyncDockerClientsHolder extends AbstractAmazonDockerClientsHolder {
@@ -138,4 +142,20 @@ public AWSLambdaAsync awsLambda() {
             LocalstackDocker::getEndpointLambda
         );
     }
+
+    @Override
+    public AWSSimpleSystemsManagement awsSimpleSystemsManagement() {
+        return decorateWithConfigsAndBuild(
+          AWSSimpleSystemsManagementClientBuilder.standard(),
+          LocalstackDocker::getEndpointSSM
+        );
+    }
+
+    @Override
+    public AWSStepFunctions awsStepFunctions() {
+        return decorateWithConfigsAndBuild(
+          AWSStepFunctionsClientBuilder.standard(),
+          LocalstackDocker::getEndpointStepFunction
+        );
+    }
 }

src/main/java/xyz/fabiano/spring/localstack/support/AmazonClientsHolder.java

@@ -12,8 +12,10 @@
 import com.amazonaws.services.route53.AmazonRoute53;
 import com.amazonaws.services.s3.AmazonS3;
 import com.amazonaws.services.simpleemail.AmazonSimpleEmailService;
+import com.amazonaws.services.simplesystemsmanagement.AWSSimpleSystemsManagement;
 import com.amazonaws.services.sns.AmazonSNS;
 import com.amazonaws.services.sqs.AmazonSQS;
+import com.amazonaws.services.stepfunctions.AWSStepFunctions;
 
 public interface AmazonClientsHolder {
 
@@ -44,4 +46,8 @@ public interface AmazonClientsHolder {
     AmazonRoute53 amazonRoute53();
 
     AWSLambda awsLambda();
+
+    AWSSimpleSystemsManagement awsSimpleSystemsManagement();
+
+    AWSStepFunctions awsStepFunctions();
 }

src/main/java/xyz/fabiano/spring/localstack/support/AmazonDockerClientsHolder.java

@@ -23,10 +23,15 @@
 import com.amazonaws.services.route53.AmazonRoute53ClientBuilder;
 import com.amazonaws.services.simpleemail.AmazonSimpleEmailService;
 import com.amazonaws.services.simpleemail.AmazonSimpleEmailServiceClientBuilder;
+import com.amazonaws.services.simplesystemsmanagement.AWSSimpleSystemsManagement;
+import com.amazonaws.services.simplesystemsmanagement.AWSSimpleSystemsManagementClientBuilder;
 import com.amazonaws.services.sns.AmazonSNS;
 import com.amazonaws.services.sns.AmazonSNSClientBuilder;
 import com.amazonaws.services.sqs.AmazonSQS;
 import com.amazonaws.services.sqs.AmazonSQSClientBuilder;
+import com.amazonaws.services.stepfunctions.AWSStepFunctions;
+import com.amazonaws.services.stepfunctions.AWSStepFunctionsClient;
+import com.amazonaws.services.stepfunctions.AWSStepFunctionsClientBuilder;
 import xyz.fabiano.spring.localstack.legacy.LocalstackDocker;
 
 public final class AmazonDockerClientsHolder extends AbstractAmazonDockerClientsHolder {
@@ -138,4 +143,20 @@ public AWSLambda awsLambda() {
             LocalstackDocker::getEndpointLambda
         );
     }
+
+    @Override
+    public AWSSimpleSystemsManagement awsSimpleSystemsManagement() {
+        return decorateWithConfigsAndBuild(
+          AWSSimpleSystemsManagementClientBuilder.standard(),
+          LocalstackDocker::getEndpointSSM
+        );
+    }
+
+    @Override
+    public AWSStepFunctions awsStepFunctions() {
+        return decorateWithConfigsAndBuild(
+          AWSStepFunctionsClientBuilder.standard(),
+          LocalstackDocker::getEndpointStepFunction
+        );
+    }
 }