MessageStream.hpp

Go to the documentation of this file.

 
#pragma once
 
#include <SDL3/SDL_stdinc.h>
 
#include <SDL3_net/SDL_net.h>
#include <functional>
#include <vector>

class MessageStream
{
 
public:
    MessageStream() = default;
 
    MessageStream(NET_StreamSocket* sock) : socket(sock) {}
 
    NET_StreamSocket* socket = nullptr; 

    bool send(const void* data, Uint32 size);

    bool poll(const std::function<void(const void* data, Uint32 size)>& callback);

    bool pumpReads();

    void drainComplete(const std::function<void(const void* data, Uint32 size)>& callback);
 
private:
    [[nodiscard]] Uint32 recvAvailable() const noexcept { return static_cast<Uint32>(recvBuf.size()) - recvHead; }

    [[nodiscard]] const Uint8* recvFront() const noexcept { return recvBuf.data() + recvHead; }

    void recvCompact();
 
    std::vector<Uint8> recvBuf; 
    Uint32 recvHead = 0;        
};