CppNet/cppnet/socket/socket_interface.h

#ifndef NET_SOCKET_SOCKET_INTERFACE
#define NET_SOCKET_SOCKET_INTERFACE

#include <memory>
#include <cstdint>
#include <unordered_map>

namespace cppnet {

class Buffer;
class Address;
class CppNetBase;
class EventActions;
class Socket { 
public:
    Socket(): _sock(0) {}
    virtual ~Socket() {}

    void SetSocket(const uint64_t& sock) { _sock = sock; }
    uint64_t GetSocket() { return _sock; }

    void SetAddress(std::shared_ptr<Address>& addr) { _addr = addr; }
    const std::shared_ptr<Address>& GetAddress() const { return _addr; }

    void SetCppNetBase(std::shared_ptr<CppNetBase> base) { _cppnet_base = base; }
    const std::shared_ptr<CppNetBase>& GetCppNetBase() const { return _cppnet_base.lock(); }

    void SetEventActions(std::shared_ptr<EventActions> actions) { _event_actions = actions; }
    const std::shared_ptr<EventActions>& GetEventActions() const { return _event_actions; }
    
protected:
    uint64_t _sock;
    std::shared_ptr<Address>  _addr;
    std::weak_ptr<CppNetBase> _cppnet_base;
    std::shared_ptr<EventActions> _event_actions;

    static thread_local std::unordered_map<uint64_t, std::shared_ptr<Socket>> __all_socket_map;
};

thread_local std::unordered_map<uint64_t, std::shared_ptr<Socket>> Socket::__all_socket_map;

}

#endif