thevar1able/ClickHouse
1
0
Fork 0
mirror of https://github.com/ClickHouse/ClickHouse.git synced 2024-11-26 01:22:04 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #54819 from canhld94/ch_net_interfaces

Browse Source 
Avoid excessive calls to getifaddrs in isLocalAddress
This commit is contained in:
Igor Nikonov 2023-09-21 13:29:18 +02:00 committed by GitHub
commit 127ce64298
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 40 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

43
src/Common/isLocalAddress.cpp
View File

@ -1,9 +1,14 @@
#include <Common/isLocalAddress.h>
#include <ifaddrs.h>
#include <chrono>
#include <cstring>
#include <memory>
#include <mutex>
#include <shared_mutex>
#include <optional>
#include <base/types.h>
#include <boost/core/noncopyable.hpp>
#include <Common/Exception.h>
#include <Poco/Net/IPAddress.h>
#include <Poco/Net/SocketAddress.h>
@ -20,7 +25,7 @@ namespace ErrorCodes
namespace
{
struct NetworkInterfaces
struct NetworkInterfaces : public boost::noncopyable
{
ifaddrs * ifaddr;
NetworkInterfaces()
@ -31,6 +36,13 @@ struct NetworkInterfaces
}
}
void swap(NetworkInterfaces && other)
{
auto * tmp = ifaddr;
ifaddr = other.ifaddr;
other.ifaddr = tmp;
}
bool hasAddress(const Poco::Net::IPAddress & address) const
{
ifaddrs * iface;
@ -74,6 +86,32 @@ struct NetworkInterfaces
{
freeifaddrs(ifaddr);
}
static const NetworkInterfaces & instance()
{
static constexpr int NET_INTERFACE_VALID_PERIOD_MS = 30000;
static NetworkInterfaces nf;
static std::atomic<std::chrono::steady_clock::time_point> last_updated_time = std::chrono::steady_clock::now();
static std::shared_mutex nf_mtx;
auto now = std::chrono::steady_clock::now();
auto last_updated_time_snapshot = last_updated_time.load();
if (std::chrono::duration_cast<std::chrono::milliseconds>(now - last_updated_time_snapshot).count() > NET_INTERFACE_VALID_PERIOD_MS)
{
std::unique_lock lock(nf_mtx);
if (last_updated_time.load() != last_updated_time_snapshot) /// it's possible that last_updated_time after we get the snapshot
return nf;
nf.swap(NetworkInterfaces());
last_updated_time.store(now);
return nf;
}
else
{
std::shared_lock lock(nf_mtx);
return nf;
}
}
};
}
@ -111,8 +149,7 @@ bool isLocalAddress(const Poco::Net::IPAddress & address)
}
}
NetworkInterfaces interfaces;
return interfaces.hasAddress(address);
return NetworkInterfaces::instance().hasAddress(address);
}