32 lines
900 B
C++
32 lines
900 B
C++
|
#pragma once
|
||
|
|
#include <boost/asio/ip/tcp.hpp>
|
||
|
|
#include <boost/system/error_code.hpp>
|
||
|
|
#include <kj/async-io.h>
|
||
|
|
|
||
|
|
namespace vanetza
|
||
|
|
{
|
||
|
|
namespace rpc
|
||
|
|
{
|
||
|
|
|
||
|
|
class AsioStream : public kj::AsyncIoStream
|
||
|
|
{
|
||
|
|
public:
|
||
|
|
AsioStream(boost::asio::ip::tcp::socket socket);
|
||
|
|
|
||
|
|
void shutdownWrite() override;
|
||
|
|
kj::Promise<void> write(const void* buffer, size_t size) override;
|
||
|
|
kj::Promise<void> write(kj::ArrayPtr<const kj::ArrayPtr<const kj::byte>> pieces) override;
|
||
|
|
kj::Promise<void> whenWriteDisconnected() override;
|
||
|
|
kj::Promise<size_t> tryRead(void* buffer, size_t minBytes, size_t maxBytes) override;
|
||
|
|
|
||
|
|
private:
|
||
|
|
void signalDisconnect(const boost::system::error_code& ec);
|
||
|
|
|
||
|
|
boost::asio::ip::tcp::socket socket_;
|
||
|
|
kj::Maybe<kj::ForkedPromise<void>> disconnect_promise_;
|
||
|
|
kj::Maybe<kj::Own<kj::PromiseFulfiller<void>>> disconnect_fulfiller_;
|
||
|
|
};
|
||
|
|
|
||
|
|
} // namepsace rpc
|
||
|
|
} // namespace vanetza
|