PHP利用MySQL保存session

實現環境：

PHP 5.4.24
MySQL 5.6.19
OS X 10.9.4/Apache 2.2.26

一、代碼

CREATE TABLE `session` (
  `skey` char(32) CHARACTER SET ascii NOT NULL,
  `data` text COLLATE utf8mb4_bin,
  `expire` int(11) NOT NULL,
  PRIMARY KEY (`skey`),
  KEY `index_session_expire` (`expire`) USING BTREE
) ENGINE=MyISAM DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_bin;

<?php
/*
 * 連接數據庫所需的DNS、用戶名、密碼等，一般情況不會在代碼中進行更改，
 * 所以使用常量的形式，可以避免在函數中引用而需要global。
 */
define('SESSION_DNS', 'mysql:host=localhost;dbname=db;charset=utf8mb4');
define('SESSION_USR', 'usr');
define('SESSION_PWD', 'pwd');
define('SESSION_MAXLIFETIME', get_cfg_var('session.gc_maxlifetime'));

//創建PDO連接
//持久化連接可以提供更好的效率
function getConnection() {
    try {
        $conn = new PDO(SESSION_DNS, SESSION_USR, SESSION_PWD, array(
            PDO::ATTR_PERSISTENT => TRUE,
            PDO::ATTR_ERRMODE => PDO::ERRMODE_EXCEPTION,
            PDO::ATTR_EMULATE_PREPARES => FALSE
        ));
        return $conn;
    } catch (Exception $ex) {

    }
}

//自定義的session的open函數
function sessionMysqlOpen($savePath, $sessionName) {
    return TRUE;
}

//自定義的session的close函數
function sessionMysqlClose() {
    return TRUE;
}
/*
 * 由於一般不會把用戶提交的數據直接保存到session，所以普通情況不存在注入問題。
 * 且處理session數據的SQL語句也不會多次使用。因此預處理功能的效益無法體現。
 * 所以，實際工程中可以不必教條的使用預處理功能。
 */
/*
 * sessionMysqlRead()函數中，首先通過SELECT count(*)來判斷sessionID是否存在。
 * 由於MySQL數據庫提供SELECT對PDOStatement::rowCount()的支持，
 * 因此，實際的工程中可以直接使用rowCount()進行判斷。
 */
//自定義的session的read函數
//SQL語句中增加了“expire > time()”判斷，用以避免讀取過期的session。
function sessionMysqlRead($sessionId) {
    try {
        $dbh = getConnection();
        $time = time();
        
        $sql = 'SELECT count(*) AS `count` FROM session '
                . 'WHERE skey = ? and expire > ?';
        $stmt = $dbh->prepare($sql);
        $stmt->execute(array($sessionId, $time));
        $data = $stmt->fetch(PDO::FETCH_ASSOC);
        if ($data['count'] = 0) {
            return '';
        }
        
        $sql = 'SELECT `data` FROM `session` '
                . 'WHERE `skey` = ? and `expire` > ?';
        $stmt = $dbh->prepare($sql);
        $stmt->execute(array($sessionId, $time));
        $data = $stmt->fetch(PDO::FETCH_ASSOC);
        return $data['data'];
    } catch (Exception $e) {
        return '';
    }
}

//自定義的session的write函數
//expire字段存儲的數據為當前時間+session生命期，當這個值小於time()時表明session失效。
function sessionMysqlWrite($sessionId, $data) {
    try {
        $dbh = getConnection();
        $expire = time() + SESSION_MAXLIFETIME;

        $sql = 'INSERT INTO `session` (`skey`, `data`, `expire`) '
                . 'values (?, ?, ?) '
                . 'ON DUPLICATE KEY UPDATE data = ?, expire = ?';
        $stmt = $dbh->prepare($sql);
        $stmt->execute(array($sessionId, $data, $expire, $data, $expire));
    } catch (Exception $e) {
        echo $e->getMessage();
    }
}

//自定義的session的destroy函數
function sessionMysqlDestroy($sessionId) {
    try {
        $dbh = getConnection();

        $sql = 'DELETE FROM `session` where skey = ?';
        $stmt = $dbh->prepare($sql);
        $stmt->execute(array($sessionId));
        return TRUE;
    } catch (Exception $e) {
        return FALSE;
    }
}

//自定義的session的gc函數
function sessionMysqlGc($lifetime) {
    try {
        $dbh = getConnection();

        $sql = 'DELETE FROM `session` WHERE expire < ?';
        $stmt = $dbh->prepare($sql);
        $stmt->execute(array(time()));
        $dbh = NULL;
        return TRUE;
    } catch (Exception $e) {
        return FALSE;
    }
}

//自定義的session的session id設置函數
/*
 * 由於在session_start()之前，SID和session_id()均無效，
 * 故使用$_GET[session_name()]和$_COOKIE[session_name()]進行檢測。
 * 如果此兩者均為空，則表明session尚未建立，需要為新session設置session id。
 * 通過MySQL數據庫獲取uuid作為session id可以更好的避免session id碰撞。
 */
function sessionMysqlId() {
    if (filter_input(INPUT_GET, session_name()) == '' and
            filter_input(INPUT_COOKIE, session_name()) == '') {
        try {
            $dbh = getConnection();
            $stmt = $dbh->query('SELECT uuid() AS uuid');
            $data = $stmt->fetch(PDO::FETCH_ASSOC);
            $data = str_replace('-', '', $data['uuid']);
            session_id($data);
            return TRUE;
        } catch (Exception $ex) {
            return FALSE;
        }
        
    }
}

//session啟動函數，包括了session_start()及其之前的所有步驟。
function startSession() {
    session_set_save_handler(
            'sessionMysqlOpen',
            'sessionMysqlClose',
            'sessionMysqlRead',
            'sessionMysqlWrite',
            'sessionMysqlDestroy',
            'sessionMysqlGc');
    register_shutdown_function('session_write_close');
    sessionMysqlId();
    session_start();
}

二、簡介

1. 使用MySQL保存session，需要保存三個關鍵性的數據：session id、session數據、session生命期。
2. 考慮到session的使用方式，沒必要使用InnoDB引擎，MyISAM引擎可以獲得更好的性能。如果環境允許，可以嘗試使用MEMORY引擎。
3. 保存session數據的列，有需要的話，可以使用utf8或utf8mb4字符集；保存session id的列則沒有必要，一般情況使用ascii字符集就可以了，可以節約存儲成本。
4. 保存session生命期的列，可以根據工程需要進行設計。比如datetime類型、timestamp類型、int類型。對於datetime、int類型可以保存session生成時間或過期時間。
5. 如果有必要可以擴展session表的列並修改讀、寫函數以支持（維護）相關列來保存諸如用戶名等信息。
6. 當前版本，只要通過session_set_save_handler注冊自定義的會話維護函數就可以，不需要在其之前使用session_module_name('user')函數。
7. 當read函數獲取數據並返回，PHP會自動對其進行反序列化，一般情況請不要對數據進行更改。
8. PHP傳遞給write函數的date參數是序列化之后的session數據，直接保存即可，一般情況請不要對數據進行更改。
9. 按照本段代碼的邏輯，PHP配置選項關於會話生命期的設置已經不再有效，這個值可以自行維護，不一定需要通過get_cfg_var獲取。
10. sessionMysqlId()函數是為了避免大用戶量、多台Web服務器情況下的碰撞，一般情況PHP自動生成的session id是可以滿足用戶要求的。
11. 沒了

三、需求

當用戶量非常大，需要多台服務器提供應用的時候，使用MySQL存儲會話相對使用會話文件具有一定的優越性。比如具有最小的存儲開銷，比如可以避免文件共享帶來的復雜性，比如可以更好的避免發生碰撞，比如相比會話文件共享具有更好的性能。總體上來說，當訪問量劇增的時候，如果使用數據庫保存會話帶來的問題是線性增長的，那么使用會話文件帶來的問題幾乎是爆炸性的。好吧，換一個更直白的說法吧：如果您的應用用戶量不大，其實讓PHP自己處理session就好了，沒必要考慮MySQL。

四、參考

http://cn2.php.net/manual/zh/function.session-set-save-handler.php
http://cn2.php.net/manual/zh/session.idpassing.php
http://cn2.php.net/manual/zh/pdo.connections.php
http://cn2.php.net/manual/zh/pdo.prepared-statements.php
http://dev.mysql.com/doc/refman/5.1/zh/sql-syntax.html#insert