打开连接时将用户 ID 从浏览器发送到 websocket 服务器

Question

在问这个问题之前，我尽了最大努力阅读了关于SO的严重问题（标记为Ratchet）并处理类似的问题，但无济于事。我什至问了一个没有引起注意的问题，因此我删除了它以写另一个问题（希望是更清晰）。

我的最终目标是使用 Ratchet 构建一个一对一的私人聊天应用程序。除了我无法向特定用户发送消息之外，一切工作正常。

每个登录用户在访问网站的安全区域时都会连接到 websocket 服务器：

$(document).ready(function() { 

    var conn = new WebSocket('ws://localhost:8080');
        conn.onopen = function(e) {
            console.log("Connection established!");

            // Here I need to send the logged in user_id to websocket server
            // and get it in onOpen method so that I can index my array 
            // of connections with user_id instead of
            //$connection->ResourceId, I explain more below

        };

        conn.onmessage = function(e) {
            console.log(e.data);
        };

});

当用户在聊天框中写入消息时，消息会通过 AJAX 发送到 Web 服务器，然后使用 ZeroMQ 推送到 Websocket。在控制器中：

// Persistence of Message(message_id, sender_id, receiver_id, message_text)
                .....

                $context = new \ZMQContext();
                $socket = $context->getSocket(\ZMQ::SOCKET_PUSH, 'my pusher');
                $socket->connect("tcp://localhost:5555");

                $pushData = array(
                       'receiver_id' => $receiver_id,
                       'sender_id'  => $user->getId(),
                       'message'  => $message->getMessageText(),
                    );
                $socket->send(json_encode($pushData));

所以最后，我的 websocket 服务器能够使用 JSON 知道哪个是接收者的 id。但他如何知道该用户的连接是哪个？换句话说，我需要将 websocket 连接存储在一个由用户 ID 索引的数组中。

<?php
namespace RealTime;

use Ratchet\MessageComponentInterface;
use Ratchet\ConnectionInterface;
use Ratchet\Wamp\WampServerInterface;

class Pusher implements WampServerInterface, MessageComponentInterface{

    private $clients;

    public function onOpen(ConnectionInterface $conn) {

        $this->clients[$conn->resourceId] = $conn;
        // I need here to get the user_id received from browser while opening connection
    }

    public function onMessageEntry($entry) {
        $entryData = json_decode($entry, true);

        //This is not what I need (It sends to all users in array)
        foreach ($this->clients as $key => $client) {

        $client->send($entryData['message']); 
        }
    }
    public function onMessage(ConnectionInterface $from, $msg) {
        echo $msg; 
    }
}

和 websocket 服务器：

  <?php
        require dirname(__DIR__) . '/vendor/autoload.php';
        use RealTime\Pusher;

        $loop   = React\EventLoop\Factory::create();
        $pusher = new Pusher;

        $context = new React\ZMQ\Context($loop);
        $pull = $context->getSocket(ZMQ::SOCKET_PULL);
        $pull->bind('tcp://127.0.0.1:5555'); 
        $pull->on('message', array($pusher, 'onMessageEntry'));


        $webSock = new React\Socket\Server($loop);
        $webSock->listen(8080, '0.0.0.0'); 
        $webServer = new Ratchet\Server\IoServer(
            new Ratchet\Http\HttpServer(
                new Ratchet\WebSocket\WsServer(
                    new Ratchet\Wamp\WampServer(
                        $pusher
                    )
                )
            ),
            $webSock
        );
        $loop->run();

        ?>

问题：

1. 如何user_id在打开连接时发送从客户端登录的信息。我需要在 websocket 服务器中拥有该值，以便我可以用它索引我的客户端数组（$client[user_id]=$conn而不是$client[recourceId]=$conn）。我尝试了 javascript 函数send，但我不知道在哪里接收发送的数据（甚至onMessage不打印任何内容）。

2. 为什么该onMessage方法甚至没有执行MessageComponentInterface（是因为我有onMessageEntry方法+$pull->on('message', array($pusher, 'onMessageEntry'));代码行吗？

谢谢。

Answer 1

这就是我的发现，欢迎提出任何增强此解决方案的建议。

可以使用 Ratchet SessionProvider。这将需要使用所示的 Symfony 自定义会话处理程序之一。我在下面的代码中使用 PdoSessionHandler.

<?php
    require dirname(__DIR__) . '/vendor/autoload.php';

    use YourDirectory\Pusher;
    use Symfony\Component\HttpFoundation\Session\Storage\Handler;

    use \Ratchet\Session\SessionProvider;

    $pusher = new Pusher;

    $pdo = new PDO('mysql:host=localhost;dbname=community', 'root', null);

    $pdo->setAttribute(PDO::ATTR_ERRMODE, PDO::ERRMODE_EXCEPTION);

    //This info is related to you db
    $dbOptions = array(
        'db_table'      => 'session',
        'db_id_col'     => 'sess_id',
        'db_data_col'   => 'sess_data',
        'db_time_col'   => 'sess_time',);

        $loop   = \React\EventLoop\Factory::create();
        $context = new \React\ZMQ\Context($loop);
        $pull = $context->getSocket(\ZMQ::SOCKET_PULL);
        $pull->bind('tcp://127.0.0.1:5555');
        $pull->on('message', array($pusher, 'onMessageEntry'));

        $webSock = new React\Socket\Server($loop);
        $webSock->listen(8080, '0.0.0.0'); // Binding to 0.0.0.0 means remotes can connect
        $webServer = new Ratchet\Server\IoServer(
            new Ratchet\Http\HttpServer(
                new Ratchet\WebSocket\WsServer(
                    new SessionProvider(
                        new Ratchet\Wamp\WampServer(
                            $pusher
                        ),new Handler\PdoSessionHandler($pdo,$dbOptions)
                    )
                )
            ),
            $webSock
        );

        $loop->run();
    ?>

然后我的存根类将变成：

   public function onOpen(ConnectionInterface $conn) {  
        $this->clients[$conn->Session->get('current_user_id')] = $conn;
    }

public function onMessageEntry($entry) {

            $entryData = json_decode($entry, true);
            $ReceiverConnection=$this->clients[$entryData['receiver_id']];
            $ReceiverConnection->send($entryData['message']);                  
        }

但之前，我已将用户 ID 添加到 Web 服务器中的会话中（在返回初始页面的控制器中）

$user = $this->getUser();
$request->getSession()->set('current_user_id', $user->getId()); 

附：

1. 可以通过实现这个（Symfony）来转移到 PdoSessionHandler。

2. 我仍然无法回答2，但所有可以放置的逻辑onMessage现在都已转移到onMessageEntry暂时满足需求的位置。