http_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 <string>
#include <unordered_map>
 
namespace kcenon::network::core {
    class http_server;
}
 
namespace kcenon::network::internal::adapters {
 
class http_request_session : public interfaces::i_session {
public:
    http_request_session(
        std::string_view session_id,
        std::string_view client_address,
        uint16_t client_port,
        std::weak_ptr<core::http_server> server);
 
    [[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;
 
    auto set_response_data(std::vector<uint8_t> data) -> void;
 
    [[nodiscard]] auto get_response_data() const -> const std::vector<uint8_t>&;
 
private:
    std::string session_id_;
    std::string client_address_;
    uint16_t client_port_;
    std::weak_ptr<core::http_server> server_;
    std::vector<uint8_t> response_data_;
    std::atomic<bool> is_connected_{true};
    mutable std::mutex data_mutex_;
};
 
class http_server_adapter : public interfaces::i_protocol_server {
public:
    explicit http_server_adapter(std::string_view server_id);
 
    ~http_server_adapter() override;
 
    // Non-copyable, non-movable
    http_server_adapter(const http_server_adapter&) = delete;
    http_server_adapter& operator=(const http_server_adapter&) = delete;
    http_server_adapter(http_server_adapter&&) = delete;
    http_server_adapter& operator=(http_server_adapter&&) = delete;
 
    // =========================================================================
    // 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_routes() -> void;
 
    static auto make_session_id(const std::string& address, uint16_t port, uint64_t request_id) -> std::string;
 
    auto get_or_create_session(const std::string& address, uint16_t port)
        -> std::shared_ptr<http_request_session>;
 
    auto remove_session(const std::string& session_id) -> void;
 
    std::string server_id_;
    std::shared_ptr<core::http_server> server_;
 
    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_;
 
    mutable std::mutex sessions_mutex_;
    std::unordered_map<std::string, std::shared_ptr<http_request_session>> sessions_;
    std::atomic<uint64_t> request_counter_{0};
    std::atomic<bool> is_running_{false};
};
 
} // namespace kcenon::network::internal::adapters