kiwix/kiwix-tools
1
0
Fork 0
mirror of https://github.com/kiwix/kiwix-tools.git synced 2025-09-22 03:12:20 -04:00
Code Issues Packages Projects Releases Wiki Activity

Handling of SIGTERM & SIGINT in kiwix-serve

Browse Source
This commit is contained in:
Veloman Yunkan 2021-10-09 23:13:24 +04:00
parent d73fcb357f
commit 4faaa978f7
1 changed files with 34 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

35
src/server/kiwix-serve.cpp
View File

@ -27,6 +27,7 @@
# include <windows.h>
#else
# include <unistd.h>
# include <signal.h>
#endif
#ifdef __APPLE__
@ -97,8 +98,40 @@ string loadCustomTemplate (string customIndexPath) {
return indexTemplateString;
}
volatile sig_atomic_t waiting = false;
#ifndef _WIN32
void handle_sigterm(int signum)
{
if ( waiting == false ) {
_exit(signum);
}
waiting = false;
}
typedef void (*SignalHandler)(int);
void set_signal_handler(int sig, SignalHandler handler)
{
struct sigaction sa;
sigaction(sig, NULL, &sa);
sa.sa_handler = handler;
sigaction(sig, &sa, NULL);
}
void setup_sighandlers()
{
set_signal_handler(SIGTERM, &handle_sigterm);
set_signal_handler(SIGINT, &handle_sigterm);
}
#endif
int main(int argc, char** argv)
{
#ifndef _WIN32
setup_sighandlers();
#endif
std::string rootLocation = "";
kiwix::Library library;
unsigned int nb_threads = DEFAULT_THREADS;
@ -298,7 +331,7 @@ int main(int argc, char** argv)
}
/* Run endless (until PPID dies) */
bool waiting = true;
waiting = true;
do {
if (PPID > 0) {
#ifdef _WIN32