mirror of
https://github.com/facebook/sapling.git
synced 2024-10-07 23:38:50 +03:00
13da025c6c
Summary: This diff adds `ServiceAddress` to utils. This is a class that encapsulates an address of a service that could be a traditional hostname & port pair as well as an SMC tier name. This eliminates the needs of the manually resolving network address for remote services. Reviewed By: chadaustin Differential Revision: D14845257 fbshipit-source-id: 9fe9847cca4bba0170be94b9c209247342708574
81 lines
2.3 KiB
C++
81 lines
2.3 KiB
C++
/*
|
|
* Copyright (c) 2019-present, Facebook, Inc.
|
|
* All rights reserved.
|
|
*
|
|
* This source code is licensed under the BSD-style license found in the
|
|
* LICENSE file in the root directory of this source tree. An additional grant
|
|
* of patent rights can be found in the PATENTS file in the same directory.
|
|
*
|
|
*/
|
|
#include "eden/fs/utils/ServiceAddress.h"
|
|
|
|
#include <folly/Random.h>
|
|
#include <folly/SocketAddress.h>
|
|
#include <folly/String.h>
|
|
#include <folly/logging/xlog.h>
|
|
#include <optional>
|
|
#include "eden/fs/eden-config.h"
|
|
|
|
#ifdef EDEN_HAVE_SERVICEROUTER
|
|
#include <servicerouter/client/cpp2/ServiceRouter.h>
|
|
#endif
|
|
|
|
namespace facebook {
|
|
namespace eden {
|
|
|
|
ServiceAddress::ServiceAddress(std::string name) : name_(std::move(name)){};
|
|
|
|
ServiceAddress::ServiceAddress(std::string hostname, uint16_t port)
|
|
: name_(std::make_pair(std::move(hostname), port)){};
|
|
|
|
std::optional<SocketAddressWithHostname>
|
|
ServiceAddress::getSocketAddressBlocking() {
|
|
if (std::holds_alternative<HostPortPair>(name_)) {
|
|
return addressFromHostname();
|
|
}
|
|
return addressFromSMCTier();
|
|
}
|
|
|
|
std::optional<SocketAddressWithHostname> ServiceAddress::addressFromHostname() {
|
|
auto hostPort = std::get<HostPortPair>(name_);
|
|
auto addr = folly::SocketAddress();
|
|
addr.setFromHostPort(hostPort.first, hostPort.second);
|
|
return std::make_pair(addr, hostPort.first);
|
|
}
|
|
|
|
std::optional<SocketAddressWithHostname> ServiceAddress::addressFromSMCTier(
|
|
std::shared_ptr<facebook::servicerouter::ServiceCacheIf> selector) {
|
|
#ifdef EDEN_HAVE_SERVICEROUTER
|
|
auto selection = selector->getSelection(std::get<std::string>(name_));
|
|
|
|
if (selection.hosts.empty()) {
|
|
return std::nullopt;
|
|
}
|
|
|
|
// TODO(zeyi, t42568801): better host selection algorithm
|
|
auto selected = folly::Random::rand32(selection.hosts.size());
|
|
const auto& host = selection.hosts[selected];
|
|
auto location = host->location();
|
|
|
|
return std::make_pair(
|
|
folly::SocketAddress(location.getIpAddress(), location.getPort()),
|
|
location.getHostname());
|
|
#else
|
|
return std::nullopt;
|
|
#endif
|
|
}
|
|
|
|
std::optional<SocketAddressWithHostname> ServiceAddress::addressFromSMCTier() {
|
|
#ifdef EDEN_HAVE_SERVICEROUTER
|
|
auto& factory = servicerouter::cpp2::getClientFactory();
|
|
auto selector = factory.getSelector();
|
|
|
|
return addressFromSMCTier(selector);
|
|
#else
|
|
return std::nullopt;
|
|
#endif
|
|
}
|
|
|
|
} // namespace eden
|
|
} // namespace facebook
|