apple/foundationdb
1
0
Fork 0
mirror of https://github.com/apple/foundationdb.git synced 2025-05-14 18:02:31 +08:00
Code Issues Packages Projects Releases Wiki Activity

Make fdbmonitor IDs unique per section rather than globally unique

Browse Source
This commit is contained in:
A.J. Beamon 2021-07-13 16:13:48 -04:00
parent e0f3b2718a
commit 5832e61c0c
1 changed files with 29 additions and 34 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

63
fdbmonitor/fdbmonitor.cpp
View File

@ -137,6 +137,8 @@ int severity_to_priority(Severity severity) {
} }
} }
typedef std::string ProcessID;
bool daemonize = false; bool daemonize = false;
std::string logGroup = "default"; std::string logGroup = "default";
@ -390,11 +392,11 @@ public:
int pipes[2][2]; int pipes[2][2];
Command() : argv(nullptr) {} Command() : argv(nullptr) {}
Command(const CSimpleIni& ini, std::string _section, uint64_t id, fdb_fd_set fds, int* maxfd) Command(const CSimpleIni& ini, std::string _section, ProcessID id, fdb_fd_set fds, int* maxfd)
: section(_section), argv(nullptr), fork_retry_time(-1), quiet(false), delete_envvars(nullptr), fds(fds), : section(_section), argv(nullptr), fork_retry_time(-1), quiet(false), delete_envvars(nullptr), fds(fds),
deconfigured(false), kill_on_configuration_change(true) { deconfigured(false), kill_on_configuration_change(true) {
char _ssection[strlen(section.c_str()) + 22]; char _ssection[strlen(section.c_str()) + 22];
snprintf(_ssection, strlen(section.c_str()) + 22, "%s.%" PRIu64, section.c_str(), id); snprintf(_ssection, strlen(section.c_str()) + 22, "%s", id.c_str());
ssection = _ssection; ssection = _ssection;
for (auto p : pipes) { for (auto p : pipes) {
@ -593,9 +595,9 @@ public:
} }
}; };
std::unordered_map<uint64_t, std::unique_ptr<Command>> id_command; std::unordered_map<ProcessID, std::unique_ptr<Command>> id_command;
std::unordered_map<pid_t, uint64_t> pid_id; std::unordered_map<pid_t, ProcessID> pid_id;
std::unordered_map<uint64_t, pid_t> id_pid; std::unordered_map<ProcessID, pid_t> id_pid;
enum { OPT_CONFFILE, OPT_LOCKFILE, OPT_LOGGROUP, OPT_DAEMONIZE, OPT_HELP }; enum { OPT_CONFFILE, OPT_LOCKFILE, OPT_LOGGROUP, OPT_DAEMONIZE, OPT_HELP };
@ -608,7 +610,7 @@ CSimpleOpt::SOption g_rgOptions[] = { { OPT_CONFFILE, "--conffile", SO_REQ_SEP }
{ OPT_HELP, "--help", SO_NONE }, { OPT_HELP, "--help", SO_NONE },
SO_END_OF_OPTIONS }; SO_END_OF_OPTIONS };
void start_process(Command* cmd, uint64_t id, uid_t uid, gid_t gid, int delay, sigset_t* mask) { void start_process(Command* cmd, ProcessID id, uid_t uid, gid_t gid, int delay, sigset_t* mask) {
if (!cmd->argv) if (!cmd->argv)
return; return;
@ -758,7 +760,7 @@ bool argv_equal(const char** a1, const char** a2) {
return true; return true;
} }
void kill_process(uint64_t id, bool wait = true) { void kill_process(ProcessID id, bool wait = true) {
pid_t pid = id_pid[id]; pid_t pid = id_pid[id];
log_msg(SevInfo, "Killing process %d\n", pid); log_msg(SevInfo, "Killing process %d\n", pid);
@ -815,7 +817,7 @@ void load_conf(const char* confpath, uid_t& uid, gid_t& gid, sigset_t* mask, fdb
/* Any change to uid or gid requires the process to be restarted to take effect */ /* Any change to uid or gid requires the process to be restarted to take effect */
if (uid != _uid || gid != _gid) { if (uid != _uid || gid != _gid) {
std::vector<uint64_t> kill_ids; std::vector<ProcessID> kill_ids;
for (auto i : id_pid) { for (auto i : id_pid) {
if (id_command[i.first]->kill_on_configuration_change) { if (id_command[i.first]->kill_on_configuration_change) {
kill_ids.push_back(i.first); kill_ids.push_back(i.first);
@ -831,8 +833,8 @@ void load_conf(const char* confpath, uid_t& uid, gid_t& gid, sigset_t* mask, fdb
gid = _gid; gid = _gid;
} }
std::list<uint64_t> kill_ids; std::list<ProcessID> kill_ids;
std::list<std::pair<uint64_t, Command*>> start_ids; std::list<std::pair<ProcessID, Command*>> start_ids;
for (auto i : id_pid) { for (auto i : id_pid) {
if (!loadedConf || ini.GetSectionSize(id_command[i.first]->ssection.c_str()) == -1) { if (!loadedConf || ini.GetSectionSize(id_command[i.first]->ssection.c_str()) == -1) {
@ -881,31 +883,24 @@ void load_conf(const char* confpath, uid_t& uid, gid_t& gid, sigset_t* mask, fdb
ini.GetAllSections(sections); ini.GetAllSections(sections);
for (auto i : sections) { for (auto i : sections) {
if (auto dot = strrchr(i.pItem, '.')) { if (auto dot = strrchr(i.pItem, '.')) {
char* strtol_end; ProcessID id = i.pItem;
if (!id_pid.count(id)) {
/* Found something we haven't yet started */
Command* cmd;
uint64_t id = strtoull(dot + 1, &strtol_end, 10); auto itr = id_command.find(id);
if (itr != id_command.end()) {
cmd = itr->second.get();
} else {
std::string section(i.pItem, dot - i.pItem);
auto p = std::make_unique<Command>(ini, section, id, rfds, maxfd);
cmd = p.get();
id_command[id] = std::move(p);
}
if (*strtol_end != '\0' || !(id > 0)) { if (cmd->fork_retry_time <= timer()) {
log_msg(SevError, "Found bogus id in %s\n", i.pItem); log_msg(SevInfo, "Starting %s\n", i.pItem);
} else { start_process(cmd, id, uid, gid, 0, mask);
if (!id_pid.count(id)) {
/* Found something we haven't yet started */
Command* cmd;
auto itr = id_command.find(id);
if (itr != id_command.end()) {
cmd = itr->second.get();
} else {
std::string section(i.pItem, dot - i.pItem);
auto p = std::make_unique<Command>(ini, section, id, rfds, maxfd);
cmd = p.get();
id_command[id] = std::move(p);
}
if (cmd->fork_retry_time <= timer()) {
log_msg(SevInfo, "Starting %s\n", i.pItem);
start_process(cmd, id, uid, gid, 0, mask);
}
} }
} }
} }
@ -1724,7 +1719,7 @@ int main(int argc, char** argv) {
break; break;
} }
uint64_t id = pid_id[pid]; ProcessID id = pid_id[pid];
Command* cmd = id_command[id].get(); Command* cmd = id_command[id].get();
pid_id.erase(pid); pid_id.erase(pid);