WorkerMan中如何向某个特定客户端发送数据

使用worker来做服务器，没有用GatewayWorker，如何实现向指定用户推送消息？

  1. <?php
  2. use Workerman\Worker;
  3. use Workerman\Connection\TcpConnection;
  4. require_once __DIR__ . '/vendor/autoload.php';
  6. // 初始化一个worker容器，监听1234端口
  7. $worker = new Worker('websocket://workerman.net:1234');
  8. // ====这里进程数必须必须必须设置为1====
  9. $worker->count = 1;
  10. // 新增加一个属性，用来保存uid到connection的映射(uid是用户id或者客户端唯一标识)
  11. $worker->uidConnections = array();
  12. // 当有客户端发来消息时执行的回调函数
  13. $worker->onMessage = function(TcpConnection $connection, $data)
  14. {
  15.     global $worker;
  16.     // 判断当前客户端是否已经验证,即是否设置了uid
  17.     if(!isset($connection->uid))
  18.     {
  19.        // 没验证的话把第一个包当做uid（这里为了方便演示，没做真正的验证）
  20.        $connection->uid = $data;
  21.        /* 保存uid到connection的映射，这样可以方便的通过uid查找connection，
  22.         * 实现针对特定uid推送数据
  23.         */
  24.        $worker->uidConnections[$connection->uid] = $connection;
  25.        return $connection->send('login success, your uid is ' . $connection->uid);
  26.     }
  27.     // 其它逻辑，针对某个uid发送 或者 全局广播
  28.     // 假设消息格式为 uid:message 时是对 uid 发送 message
  29.     // uid 为 all 时是全局广播
  30.     list($recv_uid, $message) = explode(':', $data);
  31.     // 全局广播
  32.     if($recv_uid == 'all')
  33.     {
  34.         broadcast($message);
  35.     }
  36.     // 给特定uid发送
  37.     else
  38.     {
  39.         sendMessageByUid($recv_uid, $message);
  40.     }
  41. };
  43. // 当有客户端连接断开时
  44. $worker->onClose = function(TcpConnection $connection)
  45. {
  46.     global $worker;
  47.     if(isset($connection->uid))
  48.     {
  49.         // 连接断开时删除映射
  50.         unset($worker->uidConnections[$connection->uid]);
  51.     }
  52. };
  54. // 向所有验证的用户推送数据
  55. function broadcast($message)
  56. {
  57.    global $worker;
  58.    foreach($worker->uidConnections as $connection)
  59.    {
  60.         $connection->send($message);
  61.    }
  62. }
  64. // 针对uid推送数据
  65. function sendMessageByUid($uid, $message)
  66. {
  67.     global $worker;
  68.     if(isset($worker->uidConnections[$uid]))
  69.     {
  70.         $connection = $worker->uidConnections[$uid];
  71.         $connection->send($message);
  72.     }
  73. }
  75. // 运行所有的worker（其实当前只定义了一个）
  76. Worker::runAll();

说明：

以上例子可以针对uid推送，虽然是单进程，但是支持个10W在线是没问题的。

注意这个例子只能单进程，也就是$worker->count 必须是1。要支持多进程或者服务器集群的话需要Channel组件完成进程间通讯，开发也非常简单，可以参考Channel组件集群推送例子一节。

如果希望在其它系统中推送消息给客户端，可以参考在其它项目中推送一节