Swoole 自定義項(xiàng)目初始化事件處理的實(shí)現(xiàn)

最近使用基于 Swoole 開(kāi)發(fā)的 imi 框架開(kāi)發(fā)項(xiàng)目，碰到一個(gè)需求，就是想要做項(xiàng)目初始化處理。當(dāng)初始化處理完成前，不想讓 Swoole 處理請(qǐng)求。因?yàn)榭赡苡幸恍┲禌](méi)有加載進(jìn)來(lái)，處理請(qǐng)求極有可能出現(xiàn)問(wèn)題。

下面給出了思考過(guò)程及解決問(wèn)題的demo代碼。

首先分析了一下，Swoole 是多進(jìn)程模式運(yùn)行的，分為 Master、Manager、Worker 進(jìn)程。

Master 進(jìn)程就是我們啟動(dòng)服務(wù)的 cli 命令文件所在進(jìn)程，在這里面初始化有一個(gè)問(wèn)題，這里所有加載的類(lèi)、全局變量，其它 Worker 進(jìn)程里都可以使用，無(wú)法熱重啟生效。

Manager 進(jìn)程的情況基本和上面差不多。

那么只有在 Worker 進(jìn)程做處理了，但如果寫(xiě)在 WorkerStart 事件里，每個(gè) Worker 進(jìn)程都會(huì)去執(zhí)行。

WorkerStart 事件定義：

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

$worker_id是一個(gè)從0-$worker_num之間的數(shù)字，表示這個(gè)Worker進(jìn)程的ID

那這個(gè)就好辦了，直接判斷workerid為0的去觸發(fā)項(xiàng)目初始化事件。剩下還有一個(gè)問(wèn)題就是，如何在初始化執(zhí)行完成前，讓所有 Worker 進(jìn)程暫時(shí)都不處理請(qǐng)求。

思考并嘗試了一下，這個(gè)問(wèn)題可以通過(guò)協(xié)程掛起來(lái)解決，demo 代碼如下：

<?php
 
use Swoole\Coroutine;
 
$http = new swoole_http_server('127.0.0.1', 8080);
 
$http->on('WorkerStart', function(swoole_http_server $server, $workerId){
    $initFlagFile = __DIR__ . '/init.flag';
    if(0 === $server->worker_id && (!is_file($initFlagFile) || file_get_contents($initFlagFile) != $server->manager_pid))
    {
        // 處理項(xiàng)目初始化事件
        initApp();
        // 寫(xiě)入文件，保證不再重復(fù)觸發(fā)項(xiàng)目初始化事件
        file_put_contents($initFlagFile, $server->manager_pid);
        // 當(dāng)前worker進(jìn)程恢復(fù)協(xié)程
        resumeCos();
        // 通知其它worker進(jìn)程
        for($i = 1; $i < $server->setting['worker_num']; ++$i)
        {
            $server->sendMessage('init', $i);
        }
    }
});
 
$http->on('PipeMessage', function(swoole_http_server $server, $srcWorkerId, $data) {
    if(0 === $srcWorkerId && 'init' === $data && !defined('APP_INITED'))
    {
        // 其它worker進(jìn)程恢復(fù)協(xié)程
        resumeCos();
    }
});
 
$http->on('request', function (swoole_http_request $request, swoole_http_response $response) {
    // 判斷未初始化完畢，則掛起協(xié)程
    if(!defined('APP_INITED'))
    {
        $GLOBALS['WORKER_START_END_RESUME_COIDS'][] = Coroutine::getuid();
        Coroutine::suspend();
    }
    $response->header('content-type', 'text/html;charset=utf-8');
    $response->end('IMI 是一款基于 Swoole 開(kāi)發(fā)的協(xié)程 PHP 開(kāi)發(fā)框架，擁有常駐內(nèi)存、協(xié)程異步非阻塞IO等優(yōu)點(diǎn)。官方網(wǎng)站：<a  target="_blank">https://imiphp.com</a>');
});
 
$http->start();
 
/**
 * 處理項(xiàng)目初始化事件，比如這里延時(shí)5秒，模擬初始化處理
 *
 * @return void
 */
function initApp()
{
    $count = 5;
    for($i = 0; $i < $count; ++$i)
    {
        echo 'initing ', ($i + 1), '/', $count, PHP_EOL;
        sleep(1);
    }
}
 
/**
 * 恢復(fù)協(xié)程
 *
 * @return void
 */
function resumeCos()
{
    define('APP_INITED', true);
    $coids = $GLOBALS['WORKER_START_END_RESUME_COIDS'] ?? [];
    fwrite(STDOUT, 'suspend co count: ' . count($coids) . PHP_EOL);
    foreach($coids as $id)
    {
        Coroutine::resume($id);
    }
}

通過(guò)在 request 事件中判斷是否初始化完畢，如果沒(méi)有初始化完成，則掛起當(dāng)前協(xié)程，將協(xié)程ID加入全局變量。

當(dāng)?shù)?個(gè) worker 進(jìn)程執(zhí)行完初始化后，通過(guò)向其他 worker 進(jìn)程發(fā)送消息，喚醒曾經(jīng)掛起的協(xié)程們，在初始化期間進(jìn)來(lái)的請(qǐng)求，這時(shí)候會(huì)被執(zhí)行。