feat(amazon): Add Edda proxy for LaunchTemplateVersions (#5083)

* feat(amazon): Add Edda proxy for LaunchTemplateVersions * Apply formatting Co-authored-by: mergify[bot] <37929162+mergify[bot]@users.noreply.github.com>
spinnaker · Nov 9, 2020 · c9286ae · c9286ae
1 parent f8e0144
commit c9286ae
Showing 1 changed file with 45 additions and 1 deletion.
diff --git a/...m/netflix/spinnaker/clouddriver/aws/security/sdkclient/AmazonClientInvocationHandler.java b/...m/netflix/spinnaker/clouddriver/aws/security/sdkclient/AmazonClientInvocationHandler.java
@@ -45,6 +45,8 @@
 import java.lang.reflect.Method;
 import java.util.*;
 import java.util.concurrent.TimeUnit;
+import java.util.function.Function;
+import java.util.stream.Collectors;
 import org.apache.http.HttpEntity;
 import org.apache.http.HttpResponse;
 import org.apache.http.HttpStatus;
@@ -203,6 +205,40 @@ public DescribeLaunchTemplatesResult describeLaunchTemplates(
             describe(request, "launchTemplateNames", "launchTemplates", LaunchTemplate.class));
   }
 
+  private static final Function<AmazonWebServiceRequest, Collection<String>>
+      LAUNCH_TEMPLATE_VERSION_EXTRACTOR =
+          (r) -> {
+            if (r == null) {
+              return Collections.emptyList();
+            }
+            DescribeLaunchTemplateVersionsRequest req = (DescribeLaunchTemplateVersionsRequest) r;
+            if (req.getLaunchTemplateId() == null) {
+              return Collections.emptyList();
+            }
+            if (req.getVersions() == null || req.getVersions().isEmpty()) {
+              String defaultId = req.getLaunchTemplateId() + ":$Default";
+              return Collections.singletonList(defaultId);
+            }
+            return req.getVersions().stream()
+                .map(v -> req.getLaunchTemplateId() + ":" + v)
+                .collect(Collectors.toCollection(ArrayList::new));
+          };
+
+  public DescribeLaunchTemplateVersionsResult describeLaunchTemplateVersions() {
+    return describeLaunchTemplateVersions(null);
+  }
+
+  public DescribeLaunchTemplateVersionsResult describeLaunchTemplateVersions(
+      DescribeLaunchTemplateVersionsRequest request) {
+    return new DescribeLaunchTemplateVersionsResult()
+        .withLaunchTemplateVersions(
+            describe(
+                request,
+                LAUNCH_TEMPLATE_VERSION_EXTRACTOR,
+                "launchTemplateVersions",
+                LaunchTemplateVersion.class));
+  }
+
   public DescribeImagesResult describeImages() {
     return describeImages(null);
   }
@@ -349,11 +385,19 @@ private <T> List<T> describe(
       String idKey,
       final String object,
       final Class<T> singleType) {
+    return describe(request, r -> getRequestIds(r, idKey), object, singleType);
+  }
+
+  private <T> List<T> describe(
+      AmazonWebServiceRequest request,
+      Function<AmazonWebServiceRequest, Collection<String>> idExtractor,
+      final String object,
+      final Class<T> singleType) {
     lastModified.set(null);
     final Map<String, String> metricTags = new HashMap<>(this.metricTags);
     metricTags.put("collection", object);
     try {
-      final Collection<String> ids = getRequestIds(request, idKey);
+      final Collection<String> ids = idExtractor.apply(request);
       metricTags.put("collectionMode", ids.isEmpty() ? "full" : "byId");
       final JavaType singleMeta =
           objectMapper