Swoole自訂項目初始化事件處理的實現

最近使用基於 Swoole 開發的 imi 框架開發項目，碰到一個需求，就是想要做項目初始化處理。當初始化處理完成前，不想讓 Swoole 處理請求。因為可能有些值沒有載入進來，處理請求極有可能出現問題。

下面給出了思考過程及解決問題的demo程式碼。

首先分析了一下，Swoole 是多進程模式運行的，分為 Master、Manager、Worker 進程。

Master 進程就是我們啟動服務的 cli 命令檔所在進程，在這裡面初始化有一個問題，這裡所有加載的類別、全域變量，其它 Worker 進程裡都可以使用，無法熱重啟生效。

Manager 進程的情況基本上和上面差不多。

那麼只有在 Worker 程序做處理了，但如果寫在 WorkerStart 事件裡，每個 Worker 程序都會去執行。

<strong>WorkerStart</strong> 事件定義：

function onWorkerStart(swoole_server $server, int $worker_id);

$worker_id是一個從0-$worker_num之間的數字，表示這個Worker進程的ID

那這個就好辦了，直接判斷workerid為0的去觸發專案初始化事件。剩下還有一個問題就是，如何在初始化執行完成前，讓所有 Worker 程序暫時都不處理請求。

思考並嘗試了一下，這個問題可以透過協程掛起來解決，demo 程式碼如下：

<?php
 
use Swoole\Coroutine;
 
$http = new swoole_http_server(&#39;127.0.0.1&#39;, 8080);
 
$http->on(&#39;WorkerStart&#39;, function(swoole_http_server $server, $workerId){
    $initFlagFile = __DIR__ . &#39;/init.flag&#39;;
    if(0 === $server->worker_id && (!is_file($initFlagFile) || file_get_contents($initFlagFile) != $server->manager_pid))
    {
        // 处理项目初始化事件
        initApp();
        // 写入文件，保证不再重复触发项目初始化事件
        file_put_contents($initFlagFile, $server->manager_pid);
        // 当前worker进程恢复协程
        resumeCos();
        // 通知其它worker进程
        for($i = 1; $i < $server->setting[&#39;worker_num&#39;]; ++$i)
        {
            $server->sendMessage(&#39;init&#39;, $i);
        }
    }
});
 
$http->on(&#39;PipeMessage&#39;, function(swoole_http_server $server, $srcWorkerId, $data) {
    if(0 === $srcWorkerId && &#39;init&#39; === $data && !defined(&#39;APP_INITED&#39;))
    {
        // 其它worker进程恢复协程
        resumeCos();
    }
});
 
$http->on(&#39;request&#39;, function (swoole_http_request $request, swoole_http_response $response) {
    // 判断未初始化完毕，则挂起协程
    if(!defined(&#39;APP_INITED&#39;))
    {
        $GLOBALS[&#39;WORKER_START_END_RESUME_COIDS&#39;][] = Coroutine::getuid();
        Coroutine::suspend();
    }
    $response->header(&#39;content-type&#39;, &#39;text/html;charset=utf-8&#39;);
    $response->end(&#39;IMI 是一款基于 Swoole 开发的协程 PHP 开发框架，拥有常驻内存、协程异步非阻塞IO等优点。官方网站：<a href="https://imiphp.com" target="_blank">https://imiphp.com</a>&#39;);
});
 
$http->start();
 
/**
 * 处理项目初始化事件，比如这里延时5秒，模拟初始化处理
 *
 * @return void
 */
function initApp()
{
    $count = 5;
    for($i = 0; $i < $count; ++$i)
    {
        echo &#39;initing &#39;, ($i + 1), &#39;/&#39;, $count, PHP_EOL;
        sleep(1);
    }
}
 
/**
 * 恢复协程
 *
 * @return void
 */
function resumeCos()
{
    define(&#39;APP_INITED&#39;, true);
    $coids = $GLOBALS[&#39;WORKER_START_END_RESUME_COIDS&#39;] ?? [];
    fwrite(STDOUT, &#39;suspend co count: &#39; . count($coids) . PHP_EOL);
    foreach($coids as $id)
    {
        Coroutine::resume($id);
    }
}

透過在request 事件中判斷是否初始化完畢，如果沒有初始化完成，則掛起當前協程，將協程ID加入全域變數。

當第0個 worker 進程執行完初始化後，透過向其他 worker 進程發送訊息，喚醒曾經掛起的協程們，在初始化期間進來的請求，這時候會被執行。

以上是Swoole自訂項目初始化事件處理的實現的詳細內容。更多資訊請關注PHP中文網其他相關文章！