mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-09-30 13:40:50 +00:00
53 lines
1.5 KiB
C++
53 lines
1.5 KiB
C++
#pragma once
|
|
|
|
#include <Poco/Timespan.h>
|
|
#include <Interpreters/Settings.h>
|
|
|
|
namespace DB
|
|
{
|
|
|
|
struct ConnectionTimeouts
|
|
{
|
|
Poco::Timespan connection_timeout;
|
|
Poco::Timespan send_timeout;
|
|
Poco::Timespan receive_timeout;
|
|
|
|
ConnectionTimeouts() = default;
|
|
|
|
ConnectionTimeouts(const Poco::Timespan & connection_timeout_,
|
|
const Poco::Timespan & send_timeout_,
|
|
const Poco::Timespan & receive_timeout_)
|
|
: connection_timeout(connection_timeout_),
|
|
send_timeout(send_timeout_),
|
|
receive_timeout(receive_timeout_)
|
|
{
|
|
}
|
|
|
|
static Poco::Timespan saturate(const Poco::Timespan & timespan, const Poco::Timespan & limit)
|
|
{
|
|
if (limit.totalMicroseconds() == 0)
|
|
return timespan;
|
|
else
|
|
return (timespan > limit) ? limit : timespan;
|
|
}
|
|
|
|
ConnectionTimeouts getSaturated(const Poco::Timespan & limit) const
|
|
{
|
|
return ConnectionTimeouts(saturate(connection_timeout, limit),
|
|
saturate(send_timeout, limit),
|
|
saturate(receive_timeout, limit));
|
|
}
|
|
|
|
static ConnectionTimeouts getTCPTimeouts(const Settings & settings)
|
|
{
|
|
return ConnectionTimeouts(settings.connect_timeout, settings.send_timeout, settings.receive_timeout);
|
|
}
|
|
|
|
static ConnectionTimeouts getHTTPTimeouts(const Settings & settings)
|
|
{
|
|
return ConnectionTimeouts(settings.http_connection_timeout, settings.http_send_timeout, settings.http_receive_timeout);
|
|
}
|
|
};
|
|
|
|
}
|