php PDO判斷連接是否可用的方法

轉載自：傲雪星楓  原文地址： http://blog.csdn.net/fdipzone/article/details/53117541

mysql_ping() 檢查到服務器的連接是否正常。如果到服務器的連接可用，則返回true，否則返回false。

但PDO不支持mysql_ping()方法，因此需要自己編寫代碼模擬mysql_ping()方法，檢查連接是否可用。

代碼如下：

<?php
/**
 * 檢查連接是否可用
 * @param  Link $dbconn 數據庫連接
 * @return Boolean
 */
function pdo_ping($dbconn){
    try{
        $dbconn->getAttribute(PDO::ATTR_SERVER_INFO);
    } catch (PDOException $e) {
        if(strpos($e->getMessage(), 'MySQL server has gone away')!==false){
            return false;
        }
    }
    return true;
}
?>

代碼演示：

1.創建測試數據表

CREATE TABLE `user` (
 `id` int(11) unsigned NOT NULL AUTO_INCREMENT,
 `name` varchar(20) NOT NULL,
 PRIMARY KEY (`id`)
) ENGINE=InnoDB DEFAULT CHARSET=utf8;

 

2.插入測試數據

insert into user(name) values('fdipzone'),('xfdipzone'),('terry');

mysql> select * from user;
+----+-----------+
| id | name      |
+----+-----------+
|  1 | fdipzone  |
|  2 | xfdipzone |
|  3 | terry     |
+----+-----------+

 

3.演示文件

db.PHP

<?php
// 數據庫操作類
class DB{

    // 保存數據庫連接
    private static $_instance = null;

    // 連接數據庫
    public static function get_conn($config){
        if(isset(self::$_instance) && !empty(self::$_instance)){
            return self::$_instance;
        }

        $dbhost = $config['host'];
        $dbname = $config['dbname'];
        $dbuser = $config['user'];
        $dbpasswd = $config['password'];
        $pconnect = $config['pconnect'];
        $charset = $config['charset'];

        $dsn = "mysql:host=$dbhost;dbname=$dbname;";
        try {
            $h_param = array(
                    PDO::ATTR_ERRMODE => PDO::ERRMODE_EXCEPTION,
            );
            if ($charset != '') {
                $h_param[PDO::MYSQL_ATTR_INIT_COMMAND] = 'SET NAMES ' . $charset; //設置默認編碼
            }
            if ($pconnect) {
                $h_param[PDO::ATTR_PERSISTENT] = true;
            }
            $conn = new PDO($dsn, $dbuser, $dbpasswd, $h_param);

        } catch (PDOException $e) {
            throw new ErrorException('Unable to connect to db server. Error:' . $e->getMessage(), 31);
        }

        self::$_instance = $conn;
        return $conn;
    }

    // 執行查詢
    public static function query($dbconn, $sqlstr, $condparam){
        $sth = $dbconn->prepare($sqlstr);
        try{
            $sth->execute($condparam);
        } catch (PDOException $e) {
            echo $e->getMessage().PHP_EOL;
        }
        $result = $sth->fetchAll(PDO::FETCH_ASSOC);
        return $result;
    }

    // 重置連接
    public static function reset_connect(){
        self::$_instance = null;
    }

}
?>

 

test.php

<?php
require 'db.php';

// 數據庫設定
$config = array(
    'host' => 'localhost',
    'dbname' => 'user',
    'user' => 'root',
    'password' => '',
    'pconnect' => 0,
    'charset' => ''
);

// 循環執行
while(true){

    // 創建數據連接
    $dbconn = DB::get_conn($config);

    // 判斷連接是否有效
    $status = pdo_ping($dbconn);

    if($status){
        echo 'connect ok'.PHP_EOL;
    }else{
        echo 'connect failure'.PHP_EOL;

        // 重置連接
        DB::reset_connect();
        $dbconn = DB::get_conn($config);
    }

    // 執行查詢
    $sqlstr = 'select * from user where id=?';
    $condparam = array(mt_rand(1,3));
    $data = DB::query($dbconn, $sqlstr, $condparam);
    print_r($data);

    // 延時10秒
    echo 'sleep 10'.PHP_EOL.PHP_EOL;
    sleep(10);

}

/**
 * 檢查連接是否可用
 * @param  Link $dbconn 數據庫連接
 * @return Boolean
 */
function pdo_ping($dbconn){
    try{
        $dbconn->getAttribute(PDO::ATTR_SERVER_INFO);
    } catch (PDOException $e) {
        if(strpos($e->getMessage(), 'MySQL server has gone away')!==false){
            return false;
        }
    }
    return true;
}
?>

 

 

4.執行

在php cli模式下執行test.php，然后馬上執行MySQL.server stop 與 mysql.server start 模擬閃斷

mysql.server stop
Shutting down MySQL
.... SUCCESS! 
mysql.server start
Starting MySQL
 SUCCESS! 

執行輸出：

connect ok
Array
(
    [0] => Array
        (
            [id] => 2
            [name] => xfdipzone
        )

)
sleep 10

connect failure
Array
(
    [0] => Array
        (
            [id] => 3
            [name] => terry
        )

)
sleep 10

connect ok
Array
(
    [0] => Array
        (
            [id] => 3
            [name] => terry
        )

)
sleep 10

 

 

可以看到閃斷后，pdo_ping()返回false，連接失敗，然后調用自動重連，保證后面的查詢能繼續執行。