mirror of
https://github.com/gabime/spdlog.git
synced 2024-11-15 16:35:45 +08:00
Added SQLite3 sink to the solution
This commit is contained in:
parent
08bce1bb4f
commit
a7011baa53
156
include/spdlog/sinks/database_logger_sink.h
Normal file
156
include/spdlog/sinks/database_logger_sink.h
Normal file
@ -0,0 +1,156 @@
|
||||
#pragma once
|
||||
#include <spdlog\sinks\sink.h>
|
||||
#include <spdlog/details/log_msg.h>
|
||||
#include <spdlog/common.h>
|
||||
#include <vector>
|
||||
#include <sqlite3.h>
|
||||
|
||||
struct Column
|
||||
{
|
||||
enum TableColumn
|
||||
{
|
||||
TimeStamp,
|
||||
Level,
|
||||
Message,
|
||||
LoggerName,
|
||||
ThreadId
|
||||
};
|
||||
|
||||
Column(std::string columnName, TableColumn columnMap)
|
||||
{
|
||||
ColumnName = columnName;
|
||||
ColumnMap = columnMap;
|
||||
}
|
||||
|
||||
std::string ColumnName;
|
||||
TableColumn ColumnMap;
|
||||
std::string Value;
|
||||
};
|
||||
|
||||
struct database_schema
|
||||
{
|
||||
std::string TableName;
|
||||
|
||||
std::vector<Column> Columns;
|
||||
|
||||
database_schema(const std::string& tableName, const std::vector<Column>& columns)
|
||||
{
|
||||
TableName = tableName;
|
||||
Columns = columns;
|
||||
}
|
||||
|
||||
database_schema()
|
||||
{
|
||||
TableName = "Logs";
|
||||
|
||||
Columns =
|
||||
{
|
||||
Column("TimeStamp", Column::TimeStamp),
|
||||
Column("Level", Column::Level),
|
||||
Column("Message", Column::Message),
|
||||
Column("LoggerName", Column::LoggerName),
|
||||
Column("ThreadId", Column::ThreadId)
|
||||
};
|
||||
}
|
||||
};
|
||||
|
||||
namespace spdlog
|
||||
{
|
||||
namespace sinks
|
||||
{
|
||||
class database_logger_sink :
|
||||
public sink
|
||||
{
|
||||
public:
|
||||
void flush() override
|
||||
{
|
||||
sqlite3_close(_database);
|
||||
}
|
||||
|
||||
void log(const details::log_msg& msg) override
|
||||
{
|
||||
|
||||
for (auto& column : _schema.Columns)
|
||||
{
|
||||
switch (column.ColumnMap)
|
||||
{
|
||||
case Column::TimeStamp:
|
||||
{
|
||||
auto time = std::chrono::system_clock::to_time_t(msg.time);
|
||||
char str[26];
|
||||
ctime_s(str, sizeof(str), &time);
|
||||
column.Value = str;
|
||||
break;
|
||||
}
|
||||
|
||||
case Column::Level:
|
||||
{
|
||||
column.Value = level::to_str(msg.level);
|
||||
break;
|
||||
}
|
||||
case Column::Message:
|
||||
{
|
||||
column.Value = msg.raw.str();
|
||||
break;
|
||||
}
|
||||
case Column::LoggerName:
|
||||
{
|
||||
column.Value = msg.logger_name;
|
||||
break;
|
||||
}
|
||||
case Column::ThreadId:
|
||||
{
|
||||
column.Value = std::to_string(msg.thread_id);
|
||||
break;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
auto query = fmt::format("INSERT INTO {0} ({1},{3},{5},{7},{9}) VALUES ('{2}','{4}','{6}','{8}',{10})",
|
||||
_schema.TableName,
|
||||
_schema.Columns[0].ColumnName,
|
||||
_schema.Columns[0].Value,
|
||||
_schema.Columns[1].ColumnName,
|
||||
_schema.Columns[1].Value,
|
||||
_schema.Columns[2].ColumnName,
|
||||
_schema.Columns[2].Value,
|
||||
_schema.Columns[3].ColumnName,
|
||||
_schema.Columns[3].Value,
|
||||
_schema.Columns[4].ColumnName,
|
||||
_schema.Columns[4].Value);
|
||||
|
||||
char *errorMessage = nullptr;
|
||||
|
||||
if (sqlite3_exec(_database, query.c_str(), nullptr, nullptr, &errorMessage) != SQLITE_OK)
|
||||
{
|
||||
throw spdlog_ex(errorMessage);
|
||||
}
|
||||
|
||||
|
||||
}
|
||||
|
||||
explicit database_logger_sink(const std::string& databaseName)
|
||||
{
|
||||
if (sqlite3_open(databaseName.c_str(), &_database))
|
||||
throw spdlog_ex("Error opening database");
|
||||
}
|
||||
|
||||
explicit database_logger_sink(const std::string& databaseName, const database_schema& databaseSchema)
|
||||
{
|
||||
_schema = databaseSchema;
|
||||
|
||||
if (sqlite3_open(databaseName.c_str(), &_database))
|
||||
throw spdlog_ex("Error opening database");
|
||||
}
|
||||
|
||||
~database_logger_sink()
|
||||
{
|
||||
sqlite3_close(_database);
|
||||
}
|
||||
|
||||
private:
|
||||
database_schema _schema;
|
||||
sqlite3 *_database;
|
||||
};
|
||||
}
|
||||
}
|
Loading…
Reference in New Issue
Block a user