【Swoole系列2.6】Redis 服务器

2022年7月21日 285点热度 0人点赞 0条评论

Redis 服务器

Redis 服务端可不是我们说的去连接 Reids 服务器的那个东西,那个叫做 PHP 的 Reids 客户端。服务端的意思是一个可以提供服务的应用,redis-server 才是我们最熟悉的那个 Redis 的服务端。

那么在 Swoole 中,这个 Redis 服务端是个什么东西呢?其实它是一个基于 Redis 协议的服务器程序,可以让我们使用 Redis 的客户端来连接这个服务。光这么说估计大家还是一脸蒙圈,我们直接来看看效果再说。

服务端

use Swoole\Redis\Server;

define('DB_FILE'__DIR__ . '/db');

$server = new Server("0.0.0.0"9501, SWOOLE_BASE);

if (is_file(DB_FILE)) {
    $server->data = unserialize(file_get_contents(DB_FILE));
else {
    $server->data = array();
}

$server->setHandler('GET'function ($fd, $data) use ($server) {
    if (count($data) == 0) {
        return $server->send($fd, Server::format(Server::ERROR, "ERR wrong number of arguments for 'GET' command"));
    }

    $key = $data[0];
    if (empty($server->data[$key])) {
        return $server->send($fd, Server::format(Server::NIL));
    } else {
        return $server->send($fd, Server::format(Server::STRING, $server->data[$key]));
    }
});

$server->setHandler('SET'function ($fd, $data) use ($server) {
    if (count($data) < 2) {
        return $server->send($fd, Server::format(Server::ERROR, "ERR wrong number of arguments for 'SET' command"));
    }

    $key = $data[0];
    $server->data[$key] = $data[1];
    return $server->send($fd, Server::format(Server::STATUS, "OK"));
});

$server->setHandler('sAdd'function ($fd, $data) use ($server) {
    if (count($data) < 2) {
        return $server->send($fd, Server::format(Server::ERROR, "ERR wrong number of arguments for 'sAdd' command"));
    }

    $key = $data[0];
    if (!isset($server->data[$key])) {
        $array[$key] = array();
    }

    $count = 0;
    for ($i = 1; $i < count($data); $i++) {
        $value = $data[$i];
        if (!isset($server->data[$key][$value])) {
            $server->data[$key][$value] = 1;
            $count++;
        }
    }

    return $server->send($fd, Server::format(Server::INT, $count));
});

$server->setHandler('sMembers'function ($fd, $data) use ($server) {
    if (count($data) < 1) {
        return $server->send($fd, Server::format(Server::ERROR, "ERR wrong number of arguments for 'sMembers' command"));
    }
    $key = $data[0];
    if (!isset($server->data[$key])) {
        return $server->send($fd, Server::format(Server::NIL));
    }
    return $server->send($fd, Server::format(Server::SET, array_keys($server->data[$key])));
});

$server->setHandler('hSet'function ($fd, $data) use ($server) {
    if (count($data) < 3) {
        return $server->send($fd, Server::format(Server::ERROR, "ERR wrong number of arguments for 'hSet' command"));
    }

    $key = $data[0];
    if (!isset($server->data[$key])) {
        $array[$key] = array();
    }
    $field = $data[1];
    $value = $data[2];
    $count = !isset($server->data[$key][$field]) ? 1 : 0;
    $server->data[$key][$field] = $value;
    return $server->send($fd, Server::format(Server::INT, $count));
});

$server->setHandler('hGetAll'function ($fd, $data) use ($server) {
    if (count($data) < 1) {
        return $server->send($fd, Server::format(Server::ERROR, "ERR wrong number of arguments for 'hGetAll' command"));
    }
    $key = $data[0];
    if (!isset($server->data[$key])) {
        return $server->send($fd, Server::format(Server::NIL));
    }
    return $server->send($fd, Server::format(Server::MAP, $server->data[$key]));
});

$server->on('WorkerStart'function ($server) {
    $server->tick(10000function () use ($server) {
        file_put_contents(DB_FILE, serialize($server->data));
    });
});

$server->start();

查看上面的代码,我们会发现这也是一个 Server 对象。然后主要是使用 setHandler() 方法来监听 Reids 命令,在这里我们看到了熟悉的 get、set 等命令的定义。然后我们指定了 $server->data ,可以将它看成是一个数据源,直接使用的就是一个文件,直接在当前测试环境目录下创建一个叫做 db 的空文件就可以了。

在 setHandler() 方法中,我们使用 send() 方法来返回响应的命令信息,并通过 format() 方法格式化返回的响应数据。

最后,通过一个监听一个 WorkerStart 事件,设置了一个每隔一秒将数据写入到 db 文件的操作。

客户端测试

现在,运行起来这个 Swoole 写的 Redis 服务端文件吧,然后使用你的 redis-cli 来连接并测试它。

图片

神奇吗?惊喜吗?我们竟然实现了一个 Redis 服务器,再看看 db 文件中是什么内容。

图片

不出所料,db 中保存的是我们序列化之后的内容。这个东西是不是很有意思,完全可以做一个我们自己的小 Redis 来用。对于一些小网站,小应用来说,你不需要再去安装一个庞大的 Redis 服务了,直接使用 Swoole 就可以实现一个遵循 Redis 协议规范的小型缓存服务器了。

添加一个命令

话说回来,注意到上面测试的截图中我们使用了 keys 命令却返回了一个 unknow command 吗?因为在这里,所有的命令我们都要在 setHandler() 中一个一个的定义。相对来说,这东西写起来的还是比较麻烦的。最后,我们再来简单地实现一个 keys 命令好了。

$server->setHandler('KEYS'function ($fd, $data) use ($server) {
    if (count($data) == 0) {
        return $server->send($fd, Server::format(Server::ERROR, "ERR wrong number of arguments for 'GET' command"));
    }

    if(!$server->data){
        return $server->send($fd, Server::format(Server::NIL));
    }

    $key = $data[0];
    if($key == "*"){
        return $server->send($fd, Server::format(Server::SET, array_keys($server->data)));
    }else{
        $dataKeys = array_keys($server->data);
        $key = str_replace(["*""?"], [".*"".?"], $key);
        $values = array_filter($dataKeys, function($value) use ($key){
            return preg_match('/'.$key.'/i', $value, $mchs);
        });
        if($values){
            return $server->send($fd, Server::format(Server::SET, $values));
        }
    }
    return $server->send($fd, Server::format(Server::NIL));
});

并不是很规范,但功能是没问题的,大家可以测试一下效果。

图片

总结

关于 Redis 服务器这一块的内容,我们了解一下就好了,实际的使用中自己去写这些东西还是挺费劲的。到这里为止,我们的入门相关课程就学习完了。

大家还记得讲了什么吗?就是简单地搭起了 Http/TCP/UDP/WebSocket/Redis 这几个服务。通过这几个服务,我们日常的 Web 应用其实就已经全部都可以覆盖了。只是说,我们都是使用的原生的 Swoole 对象及函数来实现的非常简单的测试代码。真正在实际的项目中使用的话,一般还是会借助框架来实现。

那么我们就继续进阶方面的内容,到最后才会简单地学习了解一个 Swoole 框架。别急,基础才是王道嘛!

测试代码:

https://github.com/zhangyue0503/swoole/blob/main/2.%E5%9F%BA%E7%A1%80/source/2.6Redis%E6%9C%8D%E5%8A%A1%E5%99%A8.php

参考文档:

https://wiki.swoole.com/#/redis_server

49770【Swoole系列2.6】Redis 服务器

这个人很懒,什么都没留下

文章评论