odyssey/sources/logger.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 <time.h>
#include <unistd.h>
#include <sys/fcntl.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <sys/time.h>

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

#include "sources/macro.h"
#include "sources/pid.h"
#include "sources/id.h"
#include "sources/log_file.h"
#include "sources/log_system.h"
#include "sources/logger.h"

typedef struct
{
	char *pid;
	int   pid_len;
	char *timestamp;
	int   timestamp_len;
	char *event;
	int   event_len;
	char *id_prefix;
	char *id;
	int   id_len;
	char *context;
	int   context_len;
	char *msg;
	int   msg_len;
} od_logger_msg_t;

typedef struct {
	od_logsystem_prio_t syslog_prio;
	char *ident;
	char *ident_short;
} od_logger_ident_t;

static od_logger_ident_t od_logger_ident_tab[] =
{
	[OD_LOG]              = { OD_LOGSYSTEM_INFO,  "info",         ""      },
	[OD_LOG_ERROR]        = { OD_LOGSYSTEM_ERROR, "error",        "error" },
	[OD_LOG_CLIENT]       = { OD_LOGSYSTEM_INFO,  "client_info",  ""      },
	[OD_LOG_CLIENT_ERROR] = { OD_LOGSYSTEM_ERROR, "client_error", "error" },
	[OD_LOG_CLIENT_DEBUG] = { OD_LOGSYSTEM_DEBUG, "client_debug", "debug" },
	[OD_LOG_SERVER]       = { OD_LOGSYSTEM_INFO,  "server_info",  ""      },
	[OD_LOG_SERVER_ERROR] = { OD_LOGSYSTEM_ERROR, "server_error", "error" },
	[OD_LOG_SERVER_DEBUG] = { OD_LOGSYSTEM_DEBUG, "server_debug", "debug" }
};

void od_logger_init(od_logger_t *logger, od_pid_t *pid)
{
	logger->pid = pid;
	logger->debug = 0;
	od_logfile_init(&logger->log);
	od_logsystem_init(&logger->log_system);
}

void od_logger_set_debug(od_logger_t *logger, int enable)
{
	logger->debug = enable;
}

int od_logger_open(od_logger_t *logger, char *path)
{
	return od_logfile_open(&logger->log, path);
}

int od_logger_open_syslog(od_logger_t *logger, char *ident, char *facility)
{
	return od_logsystem_open(&logger->log_system, ident, facility);
}

void od_logger_close(od_logger_t *logger)
{
	od_logfile_close(&logger->log);
	od_logsystem_close(&logger->log_system);
}

void od_loggerv(od_logger_t *logger,
                od_logger_event_t event,
                od_id_t *id,
                char *context,
                char *fmt, va_list args)
{
	od_logger_msg_t msg;
	char buf[512];
	int  buf_len;

	/* pid */
	msg.pid = logger->pid->pid_sz;
	msg.pid_len = logger->pid->pid_len;
	buf_len = snprintf(buf, sizeof(buf), "%s ", logger->pid->pid_sz);

	/* timestamp */
	msg.timestamp = buf + buf_len;
	msg.timestamp_len = buf_len;
	struct timeval tv;
	gettimeofday(&tv, NULL);
	buf_len += strftime(buf + buf_len, sizeof(buf) - buf_len, "%d %b %H:%M:%S.",
	                    localtime(&tv.tv_sec));
	buf_len += snprintf(buf + buf_len, sizeof(buf) - buf_len, "%03d  ",
	                    (signed)tv.tv_usec / 1000);
	msg.timestamp_len = buf_len - msg.timestamp_len;

	/* ident */
	od_logger_ident_t *ident;
	ident = &od_logger_ident_tab[event];

	msg.event = ident->ident;
	msg.event_len = strlen(msg.event);
	if (*ident->ident_short) {
		buf_len += snprintf(buf + buf_len, sizeof(buf) - buf_len,
		                    "%s: ", ident->ident_short);
	}

	/* id */
	if (id) {
		msg.id_prefix = id->id_prefix;
		msg.id = id->id;
		msg.id_len = sizeof(id->id);

		buf_len += snprintf(buf + buf_len, sizeof(buf) - buf_len, "%s%.*s: ",
		                    msg.id_prefix, msg.id_len, msg.id);
	} else {
		msg.id_prefix = "";
		msg.id = "";
		msg.id_len = 0;
	}

	/* context */
	if (context) {
		msg.context = context;
		msg.context_len = strlen(context);
		buf_len += snprintf(buf + buf_len, sizeof(buf) - buf_len, "(%.*s) ",
		                    msg.context_len, msg.context);
	} else {
		msg.context = "";
		msg.context_len = 0;
	}

	/* message */
	msg.msg_len = buf_len;
	msg.msg = buf;
	buf_len += vsnprintf(buf + buf_len, sizeof(buf) - buf_len, fmt, args);
	buf_len += snprintf(buf + buf_len, sizeof(buf) - buf_len, "\n");
	msg.msg_len = buf_len - msg.msg_len;

	/* write log message */
	od_logfile_write(&logger->log, buf, buf_len);
	od_logsystem(&logger->log_system, ident->syslog_prio, buf, buf_len);

	write(0, buf, buf_len);
}