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
|
// Copyright Vespa.ai. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package com.yahoo.vespa.hosted.provision.node;
import com.yahoo.config.provision.ApplicationId;
import com.yahoo.config.provision.ClusterMembership;
import com.yahoo.config.provision.NetworkPorts;
import com.yahoo.config.provision.NodeResources;
import java.util.Optional;
/**
* The allocation of a node
*
* @author bratseth
*/
public class Allocation {
private final ApplicationId owner;
private final ClusterMembership clusterMembership;
private final NodeResources requestedResources;
/**
* Restart generation, see {@link com.yahoo.vespa.hosted.provision.node.Generation},
* wanted is increased when a restart of the services on the node is needed, current is updated
* when a restart has been done on the node.
*/
private final Generation restartGeneration;
/** This node should be removed from the cluster on the next deployment */
private final boolean removable;
/** This node can be reused (dirtied) immediately after being removed from the cluster */
private final boolean reusable;
private final Optional<NetworkPorts> networkPorts;
public Allocation(ApplicationId owner, ClusterMembership clusterMembership, NodeResources requestedResources,
Generation restartGeneration, boolean removable) {
this(owner, clusterMembership, requestedResources, restartGeneration, removable, false, Optional.empty());
}
public Allocation(ApplicationId owner, ClusterMembership clusterMembership, NodeResources requestedResources,
Generation restartGeneration, boolean removable, boolean reusable,
Optional<NetworkPorts> networkPorts) {
this.owner = owner;
this.clusterMembership = clusterMembership;
this.requestedResources = requestedResources;
this.restartGeneration = restartGeneration;
this.removable = removable;
this.reusable = reusable;
this.networkPorts = networkPorts;
if (!removable && reusable) {
throw new IllegalArgumentException("Allocation must be removable in order to be reusable");
}
}
/** Returns the id of the application this is allocated to */
public ApplicationId owner() { return owner; }
/** Returns the role this node is allocated to */
public ClusterMembership membership() { return clusterMembership; }
/** Returns the node resources requested, leading to this allocation on this node */
public NodeResources requestedResources() { return requestedResources; }
/** Returns the restart generation (wanted and current) of this */
public Generation restartGeneration() { return restartGeneration; }
/** Returns network ports allocations (or empty if not recorded) */
public Optional<NetworkPorts> networkPorts() { return networkPorts; }
/** Returns a copy of this which is retired */
public Allocation retire() {
return new Allocation(owner, clusterMembership.retire(), requestedResources, restartGeneration, removable, reusable, networkPorts);
}
/** Returns a copy of this which is not retired */
public Allocation unretire() {
return new Allocation(owner, clusterMembership.unretire(), requestedResources, restartGeneration, removable, reusable, networkPorts);
}
/** Returns whether this node is ready to be removed from the application */
public boolean removable() { return removable; }
/** Returns whether this node has fully retired and can be reused immediately */
public boolean reusable() {
return reusable;
}
public Allocation withRequestedResources(NodeResources resources) {
return new Allocation(owner, clusterMembership, resources, restartGeneration, removable, reusable, networkPorts);
}
/** Returns a copy of this with the current restart generation set to generation */
public Allocation withRestart(Generation generation) {
return new Allocation(owner, clusterMembership, requestedResources, generation, removable, reusable, networkPorts);
}
/**
* Returns a copy of this allocation where removable and reusable are set to the given values. A node which is
* reusable may be moved directly to {@link com.yahoo.vespa.hosted.provision.Node.State#dirty} after removal.
*/
public Allocation removable(boolean removable, boolean reusable) {
return new Allocation(owner, clusterMembership, requestedResources, restartGeneration, removable, reusable, networkPorts);
}
public Allocation with(ClusterMembership newMembership) {
return new Allocation(owner, newMembership, requestedResources, restartGeneration, removable, reusable, networkPorts);
}
public Allocation withNetworkPorts(NetworkPorts ports) {
return new Allocation(owner, clusterMembership, requestedResources, restartGeneration, removable, reusable, Optional.of(ports));
}
@Override
public String toString() { return "allocated to " + owner + " as '" + clusterMembership + "'"; }
}
|
