Server.hpp

#pragma once
#include <samchon/API.hpp>

#include <samchon/protocol/ClientDriver.hpp>

#include <thread>

namespace samchon
{
namespace protocol
{
    class Server
    {
    protected:
        std::unique_ptr<boost::asio::ip::tcp::acceptor> _Acceptor;

    public:
        Server()
        {
        };

        virtual ~Server()
        {
            if (_Acceptor == nullptr || _Acceptor->is_open() == false)
                return;

            close();
        };

        virtual void open(int port)
        {
            if (_Acceptor != nullptr && _Acceptor->is_open())
                return;

            boost::asio::io_service io_service;
            boost::asio::ip::tcp::endpoint endpoint(boost::asio::ip::tcp::v4(), port);
            boost::system::error_code error;

            _Acceptor.reset(new boost::asio::ip::tcp::acceptor(io_service, endpoint));

            while (true)
            {
                std::shared_ptr<boost::asio::ip::tcp::socket> socket(new boost::asio::ip::tcp::socket(io_service));
                _Acceptor->accept(*socket);

                if (error)
                    break;

                std::thread(&Server::handle_connection, this, socket).detach();
            }
        };

        virtual void close()
        {
            if (_Acceptor == nullptr)
                return;

            _Acceptor->cancel();
            _Acceptor->close();
        };

    protected:
        virtual void addClient(std::shared_ptr<ClientDriver>) = 0; //ADD_CLIENT

    private:
        virtual void handle_connection(std::shared_ptr<boost::asio::ip::tcp::socket> socket)
        {
            addClient(std::make_shared<ClientDriver>(socket));
        };
    };
};
};