handshake.h

Go to the documentation of this file.

 
#ifndef QB_IO_ASYNC_PROTOCOL_HANDSHAKE_H
#define QB_IO_ASYNC_PROTOCOL_HANDSHAKE_H
#include <functional>
#include "../async/protocol.h"
#include "../async/event/handshake.h"
namespace qb::io::protocol {

template <typename _IO_>
class handshake : public async::AProtocol<_IO_> {
    bool _handshake_done = false;
public:
    using message = qb::io::async::event::handshake; 

    handshake() = delete;

    handshake(_IO_ &io) noexcept
        : async::AProtocol<_IO_>(io) {
            this->set_should_flush(false);
        }

    std::size_t
    getMessageSize() noexcept final {
        if (_handshake_done)
            return 0;
        return static_cast<size_t>(this->_io.transport().do_handshake());
    }

    void
    onMessage(std::size_t /*size*/) noexcept final { // size is unused
        _handshake_done = true;
        this->_io.on(message{});
    }

    void
    reset() noexcept final {
        _handshake_done = false;
    }
};
 
} // namespace qb::io::protocol
 
#endif // QB_IO_ASYNC_PROTOCOL_HANDSHAKE_H