server.hxx

Go to the documentation of this file.

/*
 * Copyright (C) 2024-2026 David C. Manuelda (StormBytePP)
 *
 * This file is part of StormByte-Network.
 *
 * StormByte-Network is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Lesser General Public License version 3
 * or later, as published by the Free Software Foundation.
 *
 * StormByte-Network 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 Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public License
 * along with StormByte-Network. If not, see
 * <https://www.gnu.org/licenses/lgpl-3.0.html>.
 */
 
#pragma once
 
#include <StormByte/network/endpoint.hxx>
 
#include <atomic>
#include <mutex>
#include <thread>
#include <unordered_map>
#include <deque>
 
namespace StormByte::Network {
    namespace Connection {
        class Client;   
    }
 
    namespace Socket {
        class Server;   
    }
 
    namespace Detail {
        class Session;  
        class WorkerPool;   
    }
 
    class STORMBYTE_NETWORK_PUBLIC Server: private Endpoint {
        public:
            Server(const DeserializePacketFunction& deserialize_packet_function, std::shared_ptr<Logger::Log> logger) noexcept;
 
            Server(const Server& other) = delete;
 
            Server(Server&& other) noexcept;
 
            virtual ~Server() noexcept;
 
            Server& operator=(const Server& other) = delete;
 
            Server& operator=(Server&& other) noexcept;
 
            bool Connect(const Connection::Protocol& protocol, const std::string& address, const unsigned short& port) override;
 
            void Disconnect() noexcept override;
 
            inline Connection::Status Status() const noexcept override {
                return m_status.load();
            }
 
        protected:
            void DisconnectClient(const std::string& uuid) noexcept;
 
        private:
            std::unique_ptr<Socket::Server> m_socket_server;                                            
            std::atomic<Connection::Status> m_status;                                               
            std::thread m_accept_thread;                                                            
            Connection::HandlerType m_wakeup_read;                                              
            Connection::HandlerType m_wakeup_write;                                             
            std::unordered_map<std::string, std::shared_ptr<Detail::Session>> m_sessions;   
            std::unique_ptr<Detail::WorkerPool> m_pool;                                                     
            enum class CompletionReason: unsigned short { Success, NullHandler, Error }; 
            struct Completion {
                std::string uuid; 
                PacketPointer packet; 
                CompletionReason reason; 
            };
            std::deque<Completion> m_completions; 
            std::mutex m_completion_mutex; 
            enum class CommandType: unsigned short { DisconnectClient, DisconnectAll, Stop }; 
            struct Command { CommandType type; std::string uuid; }; 
            std::deque<Command> m_commands; 
            std::mutex m_command_mutex; 
 
            void AcceptClients() noexcept;
 
            bool CreateWakeup() noexcept;
 
            void SignalWakeup() noexcept;
 
            void CloseWakeup() noexcept;
 
            void AcceptOneClient() noexcept;
 
            void ProcessSession(const std::shared_ptr<Detail::Session>& session, bool readable, bool writable) noexcept;
 
            void PostCompletion(Completion completion) noexcept;
 
            void DrainCompletions() noexcept;
 
            void PostCommand(Command command) noexcept;
 
            void DrainCommands() noexcept;
 
            void DisconnectClientOnLoop(const std::string& uuid) noexcept;
 
            virtual PacketPointer ProcessClientPacket(const std::string& client_uuid, PacketPointer packet) noexcept = 0;
    };
}