mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-11-29 02:52:13 +00:00
46 lines
1.3 KiB
C++
46 lines
1.3 KiB
C++
#include <IO/TimeoutSetter.h>
|
|
|
|
#include <base/logger_useful.h>
|
|
|
|
|
|
namespace DB
|
|
{
|
|
|
|
TimeoutSetter::TimeoutSetter(Poco::Net::StreamSocket & socket_,
|
|
Poco::Timespan send_timeout_,
|
|
Poco::Timespan receive_timeout_,
|
|
bool limit_max_timeout)
|
|
: socket(socket_), send_timeout(send_timeout_), receive_timeout(receive_timeout_)
|
|
{
|
|
old_send_timeout = socket.getSendTimeout();
|
|
old_receive_timeout = socket.getReceiveTimeout();
|
|
|
|
if (!limit_max_timeout || old_send_timeout > send_timeout)
|
|
socket.setSendTimeout(send_timeout);
|
|
|
|
if (!limit_max_timeout || old_receive_timeout > receive_timeout)
|
|
socket.setReceiveTimeout(receive_timeout);
|
|
}
|
|
|
|
TimeoutSetter::TimeoutSetter(Poco::Net::StreamSocket & socket_, Poco::Timespan timeout_, bool limit_max_timeout)
|
|
: TimeoutSetter(socket_, timeout_, timeout_, limit_max_timeout)
|
|
{
|
|
}
|
|
|
|
TimeoutSetter::~TimeoutSetter()
|
|
{
|
|
try
|
|
{
|
|
socket.setSendTimeout(old_send_timeout);
|
|
socket.setReceiveTimeout(old_receive_timeout);
|
|
}
|
|
catch (...)
|
|
{
|
|
/// Sometimes caught on Mac OS X. This message can be safely ignored.
|
|
/// If you are developer using Mac, please debug this error message by yourself.
|
|
tryLogCurrentException("Client", "TimeoutSetter: Can't reset timeouts");
|
|
}
|
|
}
|
|
|
|
}
|