accept_worker.h

Go to the documentation of this file.

// BSD 3-Clause License
// Copyright (c) 2021-2025, 🍀☀🌕🌥 🌊
// See the LICENSE file in the project root for full license information.
 
#ifndef PACS_NETWORK_DETAIL_ACCEPT_WORKER_HPP
#define PACS_NETWORK_DETAIL_ACCEPT_WORKER_HPP
 
#include <kcenon/thread/core/thread_base.h>
#include <kcenon/common/patterns/result.h>
 
#include <atomic>
#include <chrono>
#include <functional>
#include <memory>
#include <string>
 
#ifdef _WIN32
#include <winsock2.h>
#include <ws2tcpip.h>
#pragma comment(lib, "ws2_32.lib")
#else
#include <arpa/inet.h>
#include <netinet/in.h>
#include <sys/socket.h>
#include <unistd.h>
#endif
 
namespace kcenon::pacs::network::detail {
 
namespace common = kcenon::common;
 
class accept_worker : public kcenon::thread::thread_base {
public:
    // =========================================================================
    // Type Aliases
    // =========================================================================
 
    using result_void = common::VoidResult;
 
    using connection_callback = std::function<void(uint64_t session_id)>;
 
    using maintenance_callback = std::function<void()>;
 
    // =========================================================================
    // Construction / Destruction
    // =========================================================================
 
    explicit accept_worker(
        uint16_t port,
        connection_callback on_connection,
        maintenance_callback on_maintenance = nullptr);
 
    ~accept_worker() override;
 
    // Non-copyable, non-movable (owns thread resources)
    accept_worker(const accept_worker&) = delete;
    accept_worker& operator=(const accept_worker&) = delete;
    accept_worker(accept_worker&&) = delete;
    accept_worker& operator=(accept_worker&&) = delete;
 
    // =========================================================================
    // Configuration
    // =========================================================================
 
    void set_max_pending_connections(int backlog);
 
    [[nodiscard]] uint16_t port() const noexcept;
 
    [[nodiscard]] int max_pending_connections() const noexcept;
 
    // =========================================================================
    // Status
    // =========================================================================
 
    [[nodiscard]] bool is_accepting() const noexcept;
 
    [[nodiscard]] std::string to_string() const override;
 
protected:
    // =========================================================================
    // thread_base Overrides
    // =========================================================================
 
    result_void before_start() override;
 
    result_void do_work() override;
 
    result_void after_stop() override;
 
    [[nodiscard]] bool should_continue_work() const override;
 
    void on_stop_requested() override;
 
private:
    // =========================================================================
    // Private Methods
    // =========================================================================
 
    [[nodiscard]] uint64_t next_session_id();
 
    // =========================================================================
    // Member Variables
    // =========================================================================
 
    uint16_t port_;
 
    connection_callback on_connection_;
 
    maintenance_callback on_maintenance_;
 
    int backlog_{128};
 
    std::atomic<uint64_t> session_id_counter_{0};
 
    std::atomic<bool> accepting_{false};
 
#ifdef _WIN32
    SOCKET listen_socket_{INVALID_SOCKET};
 
    bool wsa_initialized_{false};
#else
    int listen_socket_{-1};
#endif
};
 
}  // namespace kcenon::pacs::network::detail
 
#endif  // PACS_NETWORK_DETAIL_ACCEPT_WORKER_HPP