odyssey/sources/grac_shutdown_worker.c

#include <stdlib.h>
#include <stdarg.h>
#include <stdint.h>
#include <stdio.h>
#include <string.h>
#include <ctype.h>
#include <inttypes.h>
#include <assert.h>

#include <unistd.h>
#include <signal.h>
#include <sys/stat.h>
#include <errno.h>

#include <machinarium.h>
#include <kiwi.h>
#include <odyssey.h>
#include <sys/prctl.h>

static inline int
od_system_server_complete_stop(od_system_server_t *server)
{
	/* shutdown */
	int rc;
	rc = machine_shutdown(server->io);

	if (rc == -1)
		return NOT_OK_RESPONSE;
	return OK_RESPONSE;
}

void
od_grac_shutdown_worker(void *arg)
{

	od_system_t *system     = arg;
	od_instance_t *instance = system->global->instance;
	od_log(&instance->logger,
	       "config",
	       NULL,
	       NULL,
	       "stop to accepting new connections");

	od_setproctitlef(&instance->orig_argv_ptr,
	                 "odyssey: version %s stop accepting any connections and "
	                 "working with old transactions",
	                 OD_VERSION_NUMBER);

	od_router_t *router = system->global->router;

	od_list_t *i;
	od_list_foreach(&router->servers, i)
	{
		od_system_server_t *server;
		server         = od_container_of(i, od_system_server_t, link);
		server->closed = true;
	}

	od_dbg_printf_on_dvl_lvl(1, "servers closed\n", 0);

	/* wait for all servers to complete old transations */
	od_list_foreach(&router->servers, i)
	{
		od_system_server_t *server;
		server = od_container_of(i, od_system_server_t, link);

		while (od_atomic_u32_of(&router->clients_routing) ||
		       od_atomic_u32_of(&router->clients)) {
			od_dbg_printf_on_dvl_lvl(1, "waiting for %s\n", server->sid.id);
			machine_sleep(1000);
		}

		od_dbg_printf_on_dvl_lvl(1, "server shutdown ok %s\n", server->sid.id);
	}

	od_dbg_printf_on_dvl_lvl(1, "shutting down sockets\n", "");

	/* close sockets */
	od_list_foreach(&router->servers, i)
	{
		od_system_server_t *server;
		server = od_container_of(i, od_system_server_t, link);
		od_system_server_complete_stop(server);
	}

	machine_stop(system->machine);
	od_dbg_printf_on_dvl_lvl(
	  1, "waiting done, sending sigint to own process %d\n", instance->pid.pid);

	kill(instance->pid.pid, SIGINT);
}