Ratchet实现PHP WebSocket多人聊天功能的示例

  

• composer 安装ratchet

  composer require cboden/ratchet

• 使用PDO连接数据库，创建mysql命令如下

  CREATE TABLE messages (
      id INT AUTO_INCREMENT PRIMARY KEY,
      message TEXT NOT NULL,
      created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
  );

• 使用Redis存储消息列表

这个示例代码中，PHP代码使用Ratchet来创建WebSocket服务器，并实现了简单的聊天功能。HTML代码使用JavaScript来建立WebSocket连接，并处理消息传输和用户输入。要运行此代码，请确保已安装Ratchet并在终端中运行PHP文件。然后，通过打开浏览器并访问HTML代码所在的地址，就可以开始聊天了。

 在onMessage方法中，我们首先将接收到的消息存入Redis列表中。然后，如果Redis中的消息数量超过1000，则将所有消息取出并依次存入MySQL中。请注意，在MySQL中执行多个INSERT语句时，最好使用事务（即BEGIN、COMMIT语句）来确保数据的完整性。

 WebSocket服务端代码：

clients = new \SplObjectStorage;

        // 连接到数据库
        $dsn = 'mysql:host=localhost;dbname=chat';
        $username = 'root';
        $password = '';
        $options = [
            \PDO::ATTR_ERRMODE => \PDO::ERRMODE_EXCEPTION,
            \PDO::ATTR_DEFAULT_FETCH_MODE => \PDO::FETCH_ASSOC
        ];
        $this->pdo = new \PDO($dsn, $username, $password, $options);

        // 连接到 Redis
        $this->redis = new \Redis();
        $this->redis->connect('localhost', 6379);
    }

    public function onOpen(ConnectionInterface $conn)
    {
        $this->clients->attach($conn);
        echo "New connection! ({$conn->resourceId})\n";
    }

    public function onMessage(ConnectionInterface $from, $msg)
    {
        foreach ($this->clients as $client) {
            if ($from !== $client) {
                $client->send($msg);
            }
        }

        // 将消息存入 Redis
        $this->redis->rpush('messages', $msg);

        // 如果 Redis 中的消息数量超过 1000，则将消息存入数据库
        if ($this->redis->llen('messages') > 1000) {
            $messages = $this->redis->lrange('messages', 0, -1);

            // 开始事务
            $this->pdo->beginTransaction();

            foreach ($messages as $message) {
                // 将消息存入数据库
                $stmt = $this->pdo->prepare('INSERT INTO messages (message) VALUES (?)');
                $stmt->execute([$message]);

                // 从 Redis 中删除已经存入数据库的消息
                $this->redis->lpop('messages');
            }

            // 提交事务
            $this->pdo->commit();
        }
    }

    public function onClose(ConnectionInterface $conn)
    {
        $this->clients->detach($conn);
        echo "Connection {$conn->resourceId} has disconnected\n";
    }

    public function onError(ConnectionInterface $conn, \Exception $e)
    {
        echo "An error has occurred: {$e->getMessage()}\n";
        $conn->close();
    }
}

$webSocketServer = new \Ratchet\WebSocket\WsServer(new Chat());
$server = \Ratchet\Server\IoServer::factory(
    new \Ratchet\Http\HttpServer($webSocketServer),
    8080
);

$server->run();

 开启socket服务命令，假设php文件名为socket.php

php ./socket.php

 HTML代码：

    
    


    

    

        
        Send
    
    
    

保证WebSocket服务一直开启，可以使用一个常驻进程管理工具supervisor，使用supervisor的示例配置链接。