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
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
|
// 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.time.Duration;
import java.util.Objects;
import java.util.Optional;
/**
* A capacity request.
*
* @author Ulf Lilleengen
* @author bratseth
*/
public final class Capacity {
/** Resources should stay between these values, inclusive */
private final ClusterResources min, max;
private final IntRange groupSize;
private final boolean required;
private final boolean canFail;
private final NodeType type;
private final Optional<CloudAccount> cloudAccount;
private final ClusterInfo clusterInfo;
private Capacity(ClusterResources min,
ClusterResources max,
IntRange groupSize,
boolean required,
boolean canFail,
NodeType type,
Optional<CloudAccount> cloudAccount,
ClusterInfo clusterInfo) {
validate(min);
validate(max);
if (max.smallerThan(min))
throw new IllegalArgumentException("The max capacity must be larger than the min capacity, but got min " +
min + " and max " + max);
if (cloudAccount.isEmpty() && ! clusterInfo.hostTTL().isZero())
throw new IllegalArgumentException("Cannot set hostTTL without a custom cloud account");
this.min = min;
this.max = max;
this.groupSize = groupSize;
this.required = required;
this.canFail = canFail;
this.type = type;
this.cloudAccount = Objects.requireNonNull(cloudAccount);
this.clusterInfo = clusterInfo;
}
private static void validate(ClusterResources resources) {
if (resources.nodes() == 0 && resources.groups() == 0) return; // unspecified
if (resources.nodes() % resources.groups() != 0)
throw new IllegalArgumentException("The number of nodes (" + resources.nodes() +
") must be divisible by the number of groups (" + resources.groups() + ")");
}
public ClusterResources minResources() { return min; }
public ClusterResources maxResources() { return max; }
public IntRange groupSize() { return groupSize; }
/** Returns whether the requested number of nodes must be met exactly for a request for this to succeed */
public boolean isRequired() { return required; }
/**
* Returns true if an exception should be thrown if the specified capacity can not be satisfied
* (to whatever policies are applied and taking required true/false into account).
* Returns false if it is preferable to still succeed with partially satisfied capacity.
*/
public boolean canFail() { return canFail; }
/**
* Returns the node type (role) requested. This is tenant nodes by default.
* If some other type is requested the node count and flavor may be ignored
* and all nodes of the requested type returned instead.
*/
public NodeType type() { return type; }
/** Returns the cloud account where this capacity is requested */
public Optional<CloudAccount> cloudAccount() {
return cloudAccount;
}
public ClusterInfo clusterInfo() { return clusterInfo; }
public Capacity withLimits(ClusterResources min, ClusterResources max) {
return withLimits(min, max, IntRange.empty());
}
public Capacity withLimits(ClusterResources min, ClusterResources max, IntRange groupSize) {
return new Capacity(min, max, groupSize, required, canFail, type, cloudAccount, clusterInfo);
}
@Override
public String toString() {
return (required ? "required " : "") +
(min.equals(max) ? min : "between " + min + " and " + max);
}
/** Create a non-required, failable capacity request */
public static Capacity from(ClusterResources resources) {
return from(resources, resources);
}
/** Create a non-required, failable capacity request */
public static Capacity from(ClusterResources min, ClusterResources max) {
return from(min, max, IntRange.empty());
}
public static Capacity from(ClusterResources min, ClusterResources max, IntRange groupSize) {
return from(min, max, groupSize, false, true, Optional.empty(), ClusterInfo.empty());
}
public static Capacity from(ClusterResources resources, boolean required, boolean canFail) {
return from(resources, required, canFail, Duration.ZERO);
}
public static Capacity from(ClusterResources resources, boolean required, boolean canFail, Duration hostTTL) {
return from(resources, required, canFail, NodeType.tenant, hostTTL);
}
public static Capacity from(ClusterResources min, ClusterResources max, IntRange groupSize, boolean required, boolean canFail,
Optional<CloudAccount> cloudAccount, ClusterInfo clusterInfo) {
return new Capacity(min, max, groupSize, required, canFail, NodeType.tenant, cloudAccount, clusterInfo);
}
// TODO: remove at some point, much later than March 2023 ... ?
public static Capacity from(ClusterResources min, ClusterResources max, boolean required, boolean canFail) {
return new Capacity(min, max, IntRange.empty(), required, canFail, NodeType.tenant, Optional.empty(), ClusterInfo.empty());
}
// TODO: remove at some point, much later than March 2023 ... ?
public static Capacity from(ClusterResources min, ClusterResources max, boolean required, boolean canFail, Optional<CloudAccount> cloudAccount) {
return new Capacity(min, max, IntRange.empty(), required, canFail, NodeType.tenant, cloudAccount, ClusterInfo.empty());
}
// TODO: remove at some point, much later than March 2023 ... ?
public static Capacity from(ClusterResources min, ClusterResources max, IntRange groupSize, boolean required, boolean canFail, Optional<CloudAccount> cloudAccount) {
return new Capacity(min, max, groupSize, required, canFail, NodeType.tenant, cloudAccount, ClusterInfo.empty());
}
/** Creates this from a node type */
public static Capacity fromRequiredNodeType(NodeType type) {
return from(new ClusterResources(0, 1, NodeResources.unspecified()), true, false, type, Duration.ZERO);
}
private static Capacity from(ClusterResources resources, boolean required, boolean canFail, NodeType type, Duration hostTTL) {
return new Capacity(resources, resources, IntRange.empty(), required, canFail, type, Optional.empty(), new ClusterInfo.Builder().hostTTL(hostTTL).build());
}
}
|
