207 lines
4.5 KiB
PHP
207 lines
4.5 KiB
PHP
|
<?php
|
||
|
|
||
|
namespace Doctrine\Common\Cache;
|
||
|
|
||
|
use SQLite3;
|
||
|
use SQLite3Result;
|
||
|
use const SQLITE3_ASSOC;
|
||
|
use const SQLITE3_BLOB;
|
||
|
use const SQLITE3_TEXT;
|
||
|
use function array_search;
|
||
|
use function implode;
|
||
|
use function serialize;
|
||
|
use function sprintf;
|
||
|
use function time;
|
||
|
use function unserialize;
|
||
|
|
||
|
/**
|
||
|
* SQLite3 cache provider.
|
||
|
*/
|
||
|
class SQLite3Cache extends CacheProvider
|
||
|
{
|
||
|
/**
|
||
|
* The ID field will store the cache key.
|
||
|
*/
|
||
|
public const ID_FIELD = 'k';
|
||
|
|
||
|
/**
|
||
|
* The data field will store the serialized PHP value.
|
||
|
*/
|
||
|
public const DATA_FIELD = 'd';
|
||
|
|
||
|
/**
|
||
|
* The expiration field will store a date value indicating when the
|
||
|
* cache entry should expire.
|
||
|
*/
|
||
|
public const EXPIRATION_FIELD = 'e';
|
||
|
|
||
|
/** @var SQLite3 */
|
||
|
private $sqlite;
|
||
|
|
||
|
/** @var string */
|
||
|
private $table;
|
||
|
|
||
|
/**
|
||
|
* Calling the constructor will ensure that the database file and table
|
||
|
* exist and will create both if they don't.
|
||
|
*
|
||
|
* @param string $table
|
||
|
*/
|
||
|
public function __construct(SQLite3 $sqlite, $table)
|
||
|
{
|
||
|
$this->sqlite = $sqlite;
|
||
|
$this->table = (string) $table;
|
||
|
|
||
|
$this->ensureTableExists();
|
||
|
}
|
||
|
|
||
|
private function ensureTableExists() : void
|
||
|
{
|
||
|
$this->sqlite->exec(
|
||
|
sprintf(
|
||
|
'CREATE TABLE IF NOT EXISTS %s(%s TEXT PRIMARY KEY NOT NULL, %s BLOB, %s INTEGER)',
|
||
|
$this->table,
|
||
|
static::ID_FIELD,
|
||
|
static::DATA_FIELD,
|
||
|
static::EXPIRATION_FIELD
|
||
|
)
|
||
|
);
|
||
|
}
|
||
|
|
||
|
/**
|
||
|
* {@inheritdoc}
|
||
|
*/
|
||
|
protected function doFetch($id)
|
||
|
{
|
||
|
$item = $this->findById($id);
|
||
|
|
||
|
if (! $item) {
|
||
|
return false;
|
||
|
}
|
||
|
|
||
|
return unserialize($item[self::DATA_FIELD]);
|
||
|
}
|
||
|
|
||
|
/**
|
||
|
* {@inheritdoc}
|
||
|
*/
|
||
|
protected function doContains($id)
|
||
|
{
|
||
|
return $this->findById($id, false) !== null;
|
||
|
}
|
||
|
|
||
|
/**
|
||
|
* {@inheritdoc}
|
||
|
*/
|
||
|
protected function doSave($id, $data, $lifeTime = 0)
|
||
|
{
|
||
|
$statement = $this->sqlite->prepare(sprintf(
|
||
|
'INSERT OR REPLACE INTO %s (%s) VALUES (:id, :data, :expire)',
|
||
|
$this->table,
|
||
|
implode(',', $this->getFields())
|
||
|
));
|
||
|
|
||
|
$statement->bindValue(':id', $id);
|
||
|
$statement->bindValue(':data', serialize($data), SQLITE3_BLOB);
|
||
|
$statement->bindValue(':expire', $lifeTime > 0 ? time() + $lifeTime : null);
|
||
|
|
||
|
return $statement->execute() instanceof SQLite3Result;
|
||
|
}
|
||
|
|
||
|
/**
|
||
|
* {@inheritdoc}
|
||
|
*/
|
||
|
protected function doDelete($id)
|
||
|
{
|
||
|
[$idField] = $this->getFields();
|
||
|
|
||
|
$statement = $this->sqlite->prepare(sprintf(
|
||
|
'DELETE FROM %s WHERE %s = :id',
|
||
|
$this->table,
|
||
|
$idField
|
||
|
));
|
||
|
|
||
|
$statement->bindValue(':id', $id);
|
||
|
|
||
|
return $statement->execute() instanceof SQLite3Result;
|
||
|
}
|
||
|
|
||
|
/**
|
||
|
* {@inheritdoc}
|
||
|
*/
|
||
|
protected function doFlush()
|
||
|
{
|
||
|
return $this->sqlite->exec(sprintf('DELETE FROM %s', $this->table));
|
||
|
}
|
||
|
|
||
|
/**
|
||
|
* {@inheritdoc}
|
||
|
*/
|
||
|
protected function doGetStats()
|
||
|
{
|
||
|
// no-op.
|
||
|
}
|
||
|
|
||
|
/**
|
||
|
* Find a single row by ID.
|
||
|
*
|
||
|
* @param mixed $id
|
||
|
*
|
||
|
* @return array|null
|
||
|
*/
|
||
|
private function findById($id, bool $includeData = true) : ?array
|
||
|
{
|
||
|
[$idField] = $fields = $this->getFields();
|
||
|
|
||
|
if (! $includeData) {
|
||
|
$key = array_search(static::DATA_FIELD, $fields);
|
||
|
unset($fields[$key]);
|
||
|
}
|
||
|
|
||
|
$statement = $this->sqlite->prepare(sprintf(
|
||
|
'SELECT %s FROM %s WHERE %s = :id LIMIT 1',
|
||
|
implode(',', $fields),
|
||
|
$this->table,
|
||
|
$idField
|
||
|
));
|
||
|
|
||
|
$statement->bindValue(':id', $id, SQLITE3_TEXT);
|
||
|
|
||
|
$item = $statement->execute()->fetchArray(SQLITE3_ASSOC);
|
||
|
|
||
|
if ($item === false) {
|
||
|
return null;
|
||
|
}
|
||
|
|
||
|
if ($this->isExpired($item)) {
|
||
|
$this->doDelete($id);
|
||
|
|
||
|
return null;
|
||
|
}
|
||
|
|
||
|
return $item;
|
||
|
}
|
||
|
|
||
|
/**
|
||
|
* Gets an array of the fields in our table.
|
||
|
*
|
||
|
* @return array
|
||
|
*/
|
||
|
private function getFields() : array
|
||
|
{
|
||
|
return [static::ID_FIELD, static::DATA_FIELD, static::EXPIRATION_FIELD];
|
||
|
}
|
||
|
|
||
|
/**
|
||
|
* Check if the item is expired.
|
||
|
*
|
||
|
* @param array $item
|
||
|
*/
|
||
|
private function isExpired(array $item) : bool
|
||
|
{
|
||
|
return isset($item[static::EXPIRATION_FIELD]) &&
|
||
|
$item[self::EXPIRATION_FIELD] !== null &&
|
||
|
$item[self::EXPIRATION_FIELD] < time();
|
||
|
}
|
||
|
}
|