# 協程：實現 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); }); ~~~