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
|
// Copyright Yahoo. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package com.yahoo.vespa.model;
import com.yahoo.cloud.config.SentinelConfig;
import com.yahoo.config.provision.ApplicationId;
import com.yahoo.config.provision.Zone;
/**
* There is one config-sentinel running on each Vespa host, and one
* instance of this class is therefore created by each instance of
* class {@link Host}.
*
* @author gjoranv
*/
public class ConfigSentinel extends AbstractService implements SentinelConfig.Producer {
static final int BASEPORT = 19097;
private final ApplicationId applicationId;
private final Zone zone;
/**
* Constructs a new ConfigSentinel for the given host.
*
* @param host Physical host on which to run.
*/
public ConfigSentinel(Host host, ApplicationId applicationId, Zone zone) {
super(host, "sentinel");
this.applicationId = applicationId;
this.zone = zone;
portsMeta.on(0).tag("rpc").tag("admin");
portsMeta.on(1).tag("telnet").tag("interactive").tag("http").tag("state");
setProp("clustertype", "hosts");
setProp("clustername", "admin");
}
@Override
public void allocatePorts(int start, PortAllocBridge from) {
if (start == 0) start = BASEPORT;
from.requirePort(start++, "rpc");
from.requirePort(start++, "http");
}
/**
* Returns the desired base port for this service.
*/
public int getWantedPort() { return 19097; }
/**
* The desired base port is the only allowed base port.
*/
public boolean requiresWantedPort() { return true; }
/**
* @return The number of ports reserved by the Sentinel.
*/
public int getPortCount() { return 2; }
@Override
public int getHealthPort() {return getRelativePort(1); }
/**
* Overrides parent method as this is named config-sentinel and not configsentinel all over Vespa
* @return service type for config-sentinel
*/
public String getServiceType(){
return "config-sentinel";
}
@Override
public void getConfig(SentinelConfig.Builder builder) {
builder.application(getApplicationConfig());
for (Service s : getHostResource().getServices()) {
s.getStartupCommand().ifPresent(command -> builder.service(getServiceConfig(s, command)));
}
}
private SentinelConfig.Application.Builder getApplicationConfig() {
SentinelConfig.Application.Builder builder = new SentinelConfig.Application.Builder();
builder.tenant(applicationId.tenant().value());
builder.name(applicationId.application().value());
builder.environment(zone.environment().value());
builder.region(zone.region().value());
builder.instance(applicationId.instance().value());
return builder;
}
private SentinelConfig.Service.Builder getServiceConfig(Service s, String startupCommand) {
SentinelConfig.Service.Builder serviceBuilder = new SentinelConfig.Service.Builder();
serviceBuilder.command(startupCommand);
serviceBuilder.name(s.getServiceName());
serviceBuilder.id(s.getConfigId());
serviceBuilder.affinity(getServiceAffinity(s));
for (var entry : s.getEnvVars().entrySet()) {
serviceBuilder.environ(b -> b.varname(entry.getKey()).varvalue(entry.getValue().toString()));
}
for (var entry : s.getLogctlSpecs()) {
serviceBuilder.logctl(b -> b.componentSpec(entry.componentSpec())
.levelsModSpec(entry.levelsModSpec().toLogctlModSpec()));
}
setPreShutdownCommand(serviceBuilder, s);
return serviceBuilder;
}
private void setPreShutdownCommand(SentinelConfig.Service.Builder serviceBuilder, Service service) {
if (service.getPreShutdownCommand().isPresent()) {
serviceBuilder.preShutdownCommand(service.getPreShutdownCommand().get());
}
}
private SentinelConfig.Service.Affinity.Builder getServiceAffinity(Service s) {
SentinelConfig.Service.Affinity.Builder builder = new SentinelConfig.Service.Affinity.Builder();
if (s.getAffinity().isPresent()) {
builder.cpuSocket(s.getAffinity().get().cpuSocket());
}
return builder;
}
}
|
