lifecycle_manager.hpp

/* -*- mode: c++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4 -*-
 *
 * Copyright (C) 2025 Marco Craveiro <marco.craveiro@gmail.com>
 *
 * This program is free software; you can redistribute it and/or modify it under
 * the terms of the GNU General Public License as published by the Free Software
 * Foundation; either version 3 of the License, or (at your option) any later
 * version.
 *
 * This program is distributed in the hope that it will be useful, but WITHOUT
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
 * FOR A PARTICULAR PURPOSE. See the GNU General Public License for more
 * details.
 *
 * You should have received a copy of the GNU General Public License along with
 * this program; if not, write to the Free Software Foundation, Inc., 51
 * Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
 *
 */
#ifndef ORES_TELEMETRY_LOG_LIFECYCLE_MANAGER_HPP
#define ORES_TELEMETRY_LOG_LIFECYCLE_MANAGER_HPP
 
#include <memory>
#include <optional>
#include <boost/shared_ptr.hpp>
#include <boost/log/sinks.hpp>
#include "ores.logging/lifecycle_manager.hpp"
#include "ores.logging/logging_options.hpp"
#include "ores.telemetry/log/telemetry_sink_backend.hpp"
#include "ores.telemetry/log/database_sink_backend.hpp"
#include "ores.telemetry/domain/resource.hpp"
 
namespace ores::telemetry::log {
 
using ores::logging::logging_options;
 
class lifecycle_manager final : public ores::logging::lifecycle_manager {
private:
    using telemetry_sink_type = boost::log::sinks::asynchronous_sink<
        telemetry_sink_backend>;
 
public:
    lifecycle_manager() = delete;
    lifecycle_manager(lifecycle_manager&&) = delete;
    lifecycle_manager& operator=(const lifecycle_manager&) = delete;
 
public:
    explicit lifecycle_manager(std::optional<logging_options> ocfg);
 
    ~lifecycle_manager() override;
 
    void add_telemetry_sink(
        std::shared_ptr<domain::resource> resource,
        telemetry_sink_backend::log_record_handler handler);
 
    void add_database_sink(
        std::shared_ptr<domain::resource> resource,
        database_log_handler handler,
        const std::string& source_type = "test",
        const std::string& source_name = "unit-test");
 
private:
    boost::shared_ptr<telemetry_sink_type> telemetry_sink_;
    using database_sink_type = boost::log::sinks::asynchronous_sink<
        database_sink_backend>;
    boost::shared_ptr<database_sink_type> database_sink_; // Added for database sink
};
 
}
 
#endif