thinkphp websocket 开发实时聊天系统的用户群组与订阅功能实现 (一)

PHP开发实时聊天系统的用户群组与订阅功能实现

在当今社交互联网时代，实时聊天系统已经成为人们日常交流的重要工具。为了提供更好的用户体验，我们需要实现用户群组与订阅功能，使得用户能够方便地创建和加入群组，并且能够订阅感兴趣的内容。

本篇文章将介绍如何使用PHP开发实时聊天系统的用户群组与订阅功能。我们将使用WebSocket来实现实时通信功能，并结合MySQL数据库来处理用户和群组信息。

首先，我们需要搭建一个基本的实时聊天系统的框架。以下是一个简单的WebSocket服务器示例，使用Ratchet库实现：

require 'vendor/autoload.php';

use RatchetMessageComponentInterface;
use RatchetConnectionInterface;
use RatchetServerIoServer;
use RatchetHttpHttpServer;
use RatchetWebSocketWsServer;

class Chat implements MessageComponentInterface {
    protected $clients;
    protected $groups;

    public function __construct() {
        $this->clients = new SplObjectStorage;
        $this->groups = array();
    }

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

    public function onMessage(ConnectionInterface $from, $msg) {
        // 处理收到的消息
    }

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

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

$server = IoServer::factory(
    new HttpServer(
        new WsServer(
            new Chat()
        )
    ),
    8080
);

$server->run();

以上示例创建了一个名为Chat的WebSocket类，处理连接的打开、消息收发和关闭事件。其中$this->clients是一个保存所有客户端连接的SplObjectStorage对象，$this->groups是一个保存群组信息的数组。

接下来，我们需要添加一些方法来实现用户加入群组和订阅功能。在Chat类中添加以下代码：

public function joinGroup(ConnectionInterface $conn, $groupId) {
    // 将用户加入群组
    $this->groups[$groupId][] = $conn;
    echo "User ({$conn->resourceId}) joined group {$groupId}
";
}

public function leaveGroup(ConnectionInterface $conn, $groupId) {
    // 将用户从群组中移除
    $index = array_search($conn, $this->groups[$groupId]);
    if ($index !== false) {
        unset($this->groups[$groupId][$index]);
        echo "User ({$conn->resourceId}) left group {$groupId}
";
    }
}

public function subscribe(ConnectionInterface $conn, $topic) {
    // 订阅某个主题
    $conn->topics[] = $topic;
    echo "User ({$conn->resourceId}) subscribed to {$topic}
";
}

public function unsubscribe(ConnectionInterface $conn, $topic) {
    // 取消订阅某个主题
    $index = array_search($topic, $conn->topics);
    if ($index !== false) {
        unset($conn->topics[$index]);
        echo "User ({$conn->resourceId}) unsubscribed from {$topic}
";
    }
}

以上代码中，joinGroup方法将用户加入指定的群组，leaveGroup方法将用户从群组中移除，subscribe方法将用户订阅指定的主题，unsubscribe方法取消用户对某个主题的订阅。在用户加入群组或订阅主题时，我们需要将相关的信息保存到数据库中。

以下是一个示例的数据库表结构，用于保存用户和群组信息：

CREATE TABLE `users` (
  `id` int(11) UNSIGNED AUTO_INCREMENT PRIMARY KEY,
  `name` varchar(255) NOT NULL,
  `email` varchar(255) NOT NULL
);

CREATE TABLE `groups` (
  `id` int(11) UNSIGNED AUTO_INCREMENT PRIMARY KEY,
  `name` varchar(255) NOT NULL
);

CREATE TABLE `group_user` (
  `group_id` int(11) UNSIGNED NOT NULL,
  `user_id` int(11) UNSIGNED NOT NULL,
  PRIMARY KEY (`group_id`, `user_id`),
  FOREIGN KEY (`group_id`) REFERENCES `groups`(`id`),
  FOREIGN KEY (`user_id`) REFERENCES `users`(`id`)
);

在用户加入群组或订阅主题时，我们可以使用以下代码将相关信息插入到数据库中，具体逻辑可以根据实际需求进行修改：

public function joinGroup(ConnectionInterface $conn, $groupId) {
    // 将用户加入群组
    // ...

    // 将用户加入数据库
    $userId = $_SESSION['user_id']; // 假设用户登录时已经保存用户ID到session中
    $query = "INSERT INTO group_user (group_id, user_id) VALUES (?, ?)";
    $stmt = $mysqli->prepare($query);
    $stmt->bind_param("ii", $groupId, $userId);
    $stmt->execute();
}

public function subscribe(ConnectionInterface $conn, $topic) {
    // 订阅某个主题
    // ...

    // 将订阅信息保存到数据库
    $userId = $_SESSION['user_id'];
    $query = "INSERT INTO subscriptions (user_id, topic) VALUES (?, ?)";
    $stmt = $mysqli->prepare($query);
    $stmt->bind_param("is", $userId, $topic);
    $stmt->execute();
}

我们还可以实现一些其他的功能，比如从数据库中获取用户所在的群组或订阅的主题，并将收到的消息发送给相关的连接。
通过以上的实现，我们已经成功地在PHP开发的实时聊天系统中实现了用户群组和订阅功能。这样，用户就可以方便地创建和加入群组，并且能够订阅感兴趣的内容，在实时聊天系统中获得更好的交流体验。