CppNet/cppnet/socket/win/win_rw_socket.h

// Use of this source code is governed by a BSD 3-Clause License
// that can be found in the LICENSE file.

// Author: caozhiyi (caozhiyi5@gmail.com)

#ifndef CPPNET_SOCKET_WIN_READ_WRITE_SOCKET
#define CPPNET_SOCKET_WIN_READ_WRITE_SOCKET

#include <mutex>
#include <atomic>
#include <unordered_set>
#include "../rw_socket.h"

namespace cppnet {

class Event;
class AlloterWrap;
class WinRWSocket:
    public RWSocket { 

public:
    WinRWSocket();
    WinRWSocket(std::shared_ptr<AlloterWrap> alloter);
    WinRWSocket(uint64_t sock, std::shared_ptr<AlloterWrap> alloter);
    virtual ~WinRWSocket();

    virtual void Read();
    virtual bool Write(const char* src, uint32_t len);
    virtual void Connect(const std::string& ip, uint16_t port);
    virtual void Disconnect();

    virtual void OnRead(Event* event, uint32_t len = 0);
    virtual void OnWrite(Event* event, uint32_t len = 0);
    virtual void OnDisConnect(Event* event, uint16_t err);

    virtual void SetShutdown() { _shutdown = true; }
    virtual bool IsShutdown() { return _shutdown; }

    virtual std::shared_ptr<BufferQueue> GetReadBuffer();

private:
    void AddEvent(Event* event);
    void RemvoeEvent(Event* event);
    bool EventEmpty();

private:
    std::atomic_bool _shutdown;
    std::atomic_bool _is_reading;

    // only need read cache. data to send is saved to event buffer.
    std::shared_ptr<BufferQueue> _read_buffer;

    // all event
    std::mutex _event_mutex;
    std::unordered_set<Event*> _event_set;
};

std::shared_ptr<RWSocket> MakeRWSocket();
std::shared_ptr<RWSocket> MakeRWSocket(std::shared_ptr<AlloterWrap> alloter);
std::shared_ptr<RWSocket> MakeRWSocket(uint64_t sock, std::shared_ptr<AlloterWrap> alloter);

}

#endif