odyssey/sources/instance.c

/*
 * Odissey.
 *
 * Advanced PostgreSQL connection pooler.
*/

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

#include <machinarium.h>
#include <shapito.h>

#include "sources/macro.h"
#include "sources/version.h"
#include "sources/atomic.h"
#include "sources/util.h"
#include "sources/error.h"
#include "sources/list.h"
#include "sources/pid.h"
#include "sources/id.h"
#include "sources/logger.h"
#include "sources/daemon.h"
#include "sources/scheme.h"
#include "sources/scheme_mgr.h"
#include "sources/config.h"
#include "sources/msg.h"
#include "sources/system.h"
#include "sources/server.h"
#include "sources/server_pool.h"
#include "sources/client.h"
#include "sources/client_pool.h"
#include "sources/route_id.h"
#include "sources/route.h"
#include "sources/route_pool.h"
#include "sources/io.h"
#include "sources/instance.h"
#include "sources/router.h"
#include "sources/console.h"
#include "sources/pooler.h"
#include "sources/periodic.h"
#include "sources/relay.h"
#include "sources/relay_pool.h"
#include "sources/frontend.h"

void od_instance_init(od_instance_t *instance)
{
	od_pid_init(&instance->pid);
	od_logger_init(&instance->logger, &instance->pid);
	od_scheme_init(&instance->scheme);
	od_schememgr_init(&instance->scheme_mgr);
	od_idmgr_init(&instance->id_mgr);
	shapito_cache_init(&instance->stream_cache);
	instance->config_file = NULL;
	instance->is_shared = 0;

	sigset_t mask;
	sigemptyset(&mask);
	sigaddset(&mask, SIGINT);
	sigaddset(&mask, SIGHUP);
	sigaddset(&mask, SIGPIPE);
	sigprocmask(SIG_BLOCK, &mask, NULL);
}

void od_instance_free(od_instance_t *instance)
{
	if (instance->scheme.pid_file)
		od_pid_unlink(&instance->pid, instance->scheme.pid_file);
	od_scheme_free(&instance->scheme);
	od_logger_close(&instance->logger);
	shapito_cache_free(&instance->stream_cache);
	machinarium_free();
}

static inline void
od_usage(od_instance_t *instance, char *path)
{
	od_log(&instance->logger, "init", NULL, NULL,
	       "odissey (git: %s %s)",
	       OD_VERSION_GIT,
	       OD_VERSION_BUILD);
	od_log(&instance->logger, "init", NULL, NULL,
	       "usage: %s <config_file>", path);
}

int od_instance_main(od_instance_t *instance, int argc, char **argv)
{
	/* validate command line options */
	if (argc != 2) {
		od_usage(instance, argv[0]);
		return -1;
	}
	if (strcmp(argv[1], "-h") == 0 ||
	    strcmp(argv[1], "--help") == 0) {
		od_usage(instance, argv[0]);
		return 0;
	}
	instance->config_file = argv[1];

	/* read config file */
	uint64_t scheme_version;
	scheme_version = od_schememgr_version(&instance->scheme_mgr);

	od_error_t error;
	od_error_init(&error);

	int rc;
	rc = od_config_load(&instance->scheme, &error,
	                     instance->config_file,
	                     scheme_version);
	if (rc == -1) {
		od_error(&instance->logger, "config", NULL, NULL, "%s", error.error);
		return -1;
	}

	/* validate configuration scheme */
	rc = od_scheme_validate(&instance->scheme, &instance->logger);
	if (rc == -1)
		return -1;

	/* set log in format */
	od_logger_set_format(&instance->logger, instance->scheme.log_format);

	/* set log debug messages */
	od_logger_set_debug(&instance->logger, instance->scheme.log_debug);

	/* set log to stdout */
	od_logger_set_stdout(&instance->logger, instance->scheme.log_to_stdout);

	/* set cache limits */
	shapito_cache_set_limit(&instance->stream_cache, instance->scheme.cache);
	shapito_cache_set_limit_size(&instance->stream_cache, instance->scheme.cache_chunk);

	/* run as daemon */
	if (instance->scheme.daemonize) {
		rc = od_daemonize();
		if (rc == -1)
			return -1;
		/* update pid */
		od_pid_init(&instance->pid);
	}

	/* init machinarium machinery */
	machinarium_init();

	/* reopen log file after config parsing */
	if (instance->scheme.log_file) {
		rc = od_logger_open(&instance->logger, instance->scheme.log_file);
		if (rc == -1) {
			od_error(&instance->logger, "init", NULL, NULL,
			         "failed to open log file '%s'",
			         instance->scheme.log_file);
			return -1;
		}
	}

	/* syslog */
	if (instance->scheme.log_syslog) {
		od_logger_open_syslog(&instance->logger,
		                      instance->scheme.log_syslog_ident,
		                      instance->scheme.log_syslog_facility);
	}
	od_log(&instance->logger, "init", NULL, NULL, "odissey (git: %s %s)",
	       OD_VERSION_GIT,
	       OD_VERSION_BUILD);
	od_log(&instance->logger, "init", NULL, NULL, "");

	/* print configuration */
	od_log(&instance->logger, "init", NULL, NULL, "using configuration file '%s'",
	       instance->config_file);
	od_log(&instance->logger, "init", NULL, NULL, "");

	if (instance->scheme.log_config)
		od_scheme_print(&instance->scheme, &instance->logger, 0);

	/* create pid file */
	if (instance->scheme.pid_file)
		od_pid_create(&instance->pid, instance->scheme.pid_file);

	/* seed id manager */
	od_idmgr_seed(&instance->id_mgr);

	/* is multi-worker deploy */
	instance->is_shared = instance->scheme.workers > 1;

	/* prepare system services */
	od_pooler_t pooler;
	od_pooler_init(&pooler, instance);

	od_router_t router;
	od_console_t console;
	od_periodic_t periodic;
	od_relaypool_t relay_pool;

	od_system_t *system;
	system = &pooler.system;
	system->instance   = instance;
	system->pooler     = &pooler;
	system->router     = &router;
	system->console    = &console;
	system->periodic   = &periodic;
	system->relay_pool = &relay_pool;

	od_router_init(&router, system);
	od_console_init(&console, system);
	od_periodic_init(&periodic, system);
	od_relaypool_init(&relay_pool);

	/* start pooler machine thread */
	rc = od_pooler_start(&pooler);
	if (rc == -1)
		return -1;
	machine_wait(pooler.machine);
	return 0;
}