quic_server_adapter.h

Go to the documentation of this file.

// BSD 3-Clause License
// Copyright (c) 2024, 🍀☀🌕🌥 🌊
// See the LICENSE file in the project root for full license information.
 
#pragma once
 
#include "kcenon/network/interfaces/i_protocol_server.h"
 
#include <atomic>
#include <memory>
#include <mutex>
#include <optional>
#include <string>
#include <unordered_map>
#include <vector>
 
namespace kcenon::network::core {
    class messaging_quic_server;
}
 
namespace kcenon::network::session {
    class quic_session;
}
 
namespace kcenon::network::internal::adapters {
 
class quic_session_wrapper : public interfaces::i_session {
public:
    quic_session_wrapper(
        std::string_view session_id,
        std::shared_ptr<session::quic_session> session);
 
    // i_session interface
    [[nodiscard]] auto id() const -> std::string_view override;
    [[nodiscard]] auto is_connected() const -> bool override;
    [[nodiscard]] auto send(std::vector<uint8_t>&& data) -> VoidResult override;
    auto close() -> void override;
 
private:
    std::string session_id_;
    std::shared_ptr<session::quic_session> session_;
    std::atomic<bool> is_connected_{true};
};
 
class quic_server_adapter : public interfaces::i_protocol_server {
public:
    explicit quic_server_adapter(std::string_view server_id);
 
    ~quic_server_adapter() override;
 
    // Non-copyable, non-movable
    quic_server_adapter(const quic_server_adapter&) = delete;
    quic_server_adapter& operator=(const quic_server_adapter&) = delete;
    quic_server_adapter(quic_server_adapter&&) = delete;
    quic_server_adapter& operator=(quic_server_adapter&&) = delete;
 
    // =========================================================================
    // QUIC-specific Configuration (REQUIRED before start)
    // =========================================================================
 
    auto set_cert_path(std::string_view path) -> void;
 
    auto set_key_path(std::string_view path) -> void;
 
    auto set_alpn_protocols(const std::vector<std::string>& protocols) -> void;
 
    auto set_ca_cert_path(std::string_view path) -> void;
 
    auto set_require_client_cert(bool require) -> void;
 
    auto set_max_idle_timeout(uint64_t timeout_ms) -> void;
 
    auto set_max_connections(size_t max) -> void;
 
    // =========================================================================
    // i_network_component interface implementation
    // =========================================================================
 
    [[nodiscard]] auto is_running() const -> bool override;
    auto wait_for_stop() -> void override;
 
    // =========================================================================
    // i_protocol_server interface implementation
    // =========================================================================
 
    [[nodiscard]] auto start(uint16_t port) -> VoidResult override;
    [[nodiscard]] auto stop() -> VoidResult override;
    [[nodiscard]] auto connection_count() const -> size_t override;
 
    auto set_connection_callback(connection_callback_t callback) -> void override;
    auto set_disconnection_callback(disconnection_callback_t callback) -> void override;
    auto set_receive_callback(receive_callback_t callback) -> void override;
    auto set_error_callback(error_callback_t callback) -> void override;
 
private:
    auto setup_internal_callbacks() -> void;
 
    auto get_or_create_wrapper(std::shared_ptr<session::quic_session> session)
        -> std::shared_ptr<quic_session_wrapper>;
 
    auto remove_wrapper(const std::string& session_id) -> void;
 
    std::string server_id_;
    std::shared_ptr<core::messaging_quic_server> server_;
    std::atomic<bool> is_running_{false};
 
    // QUIC-specific configuration
    std::string cert_path_;
    std::string key_path_;
    std::vector<std::string> alpn_protocols_;
    std::optional<std::string> ca_cert_path_;
    bool require_client_cert_{false};
    uint64_t max_idle_timeout_ms_{30000};
    size_t max_connections_{10000};
 
    // Session management
    mutable std::mutex sessions_mutex_;
    std::unordered_map<std::string, std::shared_ptr<quic_session_wrapper>> sessions_;
 
    // Callbacks
    mutable std::mutex callbacks_mutex_;
    connection_callback_t connection_callback_;
    disconnection_callback_t disconnection_callback_;
    receive_callback_t receive_callback_;
    error_callback_t error_callback_;
};
 
} // namespace kcenon::network::internal::adapters