百度360必应搜狗淘宝本站头条
当前位置:网站首页 > 技术教程 > 正文

如何用PHP自己实现一个redis client

mhr18 2024-10-21 05:47 23 浏览 0 评论

介绍

redis是一个经典的key-value缓存数据库,采用C/S架构。当我们安装成功以后,你就知道它有个服务端,启动后默认监听6379端口,然后提供一个客户端工具redis-cli。我们可以使用redis-cli然后书写命令与服务端通信。

上面我们大概知道了redis的工作模式,为了更好的认知它,我就开始思考如何自己去连接服务端呢?我想到使用Yii2时,用到redis我是没有安装官方提供的redis扩展,但是它仍然可以与redis服务端通信,于是乎便去追踪了Yii2-redis组件的源代码,看完以后,深感作者的强大。

准备

  1. 熟悉redis协议:http://redis.cn/topics/protocol
  2. Redis在TCP端口6379上监听到来的连接,客户端连接到来时,Redis服务器为此创建一个TCP连接。在客户端与服务器端之间传输的每个Redis命令或者数据都以\r\n结尾
  3. 新的统一协议已在Redis 1.2中引入,但是在Redis 2.0中,这就成为了与Redis服务器通讯的标准方式。

代码实现

<?php


/**
 * Class SocketException
 */
class SocketException extends \Exception
{
    /**
     * @var array
     */
    public $errorInfo = [];

    /**
     * @return string
     */
    public function getName()
    {
        return 'Redis Socket Exception';
    }

    /**
     * SocketException constructor.
     * @param string $message
     * @param array $errorInfo
     * @param int $code
     * @param Throwable|null $previous
     */
    public function __construct($message = "", $errorInfo = [], $code = 0, Throwable $previous = null)
    {
        $this->errorInfo = $errorInfo;
        parent::__construct($message, $code, $previous);
    }

    /**
     * @return string
     */
    public function __toString()
    {
        return parent::__toString() . PHP_EOL . 'Additional Information:' . PHP_EOL . print_r($this->errorInfo, true);; // TODO: Change the autogenerated stub
    }
}

/**
 * Class RedisConnection
 */
class RedisConnection
{
    /**
     * @var string
     */
    public $host = 'localhost';

    /**
     * @var int
     */
    public $port = 6379;

    /**
     * @var null
     */
    public $password = null;
    /**
     * @var int
     */
    public $database = 0;
    /**
     * @var
     */
    public $connectionTimeout;
    /**
     * @var null
     */
    public $dataTimeout = null;
    /**
     *
     * STREAM_CLIENT_ASYNC_CONNECT指示应打开每个后续连接,而不必等待上一个连接的完成
     * @var int
     */
    public $socketClientFlags = STREAM_CLIENT_CONNECT;

    /**
     * @var array
     */
    protected $pools = [];

    /**
     * @var int
     */
    protected $maxPoolSize = 10;

    /**
     * RedisConnection constructor.
     * @param string $host
     * @param int $port
     * @param null $password
     */
    public function __construct($host = 'localhost', $port = 6379, $password = null)
    {
        $this->host = $host;
        $this->port = $port;
        $this->password = $password;
    }

    /**
     * @return string
     */
    public function getConnectionString()
    {
        return 'tcp://' . $this->host . ':' . $this->port;
    }

    /**
     * @param int $database
     */
    public function connect($database = 0)
    {
        $this->database = $database;
        $countSize = count($this->pools);
        if ($countSize > $this->maxPoolSize) {
            return;
        }

        if ($this->getSocket() !== false) {
            return;
        }

        $connId = $this->getConnectionString();
        $connection = $connId . ', database=' . $this->database;
        try {
            $socket = stream_socket_client(
                $connId,
                $errorNumber,
                $errorDescription,
                $this->connectionTimeout ?? ini_get('default_socket_timeout'),
                $this->socketClientFlags
            );

            if ($socket) {
                $this->pools[$connId] = $socket;
                if ($this->dataTimeout !== null) {
                    $timeout = (int)$this->dataTimeout;
                    $microTimeout = (int)(($this->dataTimeout - $timeout) * 1000000);
                    stream_set_timeout($socket, $timeout, $microTimeout);
                }

                if ($this->password !== null) {
                    $this->exec('AUTH', [$this->password]);
                }

                if ($this->database !== null) {
                    $this->exec('SELECT', [$this->database]);
                }
            } else {
                $message = "无法打开redis数据库连接 ($connection): $errorNumber - $errorDescription";
                throw new Exception($message, $errorDescription, $errorNumber);
            }
        } catch (Exception $e) {
            exit($e->getMessage());
        }
    }


    /**
     * 用单行回复,回复的第一个字节将是“+”
     * 错误消息,回复的第一个字节将是“-”
     * 整型数字,回复的第一个字节将是“:”
     * 批量回复,回复的第一个字节将是“$”
     * 多个批量回复,回复的第一个字节将是“*”
     * 命令LRNGE需要返回多个值
     *      LRANGE mylist 0 3
     *      无值:*0
     *      有值:
     *           *3
     *           $1
     *           c
     *           $1
     *           b
     *           $1
     *           a
     *      sadd mylist a b c d
     *      :4
     *
     * @link  http://redis.cn/topics/protocol
     * @param $cmd
     * @param array $params
     */
    public function exec($cmd, $params = [])
    {
        //状态:-ERR Client sent AUTH, but no password is set
        //状态:+OK
        //get:   $15\r\nlemon1024026382\r\n
        //del: :1
        $params = array_merge(explode(' ', $cmd), $params);
        $command = '';
        $paramsCount = 0;
        foreach ($params as $param) {
            if ($param === null) {
                continue;
            }
            $command .= '#39; . mb_strlen($param, '8bit') . "\r\n" . $param . "\r\n";
            $paramsCount++;
        }
        $command = '*' . $paramsCount . "\r\n" . $command;
//        echo 'Executing Redis Command:', $cmd, PHP_EOL;
//        echo 'Yuan Shi Redis Cmd:', $command, PHP_EOL;
        return $this->send($command, $params);
    }


    /**
     * @param $cmd
     * @param $params
     * @return null
     * @throws SocketException
     */
    private function send($cmd, $params)
    {
        $socket = $this->getSocket();
        $written = fwrite($socket, $cmd);
        if ($written === false) {
            throw  new SocketException("无法写入到socket.\nRedis命令是: " . $cmd);
        }

        return $this->parseData($params, $cmd);
    }

    /**
     * 用单行回复,回复的第一个字节将是“+”
     * 错误消息,回复的第一个字节将是“-”
     * 整型数字,回复的第一个字节将是“:”
     * 批量回复,回复的第一个字节将是“$”
     * 多个批量回复,回复的第一个字节将是“*”
     * @link  http://redis.cn/topics/protocol
     *
     * \r\n作为分割符号,使用fgets按行读取
     * 如果是状态返回的命令或者错误返回或者返回整数这些相对简单处理
     * 批量返回时,如果是*号,第一步先读取返回的计数器,比如keys * 会返回有多少个key值,循环计数器
     * ,读取第二行就是$数字(批量回复),例如$9表示当前key的value值长度
     * 代码里有 $length = intval($line)+2; +2其实是表示\r\n这个分隔符
     * @param array $params
     * @throws SocketException
     * @return null|string|mixed
     */
    private function parseData($params)
    {
        $socket = $this->getSocket();
        $prettyCmd = implode(' ', $params);
        if (($line = fgets($socket)) === false) {// 从文件指针中读取一行,这里最合适,因为协议分割标识是\r\n
            throw new SocketException("无法从socket读取.\nRedis命令是: " . $prettyCmd);
        }

        echo '服务端响应数据:', $line, PHP_EOL;
        $type = $line[0];
        $line = mb_substr($line, 1, -2, '8bit');
        if ('+' === $type) {
            //  eg:SET ping select quit auth...
            if (in_array($line, ['Ok', 'PONG'])) {
                return true;
            } else {
                return $line;
            }
        }

        if ('-' === $type) {
            throw new SocketException("Redis 错误: " . $line . "\nRedis命令是: " . $prettyCmd);
        }

        // eg: hset zadd del
        if (':' === $type) {
            return $line;
        }

        /**
         * 例如:
         *     输入存在的key:get jie  返回:$9\r\nyangjiecheng\r\n
         *     输入:keys *   返回: *9\r\n$9\r\nchat.user\r\n$3\r\njie\r\n$4\r\ntest\r\n
         *     输入不存在的key: get 111 返回:$-1 对应redis返回:nil
         *
         */
        if ('#39; === $type) {
            if (-1 == $line) {
                return null;
            }

            $length = intval($line) + 2; //+2 表示后面的\r\n
            $data = '';
            while ($length > 0) {
                $str = fread($socket, $length);
                if ($str === false) {
                    throw new SocketException("无法从socket读取.\nRedis命令是: " . $prettyCmd);
                }
                $data .= $str;
                $length -= mb_strlen($str, '8bit');//此函数可以指定字符串编码并计算长度,8bit就是按照字节计算长度
            }

            return !empty($data) ? mb_substr($data, 0, -2, '8bit') : '';
        }

        if ('*' === $type) {
            $count = intval($line);
            $data = [];
            for ($i = 0; $i < $count; $i++) {
                $data[] = $this->parseData($params);
            }

            return $data;
        }

        throw new SocketException('从Redis读取到未能解析的数据: ' . $line . "\nRedis命令是: " . $prettyCmd);
    }

    /**
     * @return bool|mixed
     */
    public function getSocket()
    {
        $id = $this->getConnectionString();
        return isset($this->pools[$id]) ? $this->pools[$id] : false;
    }

    /**
     *
     */
    public function close()
    {
        $connectionString = $this->getConnectionString();
        foreach ($this->pools as $socket) {
            $connection = $connectionString . ', database=' . $this->database;
            echo 'Closing DB connection: ' . $connection . PHP_EOL;
            try {
                $this->exec('QUIT');
            } catch (SocketException $e) {
                // ignore errors when quitting a closed connection
            }
            fclose($socket);
        }

        $this->pools = [];
    }

    /**
     *
     */
    public function __destruct()
    {
        // TODO: Implement __destruct() method.
        $this->close();
    }
}

测试代码

/**
 * @param $file
 * @param $content
 */
function writeLog($file, $content)
{
    $str = '@@@@@@@@@@ Time Is ' . date('Y-m-d H:i:s') . ' @@@@@@@@@' . PHP_EOL;
    $str .= $content . PHP_EOL;
    $str .= '@@@@@@@@@@ End Block Log @@@@@@@@' . PHP_EOL;
    file_put_contents($file, $str, FILE_APPEND);
}

set_exception_handler(/**
 * @param Exception $exception
 */ function (\Exception $exception) {
    echoMsg($exception->getMessage());
});

/**
 * @param $msg
 * @param string $type
 */
function echoMsg($msg, $type = 'error')
{
    if ($type === 'error') {
        $msg = "\e[" . "1;37m" . $msg . "\e[0m";
    } else {
        $msg = "\e[" . "0;31m" . $msg . "\e[0m";
    }
    echo "\033[" . "41m", "Exception: ", $msg, "\033[0m", PHP_EOL;
}

set_error_handler(/**
 * @param $errno
 * @param $errstr
 * @param $errfile
 * @param $errline
 */ function ($errno, $errstr, $errfile, $errline){
    echoMsg("custom error:[$errno]  $errstr");
    echoMsg(" Error on line $errline in $errfile");
});
//register_shutdown_function
$redis = new RedisConnection();
$redis->connect();
echo $redis->getConnectionString(), PHP_EOL;
//$redis->exec('SET', ['RedisClient', 'OK']);
//$redis->exec('DEL', ['RedisClient']);
//$redis->exec('KEYS', ['*']);
var_dump($redis->exec('GET', ['access_token']));
//var_dump($redis->exec('LRANGE', ['qlist', 0, 3]));

相关推荐

Redis合集-使用benchmark性能测试

采用开源Redis的redis-benchmark工具进行压测,它是Redis官方的性能测试工具,可以有效地测试Redis服务的性能。本次测试使用Redis官方最新的代码进行编译,详情请参见Redis...

Java简历总被已读不回?面试挂到怀疑人生?这几点你可能真没做好

最近看了几十份简历,发现大部分人不是技术差,而是不会“卖自己”——一、简历死穴:你写的不是经验,是岗位说明书!反面教材:ד使用SpringBoot开发项目”ד负责用户模块功能实现”救命写法:...

redission YYDS(redission官网)

每天分享一个架构知识Redission是一个基于Redis的分布式Java锁框架,它提供了各种锁实现,包括可重入锁、公平锁、读写锁等。使用Redission可以方便地实现分布式锁。red...

从数据库行锁到分布式事务:电商库存防超卖的九重劫难与破局之道

2023年6月18日我们维护的电商平台在零点刚过3秒就遭遇了严重事故。监控大屏显示某爆款手机SKU_IPHONE13_PRO_MAX在库存仅剩500台时,订单系统却产生了1200笔有效订单。事故复盘发...

SpringBoot系列——实战11:接口幂等性的形而上思...

欢迎关注、点赞、收藏。幂等性不仅是一种技术需求,更是数字文明对确定性追求的体现。在充满不确定性的网络世界中,它为我们建立起可依赖的存在秩序,这或许正是技术哲学最深刻的价值所在。幂等性的本质困境在支付系...

如何优化系统架构设计缓解流量压力提升并发性能?Java实战分享

如何优化系统架构设计缓解流量压力提升并发性能?Java实战分享在高流量场景下。首先,我需要回忆一下常见的优化策略,比如负载均衡、缓存、数据库优化、微服务拆分这些。不过,可能还需要考虑用户的具体情况,比...

Java面试题: 项目开发中的有哪些成长?该如何回答

在Java面试中,当被问到“项目中的成长点”时,面试官不仅想了解你的技术能力,更希望看到你的问题解决能力、学习迭代意识以及对项目的深度思考。以下是回答的策略和示例,帮助你清晰、有说服力地展示成长点:一...

互联网大厂后端必看!Spring Boot 如何实现高并发抢券逻辑?

你有没有遇到过这样的情况?在电商大促时,系统上线了抢券活动,结果活动刚一开始,服务器就不堪重负,出现超卖、系统崩溃等问题。又或者用户疯狂点击抢券按钮,最后却被告知无券可抢,体验极差。作为互联网大厂的后...

每日一题 |10W QPS高并发限流方案设计(含真实代码)

面试场景还原面试官:“如果系统要承载10WQPS的高并发流量,你会如何设计限流方案?”你:“(稳住,我要从限流算法到分布式架构全盘分析)…”一、为什么需要限流?核心矛盾:系统资源(CPU/内存/数据...

Java面试题:服务雪崩如何解决?90%人栽了

服务雪崩是指微服务架构中,由于某个服务出现故障,导致故障在服务之间不断传递和扩散,最终造成整个系统崩溃的现象。以下是一些解决服务雪崩问题的常见方法:限流限制请求速率:通过限流算法(如令牌桶算法、漏桶算...

面试题官:高并发经验有吗,并发量多少,如何回复?

一、有实际高并发经验(建议结构)直接量化"在XX项目中,系统日活用户约XX万,核心接口峰值QPS达到XX,TPS处理能力为XX/秒。通过压力测试验证过XX并发线程下的稳定性。"技术方案...

瞬时流量高并发“保命指南”:这样做系统稳如泰山,老板跪求加薪

“系统崩了,用户骂了,年终奖飞了!”——这是多少程序员在瞬时大流量下的真实噩梦?双11秒杀、春运抢票、直播带货……每秒百万请求的冲击,你的代码扛得住吗?2025年了,为什么你的系统一遇高并发就“躺平”...

其实很多Java工程师不是能力不够,是没找到展示自己的正确姿势。

其实很多Java工程师不是能力不够,是没找到展示自己的正确姿势。比如上周有个小伙伴找我,五年经验但简历全是'参与系统设计''优化接口性能'这种空话。我就问他:你做的秒杀...

PHP技能评测(php等级考试)

公司出了一些自我评测的PHP题目,现将题目和答案记录于此,以方便记忆。1.魔术函数有哪些,分别在什么时候调用?__construct(),类的构造函数__destruct(),类的析构函数__cal...

你的简历在HR眼里是青铜还是王者?

你的简历在HR眼里是青铜还是王者?兄弟,简历投了100份没反应?面试总在第三轮被刷?别急着怀疑人生,你可能只是踩了这些"隐形求职雷"。帮3630+程序员改简历+面试指导和处理空窗期时间...

取消回复欢迎 发表评论: