messaging_udp_server.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 <functional>
#include <future>
#include <memory>
#include <string>
#include <string_view>
#include <vector>
 
#include <asio.hpp>
 
#include "internal/core/callback_indices.h"
#include "internal/interfaces/i_udp_server.h"
#include "kcenon/network/detail/utils/result_types.h"
#include "kcenon/network/detail/utils/lifecycle_manager.h"
#include "kcenon/network/detail/utils/callback_manager.h"
#include "kcenon/network/integration/thread_integration.h"
 
namespace kcenon::network::internal
{
    class udp_socket;
}
 
namespace kcenon::network::core
{
    class messaging_udp_server
        : public std::enable_shared_from_this<messaging_udp_server>
        , public interfaces::i_udp_server
    {
    public:
        using receive_callback_t = std::function<void(const std::vector<uint8_t>&,
                                                      const asio::ip::udp::endpoint&)>;
        using error_callback_t = std::function<void(std::error_code)>;
 
        explicit messaging_udp_server(std::string_view server_id);
 
        ~messaging_udp_server() noexcept override;
 
        // Non-copyable, non-movable
        messaging_udp_server(const messaging_udp_server&) = delete;
        messaging_udp_server& operator=(const messaging_udp_server&) = delete;
        messaging_udp_server(messaging_udp_server&&) = delete;
        messaging_udp_server& operator=(messaging_udp_server&&) = delete;
 
        // ========================================================================
        // Lifecycle Management
        // ========================================================================
 
        [[nodiscard]] auto start_server(uint16_t port) -> VoidResult;
 
        [[nodiscard]] auto stop_server() -> VoidResult;
 
        [[nodiscard]] auto server_id() const -> const std::string&;
 
        // ========================================================================
        // i_network_component interface implementation
        // ========================================================================
 
        [[nodiscard]] auto is_running() const -> bool override;
 
        auto wait_for_stop() -> void override;
 
        // ========================================================================
        // i_udp_server interface implementation
        // ========================================================================
 
        [[nodiscard]] auto start(uint16_t port) -> VoidResult override;
 
        [[nodiscard]] auto stop() -> VoidResult override;
 
        [[nodiscard]] auto send_to(
            const interfaces::i_udp_server::endpoint_info& endpoint,
            std::vector<uint8_t>&& data,
            interfaces::i_udp_server::send_callback_t handler = nullptr) -> VoidResult override;
 
        auto set_receive_callback(interfaces::i_udp_server::receive_callback_t callback) -> void override;
 
        auto set_receive_callback(receive_callback_t callback) -> void;
 
        auto set_error_callback(error_callback_t callback) -> void override;
 
    private:
        // =====================================================================
        // Internal Implementation Methods
        // =====================================================================
 
        auto do_start_impl(uint16_t port) -> VoidResult;
 
        auto do_stop_impl() -> VoidResult;
 
        // =====================================================================
        // Internal Callback Helpers
        // =====================================================================
 
        auto invoke_receive_callback(const std::vector<uint8_t>& data,
                                     const asio::ip::udp::endpoint& endpoint) -> void;
 
        auto invoke_error_callback(std::error_code ec) -> void;
 
        [[nodiscard]] auto get_receive_callback() const -> receive_callback_t;
 
        [[nodiscard]] auto get_error_callback() const -> error_callback_t;
 
        using callback_index = udp_server_callback;
 
        using callbacks_t = utils::callback_manager<
            receive_callback_t,
            error_callback_t
        >;
 
        // =====================================================================
        // Member Variables
        // =====================================================================
 
        std::string server_id_;                          
        utils::lifecycle_manager lifecycle_;             
        callbacks_t callbacks_;                          
        std::unique_ptr<asio::io_context> io_context_;   
        std::shared_ptr<internal::udp_socket> socket_;   
        std::shared_ptr<integration::thread_pool_interface> thread_pool_;   
        std::future<void> io_context_future_;            
    };
 
} // namespace kcenon::network::core