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
 |