think-swoole使用教程
核心思想是swoole只是作为一个消息转发器,业务逻辑还是通过接口来实现,发送消息也是使用接口,客户端websocket只负责创建和监听接受消息即可。
环境
- centos8
- PHP7.4
- thinkphp6.0.10
- think-swoole4.0.6
开发过程
- 安装think-swoole扩展
- 为了方便我们安装think-view扩展
配置swoole.php文件
- server.host 服务器IP
- server.port 服务器端口
- server.options.daemonize 是否进程
- websocket.enable 打开websocket
- websocket.handle 自己接管或者使用默认(默认的会给我们发送socket消息,不理会即可)
- websocket.subscribe 创建事件订阅,我这里的文件名是WebSocketEvent(也可以使用监听,只不过需要多个文件)
- 因为是多进程,我们需要共享变量,可以用MySQL、redis等,我们这里使用swoole的共享内容Table,因为同一个用户可能是多端登录,我们创建俩个Table,一个是用户映射fd,一个是fd映射用户,Table的映射是一对一的,但是一个用户可能有多个fd,所以用户映射fd的Table的值使用逗号分隔的多个值,例如用户1->fd1,fd2
配置tables俩个table,分别是m2fd、fd2m,thinkphp实现的Table如何使用请自己看代码
'tables' => [ 'm2fd' => [ 'size' => 102400, 'columns' => [ ['name' => 'fd', 'type' => \Swoole\Table::TYPE_STRING, 'size' => 50] ] ], 'fd2m' => [ 'size' => 102400, 'columns' => [ ['name' => 'member_id', 'type' => \Swoole\Table::TYPE_INT] ] ], ],
- 通过订阅实现websocket逻辑
- 把我们需要使用的类通过构造函数依赖注入,方便使用
- 我们需要WebSocket类实现通信逻辑,Table类实现用户fd映射
- 如果我们使用了type为11的绑定方式,则订阅open事件,发送给客户端
- message事件方法体留空或者不写即可,我们使用接口来实现逻辑
- close事件移除用户和fd的映射关系
我们定义一个事件,用于接口触发,从而实现发送消息逻辑,事件名称叫做ApiEvent,代码如下
<?php declare (strict_types = 1); namespace app\subscribe; use app\Request; use Swoole\Server; use think\swoole\Table; use think\swoole\Websocket; class WebSocketEvent { private $websocket = null; private $m2fd = null; private $fd2m = null; public function __construct(Websocket $websocket, Table $table) { $this->websocket = $websocket; $this->m2fd = $table->get('m2fd'); $this->fd2m = $table->get('fd2m'); } // 这里之所以注入一个请求,是因为如果我们不用type=11这种方式绑定,则可以通过new WebSocket的时候把用户ID传递过来,然后直接实现绑定 public function onOpen(Request $request) { $currentFd = $this->websocket->getSender(); $data = [ 'type' => 11, 'fd' => $currentFd ]; $this->websocket->push(json_encode($data)); } public function onClose() { $currentFd = $this->websocket->getSender(); // 通过fd找到用户ID $memberId = $this->fd2m->get((string)$currentFd, 'member_id'); // 如果没有找到映射,就说明没有绑定过,就什么不做,找到的话就解除绑定 if ($memberId) { $this->fd2m->del((string)$currentFd); // 根据用户ID找到映射的所有fd,然后把存在的当前fd移除掉 $fds = $this->m2fd->get((string)$memberId, 'fd'); if ($fds) { $fdArray = explode(',', $fds); $key = array_search($currentFd, $fdArray); unset($fdArray[$key]); if ($fdArray) { $resFds = implode(',', $fdArray); $this->m2fd->set((string)$memberId, $resFds); } else { $this->m2fd->del((string)$memberId); } } } } public function onApiEvent($data) { // $data是接口传递过来的参数,如果是11则实现绑定,是5就转发给from_id和to_id if ($data['type'] == 11) { // m2fd、fd2m俩个Table的映射 $this->fd2m->set((string)$data['fd'], ['member_id' => $data['member_id']]); // 先查找该用户ID是否已经绑定过其它fd了 $fds = $this->m2fd->get((string)$data['member_id'], 'fd'); if (!$fds) { $this->m2fd->set((string)$data['member_id'], ['fd' => $data['fd']]); } else { // 看看fd是否在已经映射的fd中,如果在就什么都不做,如果不在就追加到后面 $fdArray = explode(',', $fds); if (!in_array($data['fd'], $fdArray)) { $this->m2fd->set((string)$data['member_id'], ['fd' => $fds . ',' . $data['fd']]); } } } if ($data['type'] == 1) { // 根据from_id和to_id俩个用户ID找到对应的fd,然后发送消息 $fromFds = $this->m2fd->get((string)$data['from_id'], 'fd'); $toFds = $this->m2fd->get((string)$data['to_id'], 'fd'); $fromFdArray = $toFdArray = []; if ($fromFds) { $fromFdArray = explode(',', $fromFds); } if ($toFds) { $toFdArray = explode(',', $toFds); } // 合并所有发送者fd和接受者fd,之所以发送给发送者,一方面是简化前端工作,前端只需要接受websocket消息即可,另一方面,多端的话其它端可以可以即时看到聊天记录 $allFdArray = array_unique(array_merge($fromFdArray, $toFdArray)); // 发送消息 $this->websocket->to($allFdArray)->push(json_encode($data)); } } }
接口实现代码如下
<?php namespace app\controller; class Index { // 为了演示方便我们不使用路由了,使用控制器方法的方式访问 public function index() { // 聊天页面 return view(); } // 如果使用type=1的绑定方式就要,否则这个可以不要 public function bindMember() { // 用户ID本来是要从登录状态中获取的,这里我们是模拟演示,就让前端传 $params = request()->only(['member_id', 'fd']); // 触发ApiEvent事件,组装数据type=11、member_id、fd $data = $params; $data['type'] = 11; event('swoole.websocket.ApiEvent', $data); // 只要不抛异常就是绑定成功了 $res = [ 'code' => 1, 'msg' => '绑定成功' ]; return json($res); } // 我们只写一个发送文字消息的例子 public function sendMessage() { // 其它需要的字段我们就不写了,自己实现即可,持久化到数据库逻辑也不写了,只是写消息发送 $params = request()->only(['from_id', 'to_id', 'content']); $data = $params; $data['type'] = 1; event('swoole.websocket.ApiEvent', $data); // 不抛异常就是成功了 $res = [ 'code' => 1, 'msg' => '发送消息成功' ]; return json($res); } }
聊天页面代码
<!doctype html> <html lang="en"> <head> <meta charset="UTF-8"> <meta name="viewport" content="width=device-width, user-scalable=no, initial-scale=1.0, maximum-scale=1.0, minimum-scale=1.0"> <meta http-equiv="X-UA-Compatible" content="ie=edge"> <title>聊天页面</title> <style> #chat { height: 400px; width: 400px; background: gray; } </style> </head> <body> <div id="chat"></div> <button id="connect">链接websocket服务端</button> <input type="text" id="content" value="内容" placeholder="聊天内容"> <input type="text" id="to" placeholder="目标对象" value="1"> <input type="text" id="from" placeholder="发送对象" value="1"> <button id="submit">发送</button> <script src="https://cdn.bootcdn.net/ajax/libs/jquery/3.6.0/jquery.min.js"></script> <script> var ws = null; // 链接websocket $("#connect").click(function () { // TODO 改成自己的域名即可 ws = new WebSocket("ws://swoole.dreamphp.com.cn:8282"); ws.onmessage = function (res) { var data = JSON.parse(res.data); // 如果没有接管Handler,则消息格式就跟我们的不一样,可能会报错,不用理会的 if (data.type == 11) { $.ajax({ url: "{:url('index/bindMember')}", type: "post", data: {member_id: $("#from").val(), fd: data.fd}, dataType: "json", success: function (res) { console.log(res); } }); } if (data.type == 1) { $("#chat").append("用户" + data.from_id + ":" + data.content + "<br>"); } }; return false; }); // 发送内容 $("#submit").click(function () { var toId = $("#to").val(); var fromId = $("#from").val(); var content = $("#content").val(); $.ajax({ url: "{:url('index/sendMessage')}", type: "post", data: {from_id: fromId, to_id: toId, content: content}, dataType: "json", success: function (res) { console.log(res) } }); return false; }); </script> </body> </html>
type说明(type为1可以去掉,可以换成new websocket的时候就直接绑定)
type | 说明 | 额外说明 |
---|---|---|
11 | 通知用户要绑定了 | fd |
1 | 聊天消息 | 消息类型详细说明 |
**粗体** _斜体_ [链接](http://example.com) `代码` - 列表 > 引用
。你还可以使用@
来通知其他用户。