Create service for registry of plans

3 files changed, 133 insertions, 0 deletions

diff --git a/controller-api/src/main/java/com/yahoo/vespa/hosted/controller/api/integration/ServiceRegistry.java b/controller-api/src/main/java/com/yahoo/vespa/hosted/controller/api/integration/ServiceRegistry.java
index bf6e88a09e0..4ab09919263 100644
--- a/controller-api/src/main/java/com/yahoo/vespa/hosted/controller/api/integration/ServiceRegistry.java
+++ b/controller-api/src/main/java/com/yahoo/vespa/hosted/controller/api/integration/ServiceRegistry.java
@@ -8,6 +8,7 @@ import com.yahoo.vespa.hosted.controller.api.integration.aws.AwsEventFetcher;
 import com.yahoo.vespa.hosted.controller.api.integration.aws.ResourceTagger;
 import com.yahoo.vespa.hosted.controller.api.integration.billing.BillingController;
 import com.yahoo.vespa.hosted.controller.api.integration.billing.BillingDatabaseClient;
+import com.yahoo.vespa.hosted.controller.api.integration.billing.PlanRegistry;
 import com.yahoo.vespa.hosted.controller.api.integration.certificates.EndpointCertificateProvider;
 import com.yahoo.vespa.hosted.controller.api.integration.certificates.EndpointCertificateValidator;
 import com.yahoo.vespa.hosted.controller.api.integration.configserver.ConfigServer;
@@ -105,4 +106,6 @@ public interface ServiceRegistry {
     AccessControlService accessControlService();
 
     HorizonClient horizonClient();
+
+    PlanRegistry planRegistry();
 }
diff --git a/controller-api/src/main/java/com/yahoo/vespa/hosted/controller/api/integration/billing/PlanRegistry.java b/controller-api/src/main/java/com/yahoo/vespa/hosted/controller/api/integration/billing/PlanRegistry.java
new file mode 100644
index 00000000000..5eca502647b
--- /dev/null
+++ b/controller-api/src/main/java/com/yahoo/vespa/hosted/controller/api/integration/billing/PlanRegistry.java
@@ -0,0 +1,18 @@
+package com.yahoo.vespa.hosted.controller.api.integration.billing;
+
+import java.util.Optional;
+
+/**
+ * Registry of all current plans we have support for
+ *
+ * @author ogronnesby
+ */
+public interface PlanRegistry {
+
+    /** Get the default plan */
+    Plan defaultPlan();
+
+    /** Get a plan given a plan ID */
+    Optional<Plan> plan(PlanId planId);
+
+}
diff --git a/controller-api/src/main/java/com/yahoo/vespa/hosted/controller/api/integration/billing/PlanRegistryMock.java b/controller-api/src/main/java/com/yahoo/vespa/hosted/controller/api/integration/billing/PlanRegistryMock.java
new file mode 100644
index 00000000000..fa51196942e
--- /dev/null
+++ b/controller-api/src/main/java/com/yahoo/vespa/hosted/controller/api/integration/billing/PlanRegistryMock.java
@@ -0,0 +1,112 @@
+package com.yahoo.vespa.hosted.controller.api.integration.billing;
+
+import com.yahoo.config.provision.NodeResources;
+import com.yahoo.vespa.hosted.controller.api.integration.resource.CostInfo;
+import com.yahoo.vespa.hosted.controller.api.integration.resource.ResourceUsage;
+
+import java.math.BigDecimal;
+import java.math.RoundingMode;
+import java.util.Optional;
+import java.util.stream.Stream;
+
+public class PlanRegistryMock implements PlanRegistry {
+
+    private final Plan freeTrial = new MockPlan("trial", false, 0, 0, 0, 200, "Free Trial - for testing purposes");
+    private final Plan paidPlan  = new MockPlan("paid", true, 3, 6, 9, 500, "Paid Plan - for testing purposes");
+    private final Plan nonePlan  = new MockPlan("none", false, 0, 0, 0, 0, "None Plan - for testing purposes");
+
+    @Override
+    public Plan defaultPlan() {
+        return freeTrial;
+    }
+
+    @Override
+    public Optional<Plan> plan(PlanId planId) {
+        return Stream.of(freeTrial, paidPlan, nonePlan)
+                .filter(p -> p.id().equals(planId))
+                .findAny();
+    }
+
+    private static class MockPlan implements Plan {
+        private final PlanId planId;
+        private final String description;
+        private final CostCalculator costCalculator;
+        private final QuotaCalculator quotaCalculator;
+        private final boolean billed;
+
+        public MockPlan(String planId, boolean billed, int cpuPrice, int memPrice, int dgbPrice, int quota, String description) {
+            this.planId = PlanId.from(planId);
+            this.description = description;
+            this.costCalculator = new MockCostCalculator(BigDecimal.valueOf(cpuPrice), BigDecimal.valueOf(memPrice), BigDecimal.valueOf(dgbPrice));
+            this.quotaCalculator = () -> Quota.unlimited().withBudget(quota);
+            this.billed = billed;
+        }
+
+        @Override
+        public PlanId id() {
+            return planId;
+        }
+
+        @Override
+        public String displayName() {
+            return description;
+        }
+
+        @Override
+        public CostCalculator calculator() {
+            return costCalculator;
+        }
+
+        @Override
+        public QuotaCalculator quota() {
+            return quotaCalculator;
+        }
+
+        @Override
+        public boolean isBilled() {
+            return billed;
+        }
+    }
+
+    private static class MockCostCalculator implements CostCalculator {
+        private static final BigDecimal millisPerHour = BigDecimal.valueOf(60 * 60 * 1000);
+        private final BigDecimal cpuHourCost;
+        private final BigDecimal memHourCost;
+        private final BigDecimal dgbHourCost;
+
+        public MockCostCalculator() {
+            this(BigDecimal.valueOf(3, 2),
+                 BigDecimal.valueOf(5, 3),
+                 BigDecimal.valueOf(3, 4));
+        }
+
+        public MockCostCalculator(BigDecimal cpuPrice, BigDecimal memPrice, BigDecimal dgbPrice) {
+            this.cpuHourCost = cpuPrice;
+            this.memHourCost = memPrice;
+            this.dgbHourCost = dgbPrice;
+        }
+
+        @Override
+        public CostInfo calculate(ResourceUsage usage) {
+            var cpuCost = usage.getCpuMillis().divide(millisPerHour, RoundingMode.HALF_UP).multiply(cpuHourCost);
+            var memCost = usage.getMemoryMillis().divide(millisPerHour, RoundingMode.HALF_UP).multiply(memHourCost);
+            var dgbCost = usage.getDiskMillis().divide(millisPerHour, RoundingMode.HALF_UP).multiply(dgbHourCost);
+
+            return new CostInfo(
+                    usage.getApplicationId(),
+                    usage.getZoneId(),
+                    usage.getCpuMillis().divide(millisPerHour, RoundingMode.HALF_UP),
+                    usage.getMemoryMillis().divide(millisPerHour, RoundingMode.HALF_UP),
+                    usage.getDiskMillis().divide(millisPerHour, RoundingMode.HALF_UP),
+                    cpuCost,
+                    memCost,
+                    dgbCost
+            );
+        }
+
+        @Override
+        public double calculate(NodeResources resources) {
+            return resources.cost();
+        }
+    }
+}