aboutsummaryrefslogtreecommitdiffstats
path: root/controller-server/src/main/java/com/yahoo/vespa/hosted/controller/routing/rotation/RotationStatus.java
blob: 89247ca2a31b2b12c1fe2298bb5205ce1ced445c (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
62
63
64
65
66
67
// Copyright Yahoo. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package com.yahoo.vespa.hosted.controller.routing.rotation;

import com.yahoo.config.provision.zone.ZoneId;
import com.yahoo.vespa.hosted.controller.application.Deployment;

import java.time.Instant;
import java.util.Map;
import java.util.Objects;

/**
 * The status of all rotations assigned to an application.
 *
 * @author mpolden
 */
public record RotationStatus(Map<RotationId, Targets> status) {

    public static final RotationStatus EMPTY = new RotationStatus(Map.of());

    public RotationStatus(Map<RotationId, Targets> status) {
        this.status = Map.copyOf(Objects.requireNonNull(status));
    }

    public Map<RotationId, Targets> asMap() {
        return status;
    }

    /** Get targets of given rotation, if any */
    public Targets of(RotationId rotation) {
        return status.getOrDefault(rotation, Targets.NONE);
    }

    /** Get status of deployment in given rotation, if any */
    public RotationState of(RotationId rotation, Deployment deployment) {
        return of(rotation).asMap().entrySet().stream()
                           .filter(kv -> kv.getKey().equals(deployment.zone()))
                           .map(Map.Entry::getValue)
                           .findFirst()
                           .orElse(RotationState.unknown);
    }

    @Override
    public String toString() {
        return "rotation status " + status;
    }

    public static RotationStatus from(Map<RotationId, Targets> targets) {
        return targets.isEmpty() ? EMPTY : new RotationStatus(targets);
    }

    /** Targets of a rotation */
    public record Targets(Map<ZoneId, RotationState> targets, Instant lastUpdated) {

        public static final Targets NONE = new Targets(Map.of(), Instant.EPOCH);

        public Targets(Map<ZoneId, RotationState> targets, Instant lastUpdated) {
            this.targets = Map.copyOf(Objects.requireNonNull(targets, "states must be non-null"));
            this.lastUpdated = Objects.requireNonNull(lastUpdated, "lastUpdated must be non-null");
        }

        public Map<ZoneId, RotationState> asMap() {
            return targets;
        }

    }

}