Sqlite.php 3.76 KB
Newer Older
1 2 3 4 5 6 7
<?php
class Doctrine_Cache_Query_Sqlite implements Countable {
    /**
     * doctrine cache
     */
    const CACHE_TABLE = 'doctrine_query_cache';
    /**
doctrine's avatar
doctrine committed
8
     * @var Doctrine_Session $session       the table object this cache container belongs to
9 10 11 12 13 14 15 16
     */
    private $table;
    /**
     * @var PDO $dbh                    database handler
     */
    private $dbh;
    /**
     * constructor
zYne's avatar
zYne committed
17 18
     *
     * @param Doctrine_Connection|null $connection
19
     */
zYne's avatar
zYne committed
20
    public function __construct($connection = null) {
21
        if( ! ($connection instanceof Doctrine_Connection)) 
zYne's avatar
zYne committed
22
            $connection = Doctrine_Manager::getInstance()->getCurrentConnection();
23

zYne's avatar
zYne committed
24 25
        $this->session = $connection;
        $dir = $this->session->getAttribute(Doctrine::ATTR_CACHE_DIR);
26 27 28 29

        $this->path = $dir.DIRECTORY_SEPARATOR;
        $this->dbh  = new PDO("sqlite::memory:");

doctrine's avatar
doctrine committed
30

31
        try {
doctrine's avatar
doctrine committed
32
            if($this->session->getAttribute(Doctrine::ATTR_CREATE_TABLES) === true)
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
            {
                $columns = array();
                $columns['query_md5']       = array('string', 32, 'notnull');
                $columns['query_result']    = array('array', 100000, 'notnull');
                $columns['expires']         = array('integer', 11, 'notnull');

                $dataDict = new Doctrine_DataDict($this->dbh);
                $dataDict->createTable(self::CACHE_TABLE, $columns);
            }
        } catch(PDOException $e) {

        }
    }
    /**
     * store
     * stores a query in cache
     *
     * @param string $query
     * @param array $result
     * @param integer $lifespan
     * @return void
     */
    public function store($query, array $result, $lifespan) {
        $sql    = "INSERT INTO ".self::CACHE_TABLE." (query_md5, query_result, expires) VALUES (?,?,?)";
        $stmt   = $this->dbh->prepare($sql);
        $params = array(md5($query), serialize($result), (time() + $lifespan));
        $stmt->execute($params);
    }
    /**
     * fetch
     *
     * @param string $md5
     * @return array
     */
    public function fetch($md5) {
        $sql    = "SELECT query_result, expires FROM ".self::CACHE_TABLE." WHERE query_md5 = ?";
        $stmt   = $this->dbh->prepare($sql);
        $params = array($md5);
        $stmt->execute($params);
        $result = $stmt->fetch(PDO::FETCH_ASSOC);
        return unserialize($result['query_result']);
    }
    /**
     * deleteAll
     * returns the number of deleted rows
     *
     * @return integer
     */
    public function deleteAll() {
        $sql    = "DELETE FROM ".self::CACHE_TABLE;
        $stmt   = $this->dbh->query($sql);
        return $stmt->rowCount();
    }
zYne's avatar
zYne committed
86 87 88 89 90 91 92 93 94 95 96
    /**
     * deleteExpired
     * returns the number of deleted rows
     *
     * @return integer
     */
    public function deleteExpired() {
        $sql    = "DELETE FROM ".self::CACHE_TABLE." WHERE expired < ?";
        $stmt   = $this->dbh->prepare($sql);
        $stmt->execute(array(time()));
    }
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 123 124
    /**
     * delete
     * returns whether or not the given 
     * query was succesfully deleted
     *
     * @param string $md5
     * @return boolean
     */
    public function delete($md5) {
        $sql    = "DELETE FROM ".self::CACHE_TABLE." WHERE query_md5 = ?";
        $stmt   = $this->dbh->prepare($sql);
        $params = array($md5);
        $stmt->execute($params);
        return $stmt->rowCount();
    }
    /**
     * count
     *
     * @return integer
     */
    public function count() {
        $stmt = $this->dbh->query("SELECT COUNT(*) FROM ".self::CACHE_TABLE);
        $data = $stmt->fetch(PDO::FETCH_NUM);

        // table has three columns so we have to divide the count by two
        return $data[0];
    }
}
125