ebusd/utils/logger.cpp

/*
 * Copyright (C) Roland Jax 2012-2014 <roland.jax@liwest.at>
 *
 * This file is part of ebusd.
 *
 * ebusd is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * ebusd is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with ebusd. If not, see http://www.gnu.org/licenses/.
 */

#include "logger.h"
#include <iostream>
#include <sstream>
#include <fstream>
#include <cstdio>
#include <cstring>
#include <cmath>
#include <ctime>
#include <sys/time.h>
#include <unistd.h>

static const char* AreaNames[Size_of_Area] = { "bas", "net", "bus", "cyc" };
static const char* LevelNames[Size_of_Level] = { "error", "event", "trace", "debug" };

int calcArea(const std::string area)
{
	int m_area = 0;

	// prepare data
	std::string token;
	std::istringstream stream(area);
	std::vector<std::string> cmd;

	while (std::getline(stream, token, ',') != 0)
		cmd.push_back(token);

	for (std::vector<std::string>::iterator it = cmd.begin() ; it != cmd.end(); ++it)
		for (int i = 0; i < Size_of_Area; i++) {
			if (strcasecmp("all", it->c_str()) == 0)
				return (pow(2, (int)Size_of_Area) - 1);

			if (strcasecmp(AreaNames[i], it->c_str()) == 0)
				m_area += pow(2, i);
		}

	return m_area;
}

int calcLevel(const std::string level)
{
	int m_level = event;
	for (int i = 0; i < Size_of_Level; i++)
		if (strcasecmp(LevelNames[i], level.c_str()) == 0)
			return i;

	return m_level;
}


LogMessage::LogMessage(const int area, const int level, const std::string text, const Status status)
	: m_area(area), m_level(level), m_text(text), m_status(status)
{
	char time[24];
	struct timeval tv;
	struct timezone tz;
	struct tm* tm;

	gettimeofday(&tv, &tz);
	tm = localtime(&tv.tv_sec);

	sprintf(&time[0], "%04d-%02d-%02d %02d:%02d:%02d.%03ld",
		tm->tm_year+1900, tm->tm_mon+1, tm->tm_mday,
		tm->tm_hour, tm->tm_min, tm->tm_sec, tv.tv_usec/1000);

	m_time = std::string(time);
}



void LogSink::addMessage(const LogMessage& message)
{
	LogMessage* tmp = new LogMessage(LogMessage(message));
	m_queue.add((tmp));
}

void* LogSink::run()
{
	while (1) {
		LogMessage* message = m_queue.remove();
			if (message->getStatus() == LogMessage::End) {
				delete message;
				while (m_queue.size() == true) {
					LogMessage* message = m_queue.remove();
					write(*message);
					delete message;
				}
				return NULL;
			}

		write(*message);
		delete message;
	}
	return NULL;
}



int LogConsole::m_numInstance = 0;

void LogConsole::write(const LogMessage& message) const
{
	std::cout << message.getTime() << " ["
		  << AreaNames[(int)log2(message.getArea())] << " "
		  << LevelNames[message.getLevel()] << "] "
		  << message.getText() << std::endl;
}



int LogFile::m_numInstance = 0;

void LogFile::write(const LogMessage& message) const
{
	std::fstream file(m_filename.c_str(), std::ios::out | std::ios::app);

	if (file.is_open() == true) {
		file << message.getTime() << " ["
		     << AreaNames[(int)log2(message.getArea())] << " "
		     << LevelNames[message.getLevel()] << "] "
		     << message.getText() << std::endl;
		file.close();
	}
}



LogInstance& LogInstance::Instance()
{
	static LogInstance instance;
	return (instance);
}

LogInstance::~LogInstance()
{
	while (m_sinks.empty() == false)
		*this -= *(m_sinks.begin());
}

LogInstance& LogInstance::operator+= (LogSink* sink)
{
	sinkCI_t itEnd = m_sinks.end();
	sinkCI_t it = std::find(m_sinks.begin(), itEnd, sink);

	if (it == itEnd)
		m_sinks.push_back(sink);

	return (*this);
}

LogInstance& LogInstance::operator-= (const LogSink* sink)
{
	sinkCI_t itEnd = m_sinks.end();
	sinkCI_t it = std::find(m_sinks.begin(), itEnd, sink);

	if (it == itEnd)
		return (*this);

	m_sinks.erase(it);

	delete (sink);

	return (*this);
}

void LogInstance::log(const int area, const int level, const std::string& data, ...)
{
	if (m_running == true) {
		char* tmp;
		va_list ap;
		va_start(ap, data);

		if (vasprintf(&tmp, data.c_str(), ap) != -1) {
			std::string buffer(tmp);
			m_messages.add(new LogMessage(LogMessage(area, level, buffer, LogMessage::Run)));
		}

		va_end(ap);
		free(tmp);
	}

}

void* LogInstance::run()
{
	m_running = true;

	while (m_running == true) {
		LogMessage* message = m_messages.remove();

		sinkCI_t iter = m_sinks.begin();

		for (; iter != m_sinks.end(); ++iter) {
			if (*iter != 0) {

				if (((*iter)->getAreas() & message->getArea()
				&& (*iter)->getLevel() >= message->getLevel())
				&& message->getStatus() == LogMessage::Run) {
					(*iter)->addMessage(*message);
				} else if (message->getStatus() == LogMessage::End) {
					(*iter)->addMessage(*message);
					m_running = false;
				}

			}
		}

		delete message;

	}
	return NULL;
}

void LogInstance::stop()
{
	m_messages.add(new LogMessage(LogMessage(bas, error, "", LogMessage::End)));
	// TODO: Improve this method
	usleep(100000);
}