Swoole RPC 的实现

概述

这是关于 Swoole 学习的第七篇文章：Swoole RPC 的实现。

第六篇：Swoole 整合成一个小框架
第五篇：Swoole 多协议多端口的应用
第四篇：Swoole HTTP 的应用
第三篇：Swoole WebSocket 的应用
第二篇：Swoole Task 的应用
第一篇：Swoole Timer 的应用

有位读者说 “上篇文章，下载代码后直接运行成功，代码简洁明了，简直是 Swoole 入门最好的 Demo ”。

“哈哈哈...”

还有读者说 “有一起学习的组织群吗，可以在里面进行疑难答疑？”

这个还真没有，总觉得维护一个微信群不容易，因为自己本身就不爱在群里说话，另外，自己也在很多微信群中，开始氛围挺好的，大家都聊聊技术，后来技术聊的少了改成聊八卦啦，再后来慢慢就安静了，还有在群里起冲突的...

当然我也知道维护一个微信群的好处是非常大的，如果有这方面经验的同学，咱们一起交流交流 ~

还有出版社找我写书的.

他们也真是放心，我自己肚子里几滴墨水还是知道的，目前肯定是不行，以后嘛，再说。

还有一些大佬加了微信，可能是出于对晚辈的提携吧，偷偷告诉你，从大佬的朋友圈能学到很多东西。

我真诚的建议，做技术的应该自己多总结总结，将自己会的东西写出来分享给大家，先不说给别人带来太多的价值，反正对自己的帮助是非常非常大的，这方面想交流的同学，可以加我，我可以给你无私分享。

可能都会说时间少，时间只要挤，总会有的，每个人都 24 小时，时间对每个人是最公平的。说到这推荐大家读一下《暗时间》这本书，这是我这本书的读书笔记，大家可以瞅瞅。

开始今天的文章吧，这篇文章实现了一个简单的 RPC 远程调用，在实现之前需要先了解什么是 RPC，不清楚的可以看下之前发的这篇文章《我眼中的 RPC》。

下面的演示代码主要使用了 Swoole 的 Task 任务池，通过 OnRequest/OnReceive 获得信息交给 Task 去处理。

举个工作中的例子吧，在电商系统中的两个模块，个人中心模块和订单管理模块，这两个模块是独立部署的，可能不在一个机房，可能不是一个域名，现在个人中心需要通过用户ID 和订单类型获取订单数据。

实现效果

客户端

HTTP 请求

//代码片段
 'SW',
    'token' => 'Bb1R3YLipbkTp5p0',
    'param' => [
        'class'  => 'Order',
        'method' => 'get_list',
        'param' => [
            'uid'  => 1,
            'type' => 2,
        ],
    ],
];

$ch = curl_init();
$options = [
    CURLOPT_URL  => 'http://10.211.55.4:9509/',
    CURLOPT_POST => 1,
    CURLOPT_POSTFIELDS => json_encode($demo),
];
curl_setopt_array($ch, $options);
curl_exec($ch);
curl_close($ch);

TCP 请求

//代码片段
$demo = [
    'type'  => 'SW',
    'token' => 'Bb1R3YLipbkTp5p0',
    'param' => [
        'class'  => 'Order',
        'method' => 'get_list',
        'param' => [
            'uid'  => 1,
            'type' => 2,
        ],
    ],
];
$this->client->send(json_encode($demo));

请求方式

SW 单个请求，等待结果
发出请求后，分配给 Task ，并等待 Task 执行完成后，再返回。
SN 单个请求，不等待结果
发出请求后，分配给 Task 之后，就直接返回。

发送数据

$demo = [
    'type'  => 'SW',
    'token' => 'Bb1R3YLipbkTp5p0',
    'param' => [
        'class'  => 'Order',
        'method' => 'get_list',
        'param' => [
            'uid'  => 1,
            'type' => 2,
        ],
    ],
];

type 同步/异步设置
token 可进行权限验证
class 请求的类名
method 请求的方法名
uid 参数一
type 参数二

返回数据

request_method 请求方式
request_time 请求开始时间
response_time 请求结束时间
code 标识
msg 标识值
data 约定数据
query 请求参数

代码

OnRequest.php

rawContent());
            self::$query   = $data;
            if (empty($data)) {
                self::$code = '-1';
                self::$msg  = '非法请求';
                self::end($request, $response);
            }

            //TODO 验证Token

            switch ($data['type']) {
                case 'SW': //单个请求,等待结果
                    $task = [
                        'request' => $data,
                        'server'  => 'http'
                    ];
                    $rs = $serv->task(json_encode($task), -1, function ($serv, $task_id, $rs_data) use ($request, $response) {
                        self::$code = '1';
                        self::$msg  = '成功';
                        self::$data = $rs_data['response'];
                        self::end($request, $response);
                    });
                    if ($rs === false) {
                        self::$code = '-1';
                        self::$msg  = '失败';
                        self::end($request, $response);
                    }
                    break;

                case 'SN': //单个请求,不等待结果
                    $task = [
                        'request' => $data,
                        'server'  => 'http'
                    ];
                    $rs = $serv->task(json_encode($task));
                    if ($rs === false) {
                        self::$code = '-1';
                        self::$msg  = '失败';
                    } else {
                        self::$code = '1';
                        self::$msg  = '成功';
                    }
                    self::end($request, $response);
                    break;
                default:
                    self::$code = '-1';
                    self::$msg  = '非法请求';
                    self::end($request, $response);
            }
        } catch(Exception $e) {
        }
    }

    private static function end($request = null, $response = null)
    {
        $rs['request_method'] = $request->server['request_method'];
        $rs['request_time']   = $request->server['request_time'];
        $rs['response_time']  = time();
        $rs['code']           = self::$code;
        $rs['msg']            = self::$msg;
        $rs['data']           = self::$data;
        $rs['query']          = self::$query;
        $response->end(json_encode($rs));
        self::$data = [];
        return;
    }
}

OnReceive.php

 $fd,
                        'request'      => $data,
                        'server'       => 'tcp',
                        'request_time' => self::$request_time,
                    ];
                    $rs = $serv->task(json_encode($task));
                    if ($rs === false) {
                        self::$code = '-1';
                        self::$msg  = '失败';
                        self::handlerTask($serv, $fd);
                    }
                    break;

                case 'SN': //单个请求,不等待结果
                    $task = [
                        'fd'           => $fd,
                        'request'      => $data,
                        'server'       => 'tcp',
                        'request_time' => self::$request_time,
                    ];
                    $rs = $serv->task(json_encode($task));
                    if ($rs === false) {
                        self::$code = '-1';
                        self::$msg  = '失败';
                    } else {
                        self::$code = '1';
                        self::$msg  = '成功';
                    }
                    self::handlerTask($serv, $fd);
                    break;
                default:
                    self::$code = '-1';
                    self::$msg  = '非法请求';
                    self::handlerTask($serv, $fd);
            }
        } catch(Exception $e) {
        }
    }

    private static function handlerTask($serv, $fd)
    {
        $rs['request_method'] = 'TCP';
        $rs['request_time']   = self::$request_time;
        $rs['response_time']  = time();
        $rs['code']           = self::$code;
        $rs['msg']            = self::$msg;
        $rs['data']           = self::$data;
        $rs['query']          = self::$query;
        $serv->send($fd, json_encode($rs));
    }
}

小结

Demo 代码仅供参考，里面有很多不严谨的地方！

服务的调用方与提供方中间需要有一个服务注册中心，很显然上面的代码中没有，需要自己去实现。

服务注册中心，负责管理 IP、Port 信息，提供给调用方使用，还要能负载均衡和故障切换。

根据自己的情况，服务注册中心实现可容易可复杂，用 Redis 也行，用 Zookeeper、Consul 也行。

感兴趣的也可以了解下网关 Kong ，包括身份认证、权限认证、流量控制、监控预警...

再推荐一个 Swoole RPC 框架 Hprose，支持多语言。

就到这了，上面的 Demo 需要源码的，加我微信。（菜单-> 加我微信-> 扫我）