× 警告!旧版文档已经暂停维护,请查看新版文档。点击前往新版文档

协程:实现 sync.WaitGroup 功能

Swoole4中可以使用channel实现协程间的通信、依赖管理、协程同步。基于channel可以很容易地实现Golangsync.WaitGrup功能。

实现代码

class WaitGroup
{
    private $count = 0;
    private $chan;

    /**
     * waitgroup constructor.
     * @desc 初始化一个channel
     */
    public function __construct()
    {
        $this->chan = new chan;
    }

    public function add()
    {
        $this->count++;
    }

    public function done()
    {
        $this->chan->push(true);
    }

    public function wait()
    {
        while($this->count--)
        {
            $this->chan->pop();
        }
    }

}
  • add方法增加计数
  • done表示任务已完成
  • wait等待所有任务完成恢复当前协程的执行
  • WaitGroup对象可以复用,adddonewait之后可以再次使用

使用实例

go(function () {
    $wg = new waitgroup();
    $result = [];

    $wg->add();
    //启动第一个协程
    go(function () use ($wg,  &$result) {
        //启动一个协程客户端client,请求淘宝首页
        $cli = new Client('www.taobao.com', 443, true);
        $cli->setHeaders([
            'Host' => "www.taobao.com",
            "User-Agent" => 'Chrome/49.0.2587.3',
            'Accept' => 'text/html,application/xhtml+xml,application/xml',
            'Accept-Encoding' => 'gzip',
        ]);
        $cli->set(['timeout' => 1]);
        $cli->get('/index.php');

        $result['taobao'] = $cli->body;
        $cli->close();

        $wg->done();
    });

    $wg->add();
    //启动第二个协程
    go(function () use ($wg, &$result) {
        //启动一个协程客户端client,请求百度首页
        $cli = new Client('www.baidu.com', 443, true);
        $cli->setHeaders([
            'Host' => "www.baidu.com",
            "User-Agent" => 'Chrome/49.0.2587.3',
            'Accept' => 'text/html,application/xhtml+xml,application/xml',
            'Accept-Encoding' => 'gzip',
        ]);
        $cli->set(['timeout' => 1]);
        $cli->get('/index.php');

        $result['baidu'] = $cli->body;
        $cli->close();

        $wg->done();
    });

    //挂起当前协程,等待所有任务完成后恢复
    $wg->wait();
    //这里 $result 包含了 2 个任务执行结果
    var_dump($result);
});

  • 小鲍勃

    "在Swoole4中可以使用channel实现协程间的通信、依赖管理、协程同步。基于channel可以很容易地实现Golang的sync.WaitGrup功能 " 有个单词漏了 sync.WaitGroup

  • 西门

    这个功能还是很实用的,之前实用go语言写的时候就感觉非常棒,现在能在swoole中看到,很6

  • 叶柏寿

    & 是什么意思?, 执行的话这里报错 PHP Parse error: syntax error, unexpected 'amp' (T_STRING), expecting variable (T_VARIABLE) in /var/swoole/waitGroup.php on line 38

  • 远离颠倒梦想

    $wg, &$result 这存在多久了。

  •    意欣

    swoole封装的waitgroup的例子,类似于golang的sync.waitgroup,但是swoole的waitgroup跟golang的又略有不同,golang底层的实现使用了runtime/race做数据竞态的校验,而例子中仅仅只是维护一个count跟对应的chan,是chan底层做了竞态的校验呢还是这里疏忽了呢