Swoole 協程的併發呼叫及使用示例

阿新 • • 發佈：2020-07-15

示例一：

利用通道pop會自動掛起當前協程，等待生產者推送資料的特性，實現併發呼叫，並在協程完成後組合結果集。

$serv = new Swoole\Http\Server("127.0.0.1", 9503, SWOOLE_BASE);

$serv->on('request', function ($req, $resp) {
    $chan = new Swoole\Coroutine\Channel(2);
    go(function () use ($chan) {
        $cli = new Swoole\Coroutine\Http\Client('www.qq.com', 80);
        $cli->set(['timeout' => 10]);
        $cli->setHeaders([
            'Host' => "www.qq.com",
            'User-Agent' => 'Chrome/49.0.2587.3',
            'Accept' => 'text/html,application/xhtml+xml,application/xml',
            'Accept-Encoding' => 'gzip',
        ]);
        $ret = $cli->get('/');
        $chan->push(['www.qq.com' => $cli->body]);
    });

    go(function () use ($chan) {
        $cli = new Swoole\Coroutine\Http\Client('www.baidu.com', 80);
        $cli->set(['timeout' => 10]);
        $cli->setHeaders([
            'Host' => "www.baidu.com",
            'User-Agent' => 'Chrome/49.0.2587.3',
            'Accept' => 'text/html,application/xhtml+xml,application/xml',
            'Accept-Encoding' => 'gzip',
        ]);
        $ret = $cli->get('/');
        $chan->push(['www.baidu.com' => $cli->body]);
    });

    $result = [];
    for ($i = 0; $i < 2; $i++)
    {
        // 當通道為空時，會自動掛起當前協程，等待生產者推送資料後，重新排程進來
        $result += $chan->pop();
    }
    $resp->end(json_encode($result));
});
$serv->start();

示例二：

利用Swoole封裝好的WaitGroup類，實現併發呼叫，並在協程完成後組合結果集。

底層也是基於通道的計數、push和pop實現的。

<?php
Co\run(function () {
    $wg = new \Swoole\Coroutine\WaitGroup();
    $result = [];

    // 增加第一個計數
    $wg->add();
    // 啟動第一個協程
    go(function () use ($wg, &$result) {
        //啟動一個協程客戶端client，請求淘寶首頁
        $cli = new \Swoole\Coroutine\Http\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 \Swoole\Coroutine\Http\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);
});