blob: f0c7f806c8cb9c213ef49001f63ab588e00c4473 (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
|
package com.yahoo.vespa.hosted.controller.api.integration.billing;
import java.time.Clock;
import java.time.ZoneOffset;
import java.time.ZonedDateTime;
import java.util.Map;
import java.util.SortedMap;
import java.util.TreeMap;
/**
* @author ogronnesby
*/
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;
}
public static StatusHistory open(Clock clock) {
var now = clock.instant().atZone(ZoneOffset.UTC);
return new StatusHistory(
new TreeMap<>(Map.of(now, BillStatus.OPEN))
);
}
public BillStatus current() {
return history.get(history.lastKey());
}
public SortedMap<ZonedDateTime, BillStatus> getHistory() {
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;
};
}
}
|