Merge pull request #10272 from vespa-engine/freva/aws-limits

Add AwsLimitsFetcher interface

2 files changed, 63 insertions, 0 deletions

diff --git a/controller-api/src/main/java/com/yahoo/vespa/hosted/controller/api/integration/aws/AwsLimitsFetcher.java b/controller-api/src/main/java/com/yahoo/vespa/hosted/controller/api/integration/aws/AwsLimitsFetcher.java
new file mode 100644
index 00000000000..4e76f67e7cf
--- /dev/null
+++ b/controller-api/src/main/java/com/yahoo/vespa/hosted/controller/api/integration/aws/AwsLimitsFetcher.java
@@ -0,0 +1,14 @@
+// Copyright 2019 Oath Inc. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
+package com.yahoo.vespa.hosted.controller.api.integration.aws;
+
+/**
+ * @author freva
+ */
+public interface AwsLimitsFetcher {
+
+    /** Returns the AWS EC2 instance limits in the given AWS region */
+    Ec2InstanceCounts getEc2InstanceLimits(String awsRegion);
+
+    /** Returns the current usage of AWS EC2 instances in the given AWS region */
+    Ec2InstanceCounts getEc2InstanceUsage(String awsRegion);
+}
diff --git a/controller-api/src/main/java/com/yahoo/vespa/hosted/controller/api/integration/aws/Ec2InstanceCounts.java b/controller-api/src/main/java/com/yahoo/vespa/hosted/controller/api/integration/aws/Ec2InstanceCounts.java
new file mode 100644
index 00000000000..d5e65cfa2bd
--- /dev/null
+++ b/controller-api/src/main/java/com/yahoo/vespa/hosted/controller/api/integration/aws/Ec2InstanceCounts.java
@@ -0,0 +1,49 @@
+// Copyright 2019 Oath Inc. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
+package com.yahoo.vespa.hosted.controller.api.integration.aws;
+
+import java.util.Map;
+import java.util.Objects;
+
+/**
+ * @author freva
+ */
+public class Ec2InstanceCounts {
+    private final int totalCount;
+    private final Map<String, Integer> instanceCounts;
+
+    private Ec2InstanceCounts(int totalCount, Map<String, Integer> instanceCounts) {
+        this.totalCount = totalCount;
+        this.instanceCounts = Map.copyOf(instanceCounts);
+    }
+
+    public int getTotalCount() {
+        return totalCount;
+    }
+
+    /** Returns map of counts by instance type, e.g. 'r5.2xlarge' */
+    public Map<String, Integer> getInstanceCounts() {
+        return instanceCounts;
+    }
+
+    @Override
+    public boolean equals(Object o) {
+        if (this == o) return true;
+        if (o == null || getClass() != o.getClass()) return false;
+        Ec2InstanceCounts that = (Ec2InstanceCounts) o;
+        return totalCount == that.totalCount &&
+                instanceCounts.equals(that.instanceCounts);
+    }
+
+    @Override
+    public int hashCode() {
+        return Objects.hash(totalCount, instanceCounts);
+    }
+
+    @Override
+    public String toString() {
+        return "Ec2InstanceLimits{" +
+                "totalLimit=" + totalCount +
+                ", instanceCounts=" + instanceCounts +
+                '}';
+    }
+}