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

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

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

说明:

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

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

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