123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268 |
- <?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;
- @trigger_error('The '.__NAMESPACE__.'\PdoProfilerStorage class is deprecated since Symfony 2.8 and will be removed in 3.0. Use FileProfilerStorage instead.', E_USER_DEPRECATED);
- /**
- * Base PDO storage for profiling information in a PDO database.
- *
- * @author Fabien Potencier <fabien@symfony.com>
- * @author Jan Schumann <js@schumann-it.com>
- *
- * @deprecated Deprecated since Symfony 2.8, to be removed in Symfony 3.0.
- * Use {@link FileProfilerStorage} instead.
- */
- abstract class PdoProfilerStorage implements ProfilerStorageInterface
- {
- protected $dsn;
- protected $username;
- protected $password;
- protected $lifetime;
- protected $db;
- /**
- * Constructor.
- *
- * @param string $dsn A data source name
- * @param string $username The username for the database
- * @param string $password The password for the database
- * @param int $lifetime The lifetime to use for the purge
- */
- public function __construct($dsn, $username = '', $password = '', $lifetime = 86400)
- {
- $this->dsn = $dsn;
- $this->username = $username;
- $this->password = $password;
- $this->lifetime = (int) $lifetime;
- }
- /**
- * {@inheritdoc}
- */
- public function find($ip, $url, $limit, $method, $start = null, $end = null)
- {
- if (null === $start) {
- $start = 0;
- }
- if (null === $end) {
- $end = time();
- }
- list($criteria, $args) = $this->buildCriteria($ip, $url, $start, $end, $limit, $method);
- $criteria = $criteria ? 'WHERE '.implode(' AND ', $criteria) : '';
- $db = $this->initDb();
- $tokens = $this->fetch($db, 'SELECT token, ip, method, url, time, parent, status_code FROM sf_profiler_data '.$criteria.' ORDER BY time DESC LIMIT '.((int) $limit), $args);
- $this->close($db);
- return $tokens;
- }
- /**
- * {@inheritdoc}
- */
- public function read($token)
- {
- $db = $this->initDb();
- $args = array(':token' => $token);
- $data = $this->fetch($db, 'SELECT data, parent, ip, method, url, time FROM sf_profiler_data WHERE token = :token LIMIT 1', $args);
- $this->close($db);
- if (isset($data[0]['data'])) {
- return $this->createProfileFromData($token, $data[0]);
- }
- }
- /**
- * {@inheritdoc}
- */
- public function write(Profile $profile)
- {
- $db = $this->initDb();
- $args = array(
- ':token' => $profile->getToken(),
- ':parent' => $profile->getParentToken(),
- ':data' => base64_encode(serialize($profile->getCollectors())),
- ':ip' => $profile->getIp(),
- ':method' => $profile->getMethod(),
- ':url' => $profile->getUrl(),
- ':time' => $profile->getTime(),
- ':created_at' => time(),
- ':status_code' => $profile->getStatusCode(),
- );
- try {
- if ($this->has($profile->getToken())) {
- $this->exec($db, 'UPDATE sf_profiler_data SET parent = :parent, data = :data, ip = :ip, method = :method, url = :url, time = :time, created_at = :created_at, status_code = :status_code WHERE token = :token', $args);
- } else {
- $this->exec($db, 'INSERT INTO sf_profiler_data (token, parent, data, ip, method, url, time, created_at, status_code) VALUES (:token, :parent, :data, :ip, :method, :url, :time, :created_at, :status_code)', $args);
- }
- $this->cleanup();
- $status = true;
- } catch (\Exception $e) {
- $status = false;
- }
- $this->close($db);
- return $status;
- }
- /**
- * {@inheritdoc}
- */
- public function purge()
- {
- $db = $this->initDb();
- $this->exec($db, 'DELETE FROM sf_profiler_data');
- $this->close($db);
- }
- /**
- * Build SQL criteria to fetch records by ip and url.
- *
- * @param string $ip The IP
- * @param string $url The URL
- * @param string $start The start period to search from
- * @param string $end The end period to search to
- * @param string $limit The maximum number of tokens to return
- * @param string $method The request method
- *
- * @return array An array with (criteria, args)
- */
- abstract protected function buildCriteria($ip, $url, $start, $end, $limit, $method);
- /**
- * Initializes the database.
- *
- * @throws \RuntimeException When the requested database driver is not installed
- */
- abstract protected function initDb();
- protected function cleanup()
- {
- $db = $this->initDb();
- $this->exec($db, 'DELETE FROM sf_profiler_data WHERE created_at < :time', array(':time' => time() - $this->lifetime));
- $this->close($db);
- }
- protected function exec($db, $query, array $args = array())
- {
- $stmt = $this->prepareStatement($db, $query);
- foreach ($args as $arg => $val) {
- $stmt->bindValue($arg, $val, is_int($val) ? \PDO::PARAM_INT : \PDO::PARAM_STR);
- }
- $success = $stmt->execute();
- if (!$success) {
- throw new \RuntimeException(sprintf('Error executing query "%s"', $query));
- }
- }
- protected function prepareStatement($db, $query)
- {
- try {
- $stmt = $db->prepare($query);
- } catch (\Exception $e) {
- $stmt = false;
- }
- if (false === $stmt) {
- throw new \RuntimeException('The database cannot successfully prepare the statement');
- }
- return $stmt;
- }
- protected function fetch($db, $query, array $args = array())
- {
- $stmt = $this->prepareStatement($db, $query);
- foreach ($args as $arg => $val) {
- $stmt->bindValue($arg, $val, is_int($val) ? \PDO::PARAM_INT : \PDO::PARAM_STR);
- }
- $stmt->execute();
- return $stmt->fetchAll(\PDO::FETCH_ASSOC);
- }
- protected function close($db)
- {
- }
- protected function createProfileFromData($token, $data, $parent = null)
- {
- $profile = new Profile($token);
- $profile->setIp($data['ip']);
- $profile->setMethod($data['method']);
- $profile->setUrl($data['url']);
- $profile->setTime($data['time']);
- $profile->setCollectors(unserialize(base64_decode($data['data'])));
- if (!$parent && !empty($data['parent'])) {
- $parent = $this->read($data['parent']);
- }
- if ($parent) {
- $profile->setParent($parent);
- }
- $profile->setChildren($this->readChildren($token, $profile));
- return $profile;
- }
- /**
- * Reads the child profiles for the given token.
- *
- * @param string $token The parent token
- * @param string $parent The parent instance
- *
- * @return Profile[] An array of Profile instance
- */
- protected function readChildren($token, $parent)
- {
- $db = $this->initDb();
- $data = $this->fetch($db, 'SELECT token, data, ip, method, url, time FROM sf_profiler_data WHERE parent = :token', array(':token' => $token));
- $this->close($db);
- if (!$data) {
- return array();
- }
- $profiles = array();
- foreach ($data as $d) {
- $profiles[] = $this->createProfileFromData($d['token'], $d, $parent);
- }
- return $profiles;
- }
- /**
- * Returns whether data for the given token already exists in storage.
- *
- * @param string $token The profile token
- *
- * @return string
- */
- protected function has($token)
- {
- $db = $this->initDb();
- $tokenExists = $this->fetch($db, 'SELECT 1 FROM sf_profiler_data WHERE token = :token LIMIT 1', array(':token' => $token));
- $this->close($db);
- return !empty($tokenExists);
- }
- }
|