1. 程式人生 > >ThinkPHP5.0結合Swoole開發WebSocket線上聊天

ThinkPHP5.0結合Swoole開發WebSocket線上聊天

ThinkPHP使用Swoole需要安裝 think-swoole Composer包,前提系統已經安裝好了Swoole PECL 拓展(相關文章:Linux下原始碼包安裝使用Swoole擴充套件

在tp5的專案根目錄下執行composer命令安裝think-swoole:

composer require topthink/think-swoole

安裝成功:

話不多說,直接上程式碼:

新建WebSocket.php控制器:

(監聽埠要確認伺服器放行,寶塔環境還需要新增安全組規則)

<?php

namespace app\home\controller;
use think\swoole\Server;
class WebSocket extends Server
{
    protected $host = '0.0.0.0'; //監聽所有地址
    protected $port = 9501; //監聽9501埠
    protected $serverType = 'socket';
    protected $option = [ 
        'worker_num'=> 4, //設定啟動的Worker程序數
        'daemonize'	=> false, //守護程序化(上線改為true)
        'backlog'	=> 128, //Listen佇列長度
        'dispatch_mode' => 2, //固定模式,保證同一個連線發來的資料只會被同一個worker處理

        //心跳檢測:每60秒遍歷所有連線,強制關閉10分鐘內沒有向伺服器傳送任何資料的連線
        'heartbeat_check_interval' => 60,
        'heartbeat_idle_time' => 600
    ];

    //建立連線時回撥函式
    public function onOpen($server,$req)
    {
        $fd = $req->fd;//客戶端標識
        $uid = $req->get['uid'];//客戶端傳遞的使用者id
        $token = $req->get['token'];//客戶端傳遞的使用者登入token
        
        //省略token驗證邏輯......
        if (!$token) {
            $arr = array('status'=>2,'message'=>'token已過期');
            $server->push($fd, json_encode($arr));
            $server->close($fd);
            return;
        }
        //省略給使用者繫結fd邏輯......
        echo "使用者{$uid}建立了連線,標識為{$fd}\n";
    }

    //接收資料時回撥函式
    public function onMessage($server,$frame)
    {
        $fd = $frame->fd;
        $message = $frame->data;

        //省略通過fd查詢使用者uid邏輯......
        $uid = 666;
        $data['uid'] = $uid;
        $data['message'] = '使用者'.$uid.'傳送了:'.$message;
        $data['post_time'] = date("m/d H:i",time());
        $arr = array('status'=>1,'message'=>'success','data'=>$data);

        //僅推送給當前連線使用者
        //$server->push($fd, json_encode($arr));
        
        //推送給全部連線使用者
        foreach($server->connections as $fd) {
            $server->push($fd, json_encode($arr));
        } 
    }

    //連線關閉時回撥函式
    public function onClose($server,$fd)
    {
        echo "標識{$fd}關閉了連線\n";
    }
}

前端演示頁面:

(省略控制器判斷登入狀態、分配資料邏輯......)

<!DOCTYPE html>
<html lang="en">
<head>
<meta charset="UTF-8" />
<meta http-equiv="X-UA-Compatible" content="IE=edge,chrome=1" />
<meta name="viewport" content="width=device-width,initial-scale=1,minimum-scale=1,maximum-scale=1,user-scalable=no" />
<title>Chat</title>
<link rel="stylesheet" type="text/css" href="/static/liaotian/chat.css" />
<script src="/static/liaotian/js/jquery.min.js"></script>
<script src="/static/liaotian/js/flexible.js"></script>
</head>
<body>
    <header class="header">
        <a class="back" href="javascript:history.back()"></a>
        <h5 class="tit">線上聊天</h5>
        <a href=""><div class="right">退出</div></a>
    </header>

    <!-- 聊天內容 start-->
	<div class="message"> </div>
    <!-- 聊天內容 end-->

    <!-- 底部 start-->
    <div class="footer">
        <img id="setbtn" src="/static/liaotian/images/hua.png" alt="" />
        <img src="/static/liaotian/images/xiaolian.png" alt="" />
        <input type="text" id="msg" value="" maxlength="300">
        <p style="background: rgb(17, 79, 142);" id="sendBtn">傳送</p>
    </div>
    <!-- 底部 end-->
</body>
</html>
<script src="http://apps.bdimg.com/libs/jquery/2.1.4/jquery.min.js"></script>
<script src="https://cdn.bootcss.com/layer/3.1.0/layer.js"></script>
<script type="text/javascript">
$(function () {
    var uid = 666;//當前使用者id
    var token = 'abcdefg';//使用者token

    //判斷瀏覽器是否支援WebSocket
    var supportsWebSockets = 'WebSocket' in window || 'MozWebSocket' in window;
    if (supportsWebSockets) {
        //建立WebSocket連線(ip地址換成自己主機ip)
        var ws = new WebSocket("ws://127.0.0.1:9501?uid="+uid+"&token="+token);
        ws.onopen = function () {
            layer.msg('伺服器連線成功',{shade:0.1,icon:1,time:600});
        };
        ws.onerror = function () {
            layer.msg('伺服器連線失敗',{shade:0.1,icon:2,time:600});
        };
        ws.onmessage = function (evt) {
            var data = $.parseJSON(evt.data);
            //錯誤提示
            if(data.status != 1){
                layer.alert(data.message,{icon:2});
                return;
            }
            //訊息返回
            if (data.status==1 && data.data.message!='') {
                var html = "";
                if (data.data.uid == uid) {
                    html += "<div style='word-break:break-all' class=\"show\"><div class=\"time\">"+data.data.post_time+"</div><div class=\"msg\"><img src=\""+data.data.head_img+"\" alt=\"\" /><p><i clas=\"msg_input\"></i>"+data.data.message+"</p></div></div>";
                }else{
                    html += "<div style='word-break:break-all' class=\"send\"><div class=\"time\">"+data.data.post_time+"</div><div class=\"msg\"><img src=\""+data.data.head_img+"\" alt=\"\" /><p><i clas=\"msg_input\"></i>"+data.data.message+"</p></div></div>";
                }
            }
            $(".message").append(html);
            setTimeout(function () {
                ($('.message').children("div:last-child")[0]).scrollIntoView();//向上滾動
            },100);
        };
        ws.onclose = function (res) {
            
        };
        //按鈕傳送
        $("#sendBtn").click(function () {
            var contents = $("#msg").val().trim();
            if(contents == null || contents == ""){
                layer.msg('內容為空',{shade:0.1,icon:2,time:600});            
                return false;
            }else{
            	ws.send(contents);
            	$("#msg").val("");
            }
        });
        //回車傳送
        $("#msg").keydown(function (evel) {
            var that = $(this);
            if (evel.keyCode == 13) {
                evel.cancelBubble = true;
                evel.preventDefault();
                evel.stopPropagation();
                var contents = that.val().trim();
                if(contents == null || contents == ""){
                    layer.msg('內容為空',{shade:0.1,icon:2,time:600});              
                    return false;
                }else{
                    ws.send(contents);
                    that.val("");
                }
            }
        });
    }else{
        layer.alert("您的瀏覽器不支援 WebSocket!");
    }
});
</script>

伺服器移到專案根目錄開啟服務:

php public/index.php Websocket/start

(這裡的路徑,是因為我綁定了home模組為預設模組,tp5預設情況是:php public/index.php index/Websocket/start)

開啟成功,檢視埠已經被監聽:

lsof -i:9501

 演示效果如下:

伺服器監聽如下:

使用者每重新整理重連一次,fd標識都會改變。


(文章看似簡單,前後琢磨了三四天,熬夜有害身體健康......如果對你有所幫助,請點個關注!)