console_sink.h

Go to the documentation of this file.

// BSD 3-Clause License
// Copyright (c) 2025, 🍀☀🌕🌥 🌊
// See the LICENSE file in the project root for full license information.
 
#pragma once
 
#include <iostream>
#include <mutex>
#include "../interfaces/output_sink_interface.h"
 
namespace kcenon::logger {
 
class console_sink : public output_sink_interface {
public:
    explicit console_sink(bool use_stderr = false, bool auto_flush = false)
        : use_stderr_(use_stderr)
        , auto_flush_(auto_flush)
    {}
 
    ~console_sink() override {
        flush(); // Ensure all data is written
    }
 
    common::VoidResult write_raw(std::string_view message) override {
        std::lock_guard<std::mutex> lock(mutex_);
 
        if (use_stderr_) {
            std::cerr << message;
            if (auto_flush_) {
                std::cerr.flush();
            }
        } else {
            std::cout << message;
            if (auto_flush_) {
                std::cout.flush();
            }
        }
 
        return common::ok(); // Success
    }
 
    common::VoidResult flush() override {
        std::lock_guard<std::mutex> lock(mutex_);
 
        if (use_stderr_) {
            std::cerr.flush();
        } else {
            std::cout.flush();
        }
 
        return common::ok(); // Success
    }
 
    bool is_healthy() const override {
        return true; // Console is always available
    }
 
    std::string get_name() const override {
        return "console";
    }
 
    std::string get_info() const override {
        std::string info = use_stderr_ ? "stderr" : "stdout";
        if (auto_flush_) {
            info += ", auto-flush";
        }
        return info;
    }
 
    void set_use_stderr(bool use_stderr) {
        std::lock_guard<std::mutex> lock(mutex_);
        use_stderr_ = use_stderr;
    }
 
    void set_auto_flush(bool auto_flush) {
        std::lock_guard<std::mutex> lock(mutex_);
        auto_flush_ = auto_flush;
    }
 
private:
    bool use_stderr_;      
    bool auto_flush_;      
    mutable std::mutex mutex_; 
};
 
} // namespace kcenon::logger