MySQL資料庫對每個客戶端連線都會分配一個執行緒，所以連線非常寶貴。開發一個非同步的MySQL代理伺服器，PHP應用伺服器可以長連線到這臺Server，既減輕MYSQL的連線壓力，又使PHP保持長連線減少connect/close的網路開銷。
此Server考慮到了設定了資料庫連線池尺寸，區分忙閒，mysqli斷線重連，並設定了負載保護。基於swoole擴充套件開發，io迴圈使用epoll，是全非同步非阻塞的，可以應對大量TCP連線。
程式的邏輯是：啟動時建立N個MySQL連線，收到客戶端發來的SQL後，分配1個MySQL連線，將SQL發往資料庫伺服器。然後等待資料庫返回查詢結果。當資料庫返回結果後，再發給對應的客戶端連線。
核心的資料結構是3個PHP陣列。idle_pool是空閒的資料庫連線，當有SQL請求時從idle_pool中移到busy_pool中。當資料庫返回結果後從busy_pool中再移到idle_pool中，以供新的請求使用。當SQL請求到達時如果沒有空閒的資料庫連線，那會自動加入到wait_queue中。一旦有SQL完成操作，將自動從wait_queue中取出等待的請求進行處理。
如此迴圈使用。由於整個伺服器是非同步的單程式單執行緒所以完全不需要鎖。而且是完全非同步的，效率非常高。
當然本文的程式碼，如果要用於生產環境，還需做更多的保護機制和壓力測試。在此僅拋磚引玉，提供一個解決問題的思路。
|
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 |
class DBServer { protected $pool_size = 20; protected $idle_pool = array(); //空閒連線 protected $busy_pool = array(); //工作連線 protected $wait_queue = array(); //等待的請求 protected $wait_queue_max = 100; //等待佇列的最大長度，超過後將拒絕新的請求 /** * <a href='http://www.jobbole.com/members/variable'>@var</a> swoole_server */ protected $serv; function run() { $serv = new swoole_server("127.0.0.1", 9509); $serv->set(array( 'worker_num' => 1, )); $serv->on('WorkerStart', array($this, 'onStart')); //$serv->on('Connect', array($this, 'onConnect')); $serv->on('Receive', array($this, 'onReceive')); //$serv->on('Close', array($this, 'onClose')); $serv->start(); } function onStart($serv) { $this->serv = $serv; for ($i = 0; $i < $this->pool_size; $i++) { $db = new mysqli; $db->connect('127.0.0.1', 'root', 'root', 'test'); $db_sock = swoole_get_mysqli_sock($db); swoole_event_add($db_sock, array($this, 'onSQLReady')); $this->idle_pool[] = array( 'mysqli' => $db, 'db_sock' => $db_sock, 'fd' => 0, ); } echo "Server: start.Swoole version is [" . SWOOLE_VERSION . "]\n"; } function onSQLReady($db_sock) { $db_res = $this->busy_pool[$db_sock]; $mysqli = $db_res['mysqli']; $fd = $db_res['fd']; echo __METHOD__ . ": client_sock=$fd|db_sock=$db_sock\n"; if ($result = $mysqli->reap_async_query()) { $ret = var_export($result->fetch_all(MYSQLI_ASSOC), true) . "\n"; $this->serv->send($fd, $ret); if (is_object($result)) { mysqli_free_result($result); } } else { $this->serv->send($fd, sprintf("MySQLi Error: %s\n", mysqli_error($mysqli))); } //release mysqli object $this->idle_pool[] = $db_res; unset($this->busy_pool[$db_sock]); //這裡可以取出一個等待請求 if (count($this->wait_queue) > 0) { $idle_n = count($this->idle_pool); for ($i = 0; $i < $idle_n; $i++) { $req = array_shift($this->wait_queue); $this->doQuery($req['fd'], $req['sql']); } } } function onReceive($serv, $fd, $from_id, $data) { //沒有空閒的資料庫連線 if (count($this->idle_pool) == 0) { //等待佇列未滿 if (count($this->wait_queue) < $this->wait_queue_max) { $this->wait_queue[] = array( 'fd' => $fd, 'sql' => $data, ); } else { $this->serv->send($fd, "request too many, Please try again later."); } } else { $this->doQuery($fd, $data); } } function doQuery($fd, $sql) { //從空閒池中移除 $db = array_pop($this->idle_pool); /** * <a href='http://www.jobbole.com/members/variable'>@var</a> mysqli */ $mysqli = $db['mysqli']; for ($i = 0; $i < 2; $i++) { $result = $mysqli->query($sql, MYSQLI_ASYNC); if ($result === false) { if ($mysqli->errno == 2013 or $mysqli->errno == 2006) { $mysqli->close(); $r = $mysqli->connect(); if ($r === true) continue; } } break; } $db['fd'] = $fd; //加入工作池中 $this->busy_pool[$db['db_sock']] = $db; } } $server = new DBServer(); $server->run(); |