From 7003febae5464929d72d7fdf66925d8fd3777f8b Mon Sep 17 00:00:00 2001
From: gjoranv <gjoranv@gmail.com>
Date: Thu, 5 Oct 2023 12:36:10 +0200
Subject: Allow maintaining and updating invoices.

---
 .../api/integration/billing/BillingReporter.java   |  3 ++
 .../api/integration/billing/InvoiceUpdate.java     | 45 ++++++++++++++++++++++
 2 files changed, 48 insertions(+)
 create mode 100644 controller-api/src/main/java/com/yahoo/vespa/hosted/controller/api/integration/billing/InvoiceUpdate.java

(limited to 'controller-api')

diff --git a/controller-api/src/main/java/com/yahoo/vespa/hosted/controller/api/integration/billing/BillingReporter.java b/controller-api/src/main/java/com/yahoo/vespa/hosted/controller/api/integration/billing/BillingReporter.java
index 7339555e578..fe19b1187f6 100644
--- a/controller-api/src/main/java/com/yahoo/vespa/hosted/controller/api/integration/billing/BillingReporter.java
+++ b/controller-api/src/main/java/com/yahoo/vespa/hosted/controller/api/integration/billing/BillingReporter.java
@@ -6,4 +6,7 @@ import com.yahoo.vespa.hosted.controller.tenant.CloudTenant;
 
 public interface BillingReporter {
     BillingReference maintainTenant(CloudTenant tenant);
+
+    default void maintainInvoices() { }
+
 }
diff --git a/controller-api/src/main/java/com/yahoo/vespa/hosted/controller/api/integration/billing/InvoiceUpdate.java b/controller-api/src/main/java/com/yahoo/vespa/hosted/controller/api/integration/billing/InvoiceUpdate.java
new file mode 100644
index 00000000000..20d0dac02b6
--- /dev/null
+++ b/controller-api/src/main/java/com/yahoo/vespa/hosted/controller/api/integration/billing/InvoiceUpdate.java
@@ -0,0 +1,45 @@
+package com.yahoo.vespa.hosted.controller.api.integration.billing;
+
+/**
+ * Helper to track changes to an invoice.
+ *
+ * @author gjoranv
+ */
+public record InvoiceUpdate(int itemsAdded, int itemsRemoved, int itemsModified) {
+    public boolean isEmpty() {
+        return itemsAdded == 0 && itemsRemoved == 0 && itemsModified == 0;
+    }
+
+    public static InvoiceUpdate empty() {
+        return new InvoiceUpdate(0, 0, 0);
+    }
+
+    public InvoiceUpdate add(InvoiceUpdate other) {
+        return new InvoiceUpdate(itemsAdded + other.itemsAdded,
+                                 itemsRemoved + other.itemsRemoved,
+                                 itemsModified + other.itemsModified);
+    }
+
+    public static class Counter {
+        private int itemsAdded = 0;
+        private int itemsRemoved = 0;
+        private int itemsModified = 0;
+
+        public void addedItem() {
+            itemsAdded++;
+        }
+
+        public void removedItem() {
+            itemsRemoved++;
+        }
+
+        public void modifiedItem() {
+            itemsModified++;
+        }
+
+        public InvoiceUpdate finish() {
+            return new InvoiceUpdate(itemsAdded, itemsRemoved, itemsModified);
+        }
+    }
+
+}
-- 
cgit v1.2.3