前言
前面有寫一個非同步簡單的聊天室,然後想著,就把協程的也弄了吧所以就有了這個文章,其實所有的功能都大差不差,就僅僅幾個地方不一樣而已,也都是簡單的地方。
部落格:利用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> <div id="content" class="content"> <p>聊天区域</p> </div> 你好打工人:<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 div = document.getElementById("content"); div.scrollTop = div.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>
登入後複製
服務端程式碼
協程,都需要在 Co\run(function () {})
裡。
<?php //定义获取当前的id函数 function getObjectId(\Swoole\Http\Response $response) { if (PHP_VERSION_ID < 70200) { $id = spl_object_hash($response); } else { $id = spl_object_id($response); } return $id; } Co\run(function () { $server = new Co\Http\Server('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) === Swoole\WebSocket\CloseFrame::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 檔案就行。
然後前台直接訪問你的網站地址,我的是本地127.0.0.1
多開幾個視窗模擬多個用戶,然後發送訊息測試即可:
你好,打工人。
程式碼很簡單,難度不大,但可以很簡潔的反應出 webScoket 和 Swoole 的一種強大。