<?php
/**
* 对服务器进行一致性hash分布算法
*/
class HashRing
{
private $servers = array();
private $nodeList = array();
private $nodeHashList = array();
private $nodeTotalNum = 0;
private $virtualNodeNum = 32; //虚拟节点
private $keyHash = '';
public function __construct($servers)
{
$this->servers = $servers;
foreach ($servers as $server) {
for ($i = 0; $i < $this->virtualNodeNum; $i++) {
$this->nodeList[sprintf("%u", crc32($server.'-'.$i))] = array($server, $i);
}
}
ksort($this->nodeList);
$this->nodeHashList = array_keys($this->nodeList);
}
private function getNodeIndex($key)
{
$this->keyHash = sprintf("%u", crc32($key));
if ($this->keyHash > end($this->nodeHashList)) {
$this->keyHash = $this->keyHash % end($this->nodeHashList);
}
if ($this->keyHash <= reset($this->nodeHashList)) {
return 0;
}
$this->nodeTotalNum = count($this->nodeHashList);
return $this->binaryChopIndex(0, $this->nodeTotalNum);
}
private function binaryChopIndex($l=0, $r=0)
{
if ($l < $r) {
$avg = intval(($l+$r) / 2);
if ($this->nodeHashList[$avg] == $this->keyHash) {
return $avg;
} elseif ($this->keyHash < $this->nodeHashList[$avg] && ($avg > 0)) {
return $this->binaryChopIndex($l, $avg-1);
} else {
return $this->binaryChopIndex($avg+1, $r);
}
} else {
return $l;
}
}
public function getServersByKey($key, $num=1)
{
$index = $this->getNodeIndex($key);
$server = $this->nodeList[$this->nodeHashList[$index]];
if ($num == 1) {
return $server[0];
}
if ($num >= count($this->servers)) {
$num = count($this->servers);
}
$result = array($server[0]);
for ($i=$index+1; true; $i++) {
if ($i >= $this->nodeTotalNum) {
$i = 0;
}
$nextServer = $this->nodeList[$this->nodeHashList[$i]];
if (!in_array($nextServer[0], $result)) {
$result[] = $nextServer[0];
}
if (count($result) == $num) {
break;
}
}
return $result;
}
}
//示例
$servers = array(
'127.0.0.1:11211',
'127.0.0.1:11212',
'127.0.0.1:11213',
'127.0.0.1:11214',
'127.0.0.1:11215'
);
$obj = new HashRing($servers);
$servers = $obj->getServersByKey('testkey', 2);
print_r($servers);
echo "\n";