/protected/extensions/doctrine/vendors/Symfony/Component/HttpKernel/Profiler/SqliteProfilerStorage.php
PHP | 124 lines | 89 code | 15 blank | 20 comment | 15 complexity | c3f5eb478567e518dfe7bfca3b745cf0 MD5 | raw file
Possible License(s): BSD-2-Clause, LGPL-2.1, BSD-3-Clause
- <?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('You are trying to use Sqlite with a wrong dsn. "'.$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, url STRING, time INTEGER, parent STRING, created_at 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_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, $limit)
- {
- $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, '%_\\').'%';
- }
- return array($criteria, $args);
- }
- protected function close($db)
- {
- if ($db instanceof \SQLite3) {
- $db->close();
- }
- }
- }