Swoole自定义项目初始化事件处理如何实现（代码）

本篇文章给大家带来的内容是关于swoole自定义项目初始化事件处理如何实现（代码），有一定的参考价值，有需要的朋友可以参考一下，希望对你有所帮助。

最近使用基于 Swoole 开发的 imi 框架 开发项目，碰到一个需求，就是想要做项目初始化处理。当初始化处理完成前，不想让 Swoole 处理请求。因为可能有一些值没有加载进来，处理请求极有可能出现问题。

下面给出了思考过程及解决问题的demo代码。

首先分析了一下，Swoole 是多进程模式运行的，分为 Master、Manager、Worker 进程。

Master 进程就是我们启动服务的 cli 命令文件所在进程，在这里面初始化有一个问题，这里所有加载的类、全局变量，其它 Worker 进程里都可以使用，无法热重启生效。

Manager 进程的情况基本和上面差不多。

那么只有在 Worker 进程做处理了，但如果写在 WorkerStart 事件里，每个 Worker 进程都会去执行。

WorkerStart 事件定义：

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

那这个就好办了，直接判断workerid为0的去触发项目初始化事件。剩下还有一个问题就是，如何在初始化执行完成前，让所有 Worker 进程暂时都不处理请求。

WeShop唯象

WeShop唯象是国内首款AI商拍工具，专注电商产品图片的智能生成。

下载

思考并尝试了一下，这个问题可以通过协程挂起来解决，demo 代码如下：

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))
    {
        // 处理项目初始化事件
        initApp();
        // 写入文件，保证不再重复触发项目初始化事件
        file_put_contents($initFlagFile, $server->manager_pid);
        // 当前worker进程恢复协程
        resumeCos();
        // 通知其它worker进程
        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进程恢复协程
        resumeCos();
    }
});

$http->on('request', function (swoole_http_request $request, swoole_http_response $response) {
    // 判断未初始化完毕，则挂起协程
    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 开发的协程 PHP 开发框架，拥有常驻内存、协程异步非阻塞IO等优点。官方网站：https://imiphp.com');
});

$http->start();

/**
 * 处理项目初始化事件，比如这里延时5秒，模拟初始化处理
 *
 * @return void
 */
function initApp()
{
    $count = 5;
    for($i = 0; $i < $count; ++$i)
    {
        echo 'initing ', ($i + 1), '/', $count, PHP_EOL;
        sleep(1);
    }
}

/**
 * 恢复协程
 *
 * @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);
    }
}

通过在 request 事件中判断是否初始化完毕，如果没有初始化完成，则挂起当前协程，将协程ID加入全局变量。

当第0个 worker 进程执行完初始化后，通过向其他 worker 进程发送消息，唤醒曾经挂起的协程们，在初始化期间进来的请求，这时候会被执行。

相关推荐：

详解javascript实现自定义事件_javascript技巧

Angularjs自定义指令在项目中如何使用