Workerman中如何向某個(gè)特定客戶端發(fā)送數(shù)據(jù)
使用worker來做服務(wù)器,沒有用GatewayWorker,如何實(shí)現(xiàn)向指定用戶推送消息?
<?php
use Workerman\Worker;
use Workerman\Connection\TcpConnection;
require_once __DIR__ . '/vendor/autoload.php';
// 初始化一個(gè)worker容器,監(jiān)聽1234端口
$worker = new Worker('websocket://workerman.net:1234');
// ====這里進(jìn)程數(shù)必須必須必須設(shè)置為1====
$worker->count = 1;
// 新增加一個(gè)屬性,用來保存uid到connection的映射(uid是用戶id或者客戶端唯一標(biāo)識(shí))
$worker->uidConnections = array();
// 當(dāng)有客戶端發(fā)來消息時(shí)執(zhí)行的回調(diào)函數(shù)
$worker->onMessage = function(TcpConnection $connection, $data)
{
global $worker;
// 判斷當(dāng)前客戶端是否已經(jīng)驗(yàn)證,即是否設(shè)置了uid
if(!isset($connection->uid))
{
// 沒驗(yàn)證的話把第一個(gè)包當(dāng)做uid(這里為了方便演示,沒做真正的驗(yàn)證)
$connection->uid = $data;
/* 保存uid到connection的映射,這樣可以方便的通過uid查找connection,
* 實(shí)現(xiàn)針對特定uid推送數(shù)據(jù)
*/
$worker->uidConnections[$connection->uid] = $connection;
return $connection->send('login success, your uid is ' . $connection->uid);
}
// 其它邏輯,針對某個(gè)uid發(fā)送 或者 全局廣播
// 假設(shè)消息格式為 uid:message 時(shí)是對 uid 發(fā)送 message
// uid 為 all 時(shí)是全局廣播
list($recv_uid, $message) = explode(':', $data);
// 全局廣播
if($recv_uid == 'all')
{
broadcast($message);
}
// 給特定uid發(fā)送
else
{
sendMessageByUid($recv_uid, $message);
}
};
// 當(dāng)有客戶端連接斷開時(shí)
$worker->onClose = function(TcpConnection $connection)
{
global $worker;
if(isset($connection->uid))
{
// 連接斷開時(shí)刪除映射
unset($worker->uidConnections[$connection->uid]);
}
};
// 向所有驗(yàn)證的用戶推送數(shù)據(jù)
function broadcast($message)
{
global $worker;
foreach($worker->uidConnections as $connection)
{
$connection->send($message);
}
}
// 針對uid推送數(shù)據(jù)
function sendMessageByUid($uid, $message)
{
global $worker;
if(isset($worker->uidConnections[$uid]))
{
$connection = $worker->uidConnections[$uid];
$connection->send($message);
}
}
// 運(yùn)行所有的worker(其實(shí)當(dāng)前只定義了一個(gè))
Worker::runAll();
說明:
以上例子可以針對uid推送,雖然是單進(jìn)程,但是支持個(gè)10W在線是沒問題的。
注意這個(gè)例子只能單進(jìn)程,也就是$worker->count 必須是1。要支持多進(jìn)程或者服務(wù)器集群的話需要Channel組件完成進(jìn)程間通訊,開發(fā)也非常簡單,可以參考Channel組件集群推送例子一節(jié)。
如果希望在其它系統(tǒng)中推送消息給客戶端,可以參考在其它項(xiàng)目中推送一節(jié)