优化过的redis封装类

转http://www.cnblogs.com/jackluo/p/3410192.html

<?php

/**
* RedisCluster 群redius操作类
*
* //创建连接
* $rc = new RedisCluster();
* $rc->connect(array("host" => "127.0.0.1", "port" => 6379)); //php客户端设置的ip及端口
* //存储值
* $rc->set($key,$value);
* $rc->set($array);
* $rc->setnx($key,$value);//只创不更
* $rc->setnx($array);//只创不更
* //获取值
* $rc->get($key);
* $rc->get($array);
* //删除
* $rc->delete($key);
* $rc->delete($array);
* //删除所有
* $rc->deleteAll();
* //+1或者其他
* $rc->incr($key[,$othervar]);
* //-1或者其他
* $rc->decr($key[,$othervar]);
* //关闭连接 0.主服务器，1.从服务器，2.全部
* $rc->close(2);
*
* hujiande 297538600@qq.com
* @date 2014.8.15
*/
class RedisCluster extends NoRewindIterator {

// 是否使用 M/S 的读写集群方案
private $_isUseCluster = false;
// Slave 句柄标记
private $_sn = 0;
// 服务器连接句柄
private $_linkHandle = array(
'master' => null, // 只支持一台 Master
'slave' => array(), // 可以有多台 Slave
);

/**
* 构造函数
*
* @access $rc = new RedisCluster($isUseCluster=false);
* @param boolean $isUseCluster 是否采用 M/S 方案
*/
public function __construct($isUseCluster = false) {
$this->_isUseCluster = $isUseCluster;
}

/**
* 连接服务器,注意：这里使用长连接，提高效率，但不会自动关闭
*
* @access $rc->connect($config=array('host'=>'127.0.0.1','port'=>6379), $isMaster=true);
* @param array $config Redis服务器配置
* @param boolean $isMaster 当前添加的服务器是否为 Master 服务器
* @return boolean
*/
public function connect($config = array('host' => '127.0.0.1', 'port' => 6379), $isMaster = true) {
// default port
if (!isset($config['port'])) {
$config['port'] = 6379;
}
// 设置 Master 连接
if ($isMaster) {
$this->_linkHandle['master'] = new Redis();
$ret = $this->_linkHandle['master']->pconnect($config['host'], $config['port']);
}
else {
// 多个 Slave 连接
$this->_linkHandle['slave'][$this->_sn] = new Redis();
$ret = $this->_linkHandle['slave'][$this->_sn]->pconnect($config['host'], $config['port']);
++$this->_sn;
}
return $ret;
}

/**
* 关闭连接
*
* @access $rc->connect($flag=2);
* @param int $flag 关闭选择 0:关闭 Master 1:关闭 Slave 2:关闭所有
* @return boolean
*/
public function close($flag = 2) {
switch ($flag) {
// 关闭 Master
case 0:
$this->getRedis()->close();
break;
// 关闭 Slave
case 1:
for ($i = 0; $i < $this->_sn; ++$i) {
$this->_linkHandle['slave'][$i]->close();
}
break;
// 关闭所有
case 1:
$this->getRedis()->close();
for ($i = 0; $i < $this->_sn; ++$i) {
$this->_linkHandle['slave'][$i]->close();
}
break;
}
return true;
}

/**
* 获取Redis 原始对象
*
* @access $rc->getRedis(true, true);
* @param boolean $isMaster 返回服务器的类型 true:返回Master false:返回Slave
* @param boolean $slaveOne 返回的Slave选择 true:负载均衡随机返回一个Slave选择 false:返回所有的Slave选择
* @return redis object
*/
public function getRedis($isMaster = true, $slaveOne = true) {
// 只返回 Master
if ($isMaster) {
return $this->_linkHandle['master'];
}
else {
return $slaveOne ? $this->_getSlaveRedis() : $this->_linkHandle['slave'];
}
}

/**
* 写缓存
*
* @access $rc->set($key, $value, $expire);//存储单个键值
* @access $rc->set($array, $expire);//存储多个键值
* @param string $key 组存KEY
* @param string $value 缓存值
* @param int $expire 过期时间，0:表示无过期时间 单位m秒
* @param boolean $is_nx 是否只创建不更新
*/
public function set($key, $value = '', $expire = 0) {
//存储多个键值
if (is_array($key)) {
$array = $key;
$expire = (int) $value;

// 永不超时
if ($expire == 0) {
$ret = $this->getRedis()->mset($array);
}
else {
$ret = true;

foreach ($array as $key1 => $value1) {
$ret = $this->set($key1, $value1, $expire, $is_nx) && $ret;
}

}

}
//存储单个键值
else {
// 永不超时
if ($expire == 0) {
$ret = $this->getRedis()->set($key, $value);
}
else {
$ret = $this->getRedis()->setex($key, $expire, $value);
}

}

return $ret;
}

/**
* 只创建$key的值，不更新已存在的$key值
*
* @access $rc->setnx($not_exist_key, $value);//不更新已存在的$key值
* @access $rc->setnx($array);//不更新已存在的$key值
* @param string $not_exist_key 缓存KEY
* @param string $value 缓存值
* @return boolean
*/
public function setnx($not_exist_key, $value = '') {

if (is_array($not_exist_key)) {

return $this->getRedis()->msetnx($not_exist_key);
} else {

return $this->getRedis()->setnx($not_exist_key, $value);
}

}

/**
* 读缓存
*
* @access $rc->get($key);
* @param string $key 缓存KEY,支持一次取多个 $key = array('key1','key2')
* @return string || boolean 失败返回 false, 成功返回字符串
*/
public function get($key) {
// 是否一次取多个值
$func = is_array($key) ? 'mGet' : 'get';
// 没有使用M/S
if (!$this->_isUseCluster) {
return $this->getRedis()->{$func}($key);
}
// 使用了 M/S
return $this->_getSlaveRedis()->{$func}($key);
}

/**
* 删除缓存
*
* @access $rc->delete($key);
* @param string || array $key 缓存KEY，支持单个健:"key1" 或多个健:array('key1','key2')
* @return int 删除的健的数量
*/
public function delete($key) {
return $this->getRedis()->delete($key);
}

/**
* 清空当前数据库
*
* @access $rc->deleteAll();
* @return boolean
*/
public function deleteAll() {
return $this->getRedis()->flushDB();
}

/**
* 值加加操作,类似 ++$i ,如果 key 不存在时自动设置为 0 后进行加加操作
*
* @access $rc->incr($key, $default=1);
* @param string $key 缓存KEY
* @param int $default 操作时的默认值
* @return int　	操作后的值
*/
public function incr($key, $default = 1) {
if ($default == 1) {
return $this->getRedis()->incr($key);
}
else {
return $this->getRedis()->incrBy($key, $default);
}
}

/**
* 值减减操作,类似 --$i ,如果 key 不存在时自动设置为 0 后进行减减操作
*
* @access $rc->decr($key, $default=1);
* @param string $key 缓存KEY
* @param int $default 操作时的默认值
* @return int　 操作后的值
*/
public function decr($key, $default = 1) {
if ($default == 1) {
return $this->getRedis()->decr($key);
}
else {
return $this->getRedis()->decrBy($key, $default);
}
}

/* =================== 以下私有方法 =================== */

/**
* 随机 HASH 得到 Redis Slave 服务器句柄
* @access $this->_getSlaveRedis();
* @return redis object
*/
private function _getSlaveRedis() {
// 就一台 Slave 机直接返回
if ($this->_sn <= 1) {
return $this->_linkHandle['slave'][0];
}
// 随机 Hash 得到 Slave 的句柄
$hash = $this->_hashId(mt_rand(), $this->_sn);
return $this->_linkHandle['slave'][$hash];
}

/**
* 根据ID得到 hash 后 0～m-1 之间的值
* @access $this->_hashId($id, $m=10);
* @param string $id
* @param int $m
* @return int
*/
private function _hashId($id, $m = 10) {
//把字符串K转换为 0～m-1 之间的一个值作为对应记录的散列地址
$k = md5($id);
$md5_len = 32;
$b = bin2hex($k);
$h = 0;
for ($i = 0; $i < $md5_len; $i++) {
//相加模式HASH
$h += substr($b, $i * 2, 2);
}
$hash = ($h * 1) % $m;
return $hash;
}

}