2019-05-11 13:06:17 -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-04-28 18:31:09 -04:00
|
|
|
#include "spdlog/details/null_mutex.h"
|
|
|
|
#include "spdlog/sinks/base_sink.h"
|
2016-04-20 04:57:49 -04:00
|
|
|
|
|
|
|
#include <mutex>
|
2018-03-09 08:26:33 -05:00
|
|
|
#include <ostream>
|
2016-04-20 04:57:49 -04:00
|
|
|
|
2018-03-17 06:47:46 -04:00
|
|
|
namespace spdlog {
|
|
|
|
namespace sinks {
|
2018-07-14 09:21:53 -04:00
|
|
|
template<typename Mutex>
|
2018-09-26 07:45:38 -04:00
|
|
|
class ostream_sink final : public base_sink<Mutex>
|
2016-04-20 04:57:49 -04:00
|
|
|
{
|
|
|
|
public:
|
2018-03-09 08:26:33 -05:00
|
|
|
explicit ostream_sink(std::ostream &os, bool force_flush = false)
|
2018-06-10 15:59:17 -04:00
|
|
|
: ostream_(os)
|
|
|
|
, force_flush_(force_flush)
|
2019-05-11 17:22:39 -04:00
|
|
|
{}
|
2018-03-09 08:26:33 -05:00
|
|
|
ostream_sink(const ostream_sink &) = delete;
|
|
|
|
ostream_sink &operator=(const ostream_sink &) = delete;
|
2016-04-20 04:57:49 -04:00
|
|
|
|
|
|
|
protected:
|
2018-07-14 09:21:53 -04:00
|
|
|
void sink_it_(const details::log_msg &msg) override
|
2016-04-20 04:57:49 -04:00
|
|
|
{
|
2018-07-14 09:21:53 -04:00
|
|
|
fmt::memory_buffer formatted;
|
|
|
|
sink::formatter_->format(msg, formatted);
|
2018-08-16 17:32:13 -04:00
|
|
|
ostream_.write(formatted.data(), static_cast<std::streamsize>(formatted.size()));
|
2018-06-10 15:59:17 -04:00
|
|
|
if (force_flush_)
|
2018-10-05 08:19:03 -04:00
|
|
|
{
|
2018-06-10 15:59:17 -04:00
|
|
|
ostream_.flush();
|
2018-10-05 08:19:03 -04:00
|
|
|
}
|
2016-04-20 04:57:49 -04:00
|
|
|
}
|
|
|
|
|
2018-06-10 15:59:17 -04:00
|
|
|
void flush_() override
|
2016-04-20 04:57:49 -04:00
|
|
|
{
|
2018-06-10 15:59:17 -04:00
|
|
|
ostream_.flush();
|
2016-04-20 04:57:49 -04:00
|
|
|
}
|
|
|
|
|
2018-06-10 15:59:17 -04:00
|
|
|
std::ostream &ostream_;
|
|
|
|
bool force_flush_;
|
2016-04-20 04:57:49 -04:00
|
|
|
};
|
|
|
|
|
2018-02-24 16:35:09 -05:00
|
|
|
using ostream_sink_mt = ostream_sink<std::mutex>;
|
|
|
|
using ostream_sink_st = ostream_sink<details::null_mutex>;
|
|
|
|
|
2018-03-17 06:47:46 -04:00
|
|
|
} // namespace sinks
|
|
|
|
} // namespace spdlog
|