mirror of
https://github.com/laomms/wxhelper.git
synced 2024-12-23 19:49:26 +08:00
104 lines
4.1 KiB
C
104 lines
4.1 KiB
C
|
// Copyright(c) 2015-present, Gabi Melman & spdlog contributors.
|
||
|
// Distributed under the MIT License (http://opensource.org/licenses/MIT)
|
||
|
|
||
|
#pragma once
|
||
|
|
||
|
#include <spdlog/details/null_mutex.h>
|
||
|
#include <spdlog/details/synchronous_factory.h>
|
||
|
#include <spdlog/sinks/base_sink.h>
|
||
|
|
||
|
#include <array>
|
||
|
#include <string>
|
||
|
#include <syslog.h>
|
||
|
|
||
|
namespace spdlog {
|
||
|
namespace sinks {
|
||
|
/**
|
||
|
* Sink that write to syslog using the `syscall()` library call.
|
||
|
*/
|
||
|
template <typename Mutex>
|
||
|
class syslog_sink : public base_sink<Mutex> {
|
||
|
public:
|
||
|
syslog_sink(std::string ident, int syslog_option, int syslog_facility, bool enable_formatting)
|
||
|
: enable_formatting_{enable_formatting},
|
||
|
syslog_levels_{{/* spdlog::level::trace */ LOG_DEBUG,
|
||
|
/* spdlog::level::debug */ LOG_DEBUG,
|
||
|
/* spdlog::level::info */ LOG_INFO,
|
||
|
/* spdlog::level::warn */ LOG_WARNING,
|
||
|
/* spdlog::level::err */ LOG_ERR,
|
||
|
/* spdlog::level::critical */ LOG_CRIT,
|
||
|
/* spdlog::level::off */ LOG_INFO}},
|
||
|
ident_{std::move(ident)} {
|
||
|
// set ident to be program name if empty
|
||
|
::openlog(ident_.empty() ? nullptr : ident_.c_str(), syslog_option, syslog_facility);
|
||
|
}
|
||
|
|
||
|
~syslog_sink() override { ::closelog(); }
|
||
|
|
||
|
syslog_sink(const syslog_sink &) = delete;
|
||
|
syslog_sink &operator=(const syslog_sink &) = delete;
|
||
|
|
||
|
protected:
|
||
|
void sink_it_(const details::log_msg &msg) override {
|
||
|
string_view_t payload;
|
||
|
memory_buf_t formatted;
|
||
|
if (enable_formatting_) {
|
||
|
base_sink<Mutex>::formatter_->format(msg, formatted);
|
||
|
payload = string_view_t(formatted.data(), formatted.size());
|
||
|
} else {
|
||
|
payload = msg.payload;
|
||
|
}
|
||
|
|
||
|
size_t length = payload.size();
|
||
|
// limit to max int
|
||
|
if (length > static_cast<size_t>(std::numeric_limits<int>::max())) {
|
||
|
length = static_cast<size_t>(std::numeric_limits<int>::max());
|
||
|
}
|
||
|
|
||
|
::syslog(syslog_prio_from_level(msg), "%.*s", static_cast<int>(length), payload.data());
|
||
|
}
|
||
|
|
||
|
void flush_() override {}
|
||
|
bool enable_formatting_ = false;
|
||
|
|
||
|
//
|
||
|
// Simply maps spdlog's log level to syslog priority level.
|
||
|
//
|
||
|
int syslog_prio_from_level(const details::log_msg &msg) const {
|
||
|
return syslog_levels_.at(static_cast<levels_array::size_type>(msg.level));
|
||
|
}
|
||
|
|
||
|
private:
|
||
|
using levels_array = std::array<int, 7>;
|
||
|
levels_array syslog_levels_;
|
||
|
// must store the ident because the man says openlog might use the pointer as
|
||
|
// is and not a string copy
|
||
|
const std::string ident_;
|
||
|
};
|
||
|
|
||
|
using syslog_sink_mt = syslog_sink<std::mutex>;
|
||
|
using syslog_sink_st = syslog_sink<details::null_mutex>;
|
||
|
} // namespace sinks
|
||
|
|
||
|
// Create and register a syslog logger
|
||
|
template <typename Factory = spdlog::synchronous_factory>
|
||
|
inline std::shared_ptr<logger> syslog_logger_mt(const std::string &logger_name,
|
||
|
const std::string &syslog_ident = "",
|
||
|
int syslog_option = 0,
|
||
|
int syslog_facility = LOG_USER,
|
||
|
bool enable_formatting = false) {
|
||
|
return Factory::template create<sinks::syslog_sink_mt>(logger_name, syslog_ident, syslog_option,
|
||
|
syslog_facility, enable_formatting);
|
||
|
}
|
||
|
|
||
|
template <typename Factory = spdlog::synchronous_factory>
|
||
|
inline std::shared_ptr<logger> syslog_logger_st(const std::string &logger_name,
|
||
|
const std::string &syslog_ident = "",
|
||
|
int syslog_option = 0,
|
||
|
int syslog_facility = LOG_USER,
|
||
|
bool enable_formatting = false) {
|
||
|
return Factory::template create<sinks::syslog_sink_st>(logger_name, syslog_ident, syslog_option,
|
||
|
syslog_facility, enable_formatting);
|
||
|
}
|
||
|
} // namespace spdlog
|