protected function doDeduplicateRootJob( Job $job ) {
$params = $job->getRootJobParams();
$partitions = $this->partitionPushRing->getLocations( $params['rootJobSignature'], 2 );
try {
return $this->partitionQueues[$partitions[0]]->doDeduplicateRootJob( $job );
} catch ( JobQueueError $e ) {
if ( isset( $partitions[1] ) ) { // check fallback partition
return $this->partitionQueues[$partitions[1]]->doDeduplicateRootJob( $job );
}
}
return false;
}
/**
* @covers HashRing
*/
public function testHashRing()
{
$ring = new HashRing(array('s1' => 1, 's2' => 1, 's3' => 2, 's4' => 2, 's5' => 2, 's6' => 3));
$locations = array();
for ($i = 0; $i < 20; $i++) {
$locations["hello{$i}"] = $ring->getLocation("hello{$i}");
}
$expectedLocations = array("hello0" => "s5", "hello1" => "s6", "hello2" => "s2", "hello3" => "s5", "hello4" => "s6", "hello5" => "s4", "hello6" => "s5", "hello7" => "s4", "hello8" => "s5", "hello9" => "s5", "hello10" => "s3", "hello11" => "s6", "hello12" => "s1", "hello13" => "s3", "hello14" => "s3", "hello15" => "s5", "hello16" => "s4", "hello17" => "s6", "hello18" => "s6", "hello19" => "s3");
$this->assertEquals($expectedLocations, $locations, 'Items placed at proper locations');
$locations = array();
for ($i = 0; $i < 5; $i++) {
$locations["hello{$i}"] = $ring->getLocations("hello{$i}", 2);
}
$expectedLocations = array("hello0" => array("s5", "s6"), "hello1" => array("s6", "s4"), "hello2" => array("s2", "s1"), "hello3" => array("s5", "s6"), "hello4" => array("s6", "s4"));
$this->assertEquals($expectedLocations, $locations, 'Items placed at proper locations');
}
/**
* @return Status Uses RediConnRef as value on success
*/
protected function getConnection()
{
if (!isset($this->conn)) {
$conn = false;
$servers = $this->ring->getLocations($this->key, 3);
ArrayUtils::consistentHashSort($servers, $this->key);
foreach ($servers as $server) {
$conn = $this->pool->getConnection($this->serversByLabel[$server], $this->logger);
if ($conn) {
break;
}
}
if (!$conn) {
return Status::newFatal('pool-servererror', implode(', ', $servers));
}
$this->conn = $conn;
}
return Status::newGood($this->conn);
}
