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

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

好了 Swoole PECL 拓展

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

1 composer require topthink/think-swoole

话不多说，直接上代码：

新建 WebSocket.php 控制器：

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

<?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";
}
}

前端演示页面：

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

<!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>
<!-- 聊天内容 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/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>