aboutsummaryrefslogtreecommitdiffstats
path: root/config-provisioning/src/main/java/com/yahoo/config/provision/NodeType.java
blob: 0eac568ec4535e7c42695d9ae12b2077934665e5 (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
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
// Copyright Vespa.ai. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package com.yahoo.config.provision;

import java.util.List;

/**
 * The possible types of nodes in the node repository
 * 
 * @author bratseth
 */
public enum NodeType {

    /** Node assignable to a tenant to run application workloads */
    tenant("Tenant node"),

    /** Host of a tenant nodes */
    host("Tenant host", tenant),

    /** Node serving the shared proxy layer */
    proxy("Proxy node"),

    /** Host of a proxy node */
    proxyhost("Proxy host", proxy),

    /** Config server node */
    config("Config server node"),

    /** Host of a config server node */
    confighost("Config server host", config),

    /** Controller node */
    controller("Controller node"),

    /** Host of a controller node */
    controllerhost("Controller host", controller);

    private final String description;
    private final List<NodeType> childNodeTypes;

    NodeType(String description, NodeType... childNodeTypes) {
        this.childNodeTypes = List.of(childNodeTypes);
        this.description = description;
    }

    public boolean isHost() {
        return !childNodeTypes.isEmpty();
    }

    /** either config server or controller */
    public boolean isConfigServerLike() {
        return this == config || this == controller;
    }

    /** either config server host or controller host */
    public boolean isConfigServerHostLike() {
        return this == confighost || this == controllerhost;
    }

    /** Returns whether this supports host sharing */
    public boolean isSharable() {
        return this == NodeType.host;
    }

    public String description() {
        return description;
    }

    /**
     * @return {@link NodeType} of the node(s) that run on this host
     * @throws IllegalStateException if this type is not a host
     */
    public NodeType childNodeType() {
        return childNodeTypes().get(0);
    }

    /**
     * @return all {@link NodeType}s that can run on this host
     * @throws IllegalStateException if this type is not a host
     */
    public List<NodeType> childNodeTypes() {
        if (! isHost())
            throw new IllegalStateException(this + " has no children");
        return childNodeTypes;
    }

    /** Returns whether given node type can run on this */
    public boolean canRun(NodeType type) {
        return childNodeTypes.contains(type);
    }

    /** Returns the parent host type. */
    public NodeType parentNodeType() {
        for (var type : values()) {
            if (type.childNodeTypes.contains(this)) return type;
        }
        throw new IllegalStateException(this + " has no parent");
    }

    /** Returns the host type of this */
    public NodeType hostType() {
        if (isHost()) return this;
        for (NodeType nodeType : values()) {
            // Ignore host types that support multiple node types
            if (nodeType.childNodeTypes.size() == 1 && nodeType.canRun(this)) {
                return nodeType;
            }
        }
        throw new IllegalStateException("No host of " + this + " exists");
    }

}