ClickHouse/src/IO/ReadBufferFromPocoSocket.h

Ignoring revisions in .git-blame-ignore-revs. Click here to bypass and see the normal blame view.

38 lines
1018 B
C++
Raw Normal View History

2012-03-09 15:46:52 +00:00
#pragma once
#include <IO/BufferWithOwnMemory.h>
#include <IO/ReadBuffer.h>
2023-03-03 19:30:43 +00:00
#include <Common/AsyncTaskExecutor.h>
#include <Poco/Net/Socket.h>
2012-03-09 15:46:52 +00:00
namespace DB
{
/// Works with the ready Poco::Net::Socket. Blocking operations.
2012-03-09 15:46:52 +00:00
class ReadBufferFromPocoSocket : public BufferWithOwnMemory<ReadBuffer>
{
protected:
Poco::Net::Socket & socket;
2017-05-28 14:29:40 +00:00
/** For error messages. It is necessary to receive this address in advance, because,
* for example, if the connection is broken, the address will not be received anymore
* (getpeername will return an error).
*/
Poco::Net::SocketAddress peer_address;
bool nextImpl() override;
2012-03-09 15:46:52 +00:00
public:
explicit ReadBufferFromPocoSocket(Poco::Net::Socket & socket_, size_t buf_size = DBMS_DEFAULT_BUFFER_SIZE);
2012-05-09 08:16:09 +00:00
bool poll(size_t timeout_microseconds) const;
2020-12-02 17:02:14 +00:00
2021-01-19 19:21:06 +00:00
void setAsyncCallback(AsyncCallback async_callback_) { async_callback = std::move(async_callback_); }
2020-12-02 17:02:14 +00:00
private:
2021-01-19 19:21:06 +00:00
AsyncCallback async_callback;
2021-02-06 00:54:27 +00:00
std::string socket_description;
2012-03-09 15:46:52 +00:00
};
}