协程：实现 sync.WaitGroup 功能
最后更新于：2022-04-02 06:16:36
# 协程：实现 sync.WaitGroup 功能
[TOC]
在`Swoole4`中可以使用`channel`实现协程间的通信、依赖管理、协程同步。基于`channel`可以很容易地实现`Golang`的`sync.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`对象可以复用，`add`、`done`、`wait`之后可以再次使用
## 使用实例
~~~
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);
});
~~~
';