314 lines
9.5 KiB
PHP
314 lines
9.5 KiB
PHP
|
<?php
|
||
|
/**
|
||
|
* CDbCache class file
|
||
|
*
|
||
|
* @author Qiang Xue <qiang.xue@gmail.com>
|
||
|
* @link http://www.yiiframework.com/
|
||
|
* @copyright 2008-2013 Yii Software LLC
|
||
|
* @license http://www.yiiframework.com/license/
|
||
|
*/
|
||
|
|
||
|
/**
|
||
|
* CDbCache implements a cache application component by storing cached data in a database.
|
||
|
*
|
||
|
* CDbCache stores cache data in a DB table named {@link cacheTableName}.
|
||
|
* If the table does not exist, it will be automatically created.
|
||
|
* By setting {@link autoCreateCacheTable} to false, you can also manually create the DB table.
|
||
|
*
|
||
|
* CDbCache relies on {@link http://www.php.net/manual/en/ref.pdo.php PDO} to access database.
|
||
|
* By default, it will use a SQLite3 database under the application runtime directory.
|
||
|
* You can also specify {@link connectionID} so that it makes use of
|
||
|
* a DB application component to access database.
|
||
|
*
|
||
|
* See {@link CCache} manual for common cache operations that are supported by CDbCache.
|
||
|
*
|
||
|
* @property integer $gCProbability The probability (parts per million) that garbage collection (GC) should be performed
|
||
|
* when storing a piece of data in the cache. Defaults to 100, meaning 0.01% chance.
|
||
|
* @property CDbConnection $dbConnection The DB connection instance.
|
||
|
*
|
||
|
* @author Qiang Xue <qiang.xue@gmail.com>
|
||
|
* @package system.caching
|
||
|
* @since 1.0
|
||
|
*/
|
||
|
class CDbCache extends CCache
|
||
|
{
|
||
|
/**
|
||
|
* @var string the ID of the {@link CDbConnection} application component. If not set,
|
||
|
* a SQLite3 database will be automatically created and used. The SQLite database file
|
||
|
* is <code>protected/runtime/cache-YiiVersion.db</code>.
|
||
|
*/
|
||
|
public $connectionID;
|
||
|
/**
|
||
|
* @var string name of the DB table to store cache content. Defaults to 'YiiCache'.
|
||
|
* Note, if {@link autoCreateCacheTable} is false and you want to create the DB table
|
||
|
* manually by yourself, you need to make sure the DB table is of the following structure:
|
||
|
* <pre>
|
||
|
* (id CHAR(128) PRIMARY KEY, expire INTEGER, value BLOB)
|
||
|
* </pre>
|
||
|
* Note, some DBMS might not support BLOB type. In this case, replace 'BLOB' with a suitable
|
||
|
* binary data type (e.g. LONGBLOB in MySQL, BYTEA in PostgreSQL.)
|
||
|
* @see autoCreateCacheTable
|
||
|
*/
|
||
|
public $cacheTableName='YiiCache';
|
||
|
/**
|
||
|
* @var boolean whether the cache DB table should be created automatically if it does not exist. Defaults to true.
|
||
|
* If you already have the table created, it is recommended you set this property to be false to improve performance.
|
||
|
* @see cacheTableName
|
||
|
*/
|
||
|
public $autoCreateCacheTable=true;
|
||
|
/**
|
||
|
* @var CDbConnection the DB connection instance
|
||
|
*/
|
||
|
private $_db;
|
||
|
private $_gcProbability=100;
|
||
|
private $_gced=false;
|
||
|
|
||
|
/**
|
||
|
* Initializes this application component.
|
||
|
*
|
||
|
* This method is required by the {@link IApplicationComponent} interface.
|
||
|
* It ensures the existence of the cache DB table.
|
||
|
* It also removes expired data items from the cache.
|
||
|
*/
|
||
|
public function init()
|
||
|
{
|
||
|
parent::init();
|
||
|
|
||
|
$db=$this->getDbConnection();
|
||
|
$db->setActive(true);
|
||
|
if($this->autoCreateCacheTable)
|
||
|
{
|
||
|
$sql="DELETE FROM {$this->cacheTableName} WHERE expire>0 AND expire<".time();
|
||
|
try
|
||
|
{
|
||
|
$db->createCommand($sql)->execute();
|
||
|
}
|
||
|
catch(Exception $e)
|
||
|
{
|
||
|
$this->createCacheTable($db,$this->cacheTableName);
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
/**
|
||
|
* @return integer the probability (parts per million) that garbage collection (GC) should be performed
|
||
|
* when storing a piece of data in the cache. Defaults to 100, meaning 0.01% chance.
|
||
|
*/
|
||
|
public function getGCProbability()
|
||
|
{
|
||
|
return $this->_gcProbability;
|
||
|
}
|
||
|
|
||
|
/**
|
||
|
* @param integer $value the probability (parts per million) that garbage collection (GC) should be performed
|
||
|
* when storing a piece of data in the cache. Defaults to 100, meaning 0.01% chance.
|
||
|
* This number should be between 0 and 1000000. A value 0 meaning no GC will be performed at all.
|
||
|
*/
|
||
|
public function setGCProbability($value)
|
||
|
{
|
||
|
$value=(int)$value;
|
||
|
if($value<0)
|
||
|
$value=0;
|
||
|
if($value>1000000)
|
||
|
$value=1000000;
|
||
|
$this->_gcProbability=$value;
|
||
|
}
|
||
|
|
||
|
/**
|
||
|
* Creates the cache DB table.
|
||
|
* @param CDbConnection $db the database connection
|
||
|
* @param string $tableName the name of the table to be created
|
||
|
*/
|
||
|
protected function createCacheTable($db,$tableName)
|
||
|
{
|
||
|
$driver=$db->getDriverName();
|
||
|
if($driver==='mysql')
|
||
|
$blob='LONGBLOB';
|
||
|
elseif($driver==='pgsql')
|
||
|
$blob='BYTEA';
|
||
|
else
|
||
|
$blob='BLOB';
|
||
|
$sql=<<<EOD
|
||
|
CREATE TABLE $tableName
|
||
|
(
|
||
|
id CHAR(128) PRIMARY KEY,
|
||
|
expire INTEGER,
|
||
|
value $blob
|
||
|
)
|
||
|
EOD;
|
||
|
$db->createCommand($sql)->execute();
|
||
|
}
|
||
|
|
||
|
/**
|
||
|
* @return CDbConnection the DB connection instance
|
||
|
* @throws CException if {@link connectionID} does not point to a valid application component.
|
||
|
*/
|
||
|
public function getDbConnection()
|
||
|
{
|
||
|
if($this->_db!==null)
|
||
|
return $this->_db;
|
||
|
elseif(($id=$this->connectionID)!==null)
|
||
|
{
|
||
|
if(($this->_db=Yii::app()->getComponent($id)) instanceof CDbConnection)
|
||
|
return $this->_db;
|
||
|
else
|
||
|
throw new CException(Yii::t('yii','CDbCache.connectionID "{id}" is invalid. Please make sure it refers to the ID of a CDbConnection application component.',
|
||
|
array('{id}'=>$id)));
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
$dbFile=Yii::app()->getRuntimePath().DIRECTORY_SEPARATOR.'cache-'.Yii::getVersion().'.db';
|
||
|
return $this->_db=new CDbConnection('sqlite:'.$dbFile);
|
||
|
}
|
||
|
}
|
||
|
|
||
|
/**
|
||
|
* Sets the DB connection used by the cache component.
|
||
|
* @param CDbConnection $value the DB connection instance
|
||
|
* @since 1.1.5
|
||
|
*/
|
||
|
public function setDbConnection($value)
|
||
|
{
|
||
|
$this->_db=$value;
|
||
|
}
|
||
|
|
||
|
/**
|
||
|
* Retrieves a value from cache with a specified key.
|
||
|
* This is the implementation of the method declared in the parent class.
|
||
|
* @param string $key a unique key identifying the cached value
|
||
|
* @return string|boolean the value stored in cache, false if the value is not in the cache or expired.
|
||
|
*/
|
||
|
protected function getValue($key)
|
||
|
{
|
||
|
$time=time();
|
||
|
$sql="SELECT value FROM {$this->cacheTableName} WHERE id='$key' AND (expire=0 OR expire>$time)";
|
||
|
$db=$this->getDbConnection();
|
||
|
if($db->queryCachingDuration>0)
|
||
|
{
|
||
|
$duration=$db->queryCachingDuration;
|
||
|
$db->queryCachingDuration=0;
|
||
|
$result=$db->createCommand($sql)->queryScalar();
|
||
|
$db->queryCachingDuration=$duration;
|
||
|
return $result;
|
||
|
}
|
||
|
else
|
||
|
return $db->createCommand($sql)->queryScalar();
|
||
|
}
|
||
|
|
||
|
/**
|
||
|
* Retrieves multiple values from cache with the specified keys.
|
||
|
* @param array $keys a list of keys identifying the cached values
|
||
|
* @return array a list of cached values indexed by the keys
|
||
|
*/
|
||
|
protected function getValues($keys)
|
||
|
{
|
||
|
if(empty($keys))
|
||
|
return array();
|
||
|
|
||
|
$ids=implode("','",$keys);
|
||
|
$time=time();
|
||
|
$sql="SELECT id, value FROM {$this->cacheTableName} WHERE id IN ('$ids') AND (expire=0 OR expire>$time)";
|
||
|
|
||
|
$db=$this->getDbConnection();
|
||
|
if($db->queryCachingDuration>0)
|
||
|
{
|
||
|
$duration=$db->queryCachingDuration;
|
||
|
$db->queryCachingDuration=0;
|
||
|
$rows=$db->createCommand($sql)->queryAll();
|
||
|
$db->queryCachingDuration=$duration;
|
||
|
}
|
||
|
else
|
||
|
$rows=$db->createCommand($sql)->queryAll();
|
||
|
|
||
|
$results=array();
|
||
|
foreach($keys as $key)
|
||
|
$results[$key]=false;
|
||
|
foreach($rows as $row)
|
||
|
$results[$row['id']]=$row['value'];
|
||
|
return $results;
|
||
|
}
|
||
|
|
||
|
/**
|
||
|
* Stores a value identified by a key in cache.
|
||
|
* This is the implementation of the method declared in the parent class.
|
||
|
*
|
||
|
* @param string $key the key identifying the value to be cached
|
||
|
* @param string $value the value to be cached
|
||
|
* @param integer $expire the number of seconds in which the cached value will expire. 0 means never expire.
|
||
|
* @return boolean true if the value is successfully stored into cache, false otherwise
|
||
|
*/
|
||
|
protected function setValue($key,$value,$expire)
|
||
|
{
|
||
|
$this->deleteValue($key);
|
||
|
return $this->addValue($key,$value,$expire);
|
||
|
}
|
||
|
|
||
|
/**
|
||
|
* Stores a value identified by a key into cache if the cache does not contain this key.
|
||
|
* This is the implementation of the method declared in the parent class.
|
||
|
*
|
||
|
* @param string $key the key identifying the value to be cached
|
||
|
* @param string $value the value to be cached
|
||
|
* @param integer $expire the number of seconds in which the cached value will expire. 0 means never expire.
|
||
|
* @return boolean true if the value is successfully stored into cache, false otherwise
|
||
|
*/
|
||
|
protected function addValue($key,$value,$expire)
|
||
|
{
|
||
|
if(!$this->_gced && mt_rand(0,1000000)<$this->_gcProbability)
|
||
|
{
|
||
|
$this->gc();
|
||
|
$this->_gced=true;
|
||
|
}
|
||
|
|
||
|
if($expire>0)
|
||
|
$expire+=time();
|
||
|
else
|
||
|
$expire=0;
|
||
|
$sql="INSERT INTO {$this->cacheTableName} (id,expire,value) VALUES ('$key',$expire,:value)";
|
||
|
try
|
||
|
{
|
||
|
$command=$this->getDbConnection()->createCommand($sql);
|
||
|
$command->bindValue(':value',$value,PDO::PARAM_LOB);
|
||
|
$command->execute();
|
||
|
return true;
|
||
|
}
|
||
|
catch(Exception $e)
|
||
|
{
|
||
|
return false;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
/**
|
||
|
* Deletes a value with the specified key from cache
|
||
|
* This is the implementation of the method declared in the parent class.
|
||
|
* @param string $key the key of the value to be deleted
|
||
|
* @return boolean if no error happens during deletion
|
||
|
*/
|
||
|
protected function deleteValue($key)
|
||
|
{
|
||
|
$sql="DELETE FROM {$this->cacheTableName} WHERE id='$key'";
|
||
|
$this->getDbConnection()->createCommand($sql)->execute();
|
||
|
return true;
|
||
|
}
|
||
|
|
||
|
/**
|
||
|
* Removes the expired data values.
|
||
|
*/
|
||
|
protected function gc()
|
||
|
{
|
||
|
$this->getDbConnection()->createCommand("DELETE FROM {$this->cacheTableName} WHERE expire>0 AND expire<".time())->execute();
|
||
|
}
|
||
|
|
||
|
/**
|
||
|
* Deletes all values from cache.
|
||
|
* This is the implementation of the method declared in the parent class.
|
||
|
* @return boolean whether the flush operation was successful.
|
||
|
* @since 1.1.5
|
||
|
*/
|
||
|
protected function flushValues()
|
||
|
{
|
||
|
$this->getDbConnection()->createCommand("DELETE FROM {$this->cacheTableName}")->execute();
|
||
|
return true;
|
||
|
}
|
||
|
}
|