专注收集记录技术开发学习笔记、技术难点、解决方案
网站信息搜索 >> 请输入关键词:
您当前的位置: 首页 > Go

thinkphp框架下session 惠存mongo

发布时间:2011-06-29 18:27:58 文章来源:www.iduyao.cn 采编人员:星星草
thinkphp框架下session 存入mongo


最近做系统的时候处理到session入数据库问题 , 由于是在thinkphp框架下, 查看了下框架session相关代码, 发现原框架默认支持mysql数据库 , 

于是对原Session驱动做了相应修改 , 使其支持mongo数据库存储session

修改的相关文件是\Think\Session\Driver\Db\Db.class.php , 本人使用的是TP3.2,2版本 , 修改后的Db.class.php代码如下:


<?php
namespace Think\Session\Driver;

class Db
{
    const MONGO_EXPIRY = 3;
    private $host = '';
    private $port = '';
    private $username = '';
    private $password = '';
    private $dbname = '';
    private $dbsession = '';
    private $__mongo_collection = NULL;
    private $__current_session = NULL;

    /**
     * Default constructor set default parameter
     * @access public
     */
    public function __construct()
    {
        $this->host = C('DB_HOST');
        $this->port = C('DB_PORT');
        $this->username = C('DB_USER');
        $this->password = C('DB_PWD');
        $this->dbname = C('DB_NAME');
        $this->dbsession = C('SESSION_TABLE');
        $this->__connect();
        session_set_save_handler(
            array(&$this, 'open'),
            array(&$this, 'close'),
            array(&$this, 'read'),
            array(&$this, 'write'),
            array(&$this, 'destroy'),
            array(&$this, 'gc')
        );
    }

    /**
     * connectes the mongo database and create collection
     * @access private
     */
    private function __connect()
    {
        $connection_string = sprintf('mongodb://%s:%s', $this->host, $this->port);
        if ($this->username != null && $this->password != null) {
            $connection_string = "mongodb://{$this->username}:{$this->password}@{$this->host}:{$this->port}";
            $connection_string = sprintf('mongodb://%s:%s@%s:%s', $this->username, $this->password, $this->username, $this->password);
        }
        $object_mongo = new \Mongo($connection_string);
        $object_mongo = $object_mongo->{$this->dbname};
        $this->__mongo_collection = $object_mongo->{$this->dbsession};
    }


    /**
     *
     * check for collection object
     * @access public
     * @param string $session_path
     * @param string $session_name
     * @return boolean
     */
    public function open($session_path, $session_name)
    {
        $result = false;
        if ($this->__mongo_collection != NULL) {
            $result = false;
        }
        return $result;
    }

    /**
     *
     * doing noting
     * @access public
     * @return boolean
     */
    public function close()
    {
        return true;
    }

    /**
     *
     * Reading session data based on id
     * @access public
     * @param string $session_id
     * @return mixed
     */
    public function read($session_id)
    {
        $result = NULL;
        $expiry = time();
        $query['_id'] = $session_id;
        $query['expiry'] = array('$gte' => $expiry);
        $result = $this->__mongo_collection->findone($query);
        if ($result) {
            $this->__current_session = $result;
        }
        return $result['data'];
    }

    /**
     *
     * Writing session data
     * @access public
     * @param string $session_id
     * @param mixed $data
     * @return boolean
     */
    public function write($session_id, $data)
    {
        $result = true;
        $expiry = $this->__getExpriry();
        $session_data = array();
        if (empty($this->__current_session)) {
            $session_id = $session_id;
            $session_data['_id'] = $session_id;
            $session_data['data'] = $data;
            $session_data['expiry'] = $expiry;
        } else {
            $session_data = (array)$this->__current_session;
            $session_data['data'] = $data;
            $session_data['expiry'] = $expiry;

        }
        $query['_id'] = $session_id;
        $record = $this->__mongo_collection->findOne($query);
        if ($record == null) {
            $this->__mongo_collection->insert($session_data);
        } else {
            $record['data'] = $data;
            $record['expiry'] = $expiry;
            $this->__mongo_collection->save($record);
        }
        return true;
    }

    /**
     *
     * remove session data
     * @access public
     * @param string $session_id
     * @return boolean
     */
    public function destroy($session_id)
    {
        $query['_id'] = $session_id;
        $this->__mongo_collection->remove($query, true);
        return true;
    }

    /**
     *
     * Garbage collection
     * @access public
     * @return boolean
     */
    public function gc()
    {
        $query = array();
        $query['expiry'] = array(':lt' => time());
        $this->__mongo_collection->remove($query, array('justOne' => false));
        return true;

    }

    /**
     * get expiry
     * @access private
     * @return int
     */
    private function __getExpriry()
    {
        return time() + self::MONGO_EXPIRY;
    }
}



友情提示:
信息收集于互联网,如果您发现错误或造成侵权,请及时通知本站更正或删除,具体联系方式见页面底部联系我们,谢谢。

其他相似内容:

  • ModernUI课程:定义一个Logo

    ModernUI教程:定义一个Logo ModernWindow的标题栏包含了一块区域用来显示自定义的窗体Logo: 这个窗体logo通过ModernWindow.LogoD...

  • Django忘记管理员账号和密码的解决方法

    Django忘记管理员账号和密码的解决办法 看着Django的教程学习搭建网站,结果忘记第一次创建的账号和密码了。结果搭建成功以后,一直...

  • GO语言小结(1)——基本知识

    GO语言总结(1)——基本知识 1、注释(与C++一样)   行注释://  块注释:/*   ...  */ 2、标识符   可以这么说,除了数字开头...

  • golang 惯用的文件读取方式

    golang 常用的文件读取方式 Golang 的文件读取方法很多,刚上手时不知道怎么选择,所以贴在此处便后速查。 一次性读取 小文件推荐一...

  • 查询深圳市通相关信息

    查询深圳通相关信息 用 HTTP.GET 从开放 API 中查询深圳通信息,然后将 JSON 数据存入结构体中,再格式化输出。 注意:获取的并不是实...

  • Go语言设计模式实践:结合(Composite)

    Go语言设计模式实践:组合(Composite) 关于本系列 这个系列首先是关于Go语言实践的。在项目中实际使用Go语言也有段时间了,一个体会就...

  • 列出索引和遍历目录

    列出目录和遍历目录 获取目录列表用 ioutil.ReadDir(),遍历目录用 filepath.Walk(),使用方法请参考文章示例。 示例代码: package ma...

  • io 包的惯用接口速记

    io 包的常用接口速记 我没有 C/C++ 基础,没有接口的概念,且从 Python 投奔而来,Python 的极简主义(一个结果往往只提供一个方法),让我在...

  • 代理服务扩充

    代理服务扩展 之前自己实现了一个代理服务,当时考虑的是只要支持SOCKS5就好了,因为我经常用CHROME,配合着SwitchySharp,体验还是很棒...

  • 文件的创造与打开

    文件的创建与打开 文件操作是个很重要的话题,使用也非常频繁,熟悉如何操作文件是必不可少的。Golang 对文件的支持是在 os package ...

热门推荐: