mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-09-29 21:20:49 +00:00
82c79fe4ce
In #21643 async_socket_for_remote=1 was fixed to avoid leaving the connection in the unsynchronised state. But one should not try to wait for the current packet in case of timeout because this will exceed the timeout. Anyway if the timeout is exceeded, then the connection will be shutdown (disconnected), so it will not left in an unsynchronised state.
82 lines
1.9 KiB
C++
82 lines
1.9 KiB
C++
#pragma once
|
|
|
|
#if defined(OS_LINUX)
|
|
|
|
#include <mutex>
|
|
#include <atomic>
|
|
#include <Common/Fiber.h>
|
|
#include <Common/FiberStack.h>
|
|
#include <Common/TimerDescriptor.h>
|
|
#include <Common/Epoll.h>
|
|
#include <Client/Connection.h>
|
|
#include <Client/IConnections.h>
|
|
#include <Poco/Timespan.h>
|
|
|
|
namespace Poco::Net
|
|
{
|
|
class Socket;
|
|
}
|
|
|
|
namespace DB
|
|
{
|
|
|
|
class MultiplexedConnections;
|
|
|
|
class RemoteQueryExecutorReadContext
|
|
{
|
|
public:
|
|
std::atomic_bool is_read_in_progress = false;
|
|
Packet packet;
|
|
|
|
std::exception_ptr exception;
|
|
FiberStack stack;
|
|
boost::context::fiber fiber;
|
|
/// This mutex for fiber is needed because fiber could be destroyed in cancel method from another thread.
|
|
std::mutex fiber_lock;
|
|
|
|
Poco::Timespan receive_timeout;
|
|
IConnections & connections;
|
|
Poco::Net::Socket * last_used_socket = nullptr;
|
|
|
|
/// Here we have three descriptors we are going to wait:
|
|
/// * connection_fd is a descriptor of connection. It may be changed in case of reading from several replicas.
|
|
/// * timer is a timerfd descriptor to manually check socket timeout
|
|
/// * pipe_fd is a pipe we use to cancel query and socket polling by executor.
|
|
/// We put those descriptors into our own epoll which is used by external executor.
|
|
TimerDescriptor timer{CLOCK_MONOTONIC, 0};
|
|
bool is_timer_alarmed = false;
|
|
int connection_fd = -1;
|
|
int pipe_fd[2] = { -1, -1 };
|
|
|
|
Epoll epoll;
|
|
|
|
std::string connection_fd_description;
|
|
|
|
explicit RemoteQueryExecutorReadContext(IConnections & connections_);
|
|
~RemoteQueryExecutorReadContext();
|
|
|
|
bool checkTimeout(bool blocking = false);
|
|
bool checkTimeoutImpl(bool blocking);
|
|
|
|
void setConnectionFD(int fd, const Poco::Timespan & timeout = 0, const std::string & fd_description = "");
|
|
void setTimer() const;
|
|
|
|
bool resumeRoutine();
|
|
void cancel();
|
|
};
|
|
|
|
}
|
|
|
|
#else
|
|
|
|
namespace DB
|
|
{
|
|
class RemoteQueryExecutorReadContext
|
|
{
|
|
public:
|
|
void cancel() {}
|
|
};
|
|
|
|
}
|
|
#endif
|