doctrine2/lib/Doctrine/Cache.php

<?php
/*
 *  $Id$
 *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
 * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
 * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
 * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 *
 * This software consists of voluntary contributions made by many individuals
 * and is licensed under the LGPL. For more information, see
 * <http://www.phpdoctrine.com>.
 */
/**
 * Doctrine_Cache
 *
 * @package     Doctrine
 * @subpackage  Doctrine_Cache
 * @author      Konsta Vesterinen <kvesteri@cc.hut.fi>
 * @license     http://www.opensource.org/licenses/lgpl-license.php LGPL
 * @category    Object Relational Mapping
 * @link        www.phpdoctrine.com
 * @since       1.0
 * @version     $Revision$
 */
class Doctrine_Cache extends Doctrine_Db_EventListener implements Countable, IteratorAggregate
{
    /**
     * @var array $_options                         an array of general caching options
     */
    protected $_options = array('size'                  => 1000,
                                'lifeTime'              => 3600,
                                'addStatsPropability'   => 0.25,
                                'savePropability'       => 0.10,
                                'cleanPropability'      => 0.01,
                                'statsFile'             => '../data/stats.cache',
                                );
    /**
     * @var array $_queries                         query stack
     */
    protected $_queries = array();
    /**
     * @var Doctrine_Cache_Interface $_driver       the cache driver object
     */
    protected $_driver;
    /**
     * @var array $data                             current cache data array
     */
    protected $_data;
    /**
     * @var boolean $success                        the success of last operation
     */
    protected $_success = false;
    /**
     * constructor
     *
     * @param Doctrine_Cache_Interface|string $driver       cache driver name or a driver object
     * @param array $options                                cache driver options
     */
    public function __construct($driver, $options = array())
    {
    	if (is_object($driver)) {
    	   if ( ! ($driver instanceof Doctrine_Cache_Interface)) {
    	       throw new Doctrine_Cache_Exception('Driver should implement Doctrine_Cache_Interface.');
    	   }
    	   
    	   $this->_driver = $driver;
    	   $this->_driver->setOptions($options);
        } else {
            $class = 'Doctrine_Cache_' . ucwords(strtolower($driver));
    
            if ( ! class_exists($class)) {
                throw new Doctrine_Cache_Exception('Cache driver ' . $driver . ' could not be found.');
            }
    
            $this->_driver = new $class($options);
        }
    }
    /**
     * getDriver
     * returns the current cache driver
     *
     * @return Doctrine_Cache_Driver
     */
    public function getDriver()
    {
        return $this->_driver;
    }
    /**
     * setOption
     *
     * @param mixed $option     the option name
     * @param mixed $value      option value
     * @return boolean          TRUE on success, FALSE on failure
     */
    public function setOption($option, $value)
    {
    	// sanity check (we need this since we are using isset() instead of array_key_exists())
    	if ($value === null) {
            throw new Doctrine_Cache_Exception('Null values not accepted for options.');
    	}

    	if (isset($this->_options[$option])) {
            $this->_options[$option] = $value;
            return true;
        }
        return false;
    }
    /**
     * getOption
     * 
     * @param mixed $option     the option name
     * @return mixed            option value
     */
    public function getOption($option)
    {
        if ( ! isset($this->_options[$option])) {
            throw new Doctrine_Cache_Exception('Unknown option ' . $option);
        }

        return $this->_options[$option];
    }
    /**
     * add
     * adds a query to internal query stack
     *
     * @param string|array $query           sql query string
     * @param string $namespace             connection namespace
     * @return void
     */
    public function add($query, $namespace = null)
    {
    	if (isset($namespace)) {
            $this->_queries[$namespace][] = $query;
        } else {
            $this->_queries[] = $query;
        }
    }
    /**
     * getQueries
     *
     * @param string $namespace     optional query namespace
     * @return array                an array of sql query strings
     */
    public function getAll($namespace = null)
    {
        if (isset($namespace)) {
            if( ! isset($this->_queries[$namespace])) {
                return array();
            }

            return $this->_queries[$namespace];
        }
        
        return $this->_queries;
    }
    /**
     * pop
     *
     * pops a query from the stack
     * @return string
     */
    public function pop()
    {
        return array_pop($this->_queries);
    }
    /**
     * reset
     *
     * removes all queries from the query stack
     * @return void
     */
    public function reset()
    {
        $this->_queries = array();
    }
    /**
     * count
     *
     * @return integer          the number of queries in the stack
     */
    public function count() 
    {
        return count($this->_queries);
    }
    /**
     * getIterator
     *
     * @return ArrayIterator    an iterator that iterates through the query stack
     */
    public function getIterator()
    {
        return new ArrayIterator($this->_queries);
    }
    /**
     * @return boolean          whether or not the last cache operation was successful
     */
    public function isSuccessful() 
    {
        return $this->_success;
    }
    /**
     * save
     *
     * @return boolean
     */
    public function clean()
    {
        $rand = (mt_rand() / mt_getrandmax());

    	if ($rand <= $this->_options['cleanPropability']) {
            $content = file_get_contents($this->_statsFile);
            $queries = explode("\n", $content);

            $stats   = array();
    
            foreach ($queries as $query) {
                if (is_array($query)) {
                    $query = $query[0];
                }
                if (isset($stats[$query])) {
                    $stats[$query]++;
                } else {
                    $stats[$query] = 1;
                }
            }
            sort($stats);
    
            $i = $this->_options['size'];
    
            while ($i--) {
                $element = next($stats);
                $query   = key($stats);
    
                if (is_array($query)) {
                    $hash = md5(serialize($query));
                } else {
                    $hash = md5($query);
                }
    
                $this->_driver->delete($hash);
            }
        }
    }
    /**
     * appendStats
     *
     * adds all queries to stats file
     * @return void
     */
    public function appendStats()
    {
    	if ($this->_options['statsFile'] !== false) {

            if ( ! file_exists($this->_options['statsFile'])) {
                throw new Doctrine_Cache_Exception("Couldn't save cache statistics. Cache statistics file doesn't exists!");
            }
            
            $rand = (mt_rand() / mt_getrandmax());

            if ($rand <= $this->_options['addStatsPropability']) {
                file_put_contents($this->_options['statsFile'], implode("\n", $this->_queries));
            }
        }
    }
    /**
     * onPreQuery
     * listens the onPreQuery event of Doctrine_Db
     *
     * adds the issued query to internal query stack
     * and checks if cached element exists
     *
     * @return boolean
     */
    public function onPreQuery(Doctrine_Db_Event $event)
    { 
        $query = $event->getQuery();

        $data  = false;
        // only process SELECT statements
        if (substr(trim(strtoupper($query)), 0, 6) == 'SELECT') {

            $this->add($query, $event->getInvoker()->getName());

            $data = $this->_driver->fetch(md5($query));

            $this->success = ($data) ? true : false;

            if ( ! $data) {
                $rand = (mt_rand() / mt_getrandmax());

                if ($rand < $this->_options['savePropability']) {
                    $stmt = $event->getInvoker()->getAdapter()->query($query);

                    $data = $stmt->fetchAll(Doctrine::FETCH_ASSOC);

                    $this->success = true;

                    $this->_driver->save(md5($query), $data);
                }
            }
            $this->_data = $data;
        }
        return (bool) $data;
    }
    /**
     * onPreFetch
     * listens the onPreFetch event of Doctrine_Db_Statement
     *
     * advances the internal pointer of cached data and returns 
     * the current element
     *
     * @return array
     */
    public function onPreFetch(Doctrine_Db_Event $event)
    {
        $ret = current($this->_data);
    	next($this->_data);
        return $ret;
    }
    /**
     * onPreFetch
     * listens the onPreFetchAll event of Doctrine_Db_Statement
     *
     * returns the current cache data array
     *
     * @return array
     */
    public function onPreFetchAll(Doctrine_Db_Event $event)
    {
        return $this->_data;
    }
    /**
     * onPreExecute
     * listens the onPreExecute event of Doctrine_Db_Statement
     *
     * adds the issued query to internal query stack
     * and checks if cached element exists
     *
     * @return boolean
     */
    public function onPreExecute(Doctrine_Db_Event $event)
    {
        $query = $event->getQuery();

        $data  = false;

        // only process SELECT statements
        if (substr(trim(strtoupper($query)), 0, 6) == 'SELECT') {

            $this->add($query, $event->getInvoker()->getDbh()->getName());

            $data = $this->_driver->fetch(md5(serialize(array($query, $event->getParams()))));

            $this->success = ($data) ? true : false;

            if ( ! $data) {
                $rand = (mt_rand() / mt_getrandmax());

                if ($rand <= $this->_options['savePropability']) {

                    $stmt = $event->getInvoker()->getStatement();

                    $stmt->execute($event->getParams());

                    $data = $stmt->fetchAll(Doctrine::FETCH_ASSOC);

                    $this->success = true;

                    $this->_driver->save(md5(serialize(array($query, $event->getParams()))), $data);
                }
            }

            $this->_data = $data;

        }
        return (bool) $data;
    }
}