关于Worker如何向指定Client推送消息

aaron

目前我用worker来做服务器,没有用GatewayWorker,请问walkor大哥,如何实现向指定用户推送消息?

以下是我的代码:

$client_id = '';
$worker->onConnection( $connection ){
    global $client_id;
    $client_id = $connection->id;
};
$worker->onMessage( $connection, $data ){
global $client_id;
   foreach( $connection->worker->connections as $con ){
       if( $con->id == $client_id ){
           $con->send( $data );
           break;
       }
   }
};
10814 7 0
7个回答

aaron

注:在触发onConnect的时候,取得ClientID,然后在onMessage内将ClientID和用户其他资料绑定并储存到数据库。上述代码临时写出来的,我实际的需求是,基于worker向指定ClientID的在线用户推送消息,上面代码可能写成向当前用户推送了。

  • 暂无评论
walkor
<?php
use Workerman\Worker;
require_once './Workerman/Autoloader.php';
// 初始化一个worker容器,监听1234端口
$worker = new Worker('websocket://workerman.net:1234');
// 进程数设置为1
$worker->count = 1;
// 新增加一个属性,用来保存uid到connection的映射
$worker->uidConnections = array();
// 当有客户端发来消息时执行的回调函数
$worker->onMessage = function($connection, $data)use($worker)
{
    // 判断当前客户端是否已经验证,既是否设置了uid
    if(!isset($connection->uid))
    {
        // 没验证的话把第一个包当做uid(这里为了方便演示,没做真正的验证)
        $connection->uid = $data;
        /* 保存uid到connection的映射,这样可以方便的通过uid查找connection,
         * 实现针对特定uid推送数据
         */
        $worker->uidConnections = $connection;
        return $connection->send('login success, your uid is ' . $connection->uid);
    }
    // 其它罗辑,针对某个uid发送 或者 全局广播
    // 假设消息格式为 uid:message 时是对 uid 发送 message
    // uid 为 all 时是全局广播
    list($recv_uid, $message) = explode(':', $data);
    // 全局广播
    if($recv_uid == 'all')
    {
        broadcast($message);
    }
    // 给特定uid发送
    else
    {
        sendMessageByUid($recv_uid, $message);
    }
};

// 当有客户端连接断开时
$worker->onClose = function($connection)use($worker)
{
    global $worker;
    if(isset($connection->uid))
    {
        // 连接断开时删除映射
        unset($worker->uidConnections);
    }
};

// 向所有验证的用户推送数据
function broadcast($message)
{
    global $worker;
    foreach($worker->uidConnections as $connection)
    {
        $connection->send($message);
    }
}

// 针对uid推送数据
function sendMessageByUid($uid, $message)
{
    global $worker;
    if(isset($worker->uidConnections))
    {
        $connection = $worker->uidConnections;
        $connection->send($message);
    }
}

// 运行所有的worker(其实当前只定义了一个)
Worker::runAll();

可以针对uid推送了,虽然是单进程,但是支持个1W在线是没问题的

  • 暂无评论
walkor

注意只能单进程,多进程的话连接可能被分配到不同的进程,而不同的进程间是无法直接通讯的

  • leo 2015-12-16

    有没有办法可以实现多进程?

  • walkor 2015-12-16

    可以多进程,类似GatewayWorker进程模型,或者用一个代理进程转发给其它进程

aaron

收到了,谢谢walkor,另外还有一个问题:控制客户端close是否能够通过上述代码来实现?

  • 暂无评论
walkor

找到$connection,调用$connection->close();即可

  • 暂无评论
aaron
$worker->onMessage = function( $connection, $data ){
    if( $data数据不合法时 ){
        $connection->close();
    }
}

使用类似代码时,貌似是全体在线用户均会触发close事件。

  • 暂无评论
walkor

不会

  • 暂无评论
年代过于久远,无法发表回答
🔝