diff options
Diffstat (limited to 'controller-api')
2 files changed, 10 insertions, 0 deletions
diff --git a/controller-api/src/main/java/com/yahoo/vespa/hosted/controller/api/integration/billing/BillingController.java b/controller-api/src/main/java/com/yahoo/vespa/hosted/controller/api/integration/billing/BillingController.java index 0fc20095b41..1bd1faf5dd1 100644 --- a/controller-api/src/main/java/com/yahoo/vespa/hosted/controller/api/integration/billing/BillingController.java +++ b/controller-api/src/main/java/com/yahoo/vespa/hosted/controller/api/integration/billing/BillingController.java @@ -16,6 +16,8 @@ public interface BillingController { PlanId getPlan(TenantName tenant); + Map<TenantName, PlanId> getPlans(List<TenantName> tenants); + String getPlanDisplayName(PlanId planId); Quota getQuota(TenantName tenant); diff --git a/controller-api/src/main/java/com/yahoo/vespa/hosted/controller/api/integration/billing/MockBillingController.java b/controller-api/src/main/java/com/yahoo/vespa/hosted/controller/api/integration/billing/MockBillingController.java index 21eada37ab1..3f241510ed6 100644 --- a/controller-api/src/main/java/com/yahoo/vespa/hosted/controller/api/integration/billing/MockBillingController.java +++ b/controller-api/src/main/java/com/yahoo/vespa/hosted/controller/api/integration/billing/MockBillingController.java @@ -34,6 +34,14 @@ public class MockBillingController implements BillingController { } @Override + public Map<TenantName, PlanId> getPlans(List<TenantName> tenants) { + return tenants.stream().collect(Collectors.toMap( + (TenantName t) -> t, + (TenantName t) -> plans.getOrDefault(t, PlanId.from("trial")) + )); + } + + @Override public String getPlanDisplayName(PlanId planId) { return "Plan with id: " + planId.value(); } |