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);

$worker_id 是一个从 0-$worker_num 之间的数字，表示这个 Worker 进程的 ID

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

思考并尝试了一下，这个问题可以通过协程挂起来解决，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))
	{
		// 处理项目初始化事件
		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 等优点。官方网站：<a href="https://imiphp.com" target="_blank">https://imiphp.com</a>');
});

$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 进程发送消息，唤醒曾经挂起的协程们，在初始化期间进来的请求，这时候会被执行。

原文地址： https://blog.yurunsoft.com/a/121.html