利用 webSocket 与 Swoole 打造一个小型聊天室（协程）

2022-02-22 09:30:37 IT技术网互联网

浏览

前言

前面有写一个异步简单的聊天室，然后想着，就把协程的也弄了吧所以就有了这个文章，其实所有的功能都大差不差，就仅仅几个地方不一样而已，也都是简单的地方。
博文地址：利用 webSocket 与 Swoole 打造一个小型聊天室 (异步)
本次也没增加功能，就是增加了一个心跳，从前端定时发送一个 ping ，服务端不作反应，仅此而已。

前端页面代码：

<!DOCTYPE html><html lang="en"><head>    <meta charset="UTF-8">    <title>打工人聊天室</title>   <!--需要引入jq 文件--></head><style>    .content {        height: 400px;        max-width: 400px;        overflow: auto;        border-radius: 5px;        border: 1px solid #f0f0f0;    }</style><body>            <p id="content" class="content">                <p>聊天区域</p>            </p>            你好打工人：<samp id="nickname">昵称</samp> <br>            本次连接FD: <samp id="fd-samp"></samp> <br>            <input type="text" id="msg">            <input type="hidden" id="fd" value="">            <button id="send" onclick="send()">发送</button></body></html>

JS 代码：

在服务器信息回执时，会有第一次连接回执，还是服务端发送消息回执的状态区别，通过 msgType 来分辨，如果是第一次连接的回执消息，则把 FD 做一个页面留存，并不显示在聊天消息区，如果收到的是消息回执，就直接显示到聊天消息区。

还有就是，前后端相互通信发送的东西，都是字符串性质最优，我前端处理的方法是先组合成一个对象，然后转 JSON 串。

<script>    //滚动条最底部    function scrolltest() {        var p = document.getElementById("content");        p.scrollTop = p.scrollHeight;    }    var wsServer = 'ws://127.0.0.1:9502/websocket';    var websocket = new WebSocket(wsServer);    var nickname = Math.random().toString(36).substr(2);    thisFd = '';    $('#nickname').html(nickname);    //点击发送    function send() {        var msg = $('#msg').val();        var data = {            'nickname': nickname,            'fd': thisFd,            'data': msg        }        //生成json 方便后台接收以及使用        var data = JSON.stringify(data);        websocket.send(data);        //然后清空        $('#msg').val('');    }    //链接成功    websocket.onopen = function (evt) {        var data = {            'msgType': 'open'        }        var data = JSON.stringify(data);        $("#content >p:last-child").after('<p> 服务器已连接,开始聊天吧 </p>');        websocket.send(data);    };    //链接断开    websocket.onclose = function (evt) {        $("#content >p:last-child").after('<p> 服务器已断开,请重新连接 </p>');    };    //收到服务器消息    websocket.onmessage = function (evt) {        //握手成功后，会接受到服务端返回的fd ，msgType = 1        //字符串格式化成json        var data = eval('(' + evt.data + ')');        // console.log(evt.data);        switch (data.msgType) {            case 1:                thisFd = data.fd;                $('#fd-samp').html(thisFd);                $('#fd').val(thisFd);                break;            case 2:                if (data.nickname == nickname) {                    data.nickname = '我';                }                $("#content >p:last-child").after('<p>' + data.nickname + ' 在 ' + data.time + ' 说：<br>' + data.data + '</p>');                //接收到消息自动触底                scrolltest();                break;        }    };    //服务器异常    websocket.onerror = function (evt, e) {        $("#content >p:last-child").after('<p> 服务器异常 </p>');    };    //心跳,本次新增    function heartbeat() {        var data = {            'msgType': 'ping',        }        //生成json 方便后台接收以及使用        var data = JSON.stringify(data);        websocket.send(data);    }    //30 秒一次    setInterval(heartbeat, 30000);</script>

服务端代码
协程，都需要在 Corun(function () {}) 里。

<?php    //定义获取当前的id函数    function getObjectId(SwooleHttpResponse $response) {        if (PHP_VERSION_ID < 70200) {            $id = spl_object_hash($response);        } else {            $id = spl_object_id($response);        }        return $id;    }    Corun(function () {        $server = new CoHttpServer('127.0.0.1', 9502, false);        $server->set([            'heartbeat_idle_time'      => 600, // 表示一个连接如果600秒内未向服务器发送任何数据，此连接将被强制关闭            'heartbeat_check_interval' => 60,  // 表示每60秒遍历一次        ]);        $server->handle('/websocket', function ($request, $ws) {            $ws->upgrade();            global $wsObjects;            $objectId = getObjectId($ws);            $wsObjects[$objectId] = $ws;            while (true) {                $frame = $ws->recv();                if ($frame === '') {                    unset($wsObjects[$objectId]);                    $ws->close();                    break;                } else if ($frame === false) {                    echo 'error : ' . swoole_last_error() . "n";                    break;                } else {                    if ($frame->data == 'close' || get_class($frame) === SwooleWebSocketCloseFrame::class) {                        unset($wsObjects[$objectId]);                        $ws->close();                        return;                    }                    //格式化接收到json                    $data = json_decode($frame->data);                    switch ($data->msgType){                        case 'open':                            //链接第一次                            $data = json_encode([                                'fd' => $objectId,                                'msgType' => 1  //代表第一次连接，前端处理fd                            ]);                            $ws->push($data);                            break;                        case 'ping':                            //接收到心跳 不作回复//                            echo  $data->msgType;                            break;                        default :                            // 原基础上不动，增加一些自定义                            $data->msgType = 2; //代表服务器端回复                            $data->time = date('Y-m-d H-i-s');                            $data = json_encode($data);                            foreach ($wsObjects as $obj) {                                $obj->push($data);                            }                    }                }            }        });        $server->start();    });

代码齐全之后，接下来就只需要在控制台执行以下 PHP 文件就行。