class SqliteProfilerStorage extends PdoProfilerStorage {
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')) {
$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)');
$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;
}
protected function buildCriteria($ip, $url, $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;
}
return array(
$criteria,
$args,
);
}
protected function close($db) {
if ($db instanceof \SQLite3) {
$db
->close();
}
}
}
