2019-06-03 17:09:16 -04:00
|
|
|
// Copyright(c) 2015-present, Gabi Melman & spdlog contributors.
|
2016-04-20 04:57:49 -04:00
|
|
|
// Distributed under the MIT License (http://opensource.org/licenses/MIT)
|
|
|
|
|
|
|
|
#pragma once
|
|
|
|
|
2018-07-07 07:04:31 -04:00
|
|
|
#include "spdlog/details/console_globals.h"
|
2018-04-28 18:31:09 -04:00
|
|
|
#include "spdlog/details/null_mutex.h"
|
2019-05-12 16:08:14 -04:00
|
|
|
#include "spdlog/details/pattern_formatter.h"
|
2019-06-03 16:56:18 -04:00
|
|
|
#include "spdlog/details/synchronous_factory.h"
|
2016-04-20 04:57:49 -04:00
|
|
|
|
|
|
|
#include <cstdio>
|
|
|
|
#include <memory>
|
|
|
|
#include <mutex>
|
|
|
|
|
2018-03-17 06:47:46 -04:00
|
|
|
namespace spdlog {
|
2018-04-20 06:20:19 -04:00
|
|
|
|
|
|
|
namespace sinks {
|
|
|
|
|
2018-07-14 09:21:53 -04:00
|
|
|
template<typename TargetStream, typename ConsoleMutex>
|
2018-09-26 08:24:22 -04:00
|
|
|
class stdout_sink final : public sink
|
2018-04-20 06:20:19 -04:00
|
|
|
{
|
|
|
|
public:
|
2018-07-07 07:04:31 -04:00
|
|
|
using mutex_t = typename ConsoleMutex::mutex_t;
|
2018-04-20 06:20:19 -04:00
|
|
|
stdout_sink()
|
2018-07-14 09:21:53 -04:00
|
|
|
: mutex_(ConsoleMutex::mutex())
|
2018-07-07 07:04:31 -04:00
|
|
|
, file_(TargetStream::stream())
|
2019-05-11 17:22:39 -04:00
|
|
|
{}
|
2018-08-13 03:30:02 -04:00
|
|
|
~stdout_sink() override = default;
|
2018-04-20 06:20:19 -04:00
|
|
|
|
|
|
|
stdout_sink(const stdout_sink &other) = delete;
|
|
|
|
stdout_sink &operator=(const stdout_sink &other) = delete;
|
|
|
|
|
|
|
|
void log(const details::log_msg &msg) override
|
|
|
|
{
|
2018-06-10 15:59:17 -04:00
|
|
|
std::lock_guard<mutex_t> lock(mutex_);
|
2018-07-07 07:04:31 -04:00
|
|
|
fmt::memory_buffer formatted;
|
|
|
|
formatter_->format(msg, formatted);
|
|
|
|
fwrite(formatted.data(), sizeof(char), formatted.size(), file_);
|
|
|
|
fflush(TargetStream::stream());
|
2018-04-20 06:20:19 -04:00
|
|
|
}
|
|
|
|
|
|
|
|
void flush() override
|
|
|
|
{
|
2018-06-10 15:59:17 -04:00
|
|
|
std::lock_guard<mutex_t> lock(mutex_);
|
2018-07-14 09:21:53 -04:00
|
|
|
fflush(file_);
|
|
|
|
}
|
|
|
|
|
2018-08-13 03:30:02 -04:00
|
|
|
void set_pattern(const std::string &pattern) override
|
2018-07-14 09:21:53 -04:00
|
|
|
{
|
|
|
|
std::lock_guard<mutex_t> lock(mutex_);
|
|
|
|
formatter_ = std::unique_ptr<spdlog::formatter>(new pattern_formatter(pattern));
|
|
|
|
}
|
|
|
|
|
2018-08-13 03:30:02 -04:00
|
|
|
void set_formatter(std::unique_ptr<spdlog::formatter> sink_formatter) override
|
2018-07-14 09:21:53 -04:00
|
|
|
{
|
|
|
|
std::lock_guard<mutex_t> lock(mutex_);
|
|
|
|
formatter_ = std::move(sink_formatter);
|
2018-04-20 06:20:19 -04:00
|
|
|
}
|
|
|
|
|
|
|
|
private:
|
2018-06-10 15:59:17 -04:00
|
|
|
mutex_t &mutex_;
|
|
|
|
FILE *file_;
|
2018-04-20 06:20:19 -04:00
|
|
|
};
|
|
|
|
|
2018-07-14 09:21:53 -04:00
|
|
|
using stdout_sink_mt = stdout_sink<details::console_stdout, details::console_mutex>;
|
|
|
|
using stdout_sink_st = stdout_sink<details::console_stdout, details::console_nullmutex>;
|
2018-07-07 07:04:31 -04:00
|
|
|
|
2018-07-14 09:21:53 -04:00
|
|
|
using stderr_sink_mt = stdout_sink<details::console_stderr, details::console_mutex>;
|
|
|
|
using stderr_sink_st = stdout_sink<details::console_stderr, details::console_nullmutex>;
|
2018-04-20 06:20:19 -04:00
|
|
|
|
|
|
|
} // namespace sinks
|
|
|
|
|
|
|
|
// factory methods
|
2019-06-03 16:56:18 -04:00
|
|
|
template<typename Factory = spdlog::synchronous_factory>
|
2018-04-20 06:20:19 -04:00
|
|
|
inline std::shared_ptr<logger> stdout_logger_mt(const std::string &logger_name)
|
|
|
|
{
|
2018-04-20 07:03:15 -04:00
|
|
|
return Factory::template create<sinks::stdout_sink_mt>(logger_name);
|
2018-04-20 06:20:19 -04:00
|
|
|
}
|
|
|
|
|
2019-06-03 16:56:18 -04:00
|
|
|
template<typename Factory = spdlog::synchronous_factory>
|
2018-04-20 06:20:19 -04:00
|
|
|
inline std::shared_ptr<logger> stdout_logger_st(const std::string &logger_name)
|
|
|
|
{
|
2018-04-20 07:03:15 -04:00
|
|
|
return Factory::template create<sinks::stdout_sink_st>(logger_name);
|
2018-04-20 06:20:19 -04:00
|
|
|
}
|
|
|
|
|
2019-06-03 16:56:18 -04:00
|
|
|
template<typename Factory = spdlog::synchronous_factory>
|
2018-04-20 06:20:19 -04:00
|
|
|
inline std::shared_ptr<logger> stderr_logger_mt(const std::string &logger_name)
|
|
|
|
{
|
2018-04-20 07:03:15 -04:00
|
|
|
return Factory::template create<sinks::stderr_sink_mt>(logger_name);
|
2018-04-20 06:20:19 -04:00
|
|
|
}
|
|
|
|
|
2019-06-03 16:56:18 -04:00
|
|
|
template<typename Factory = spdlog::synchronous_factory>
|
2018-04-20 06:20:19 -04:00
|
|
|
inline std::shared_ptr<logger> stderr_logger_st(const std::string &logger_name)
|
|
|
|
{
|
2018-04-20 07:03:15 -04:00
|
|
|
return Factory::template create<sinks::stderr_sink_st>(logger_name);
|
2018-04-20 06:20:19 -04:00
|
|
|
}
|
2018-03-17 06:47:46 -04:00
|
|
|
} // namespace spdlog
|