協(xié)程等待組 WaitGroup

WaitGroup與Barrier相似，是一個(gè)用于協(xié)程同步的工具，允許在異步任務(wù)中等待所有協(xié)程執(zhí)行完成后再繼續(xù)后續(xù)邏輯。

與Barrier區(qū)別的是WaitGroup可以由開發(fā)者自行控制計(jì)數(shù)的增加和減少。

注意
底層自動(dòng)識(shí)別驅(qū)動(dòng)類型，僅支持Swoole/Swow/Fiber驅(qū)動(dòng)

提示
此特性需要 workerman>=5.1.0

<?php
use Workerman\Connection\TcpConnection;
use Workerman\Coroutine\Barrier;
use Workerman\Coroutine;
use Workerman\Events\Swoole;
use Workerman\Protocols\Http\Request;
use Workerman\Worker;
require_once __DIR__ . '/vendor/autoload.php';

// Http Server
$worker = new Worker('http://0.0.0.0:8001');
$worker->eventLoop = Swoole::class; // Or Swow::class or Fiber::class
$worker->onMessage = function (TcpConnection $connection, Request $request) {
    $wg = new WaitGroup();
    for ($i=1; $i<5; $i++) {
        $wg->add();
        Coroutine::create(function () use ($wg, $i) {
            try {
            // Do something
            } finally {
                $wg->done();
            }
        });
    }
    // Wait all coroutine done, timout after 10 seconds
    $result = $wg->wait(10.0);
    if (!$result) {
        $connection->send('WaitGroup Timeout');
        return;
    }
    $connection->send('All Task Done');
};

Worker::runAll();

接口說明

interface WaitGroupInterface
{

    /**
     * 增加計(jì)數(shù)
     *
     * @param int $delta
     * @return bool
     */
    public function add(int $delta = 1): bool;

    /**
     * 完成計(jì)數(shù)
     *
     * @return bool
     */
    public function done(): bool;

    /**
     * 返回計(jì)數(shù)
     *
     * @return int
     */
    public function count(): int;

    /**
     * 協(xié)程等待
     *
     * @param int|float $timeout second
     * @return bool timeout:false success:true
     */
    public function wait(int|float $timeout = -1): bool;
}