V2EX = way to explore
V2EX 是一个关于分享和探索的地方
现在注册
已注册用户请  登录
yurun
V2EX  ›  PHP

Swoole 自定义项目初始化事件处理的实现

  •  
  •   yurun · 2018-09-13 14:41:46 +08:00 · 1840 次点击
    这是一个创建于 2023 天前的主题,其中的信息可能已经有所发展或是发生改变。

    最近使用基于 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

    4 条回复    2018-09-17 10:11:28 +08:00
    zn
        1
    zn  
       2018-09-13 14:51:07 +08:00 via iPhone
    参考价值不大,但贵在真实。
    iRiven
        2
    iRiven  
       2018-09-14 06:42:59 +08:00 via Android
    我们是一个群的 所以帮你顶一下
    sagaxu
        3
    sagaxu  
       2018-09-16 14:45:43 +08:00 via Android
    应该等初始化完成后,再监听端口
    yurun
        4
    yurun  
    OP
       2018-09-17 10:11:28 +08:00
    @sagaxu 主进程里不想提前引入过多的类,后面无法热更新生效
    关于   ·   帮助文档   ·   博客   ·   API   ·   FAQ   ·   我们的愿景   ·   实用小工具   ·   3252 人在线   最高记录 6543   ·     Select Language
    创意工作者们的社区
    World is powered by solitude
    VERSION: 3.9.8.5 · 27ms · UTC 13:10 · PVG 21:10 · LAX 06:10 · JFK 09:10
    Developed with CodeLauncher
    ♥ Do have faith in what you're doing.