HostAddress * HostCatcher::getHost( ) {
mLock->lock();
int numHosts = mHostVector->size();
if( numHosts == 0 ) {
mLock->unlock();
return NULL;
}
// remove random host from queue
int index = mRandSource->getRandomBoundedInt( 0, numHosts - 1 );
HostAddress *host = *( mHostVector->getElement( index ) );
mHostVector->deleteElement( index );
// add host to end of queue
mHostVector->push_back( host );
HostAddress *hostCopy = host->copy();
mLock->unlock();
return hostCopy;
}
HostAddress * HostCatcher::getHostOrdered( ) {
mLock->lock();
int numHosts = mHostVector->size();
if( numHosts == 0 ) {
mLock->unlock();
return NULL;
}
// remove first host from queue
HostAddress *host = *( mHostVector->getElement( 0 ) );
mHostVector->deleteElement( 0 );
// add host to end of queue
mHostVector->push_back( host );
HostAddress *hostCopy = host->copy();
mLock->unlock();
return hostCopy;
}
void HostCatcher::addHost( HostAddress * inHost ) {
// convert to numerical form once and for all here
// (to avoid converting over and over in equals checks below)
HostAddress *numericalAddress = inHost->getNumericalAddress();
if( numericalAddress != NULL ) {
mLock->lock();
// make sure this host doesn't already exist in our list
char exists = false;
int numHosts = mHostVector->size();
for( int i=0; i<numHosts; i++ ) {
HostAddress *otherHost = *( mHostVector->getElement( i ) );
if( otherHost->equals( numericalAddress ) ) {
exists = true;
// jump out of loop
i = numHosts;
}
}
if( !exists ) {
mHostVector->push_back( numericalAddress->copy() );
}
while( mHostVector->size() > mMaxListSize ) {
// remove first host from queue
HostAddress *host = *( mHostVector->getElement( 0 ) );
mHostVector->deleteElement( 0 );
delete host;
}
mLock->unlock();
delete numericalAddress;
}
}
