blob: 875b6190763f94912bc70b0f08033b61a6bff905 (
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
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
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
|
/*
* Copyright 2019 Oath Inc. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
*/
package ai.vespa.metricsproxy.service;
import com.google.inject.Inject;
import com.yahoo.log.LogLevel;
import java.io.BufferedReader;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.io.PrintStream;
import java.util.ArrayList;
import java.util.List;
import java.util.logging.Logger;
/**
* Connects to the config sentinel and gets information like pid for the services on the node
*/
public class ConfigSentinelClient {
private final static Logger log = Logger.getLogger(ConfigSentinelClient.class.getName());
private final CmdClient client;
@Inject
public ConfigSentinelClient() {
this.client = new CmdClient();
}
/**
* Update all services reading from config sentinel
*
* @param services The list of services
*/
synchronized void updateServiceStatuses(List<VespaService> services) {
try {
setStatus(services);
} catch (Exception e) {
log.log(LogLevel.ERROR, "Unable to update service pids from sentinel", e);
}
}
/**
* Update status
*
* @param s The service to update the status for
*/
public synchronized void ping(VespaService s) {
List<VespaService> services = new ArrayList<>();
services.add(s);
log.log(LogLevel.DEBUG, "Ping for service " + s);
try {
setStatus(services);
} catch (Exception e) {
log.log(LogLevel.ERROR, "Unable to update service pids from sentinel", e);
}
}
/**
* Update the status (pid check etc)
*
* @param services list of services
* @throws Exception if something went wrong
*/
protected synchronized void setStatus(List<VespaService> services) throws Exception {
InputStream in;
PrintStream out;
client.connect();
in = client.getInputStream();
BufferedReader reader = new BufferedReader(new InputStreamReader(in));
String line;
List<VespaService> updatedServices = new ArrayList<>();
while ((line = reader.readLine()) != null) {
if (line.equals("")) {
break;
}
VespaService s = parseServiceString(line, services);
if (s != null) {
updatedServices.add(s);
}
}
//Check if there are services that were not found in output
//from the sentinel
for (VespaService s : services) {
if ((!s.getServiceName().equals("configserver")) && !updatedServices.contains(s)) {
log.log(LogLevel.DEBUG,"Service " + s + " is no longer found with sentinel - setting alive = false");
s.setAlive(false);
}
}
//Close streams
reader.close();
client.disconnect();
}
protected static VespaService parseServiceString(String line, List<VespaService> services) {
String[] parts = line.split(" ");
if (parts.length < 3)
return null;
String name = parts[0];
int pid = -1;
String state = null;
VespaService service = null;
for (VespaService s : services) {
if (s.getInstanceName().compareToIgnoreCase(name) == 0) {
service = s;
break;
}
}
//Could not find this service
//nothing wrong with that as the check is invoked per line from sentinel
if (service == null) {
return service;
}
for (int i = 1; i < parts.length; i++) {
String keyValue[] = parts[i].split("=");
String key = keyValue[0];
String value = keyValue[1];
if (key.equals("state")) {
state = value;
} else if (key.equals("pid")) {
pid = Integer.parseInt(value);
}
}
if (state != null) {
service.setState(state);
if (pid >= 0 && "RUNNING".equals(state)) {
service.setAlive(true);
service.setPid(pid);
} else {
service.setAlive(false);
}
} else {
service.setAlive(false);
}
return service;
}
static class CmdClient {
Process proc;
// NOTE: hostname/port not used yet
void connect() {
String[] args = new String[]{"vespa-sentinel-cmd", "list"};
try {
proc = Runtime.getRuntime().exec(args);
} catch (Exception e) {
log.log(LogLevel.WARNING, "could not run vespa-sentinel-cmd: "+e);
proc = null;
}
}
void disconnect() {
if (proc.isAlive()) {
proc.destroy();
}
proc = null;
}
InputStream getInputStream() {
return (proc != null)
? proc.getInputStream()
: new java.io.ByteArrayInputStream(new byte[0]);
}
}
}
|
