mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-12-16 19:32:07 +00:00
31 lines
803 B
C++
31 lines
803 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
|
|
/// 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();
|
|
|
|
Poco::Net::StreamSocket & socket;
|
|
|
|
Poco::Timespan send_timeout;
|
|
Poco::Timespan receive_timeout;
|
|
|
|
Poco::Timespan old_send_timeout;
|
|
Poco::Timespan old_receive_timeout;
|
|
};
|
|
}
|