mirror of
https://github.com/gabime/spdlog.git
synced 2024-12-25 10:01:33 +08:00
commit
9c12a44d6e
91
include/spdlog/sinks/systemd_sink.h
Normal file
91
include/spdlog/sinks/systemd_sink.h
Normal file
@ -0,0 +1,91 @@
|
||||
//
|
||||
// Copyright(c) 2019 ZVYAGIN.Alexander@gmail.com
|
||||
// Distributed under the MIT License (http://opensource.org/licenses/MIT)
|
||||
//
|
||||
|
||||
#pragma once
|
||||
|
||||
#ifndef SPDLOG_H
|
||||
#include "spdlog/spdlog.h"
|
||||
#endif
|
||||
|
||||
#include "spdlog/sinks/base_sink.h"
|
||||
|
||||
#include <array>
|
||||
#include <string>
|
||||
#include <systemd/sd-journal.h>
|
||||
|
||||
namespace spdlog {
|
||||
namespace sinks {
|
||||
|
||||
inline int syslog_level(level::level_enum l) {
|
||||
switch(l) {
|
||||
case level::off:
|
||||
case level::trace:
|
||||
case level::debug:
|
||||
return LOG_DEBUG;
|
||||
case level::info:
|
||||
return LOG_INFO;
|
||||
case level::warn:
|
||||
return LOG_WARNING;
|
||||
case level::err:
|
||||
return LOG_ERR;
|
||||
case level::critical:
|
||||
return LOG_CRIT;
|
||||
default:
|
||||
throw std::invalid_argument("systemd_sink.h syslog_level()");
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Sink that write to systemd using the `sd_journal_print()` library call.
|
||||
*
|
||||
* Locking is not needed, as `sd_journal_print()` itself is thread-safe.
|
||||
*/
|
||||
template<typename Mutex>
|
||||
class systemd_sink : public base_sink<Mutex>
|
||||
{
|
||||
public:
|
||||
//
|
||||
explicit systemd_sink(void) {}
|
||||
|
||||
~systemd_sink() override {}
|
||||
|
||||
systemd_sink(const systemd_sink &) = delete;
|
||||
systemd_sink &operator=(const systemd_sink &) = delete;
|
||||
|
||||
protected:
|
||||
void sink_it_(const details::log_msg &msg) override
|
||||
{
|
||||
if( sd_journal_print(
|
||||
syslog_level(msg.level),
|
||||
"%.*s",
|
||||
static_cast<int>(msg.payload.size()),
|
||||
msg.payload.data()
|
||||
)
|
||||
)
|
||||
throw spdlog_ex("Failed writing to systemd");
|
||||
}
|
||||
|
||||
void flush_() override {}
|
||||
};
|
||||
|
||||
using systemd_sink_mt = systemd_sink<std::mutex>;
|
||||
using systemd_sink_st = systemd_sink<details::null_mutex>;
|
||||
} // namespace sinks
|
||||
|
||||
// Create and register a syslog logger
|
||||
template<typename Factory = default_factory>
|
||||
inline std::shared_ptr<logger> systemd_logger_mt(
|
||||
const std::string &logger_name)
|
||||
{
|
||||
return Factory::template create<sinks::systemd_sink_mt>(logger_name);
|
||||
}
|
||||
|
||||
template<typename Factory = default_factory>
|
||||
inline std::shared_ptr<logger> systemd_logger_st(
|
||||
const std::string &logger_name)
|
||||
{
|
||||
return Factory::template create<sinks::systemd_sink_st>(logger_name);
|
||||
}
|
||||
} // namespace spdlog
|
35
tests/test_systemd.cpp
Normal file
35
tests/test_systemd.cpp
Normal file
@ -0,0 +1,35 @@
|
||||
#include "includes.h"
|
||||
#include <spdlog/sinks/systemd_sink.h>
|
||||
|
||||
namespace {
|
||||
const char *tested_logger_name = "main";
|
||||
}
|
||||
|
||||
void run(spdlog::logger &logger) {
|
||||
logger.debug("test debug");
|
||||
logger.error("test error");
|
||||
logger.info("test info");
|
||||
}
|
||||
|
||||
// std::shared_ptr<spdlog::logger> create_stdout_and_systemd_logger(
|
||||
// std::string name="",
|
||||
// spdlog::level::level_enum level_stdout=spdlog::level::level_enum::debug,
|
||||
// spdlog::level::level_enum level_systemd=spdlog::level::level_enum::err
|
||||
// ) {
|
||||
// auto console_sink = std::make_shared<spdlog::sinks::stdout_color_sink_mt>();
|
||||
// console_sink->set_level(level_stdout);
|
||||
// auto systemd_sink = std::make_shared<spdlog::sinks::systemd_sink_st>();
|
||||
// systemd_sink->set_level(level_systemd);
|
||||
// return std::make_shared<spdlog::logger>(name, {console_sink, systemd_sink});
|
||||
|
||||
// }
|
||||
|
||||
TEST_CASE("systemd", "[all]")
|
||||
{
|
||||
auto console_sink = std::make_shared<spdlog::sinks::stdout_color_sink_mt>();
|
||||
console_sink->set_level(spdlog::level::level_enum::debug);
|
||||
auto systemd_sink = std::make_shared<spdlog::sinks::systemd_sink_st>();
|
||||
systemd_sink->set_level(spdlog::level::level_enum::err);
|
||||
spdlog::logger logger("spdlog_systemd_test", {console_sink, systemd_sink});
|
||||
run(logger);
|
||||
}
|
Loading…
Reference in New Issue
Block a user