1. 程式人生 > >TaskWorker 實現非同步操作 Mysql

TaskWorker 實現非同步操作 Mysql

在一般的 Server 程式中都會有一些耗時的任務,比如:傳送郵件、聊天伺服器傳送廣播等。如果我們採用同步阻塞的防水去執行這些任務,那麼這肯定會非常的慢。

Swoole 的 TaskWorker 程序池可以用來執行一些非同步的任務,而且不會影響接下來的任務,很適合處理以上場景。

那麼什麼是非同步任務呢?

可以從下面的圖示中來簡單瞭解一下。(來源於網路,侵刪)

我們上一個 Swoole 的文章介紹瞭如何建立一個簡單的伺服器,並且知道了幾個核心的回撥函式的使用方法。

要實現上述的非同步處理,只需要增加兩個事件回撥即可:onTask 和 onFinish, 這兩個回撥函式分別用於執行 Task 任務和處理 Task 任務的返回結果。另外還需要在 set 方法中設定 task 程序數量。

使用示例:

<?php
class Server
{
    private $serv;
    public function __construct() {
        $this->serv = new swoole_server("0.0.0.0", 9501);
        $this->serv->set(array(
            'worker_num' => 4,
            'daemonize' => false,
            'task_worker_num' => 8 // task程序數量 即為維持的MySQL連線的數量
        ));
        $this->serv->on('Start', array($this, 'onStart'));
        $this->serv->on('Connect', array($this, 'onConnect'));
        $this->serv->on('Receive', array($this, 'onReceive'));
        $this->serv->on('Close', array($this, 'onClose'));
        $this->serv->on('Task', array($this, 'onTask'));
        $this->serv->on('Finish', array($this, 'onFinish'));
        $this->serv->start();
    }

    public function onReceive( swoole_server $serv, $fd, $from_id, $data ) {
        echo "收到資料". $data . PHP_EOL;
        // 傳送任務到Task程序
        $param = array(
            'sql' => $data, // 接收客戶端傳送的 sql 
            'fd'  => $fd
        );
        $serv->task( json_encode( $param ) );  // 向 task 投遞任務
        echo "繼續處理之後的邏輯\n";
    }

    public function onTask($serv, $task_id, $from_id, $data) {
    	$data = json_decode($data, true);
        echo "This Task {$task_id} from Worker {$from_id}\n";
        echo "recv SQL: {$data['sql']}\n";
        static $link = null;
        $sql = $data['sql'];
        $fd  = $data['fd'];
        HELL:
        if ($link == null) {
            $link = @mysqli_connect("127.0.0.1", "root", "root", "test");
        }
        $result = $link->query($sql);
        if (!$result) { //如果查詢失敗
            if(in_array(mysqli_errno($link), [2013, 2006])){
                //錯誤碼為2013,或者2006,則重連資料庫,重新執行sql
                    $link = null;
                    goto HELL;
            }
        }
        if(preg_match("/^select/i", $sql)){//如果是select操作,就返回關聯陣列
             $data = array();
                while ($fetchResult = mysqli_fetch_assoc($result) ){
                     $data['data'][] = $fetchResult;
                }                
        }else{//否則直接返回結果
            $data['data'] = $result;
        }
        $data['status'] = "OK";
        $data['fd'] = $fd;
        $serv->finish(json_encode($data));
    }
    public function onFinish($serv, $task_id, $data) {
        echo "Task {$task_id} finish\n";
        $result = json_decode($data, true);
        if ($result['status'] == 'OK') {
            $this->serv->send($result['fd'], json_encode($result['data']) . "\n");
        } else {
            $this->serv->send($result['fd'], $result);
        }
    }
    public function onStart( $serv ) {
        echo "Server Start\n";
    }
    public function onConnect( $serv, $fd, $from_id ) {
        echo "Client {$fd} connect\n";
    }
    public function onClose( $serv, $fd, $from_id ) {
        echo "Client {$fd} close connection\n";
    }
}
$server = new Server();

以上程式碼在 onReceive 時直接接收一條 sql,之後直接傳送到 Task 任務中。這個時候下一步的流程緊接著輸出,這裡也就體現出了非同步。然後 onTask 和 onFinish 分別用來向資料庫傳送 sql,處理 task 執行結果。

參考連結: