mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-12-16 03:12:43 +00:00
34 lines
785 B
C++
34 lines
785 B
C++
#pragma once
|
|
|
|
#include <Poco/Net/Socket.h>
|
|
|
|
#include <IO/ReadBuffer.h>
|
|
#include <IO/BufferWithOwnMemory.h>
|
|
|
|
|
|
namespace DB
|
|
{
|
|
|
|
/** Works with the ready Poco::Net::Socket. Blocking operations.
|
|
*/
|
|
class ReadBufferFromPocoSocket : public BufferWithOwnMemory<ReadBuffer>
|
|
{
|
|
protected:
|
|
Poco::Net::Socket & socket;
|
|
|
|
/** 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;
|
|
|
|
public:
|
|
ReadBufferFromPocoSocket(Poco::Net::Socket & socket_, size_t buf_size = DBMS_DEFAULT_BUFFER_SIZE);
|
|
|
|
bool poll(size_t timeout_microseconds);
|
|
};
|
|
|
|
}
|