Merge pull request #29077 from vespa-engine/bill-state-transitions

Add bill status logic.

2 files changed, 32 insertions, 3 deletions

diff --git a/controller-api/src/main/java/com/yahoo/vespa/hosted/controller/api/integration/billing/BillStatus.java b/controller-api/src/main/java/com/yahoo/vespa/hosted/controller/api/integration/billing/BillStatus.java
index d6c6262069b..4f35b47219a 100644
--- a/controller-api/src/main/java/com/yahoo/vespa/hosted/controller/api/integration/billing/BillStatus.java
+++ b/controller-api/src/main/java/com/yahoo/vespa/hosted/controller/api/integration/billing/BillStatus.java
@@ -5,12 +5,14 @@ package com.yahoo.vespa.hosted.controller.api.integration.billing;
  */
 public enum BillStatus {
     OPEN,       // All bills start in this state. The bill can be modified and exported/synced to external systems.
-    FROZEN,     // Syncing to external systems is switched off. Reviews should be done in this state.
+    FROZEN,     // Syncing to external systems is switched off. No changes can be made.
     CLOSED,     // End state for a valid bill.
     VOID;       // End state, indicating that the bill is not valid.
 
-    private static final String LEGACY_ISSUED = "ISSUED";         // Legacy state, used by historical bills
-    private static final String LEGACY_EXPORTED = "EXPORTED";     // Legacy state, used by historical bills
+    // Legacy states, used by historical bills
+    private static final String LEGACY_ISSUED = "ISSUED";
+    private static final String LEGACY_EXPORTED = "EXPORTED";
+    private static final String LEGACY_CANCELED = "CANCELED";
 
     private final String value;
 
@@ -24,6 +26,7 @@ public enum BillStatus {
 
     public static BillStatus from(String status) {
         if (LEGACY_ISSUED.equals(status) || LEGACY_EXPORTED.equals(status)) return OPEN;
+        if (LEGACY_CANCELED.equals(status)) return VOID;
         return Enum.valueOf(BillStatus.class, status.toUpperCase());
     }
 
diff --git a/controller-api/src/main/java/com/yahoo/vespa/hosted/controller/api/integration/billing/StatusHistory.java b/controller-api/src/main/java/com/yahoo/vespa/hosted/controller/api/integration/billing/StatusHistory.java
index 6335ada1396..f0c7f806c8c 100644
--- a/controller-api/src/main/java/com/yahoo/vespa/hosted/controller/api/integration/billing/StatusHistory.java
+++ b/controller-api/src/main/java/com/yahoo/vespa/hosted/controller/api/integration/billing/StatusHistory.java
@@ -14,6 +14,17 @@ public class StatusHistory {
     SortedMap<ZonedDateTime, BillStatus> history;
 
     public StatusHistory(SortedMap<ZonedDateTime, BillStatus> history) {
+        // Validate the given history
+        var iter = history.values().iterator();
+        BillStatus next  = iter.hasNext() ? iter.next() : null;
+        while (iter.hasNext()) {
+            var current = next;
+            next = iter.next();
+            if (! validateStatus(current, next)) {
+                throw new IllegalArgumentException("Invalid transition from " + current + " to " + next);
+            }
+        }
+
         this.history = history;
     }
 
@@ -32,4 +43,19 @@ public class StatusHistory {
         return history;
     }
 
+    public void checkValidTransition(BillStatus newStatus) {
+        if (! validateStatus(current(), newStatus)) {
+            throw new IllegalArgumentException("Invalid transition from " + current() + " to " + newStatus);
+        }
+    }
+
+    private static boolean validateStatus(BillStatus current, BillStatus newStatus) {
+        return switch(current) {
+            case OPEN -> true;
+            case FROZEN -> newStatus != BillStatus.OPEN; // This could be subject to change.
+            case CLOSED -> newStatus == BillStatus.CLOSED;
+            case VOID -> newStatus == BillStatus.VOID;
+        };
+    }
+
 }