swoole學習之: 協程初探(二)

繼續昨天的協程初探, 我們繼續來編寫數據庫等操作。

首先我們準備好數據庫賬號及表, 數據:

CREATE USER 'test'@'%' IDENTIFIED BY 'test';
GRANT ALL PRIVILEGES ON *.* TO 'test'@'%';
FLUSH PRIVILEGES;

我們建立了一個mysql的賬號test, 並設置host爲%即不限制客戶端連接, 然後賦予操作當前服務器所有數據庫的所有權限。

現在來建立數據庫和表:

CREATE DATABASE `test` DEFAULT CHARACTER SET utf8mb4; -- 如果已經有了test數據庫, 請忽略
USE test;
-- 建表 users
CREATE TABLE `users`(
    `user_id` INT UNSIGNED AUTO_INCREMENT,
    `user_name` varchar(20) NOT NULL,
    `create_time` INT NOT NULL,
    PRIMARY KEY(`user_id`)
) ENGINE=InnoDB COMMENT '用戶表';

-- 創建生成隨機字符串的函數
DROP FUNCTION IF EXISTS rand_string;
DELIMITER $$
CREATE FUNCTION rand_string(n INT) RETURNS varchar(255)
BEGIN
    DECLARE chars varchar(100) DEFAULT 'abcdefghijklmlopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789';
    DECLARE rt varchar(255) DEFAULT '';
    DECLARE i INT DEFAULT 0;
    WHILE i < n DO
        SET rt = CONCAT(rt, SUBSTRING(chars, FLOOR(rand() * 62 + 1), 1));
        SET i = i+1;
    END WHILE;
    RETURN rt;
END $$
DELIMITER ;
-- 如果創建函數時報錯Error Code: 14181, 請執行: set global log_bin_trust_function_creators=TRUE;

-- 編寫存儲過程, 實現批量插入數據功能
DROP PROCEDURE IF EXISTS mysp_batch_insert_users;
DELIMITER $$
CREATE PROCEDURE mysp_batch_insert_users(IN n INT)
BEGIN
    DECLARE i INT DEFAULT 0;
    SET autocommit = 0;-- 關閉默認的自動提交, 可極大的提高插入速度
    WHILE i < n DO
        SET i = i + 1;
        INSERT INTO `users`(user_name, create_time)
        VALUES(rand_string(5), UNIX_TIMESTAMP(now()));
    END WHILE;
    COMMIT;
END $$
DELIMITER ;

-- 執行存儲過程, 插入10k數據
CALL mysp_batch_insert_users(10000);

注意上面在大批量插入數據時, 使用了SET autocommit = 0設置, 我們可以控制何時commit, 在填充測試數時很有用, 比每次循環插入多次INSRET INTO tb()VALUES(),...都要快很多很多, 普通ssd一分鐘可以生成60w+數據

ok, 數據已經準備好，是時候讓php上場了（其實它纔是主角？）

use Swoole\Runtime;
use Swoole\Coroutine;
use function Swoole\Coroutine\run;

//開啓協程. 自動將文件操作,sleep,mysqli,pdo,streams等都變成異步IO
Runtime::enableCoroutine();

$time_start = microtime(true);

//Swoole\Coroutine\run()
run(function(){
    //5k PDO read
    echo 'PDO'.PHP_EOL;
    for($n = 50; $n --;){
        Coroutine::create(function(){
            $pdo = new PDO('mysql:host=127.0.0.1;dbname=test;charset=utf8', 'test', 'test');
            $statement = $pdo->prepare('SELECT * FROM `users` LIMIT 10');
            for($n2 = 100; $n2 --;){
                $statement->execute();
                assert(count($statement->fetchAll()) > 0); //如果獲取到的數據個數爲0, 則終止
            }
        });
        echo ($n+1) . "\t";
    }
    //5k MySQLi read
    echo PHP_EOL.'mysqli'.PHP_EOL;
    for($n = 50; $n--;){
        Coroutine::create(function(){
            $mysqli = new mysqli('127.0.0.1', 'test', 'test', 'test');
            $statement = $mysqli->prepare('SELECT user_id FROM `users` LIMIT 10');
            for($n2 = 100; $n2 --;){
                $statement->bind_result($user_id);
                $statement->execute();
                $statement->fetch();
                assert($user_id > 0);//如果user_id<=0, 則終止
            }
        });
        echo ($n+1) . "\t";
    }
});

$time_end = microtime(true);
echo 'Time elapsed: '.($time_end - $time_start).' s'.PHP_EOL;

在上面的php代碼實際執行過程中, 因爲表中有1w數據, 如果不做limit, 時間在40s以上, 做了limit 10或者limit 1, 時間都是在零點幾秒.

我們以不使用協程的做比較, 代碼如下:

$time_start = microtime(true);

//5k PDO read
echo 'PDO'.PHP_EOL;
for($n = 50; $n --;){
    $pdo = new PDO('mysql:host=127.0.0.1;dbname=test;charset=utf8', 'test', 'test');
    $statement = $pdo->prepare('SELECT * FROM `users` LIMIT 10');
    for($n2 = 100; $n2 --;){
        $statement->execute();
        assert(count($statement->fetchAll()) > 0); //如果獲取到的數據個數爲0, 則終止
    }
    echo ($n+1) . "\t";
}
//5k MySQLi read
echo PHP_EOL.'mysqli'.PHP_EOL;
for($n = 50; $n--;){
    $mysqli = new mysqli('127.0.0.1', 'test', 'test', 'test');
    $statement = $mysqli->prepare('SELECT user_id FROM `users` LIMIT 10');
    for($n2 = 100; $n2 --;){
        $statement->bind_result($user_id);
        $statement->execute();
        $statement->fetch();
        assert($user_id > 0);//如果user_id<=0, 則終止
    }
    echo ($n+1) . "\t";
}

$time_end = microtime(true);
echo 'Time elapsed: '.($time_end - $time_start).' s'.PHP_EOL;

下圖是使用協程和同步處理的執行時間比較: