hyperf-chat/app/Cache/Repository/ZSetRedis.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';

    private $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());
    }
}