12345678910111213141516171819202122232425262728293031323334353637383940414243 |
- // 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/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 {
- memory_buf_t formatted;
- base_sink<Mutex>::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
|