mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-12-11 08:52:06 +00:00
35 lines
917 B
C++
35 lines
917 B
C++
#pragma once
|
|
|
|
#include <Poco/Net/StreamSocket.h>
|
|
#include <Poco/Timespan.h>
|
|
|
|
|
|
namespace DB
|
|
{
|
|
/// Temporarily overrides socket send/receive timeouts and reset them back into destructor (or manually by calling reset method)
|
|
/// If "limit_max_timeout" is true, timeouts could be only decreased (maxed by previous value).
|
|
struct TimeoutSetter
|
|
{
|
|
TimeoutSetter(Poco::Net::StreamSocket & socket_,
|
|
Poco::Timespan send_timeout_,
|
|
Poco::Timespan receive_timeout_,
|
|
bool limit_max_timeout = false);
|
|
|
|
TimeoutSetter(Poco::Net::StreamSocket & socket_, Poco::Timespan timeout_, bool limit_max_timeout = false);
|
|
|
|
~TimeoutSetter();
|
|
|
|
/// Reset timeouts back.
|
|
void reset();
|
|
|
|
Poco::Net::StreamSocket & socket;
|
|
|
|
Poco::Timespan send_timeout;
|
|
Poco::Timespan receive_timeout;
|
|
|
|
Poco::Timespan old_send_timeout;
|
|
Poco::Timespan old_receive_timeout;
|
|
bool was_reset = false;
|
|
};
|
|
}
|