mirrors/ZeroTierOne
1
Fork 0
mirror of https://github.com/zerotier/ZeroTierOne.git synced 2026-01-23 04:51:20 -08:00
Code Issues Projects Releases Wiki Activity

Stop and restart the Prometheus worker thread when daemonizing

Browse source 
Also use _Exit() instead of returning from main() to avoid duplicate
cleanup/flusing/etc.
This commit is contained in:
Chris Spiegel 2025-11-03 17:48:04 -08:00 • committed by Chris Spiegel
parent ce5556a677
commit f231561276
2 changed files with 15 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

10
ext/prometheus-cpp-lite-1.0/core/include/prometheus/save_to_file.h
View file

@ -50,6 +50,16 @@ namespace prometheus {
public:
SaveToFile() = default;
void stop() {
must_die = true;
worker_thread.join();
}
void restart() {
must_die = false;
worker_thread = std::thread(&SaveToFile::worker_function, this);
}
~SaveToFile() {
must_die = true;
worker_thread.join();

6
one.cpp
View file

@ -2330,14 +2330,18 @@ int main(int argc, char** argv)
}
#endif // !ZT_ONE_NO_ROOT_CHECK
if (runAsDaemon) {
prometheus::simpleapi::saver.stop();
long p = (long)fork();
if (p < 0) {
fprintf(stderr, "%s: could not fork" ZT_EOL_S, argv[0]);
return 1;
}
else if (p > 0)
return 0; // forked
_Exit(0); // forked
// else p == 0, so we are daemonized
prometheus::simpleapi::saver.restart();
}
#endif // __UNIX_LIKE__