如何使用 Workman 做一个聊天室
一:首先,得简单说说 thinkphp+workerman 的安装。
安装 thinkphp5.1
composer create-project topthink/think=5.1.x-dev tp5andworkman
安装 think-worker
我的官方群点击此处。
composer require workerman/workerman
二:我们先看 think-worker 的代码
- config/worker_server.php
- 先来个服务器广播消息的示例,每10秒钟定时广播一条消息
'onWorkerStart' => function ($worker) {
\Workerman\Lib\Timer::add(10, function()use($worker){
// 遍历当前进程所有的客户端连接,发送自定义消息
foreach($worker->connections as $connection){
$send['name'] = '系统信息';
$send['content'] = '这是一个定时任务信息';
$send['time'] = time();
$connection->send(json_encode($send));
}
});
}
但是在 onMessage 时,我们获取不到 $worker 对象,所以无法广播消息。
'onMessage' => function ($connection, $data) {
$origin = json_decode($data,true);
$send['name'] = '广播数据';
$send['content'] = $origin['content'];
$message = json_encode($send);
foreach($worker->connections as $connection)
{
$connection->send($message);
}
}
尝试了各种方法,貌似都不行
'onMessage' => function ($connection, $data)use($worker) {
// 这样是获取不到 $worker 对象的
// ...省略代码
}
所以只能抛弃 thinkphp 给我们封装的 think-worker 框架,得自己写,(或者修改框架内部代码)
修改框架内部的代码:/vendor/topthink/think-worker/src/command/Server.php
,主要是把 onMessage 方法自己加进去
use() 就是把外部变量传递到函数内部使用,或者使用global $worker
$worker = new Worker($socket, $context);
$worker->onMessage = function ($connection, $data)use($worker) {
$origin = json_decode($data,true);
$send['name'] = '广播数据';
$send['content'] = $origin['content'];
$send['uid'] = $connection->uid;
$message = json_encode($send);
foreach($worker->connections as $connection)
{
$connection->send($message);
}
};
这样,我们就能够获取到 $worker 对象了
$worker->onMessage = function ($connection, $data)use($worker) { ... }
三:$connection 绑定 uid
其实你早都已经看出,$worker->connections 获取到的是当前所有用户的连接,connections 即为其中一个链接。
记录websocket连接时间:
$worker->onConnect = function ($connection) {
$connection->login_time = time();
};
获取websocket连接时间:
$worker->onMessage = function ($connection, $data)use($worker) {
$login_time = $connection->login_time;
};
由此可以看出,我们可以把数据绑定到 $connection 连接的一个属性,例如:
$connection->uid = $uid;
当JavaScript端在连接websocket服务器成功后,即把自己的 uid 立马发送服务端绑定:
$worker->onMessage = function ($connection, $data)use($worker) {
$origin = json_decode($data,true);
if(array_key_exists('bind',$origin)){
$connection->uid = $origin['uid'];
}
};
四:单播发送消息,即自定义发送
$worker->onMessage = function ($connection, $data)use($worker) {
$origin = json_decode($data,true);
$sendTo = $origin['sendto']; // 需要发送的对方的uid
$content = $origin['content']; // 需要发送到对方的内容
foreach($worker->connections as $connection)
{
if( $connection->uid == $sendTo){
$connection->send($content);
}
}
};
到此,已经完成基于 workman 的自定义对象发送消息。
由于该php文件存放于composer中,只需要把该文件复制出来,放到application/command
,修改命名空间,即可保存到自己的项目中
五:对比swoole
1、workman可以在windows系统中运行,swoole则不能。
2、workman:$worker->connections获取所有连接,$connection->id获取自己的连接id;swoole:$server->connections获取所有连接,$connection->fd获取自己的连接id。
3、workman启动时执行 onWorkerStart 方法,可以把定时器写入到里面;swoole 使用 WorkerStart 启动定时器。
仅仅于聊天室或者定时器而言,workman 还是比较方便的。
以上内容希望帮助到大家,很多PHPer在进阶的时候总会遇到一些问题和瓶颈,业务代码写多了没有方向感,不知道该从那里入手去提升,对此我整理了一些资料,包括但不限于:分布式架构、高可扩展、高性能、高并发、服务器性能调优、TP6,laravel,YII2,Redis,Swoole、Swoft、Kafka、Mysql优化、shell脚本、Docker、微服务、Nginx等多个知识点高级进阶干货需要的可以免费分享给大家,需要的可以加入我的官方群点击此处。