140 lines
4.8 KiB
PHP
140 lines
4.8 KiB
PHP
<?php
|
|
|
|
/*
|
|
* This file is part of the Symfony package.
|
|
*
|
|
* (c) Fabien Potencier <fabien@symfony.com>
|
|
*
|
|
* For the full copyright and license information, please view the LICENSE
|
|
* file that was distributed with this source code.
|
|
*/
|
|
|
|
namespace Symfony\Component\HttpKernel\Profiler;
|
|
|
|
/**
|
|
* SqliteProfilerStorage stores profiling information in a SQLite database.
|
|
*
|
|
* @author Fabien Potencier <fabien@symfony.com>
|
|
*/
|
|
class SqliteProfilerStorage extends PdoProfilerStorage
|
|
{
|
|
/**
|
|
* @throws \RuntimeException When neither of SQLite3 or PDO_SQLite extension is enabled
|
|
*/
|
|
protected function initDb()
|
|
{
|
|
if (null === $this->db || $this->db instanceof \SQLite3) {
|
|
if (0 !== strpos($this->dsn, 'sqlite')) {
|
|
throw new \RuntimeException(sprintf('Please check your configuration. You are trying to use Sqlite with an invalid dsn "%s". The expected format is "sqlite:/path/to/the/db/file".', $this->dsn));
|
|
}
|
|
if (class_exists('SQLite3')) {
|
|
$db = new \SQLite3(substr($this->dsn, 7, strlen($this->dsn)), \SQLITE3_OPEN_READWRITE | \SQLITE3_OPEN_CREATE);
|
|
if (method_exists($db, 'busyTimeout')) {
|
|
// busyTimeout only exists for PHP >= 5.3.3
|
|
$db->busyTimeout(1000);
|
|
}
|
|
} elseif (class_exists('PDO') && in_array('sqlite', \PDO::getAvailableDrivers(), true)) {
|
|
$db = new \PDO($this->dsn);
|
|
} else {
|
|
throw new \RuntimeException('You need to enable either the SQLite3 or PDO_SQLite extension for the profiler to run properly.');
|
|
}
|
|
|
|
$db->exec('PRAGMA temp_store=MEMORY; PRAGMA journal_mode=MEMORY;');
|
|
$db->exec('CREATE TABLE IF NOT EXISTS sf_profiler_data (token STRING, data STRING, ip STRING, method STRING, url STRING, time INTEGER, parent STRING, created_at INTEGER, status_code INTEGER)');
|
|
$db->exec('CREATE INDEX IF NOT EXISTS data_created_at ON sf_profiler_data (created_at)');
|
|
$db->exec('CREATE INDEX IF NOT EXISTS data_ip ON sf_profiler_data (ip)');
|
|
$db->exec('CREATE INDEX IF NOT EXISTS data_method ON sf_profiler_data (method)');
|
|
$db->exec('CREATE INDEX IF NOT EXISTS data_url ON sf_profiler_data (url)');
|
|
$db->exec('CREATE INDEX IF NOT EXISTS data_parent ON sf_profiler_data (parent)');
|
|
$db->exec('CREATE UNIQUE INDEX IF NOT EXISTS data_token ON sf_profiler_data (token)');
|
|
|
|
$this->db = $db;
|
|
}
|
|
|
|
return $this->db;
|
|
}
|
|
|
|
protected function exec($db, $query, array $args = array())
|
|
{
|
|
if ($db instanceof \SQLite3) {
|
|
$stmt = $this->prepareStatement($db, $query);
|
|
foreach ($args as $arg => $val) {
|
|
$stmt->bindValue($arg, $val, is_int($val) ? \SQLITE3_INTEGER : \SQLITE3_TEXT);
|
|
}
|
|
|
|
$res = $stmt->execute();
|
|
if (false === $res) {
|
|
throw new \RuntimeException(sprintf('Error executing SQLite query "%s"', $query));
|
|
}
|
|
$res->finalize();
|
|
} else {
|
|
parent::exec($db, $query, $args);
|
|
}
|
|
}
|
|
|
|
protected function fetch($db, $query, array $args = array())
|
|
{
|
|
$return = array();
|
|
|
|
if ($db instanceof \SQLite3) {
|
|
$stmt = $this->prepareStatement($db, $query, true);
|
|
foreach ($args as $arg => $val) {
|
|
$stmt->bindValue($arg, $val, is_int($val) ? \SQLITE3_INTEGER : \SQLITE3_TEXT);
|
|
}
|
|
$res = $stmt->execute();
|
|
while ($row = $res->fetchArray(\SQLITE3_ASSOC)) {
|
|
$return[] = $row;
|
|
}
|
|
$res->finalize();
|
|
$stmt->close();
|
|
} else {
|
|
$return = parent::fetch($db, $query, $args);
|
|
}
|
|
|
|
return $return;
|
|
}
|
|
|
|
/**
|
|
* {@inheritdoc}
|
|
*/
|
|
protected function buildCriteria($ip, $url, $start, $end, $limit, $method)
|
|
{
|
|
$criteria = array();
|
|
$args = array();
|
|
|
|
if ($ip = preg_replace('/[^\d\.]/', '', $ip)) {
|
|
$criteria[] = 'ip LIKE :ip';
|
|
$args[':ip'] = '%'.$ip.'%';
|
|
}
|
|
|
|
if ($url) {
|
|
$criteria[] = 'url LIKE :url ESCAPE "\"';
|
|
$args[':url'] = '%'.addcslashes($url, '%_\\').'%';
|
|
}
|
|
|
|
if ($method) {
|
|
$criteria[] = 'method = :method';
|
|
$args[':method'] = $method;
|
|
}
|
|
|
|
if (!empty($start)) {
|
|
$criteria[] = 'time >= :start';
|
|
$args[':start'] = $start;
|
|
}
|
|
|
|
if (!empty($end)) {
|
|
$criteria[] = 'time <= :end';
|
|
$args[':end'] = $end;
|
|
}
|
|
|
|
return array($criteria, $args);
|
|
}
|
|
|
|
protected function close($db)
|
|
{
|
|
if ($db instanceof \SQLite3) {
|
|
$db->close();
|
|
}
|
|
}
|
|
}
|