隨著現(xiàn)代web應(yīng)用程式的不斷發(fā)展，即時(shí)通訊成為了必須的功能之一。而WebSocket 技術(shù)就是其中一種最受歡迎的即時(shí)通訊方式，能夠在伺服器和用戶端之間建立持久的雙向連接，實(shí)現(xiàn)即時(shí)通訊。

ThinkPHP 是一個(gè)非常流行的 PHP 框架，ThinkPHP 6 自帶了 Swoole 擴(kuò)充包，使得在使用 WebSocket 技術(shù)時(shí)變得非常簡(jiǎn)單。本文將介紹如何使用 ThinkPHP 6 框架來(lái)實(shí)作一個(gè) WebSocket 廣播功能。

環(huán)境需求

在開(kāi)始之前，需要準(zhǔn)備以下環(huán)境：

• PHP 7.2
• Swoole 擴(kuò)充功能
##Composer
• 基本的Web 開(kāi)發(fā)知識(shí)

composer create-project topthink/think my-project

composer.json 檔案中新增Swoole 擴(kuò)充功能：

"require": {
    "topthink/think-swoole": "^2.0",
    "swoole/swoole": "^4.7"
}

composer install

app/controller 目錄下建立一個(gè)名為Websocket.php 的文件，寫入以下程式碼：

<?php

declare(strict_types=1);

namespace appcontroller;

use thinkswooleWebsocket as SwooleWebsocket;
use SwooleWebSocketFrame;

class Websocket extends SwooleWebsocket
{
    /**
     * 監(jiān)聽(tīng)連接事件
     * @param SwooleWebSocketServer $server
     * @param SwooleHttpRequest $request
     */
    public function onOpen($server, $request)
    {
    }

    /**
     * 監(jiān)聽(tīng)接收消息事件
     * @param SwooleWebSocketServer $server
     * @param Frame $frame
     */
    public function onMessage($server, Frame $frame)
    {
    }

    /**
     * 監(jiān)聽(tīng)關(guān)閉事件
     * @param SwooleWebSocketServer $server
     * @param int $fd
     */
    public function onClose($server, $fd)
    {
    }
}

thinkswooleWebsocket 類，並重寫了其中的三個(gè)方法：

• onOpen 方法用於監(jiān)聽(tīng)連接事件；
• # onMessage 方法用於監(jiān)聽(tīng)接收訊息事件；
• onClose 方法用於監(jiān)聽(tīng)關(guān)閉事件。

app/route.php 檔案中加入以下內(nèi)容：

use thinkacadeRoute;

Route::post('/ws', 'Websocket@onMessage')->middleware(    hinkmiddlewareAllowCrossDomain::class);

Route::post 方法來(lái)註冊(cè)路由。這個(gè)路由的請(qǐng)求方式是 POST，路徑為 /ws，並將請(qǐng)求對(duì)應(yīng)到了 Websocket 控制器的 onMessage 方法上。

onOpen 方法中，我們可以取得客戶端的連接對(duì)象，並將其儲(chǔ)存起來(lái)，以便後續(xù)使用。在 onMessage 方法中，我們可以取得客戶端所傳送的訊息，並將這則訊息廣播給其它客戶端。在 onClose 方法中，我們需要將客戶端從連線池中移除。

app/controller 目錄下建立一個(gè)名為WebSocketServer.php 的文件，寫入以下程式碼：

<?php

declare(strict_types=1);

namespace appcontroller;

use SwooleHttpResponse;
use SwooleWebSocketFrame;
use SwooleWebSocketServer;
use thinkswoolewebsocketHandlerInterface;

class WebSocketServer implements HandlerInterface
{
    /**
     * @var array $connections
     */
    private $connections = [];

    /**
     * 監(jiān)聽(tīng)連接事件
     * @param Server $server
     * @param SwooleHttpRequest $request
     */
    public function onOpen(Server $server, SwooleHttpRequest $request): void
    {
        $this->connections[$request->fd] = $request->fd;

        echo "client-{$request->fd} is connected
";
    }

    /**
     * 監(jiān)聽(tīng)消息事件
     * @param Server $server
     * @param Frame $frame
     */
    public function onMessage(Server $server, Frame $frame): void
    {
        foreach ($this->connections as $fd) {
            $info = $server->getClientInfo((int)$fd);
            if ($info && isset($info['websocket_status']) && $info['websocket_status'] == WEBSOCKET_STATUS_FRAME) {
                $server->push($fd, $frame->data);
            } else {
                unset($this->connections[$fd]);
            }
        }

        echo "received message from client-{$frame->fd}: {$frame->data}
";
    }

    /**
     * 監(jiān)聽(tīng)關(guān)閉事件
     * @param Server $server
     * @param int $fd
     * @param int $reactorId
     */
    public function onClose(Server $server, int $fd, int $reactorId): void
    {
        unset($this->connections[$fd]);

        echo "client-{$fd} is closed
";
    }

    /**
     * @param Response $response
     */
    public function onHandShake(Request $request, Response $response): bool
    {
        // Do nothing
        return true;
    }
}

config 目錄下建立一個(gè)名為swoole.php 的設(shè)定文件，寫入以下內(nèi)容：

return [
    'socket_type' => 'websocket',
    'host'        => '0.0.0.0',
    'port'        => 9501,
    'mode'        => SWOOLE_PROCESS,
    'sock_type'   => SWOOLE_SOCK_TCP,
    'settings' => [
        'worker_num' => swoole_cpu_num(),
    ],
    'handler' => ppcontrollerWebSocketServer::class,
];

websocket socket 類型，綁定在 0.0.0.0:9501 上，並開(kāi)啟了多進(jìn)程模式，使用 TCP 協(xié)定。 worker_num 配置項(xiàng)目設(shè)定了伺服器的進(jìn)程數(shù)，這裡使用了swoole_cpu_num() 用於返回系統(tǒng)CPU 數(shù)量；handler 配置項(xiàng)目指定了我們?cè)谏衔闹薪⒌?/code>WebSocketServer 類別。

php think swoole start --mode=websocket