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
|
// Copyright 2017 Yahoo Holdings. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
#include <vespa/config/common/exceptions.h>
#include <signal.h>
#include <unistd.h>
#include <sys/time.h>
#include <vespa/defaults.h>
#include "config-handler.h"
#include <vespa/log/log.h>
LOG_SETUP("config-sentinel");
using namespace config;
constexpr uint64_t CONFIG_TIMEOUT_MS = 3 * 60 * 1000;
static int sigPermanent(int sig, void(*handler)(int));
static void gracefulShutdown(int sig);
static void sigchldHandler(int sig);
sig_atomic_t stop = 0;
static sig_atomic_t pendingWait = 0;
int
main(int argc, char **argv)
{
int c = getopt(argc, argv, "c:");
if (c != 'c') {
LOG(error, "Usage: %s -c <config-id>", argv[0]);
EV_STOPPING("config-sentinel", "Bad arguments on command line");
exit(EXIT_FAILURE);
}
const char *configId = strdup(optarg);
const char *rootDir = getenv("ROOT");
if (!rootDir) {
rootDir = vespa::Defaults::vespaHome();
LOG(warning, "ROOT is not set, using %s", rootDir);
setenv("ROOT", rootDir, 1);
}
if (chdir(rootDir) == -1) {
LOG(error, "Fatal: Cannot cd to $ROOT (%s)", rootDir);
EV_STOPPING("config-sentinel", "Cannot cd to $ROOT");
exit(EXIT_FAILURE);
}
EV_STARTED("config-sentinel");
sigPermanent(SIGPIPE, SIG_IGN);
sigPermanent(SIGTERM, gracefulShutdown);
sigPermanent(SIGINT, gracefulShutdown);
sigPermanent(SIGCHLD, sigchldHandler);
if (setenv("LC_ALL", "C", 1) != 0) {
LOG(error, "Unable to set locale");
exit(EXIT_FAILURE);
}
setlocale(LC_ALL, "C");
sentinel::ConfigHandler handler;
LOG(debug, "Reading configuration");
try {
handler.subscribe(configId, CONFIG_TIMEOUT_MS);
} catch (ConfigTimeoutException & ex) {
LOG(warning, "Timout getting config, please check your setup. Will exit and restart: %s", ex.getMessage().c_str());
EV_STOPPING("config-sentinel", ex.what());
exit(EXIT_FAILURE);
} catch (InvalidConfigException& ex) {
LOG(error, "Fatal: Invalid configuration, please check your setup: %s", ex.getMessage().c_str());
EV_STOPPING("config-sentinel", ex.what());
exit(EXIT_FAILURE);
} catch (ConfigRuntimeException& ex) {
LOG(error, "Fatal: Could not get config, please check your setup: %s", ex.getMessage().c_str());
EV_STOPPING("config-sentinel", ex.what());
exit(EXIT_FAILURE);
}
struct timeval lastTv;
gettimeofday(&lastTv, nullptr);
while (!stop) {
try {
pendingWait = 0;
handler.doWork(); // Check for child procs & commands
} catch (InvalidConfigException& ex) {
LOG(warning, "Configuration problem: (ignoring): %s",
ex.what());
}
if (!pendingWait) {
int maxNum = 0;
fd_set fds;
FD_ZERO(&fds);
handler.updateActiveFdset(&fds, &maxNum);
struct timeval tv;
tv.tv_sec = 1;
tv.tv_usec = 0;
if (!pendingWait) {
select(maxNum, &fds, nullptr, nullptr, &tv);
}
}
struct timeval tv;
gettimeofday(&tv, nullptr);
double delta = tv.tv_sec - lastTv.tv_sec
+ 1e-6 * tv.tv_usec - lastTv.tv_usec;
if (delta < 0.01) {
usleep(12500); // Avoid busy looping;
}
lastTv = tv;
}
int rv = handler.terminate();
EV_STOPPING("config-sentinel", "normal exit");
return rv;
}
static void
gracefulShutdown(int sig)
{
(void)sig;
stop = 1;
}
static void
sigchldHandler(int sig)
{
(void)sig;
pendingWait = 1;
}
static int
sigPermanent(int sig, void(*handler)(int))
{
struct sigaction sa;
memset(&sa, 0, sizeof(sa));
sigemptyset(&sa.sa_mask);
sa.sa_flags = 0; // no SA_RESTART!
sa.sa_handler = handler;
return sigaction(sig, &sa, nullptr);
}
|
