<?php

// +----------------------------------------------------------------------
// | ThinkPHP [ WE CAN DO IT JUST THINK IT ]
// +----------------------------------------------------------------------
// | Copyright (c) 2006-2012 http://thinkphp.cn All rights reserved.
// +----------------------------------------------------------------------
// | Licensed ( http://www.apache.org/licenses/LICENSE-2.0 )
// +----------------------------------------------------------------------
// | Author: liu21st <liu21st@gmail.com>
// +----------------------------------------------------------------------
/**
 * 数据库方式Session驱动
 *    CREATE TABLE think_session (
 *      session_id varchar(255) NOT NULL,
 *      session_expire int(11) NOT NULL,
 *      session_data blob,
 *      UNIQUE KEY `session_id` (`session_id`)
 *    );.
 *
 * @category   Extend
 *
 * @author e启学束
 */
class SessionDb
{
    /**
    * Session有效时间.
    */
   protected $lifeTime = '';

   /**
    * session保存的数据库名.
    */
   protected $sessionTable = '';

   /**
    * 数据库句柄.
    */
   protected $hander = array();

    /**
     * 打开Session.
     *
     * @param string $savePath
     * @param mixed  $sessName
     */
    public function open($savePath, $sessName)
    {
        $this->lifeTime = C('SESSION_EXPIRE') ? C('SESSION_EXPIRE') : ini_get('session.gc_maxlifetime');
        $this->sessionTable = C('SESSION_TABLE') ? C('SESSION_TABLE') : C('DB_PREFIX').'session';
       //分布式数据库
       $host = explode(',', C('DB_HOST'));
        $port = explode(',', C('DB_PORT'));
        $name = explode(',', C('DB_NAME'));
        $user = explode(',', C('DB_USER'));
        $pwd = explode(',', C('DB_PWD'));
        if (1 == C('DB_DEPLOY_TYPE')) {
            //读写分离
           if (C('DB_RW_SEPARATE')) {
               $w = floor(mt_rand(0, C('DB_MASTER_NUM') - 1));
               if (is_numeric(C('DB_SLAVE_NO'))) {
                   //指定服务器读
                   $r = C('DB_SLAVE_NO');
               } else {
                   $r = floor(mt_rand(C('DB_MASTER_NUM'), count($host) - 1));
               }
               //主数据库链接
               $hander = mysql_connect(
                   $host[$w].(isset($port[$w]) ? ':'.$port[$w] : ':'.$port[0]),
                   isset($user[$w]) ? $user[$w] : $user[0],
                   isset($pwd[$w]) ? $pwd[$w] : $pwd[0]
                   );
               $dbSel = mysql_select_db(
                   isset($name[$w]) ? $name[$w] : $name[0], $hander);
               if (!$hander || !$dbSel) {
                   return false;
               }
               $this->hander[0] = $hander;
               //从数据库链接
               $hander = mysql_connect(
                   $host[$r].(isset($port[$r]) ? ':'.$port[$r] : ':'.$port[0]),
                   isset($user[$r]) ? $user[$r] : $user[0],
                   isset($pwd[$r]) ? $pwd[$r] : $pwd[0]
                   );
               $dbSel = mysql_select_db(
                   isset($name[$r]) ? $name[$r] : $name[0], $hander);
               if (!$hander || !$dbSel) {
                   return false;
               }
               $this->hander[1] = $hander;

               return true;
           }
        }
       //从数据库链接
       $r = floor(mt_rand(0, count($host) - 1));
        $hander = mysql_connect(
           $host[$r].(isset($port[$r]) ? ':'.$port[$r] : ':'.$port[0]),
           isset($user[$r]) ? $user[$r] : $user[0],
           isset($pwd[$r]) ? $pwd[$r] : $pwd[0]
           );
        $dbSel = mysql_select_db(
           isset($name[$r]) ? $name[$r] : $name[0], $hander);
        if (!$hander || !$dbSel) {
            return false;
        }
        $this->hander = $hander;

        return true;
    }

   /**
    * 关闭Session.
    */
   public function close()
   {
       if (is_array($this->hander)) {
           $this->gc($this->lifeTime);

           return mysql_close($this->hander[0]) && mysql_close($this->hander[1]);
       }
       $this->gc($this->lifeTime);

       return mysql_close($this->hander);
   }

   /**
    * 读取Session.
    *
    * @param string $sessID
    */
   public function read($sessID)
   {
       $hander = is_array($this->hander) ? $this->hander[1] : $this->hander;
       $res = mysql_query('SELECT session_data AS data FROM '.$this->sessionTable." WHERE session_id = '$sessID'   AND session_expire >".time(), $hander);
       if ($res) {
           $row = mysql_fetch_assoc($res);

           return $row['data'];
       }

       return '';
   }

   /**
    * 写入Session.
    *
    * @param string $sessID
    * @param string $sessData
    */
   public function write($sessID, $sessData)
   {
       $hander = is_array($this->hander) ? $this->hander[0] : $this->hander;
       $expire = time() + $this->lifeTime;
       mysql_query('REPLACE INTO  '.$this->sessionTable." (  session_id, session_expire, session_data)  VALUES( '$sessID', '$expire',  '$sessData')", $hander);
       if (mysql_affected_rows($hander)) {
           return true;
       }

       return false;
   }

   /**
    * 删除Session.
    *
    * @param string $sessID
    */
   public function destroy($sessID)
   {
       $hander = is_array($this->hander) ? $this->hander[0] : $this->hander;
       mysql_query('DELETE FROM '.$this->sessionTable." WHERE session_id = '$sessID'", $hander);
       if (mysql_affected_rows($hander)) {
           return true;
       }

       return false;
   }

   /**
    * Session 垃圾回收.
    *
    * @param string $sessMaxLifeTime
    */
   public function gc($sessMaxLifeTime)
   {
       $hander = is_array($this->hander) ? $this->hander[0] : $this->hander;
       mysql_query('DELETE FROM '.$this->sessionTable.' WHERE session_expire < '.time(), $hander);

       return mysql_affected_rows($hander);
   }

    /**
     * 打开Session.
     */
    public function execute()
    {
        session_set_save_handler(array(&$this, 'open'),
                         array(&$this, 'close'),
                         array(&$this, 'read'),
                         array(&$this, 'write'),
                         array(&$this, 'destroy'),
                         array(&$this, 'gc'));
    }
}
