ostream_sink.h

Go to the documentation of this file.

//
// Copyright(c) 2015 Gabi Melman.
// Distributed under the MIT License (http://opensource.org/licenses/MIT)
//
 
#pragma once
 
#ifndef SPDLOG_H
#include "spdlog/spdlog.h"
#endif
 
#include "spdlog/details/null_mutex.h"
#include "spdlog/sinks/base_sink.h"
 
#include <mutex>
#include <ostream>
 
namespace spdlog {
namespace sinks {
template<typename Mutex>
class ostream_sink final : public base_sink<Mutex>
{
public:
    explicit ostream_sink(std::ostream &os, bool force_flush = false)
        : ostream_(os)
        , force_flush_(force_flush)
    {
    }
    ostream_sink(const ostream_sink &) = delete;
    ostream_sink &operator=(const ostream_sink &) = delete;
 
protected:
    void sink_it_(const details::log_msg &msg) override
    {
        fmt::memory_buffer formatted;
        sink::formatter_->format(msg, formatted);
        ostream_.write(formatted.data(), static_cast<std::streamsize>(formatted.size()));
        if (force_flush_)
        {
            ostream_.flush();
        }
    }
 
    void flush_() override
    {
        ostream_.flush();
    }
 
    std::ostream &ostream_;
    bool force_flush_;
};
 
using ostream_sink_mt = ostream_sink<std::mutex>;
using ostream_sink_st = ostream_sink<details::null_mutex>;
 
} // namespace sinks
} // namespace spdlog