乐趣区

使用PHP协程框架开发WebSocket在线聊天通讯系统

ThinkPHP 使用 Swoole 需要安装 think-swoole Composer 包,前提系统已经安装好了 Swoole PECL 拓展

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

很多 PHPer 在进阶的时候总会遇到一些问题和瓶颈,业务代码写多了没有方向感,不知道该从那里入手去提升,对此我整理了一些资料,包括但不限于:分布式架构、高可扩展、高性能、高并发、服务器性能调优、TP6,laravel,YII2,Redis,Swoole、Swoft、Kafka、Mysql 优化、shell 脚本、Docker、微服务、Nginx 等多个知识点高级进阶干货需要的可以免费分享给大家,需要请戳这里

退出移动版