Tp6 WebSocket加入离开房间

Think-Swoole 3.0 中 Websocket 新增了 Room 聊天室功能，它主要用于群发消息，但不同Room之间的消息又是相互隔离的。当我们进入一个聊天室，那么我们的进入、离开以及发送的消息只有这个聊天室的 fd 能接收到。

config.swoole.php

'websocket'  => [
        'enable'        => true,
        'handler'       => Handler::class,
        'parser'        => Parser::class,
        'ping_interval' => 25000,
        'ping_timeout'  => 60000,
        'room'          => [
            'type'  => 'table',
            'table' => [
                'room_rows'   => 4096,
                'room_size'   => 2048,
                'client_rows' => 8192,
                'client_size' => 2048,
            ],
            'redis' => [
                'host'          => '127.0.0.1',
                'port'          => 6379,
                'max_active'    => 3,
                'max_wait_time' => 5,
            ],
        ],
        'listen'        => [],
        'subscribe'     => [],
    ],

其中有 room 配置项，里面的 type 表示使用哪种数据处理方式，下面有两种，“table” 和“redis”，table 是可以直接拿来使用的，而 redis 则需要我们的系统和项目中安装了 redis 扩展。table 是一种高性能、跨进程的内存处理服务，不同进程间可以共享数据。

创建事件

在项目根目录输入如下命令，分别创建加入房间事件、离开房间事件和房间的聊天事件：

php think make:listener WsJoin

php think make:listener WsLeave

php think make:listener RoomTest

然后在 app/event.php 中定义事件：

 [
    ],

    'listen'    => [
        'AppInit'  => [],
        'HttpRun'  => [],
        'HttpEnd'  => [],
        'LogLevel' => [],
        'LogWrite' => [],

        //监听连接，swoole 事件必须以 swoole 开头
        'swoole.websocket.Connect' => [
            app\listener\WsConnect::class
        ],
        //监听关闭
        'swoole.websocket.Close' => [
            \app\listener\WsClose::class
        ],
        //监听 Test 场景
        'swoole.websocket.Test' => [
            \app\listener\WsTest::class
        ],
        //加入房间事件
        'swoole.websocket.Join' => [
            \app\listener\WsJoin::class
        ],
        //离开房间事件
        'swoole.websocket.Leave' => [
            \app\listener\WsLeave::class
        ],
        //处理聊天室消息
        'swoole.websocket.RoomTest' => [
            \app\listener\RoomTest::class
        ],
    ],

    'subscribe' => [
    ],
];

上述的 Join、Leave和RoomTest等名称都是自定义的，要和前端发送消息的场景值对应。

当然，事件定义一样可以在 config/swoole.php 配置文件的 websocket listen 进行配置，具体参考前面文章。

H5 WebSocker 客户端方式连接

wsroot.html

<!DOCTYPE HTML>
<html>
<head>
    <meta charset="UTF-8">
    <title>Document</title>
</head>
<body>

<button onclick="join()">加入房间</button>
<button onclick="leave()">离开房间</button>
<input type="text" id="message">
<button onclick="send()">发送</button>

<script>
    var ws = new WebSocket("ws://127.0.0.1:9501/?uid=1");
    ws.onopen = function(){
        console.log('连接成功');
    }

    //数据返回的解析
    function mycallback(data){
        var start = data.indexOf('[') // 第一次出现的位置
        var start1 = data.indexOf('{')

        if(start < 0){
            start = start1;
        }
        if(start >= 0 && start1 >= 0){
            start = Math.min(start,start1);
        }

        if(start >= 0){
            console.log(data);
            var json = data.substr(start); //截取
            var json = JSON.parse(json);
            console.log(json);
            // if(json instanceof Array){
            //     window[json[0]](json[1]);
            // }
        }
    }

    function sendfd($message){
        console.log($message)
    }

    function testcallback($message){
        console.log($message)
    }

    function joincallback($message){
        // console.log($message)
        console.log(11);
    }

    function leavecallback($message){
        console.log($message)
    }

    ws.onmessage = function(data){
        // console.log(data.data);
        mycallback(data.data);
    }

    ws.onclose = function(){
        console.log('连接断开');
    }

    function join()
    {
        var room = prompt('请输入房间号');
        ws.send(JSON.stringify(['join',{
            room:room
        }])); //发送的数据必须是 ['test',数据] 这种格式
    }

    function leave()
    {
        var room = prompt('请输入要离开的房间号');
        ws.send(JSON.stringify(['leave',{
            room:room
        }])); //发送的数据必须是 ['test',数据] 这种格式
    }

    function send()
    {
        var message = document.getElementById('message').value;
        var room = prompt('请输入接收消息的房间号')
        ws.send(JSON.stringify(['RoomTest',{
            message:message,
            room:room
        }])); //发送的数据必须是 ['test',数据] 这种格式
    }
</script>
</body>
</html>

SocketIO 客户端方式连接

ioroomtest.html

<!DOCTYPE HTML>
<html>
<head>
    <meta charset="UTF-8">
    <title>Document</title>
</head>
<body>

<button onclick="join()">加入房间</button>
<button onclick="leave()">离开房间</button>
<input type="text" id="message">
<button onclick="send()">发送</button>

<script src="./socketio.js"></script>
<script>
    //http 协议
    var socket = io("http://127.0.0.1:9501?uid=1", {transports: ['websocket']});

    socket.on('connect', function(){
        console.log('connect success');
    });

    socket.on('close',function(){
       console.log('connect close')
    });

    //send_fd 为自定义的场景值，和后端对应
    socket.on("sendfd", function (data) {
        console.log(data)
    });

    //testcallback 为自定义的场景值，和后端对应
    socket.on("testcallback", function (data) {
        console.log(data)
    });

    socket.on("joincallback", function (data) {
        console.log(data)
    });

    socket.on("roomtestcallback", function (data) {
        console.log(data)
    });



    function join()
    {
        var room = prompt('请输入房间号');
        socket.emit('join',{
            room : room
        });
    }

    function leave()
    {
        var room = prompt('请输入要离开的房间号');
        socket.emit('leave',{
            room : room
        });
    }

    function send()
    {
        var message = document.getElementById('message').value;
        var room = prompt('请输入接收消息的房间号')
        socket.emit('RoomTest',{
            message : message,
            room : room
        });
    }
</script>
</body>
</html>

页面中，join()、leave()、和send() 函数中定义的场景值分别是 join、leave和RoomTest，我们在 app/leave.php 中定义了这些场景值对应的事件，因此分别触发 WsJoin.php、WsLeave.php 和 RoomTest.php 事件。

后端事件编写

app/listener/WsJoin.php

<?php
declare (strict_types = 1);

namespace app\listener;

class WsJoin
{
    /**
     * 事件监听处理
     *
     * @return mixed
     */
    public function handle($event)
    {
        $ws = app('think\swoole\Websocket');
        $roomobj = app('think\swoole\websocket\Room');

        //当前客户端加入指定 Room
        $ws -> join($event['room']);
        //同时加入多个房间
//        $ws -> join(['room1','room2']);
        //指定客户端加入指定 room
//        $ws -> setSender(2) -> join($event['room']);

        //获取当前房间所有的 fd
        $getAllFdInRoom = $roomobj -> getClients($event['room']);

        //获取指定 fd 加入哪些房间
        $getAllRoom = $roomobj -> getRooms($ws -> getSender());

        var_dump('当前房间所有 fd：',$getAllFdInRoom);
        var_dump('当前 fd 加入的所有房间：',$getAllRoom);
        var_dump('当前请求数据：',$event);
        $ws -> emit('joincallback','房间加入成功');
    }
}

app/listener/WsLeave.php

<?php
declare (strict_types = 1);

namespace app\listener;

class WsLeave
{
    /**
     * 事件监听处理
     *
     * @return mixed
     */
    public function handle($event)
    {
        $ws = app('think\swoole\Websocket');
        $roomobj = app('think\swoole\websocket\Room');

        // 当前客户端离开指定 room
        $ws -> leave($event['room']);
        // 同时离开多个 room
//        $ws -> leave(['one','two']);
        // 指定客户端离开指定 room
//        $ws -> setSender(2) -> leave($event['room']);

        // 获取指定 room 中的所有客户端 fd
        $getAllFdInRoom = $roomobj -> getClients($event['room']);
        var_dump('当前房间还剩 fd：',$getAllFdInRoom);

        $ws -> emit('leavecallback','房间离开成功');
    }
}

app/listener/RoomTest.php

<?php
declare (strict_types = 1);

namespace app\listener;

class RoomTest
{
    /**
     * 事件监听处理
     *
     * @return mixed
     */
    public function handle($event)
    {
        var_dump($event);
        $ws = app('think\swoole\Websocket');
        //给指定的 room 内所有 fd 发送消息，包括当前客户端，当前客户端没有加入该 room 也可发送
        $ws -> to($event['room']) -> emit('roomtestcallback',$event['message']);

        //指定多个 room 发送消息
        //$ws -> to(['one','two']) -> emit('客户端场景值',$event['message']);
    }
}

以上分别是前端 HTML 页面和后端的加入房间、离开房间和房间聊天事件代码，下面开始测试。

首先在项目根目录开启 Think-Swoole 服务。

浏览器访问 wsroot.html 或者 ioroomtest.html 页面，可以打开多个标签，模拟多个客户端，我们先打开三个，连接成功后，fd 分别为1、2、3，我们让 fd 1 和 2 的客户端都加入“one”,房间，fd 为 3 的客户端加入 “two” 房间，由于我们在 WsJoin.php 加入房间事件中打印了用户加入房间所有 fd，和该用户所加入的所有房间名称，这些信息会打印在命令行中，最后会发送给当前客户端聊天场景值和“加入房间成功”信息给客户端，在浏览器控制台可查看。

加入房间后，用 fd 为 1 的客户端在页面的输入框输入要发送的消息，点击发送后，输入要发送的房间名称为“one”，然后，消息就发送到“one”房间，只有“one”房间的所有客户端（fd 为1、2）能收到消息。

现在让 fd 为 2 的客户端离开 “one” 房间，由于在 WsLeave.php 离开房间事件中，我们打印了离开后剩余 fd，信息会出现在命令行中，可见“one”房间只剩下 fd 1 了，fd 1 发送信息，fd 2 就收不到了。

关于 WebSocket-Room 的其他功能，都已经在上述代码中注释了，需要可打开进行测试。

H5 WebSocket 和 SocketIO 对于消息的处理，上篇文章已经演示过了，前者对于服务端返回的消息需要手动解析才能使用，后者会根据消息的场景值接收消息，并做过处理可以直接使用。

Thinkphp6+swoole 教程（八）WebSocket-Room 加入、离开房间和房间消息发送

相关文章: