Swoole 实现在线聊天

缺乏、安全感 2023-05-31 07:57 13阅读 0赞

Swoole 跟 thinkphp5 结合开发 WebSocket 在线聊天通讯系统

ThinkPHP 使用 Swoole 需要安装 think-swoole Composer 包，前提系统已经安装

好了 Swoole PECL 拓展

tp5 的项目根目录下执行 composer 命令安装 think-swoole：

1 composer require topthink/think-swoole

话不多说，直接上代码：

新建 WebSocket.php 控制器：

（监听端口要确认服务器放行，宝塔环境还需要添加安全组规则）

![复制代码][aHR0cHM6Ly9jb21tb24uY25ibG9ncy5jb20vaW1hZ2VzL2NvcHljb2RlLmdpZg]

<?php
    namespace app\home\controller;use think\swoole\Server;class WebSocket e
    xtends 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";
    }
    }

![复制代码][aHR0cHM6Ly9jb21tb24uY25ibG9ncy5jb20vaW1hZ2VzL2NvcHljb2RlLmdpZg]

前端演示页面：

（省略控制器判断登录状态、分配数据逻辑......）

![复制代码][aHR0cHM6Ly9jb21tb24uY25ibG9ncy5jb20vaW1hZ2VzL2NvcHljb2RlLmdpZg]

<!DOCTYPE html><html lang="en"><head><meta charset="UTF-8" /><meta http-
    equiv="X-UA-Compatible" content="IE=edge,chrome=1" /><meta name="viewpor
    t" content="width=device-width,initial-scale=1,minimum-scale=1,maximum-s
    cale=1,user-scalable=no" /><title>Chat</title><link rel="stylesheet" typ
    e="text/css" href="/static/liaotian/chat.css" /><script src="/static/liaotian/js/jquery.min.js"></script><script src="/static/liaotian/js/flexib
    le.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>
    
    <div class="message"> </div>
    
    
    <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>
    </body></html><script src="http://apps.bdimg.com/lib
    s/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;//当前用户 idvar token = 'abcdefg';//用户 token
    //判断浏览器是否支持 WebSocket
    var supportsWebSockets = 'WebSocket' in window || 'MozWebSocket' in w
    indow;
    if (supportsWebSockets) {
    //建立 WebSocket 连接（ip 地址换成自己主机 ip）
    var ws = new WebSocket("ws://127.0.0.1:9501?uid="+uid+"&token="+t
    oken);
    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=\"sho
    w\"><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=\"sen
    d\"><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]).scrollIntoV
    iew();//向上滚动
    },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>

![复制代码][aHR0cHM6Ly9jb21tb24uY25ibG9ncy5jb20vaW1hZ2VzL2NvcHljb2RlLmdpZg]

服务器移到项目根目录开启服务：

php public/index.php Websocket/start

这里的路径，是因为我绑定了 home 模块为默认模块，tp5 默认情况是：php

public/index.php index/Websocket/start）

开启成功，查看端口已经被监听：

lsof -i:9501

[aHR0cHM6Ly9jb21tb24uY25ibG9ncy5jb20vaW1hZ2VzL2NvcHljb2RlLmdpZg]: /images/20230531/9cf79844957345bfa314425cf53fb00f.png