170 lines
4.0 KiB
PHP
170 lines
4.0 KiB
PHP
<?php
|
||
|
||
namespace App\Cache\Repository;
|
||
|
||
use App\Cache\Contracts\ZSetRedisInterface;
|
||
|
||
/**
|
||
* Class ZSetRedis
|
||
*
|
||
* @package App\Cache\Repository
|
||
*/
|
||
class ZSetRedis implements ZSetRedisInterface
|
||
{
|
||
use RedisTrait;
|
||
|
||
private $prefix = 'rds:zset';
|
||
|
||
public $name = 'default';
|
||
|
||
/**
|
||
* 添加有序集合元素
|
||
*
|
||
* @param string $member
|
||
* @param float $score
|
||
* @return int
|
||
*/
|
||
public function add(string $member, float $score)
|
||
{
|
||
return $this->redis()->zAdd($this->getKeyName(), $score, $member);
|
||
}
|
||
|
||
/**
|
||
* 删除有序集合元素
|
||
*
|
||
* @param string ...$member
|
||
* @return int
|
||
*/
|
||
public function rem(string ...$member)
|
||
{
|
||
return $this->redis()->zRem($this->getKeyName(), ...$member);
|
||
}
|
||
|
||
/**
|
||
* 给指定元素累加分数
|
||
*
|
||
* @param string $member 元素
|
||
* @param float $score
|
||
* @return float
|
||
*/
|
||
public function incr(string $member, float $score)
|
||
{
|
||
return $this->redis()->zIncrBy($this->getKeyName(), $score, $member);
|
||
}
|
||
|
||
/**
|
||
* 获取有序集合元素总数
|
||
*
|
||
* @return int
|
||
*/
|
||
public function count()
|
||
{
|
||
return $this->redis()->zCard($this->getKeyName());
|
||
}
|
||
|
||
/**
|
||
* 获取有序集合所有元素
|
||
*
|
||
* @param bool $asc [true:从高到低排序，false:从低到高排序]
|
||
* @param bool $is_score 是否需要分数值
|
||
* @return array
|
||
*/
|
||
public function all($asc = true, $is_score = true)
|
||
{
|
||
return $this->redis()->{$asc ? 'zRevRange' : 'zRange'}($this->getKeyName(), 0, -1, $is_score);
|
||
}
|
||
|
||
/**
|
||
* 获取排行榜单
|
||
*
|
||
* @param int $page 分页
|
||
* @param int $size 分页大小
|
||
* @param bool $asc [true:从高到低排序，false:从低到高排序]
|
||
* @return array
|
||
*/
|
||
public function rank($page = 1, $size = 10, $asc = true)
|
||
{
|
||
$count = $this->count();
|
||
|
||
[$start, $end] = $asc ? ['+inf', '-inf'] : ['-inf', '+inf'];
|
||
|
||
$rows = $this->redis()->{$asc ? 'zRevRangeByScore' : 'zRangeByScore'}($this->getKeyName(), $start, $end, [
|
||
'withscores' => true,
|
||
'limit' => [($page - 1) * $size, $size]
|
||
]);
|
||
|
||
$ranks = [];
|
||
foreach ($rows as $node => $score) {
|
||
$ranks[] = [
|
||
'rank' => $this->getMemberRank($node, $asc),
|
||
'node' => $node,
|
||
'score' => $score
|
||
];
|
||
}
|
||
|
||
return [
|
||
'count' => $count,
|
||
'page' => $page,
|
||
'size' => $size,
|
||
'ranks' => $ranks
|
||
];
|
||
}
|
||
|
||
/**
|
||
* 获取指定区间分数
|
||
*
|
||
* @param string $start 最低分值
|
||
* @param string $end 最高分值
|
||
* @param array $options
|
||
* @return array
|
||
*/
|
||
public function range(string $start, string $end, array $options = [])
|
||
{
|
||
return $this->redis()->zRangeByScore($this->getKeyName(), $start, $end, $options);
|
||
}
|
||
|
||
/**
|
||
* 获取指定元素的排名
|
||
*
|
||
* @param string $member 元素
|
||
* @param bool $asc [true:从高到低排序，false:从低到高排序]
|
||
* @return false|int
|
||
*/
|
||
public function getMemberRank(string $member, $asc = true)
|
||
{
|
||
return $this->redis()->{$asc ? 'zRevRank' : 'zRank'}($this->getKeyName(), $member) + 1;
|
||
}
|
||
|
||
/**
|
||
* 获取指定元素的分数
|
||
*
|
||
* @param string $member 元素
|
||
* @return bool|float
|
||
*/
|
||
public function getMemberScore(string $member)
|
||
{
|
||
return $this->redis()->zScore($this->getKeyName(), $member);
|
||
}
|
||
|
||
/**
|
||
* 判断是否是集合元素
|
||
*
|
||
* @param string $member
|
||
* @return bool
|
||
*/
|
||
public function isMember(string $member)
|
||
{
|
||
return $this->redis()->zScore($this->getCacheKey($this->name), $member);
|
||
}
|
||
|
||
/**
|
||
* 删除 ZSet 有序集合表
|
||
*
|
||
* @return bool
|
||
*/
|
||
public function delete()
|
||
{
|
||
return (bool)$this->redis()->del($this->getKeyName());
|
||
}
|
||
}
|