void LatencyAwarePolicy::init(const Host::Ptr& connected_host,
const HostMap& hosts,
Random* random) {
hosts_->reserve(hosts.size());
std::transform(hosts.begin(), hosts.end(), std::back_inserter(*hosts_), GetHost());
for (HostMap::const_iterator i = hosts.begin(),
end = hosts.end(); i != end; ++i) {
i->second->enable_latency_tracking(settings_.scale_ns, settings_.min_measured);
}
ChainedLoadBalancingPolicy::init(connected_host, hosts, random);
}
void LatencyAwarePolicy::init(const SharedRefPtr<Host>& connected_host, const HostMap& hosts) {
copy_hosts(hosts, hosts_);
for (HostMap::const_iterator i = hosts.begin(),
end = hosts.end(); i != end; ++i) {
i->second->enable_latency_tracking(settings_.scale_ns, settings_.min_measured);
}
ChainedLoadBalancingPolicy::init(connected_host, hosts);
}
/**
* Clears the connections kept by this pool (ie, not including the global pool)
*/
void clearPool() {
for(HostMap::iterator iter = _hosts.begin(); iter != _hosts.end(); ++iter) {
if (iter->second->avail != NULL) {
delete iter->second->avail;
}
}
_hosts.clear();
}
int DNS::connect( const std::string& domain, const LogSink& logInstance )
{
HostMap hosts = resolve( domain );
if( hosts.size() == 0 )
return -DNS_NO_HOSTS_FOUND;
struct protoent* prot;
if( ( prot = getprotobyname( "tcp" ) ) == 0)
return -DNS_COULD_NOT_RESOLVE;
int fd;
if( ( fd = socket( PF_INET, SOCK_STREAM, prot->p_proto ) ) == -1 )
return -DNS_COULD_NOT_RESOLVE;
struct hostent *h;
struct sockaddr_in target;
target.sin_family = AF_INET;
int ret = 0;
HostMap::const_iterator it = hosts.begin();
for( ; it != hosts.end(); ++it )
{
int port;
if( (*it).second == 0 )
port = XMPP_PORT;
else
port = (*it).second;
target.sin_port = htons( port );
if( ( h = gethostbyname( (*it).first.c_str() ) ) == 0 )
{
ret = -DNS_COULD_NOT_RESOLVE;
continue;
}
in_addr *addr = (in_addr*)malloc( sizeof( in_addr ) );
memcpy( addr, h->h_addr, sizeof( in_addr ) );
char *tmp = inet_ntoa( *addr );
free( addr );
std::ostringstream oss;
oss << "resolved " << (*it).first.c_str() << " to: " << tmp << ":" << port;
logInstance.log( LogLevelDebug, LogAreaClassDns, oss.str() );
if( inet_aton( tmp, &(target.sin_addr) ) == 0 )
continue;
memset( target.sin_zero, '\0', 8 );
if( ::connect( fd, (struct sockaddr *)&target, sizeof( struct sockaddr ) ) == 0 )
return fd;
close( fd );
}
if( ret )
return ret;
return -DNS_COULD_NOT_CONNECT;
}
void DCAwarePolicy::init(const SharedRefPtr<Host>& connected_host, const HostMap& hosts) {
if (local_dc_.empty() && !connected_host->dc().empty()) {
LOG_INFO("Using '%s' for the local data center "
"(if this is incorrect, please provide the correct data center)",
connected_host->dc().c_str());
local_dc_ = connected_host->dc();
}
for (HostMap::const_iterator i = hosts.begin(),
end = hosts.end(); i != end; ++i) {
on_add(i->second);
}
}
int DNS::connect( const std::string& host, const LogSink& logInstance )
{
HostMap hosts = resolve( host, logInstance );
if( hosts.size() == 0 )
return -ConnDnsError;
HostMap::const_iterator it = hosts.begin();
for( ; it != hosts.end(); ++it )
{
int fd = DNS::connect( (*it).first, (*it).second, logInstance );
if( fd >= 0 )
return fd;
}
return -ConnConnectionRefused;
}
void ListPolicy::init(const Host::Ptr& connected_host,
const HostMap& hosts,
Random* random) {
HostMap valid_hosts;
for (HostMap::const_iterator i = hosts.begin(),
end = hosts.end(); i != end; ++i) {
const Host::Ptr& host = i->second;
if (is_valid_host(host)) {
valid_hosts.insert(HostPair(i->first, host));
}
}
if (valid_hosts.empty()) {
LOG_ERROR("No valid hosts available for list policy");
}
ChainedLoadBalancingPolicy::init(connected_host, valid_hosts, random);
}