DatabaseProfilerStorage.php in Devel 8.3

Namespace

Drupal\webprofiler\Profiler

File

webprofiler/src/Profiler/DatabaseProfilerStorage.php

<?php

namespace Drupal\webprofiler\Profiler;

use Drupal\Core\Database\Connection;
use Symfony\Component\HttpKernel\Profiler\Profile;
use Symfony\Component\HttpKernel\Profiler\ProfilerStorageInterface;

/**
 * Implements a profiler storage using the DBTNG query api.
 */
class DatabaseProfilerStorage implements ProfilerStorageInterface {

  /**
   * The database connection.
   *
   * @var \Drupal\Core\Database\Connection
   */
  protected $database;

  /**
   * Constructs a new DatabaseProfilerStorage instance.
   *
   * @param \Drupal\Core\Database\Connection $database
   *   The database connection.
   */
  public function __construct(Connection $database) {
    $this->database = $database;
  }

  /**
   * {@inheritdoc}
   */
  public function find($ip, $url, $limit, $method, $start = NULL, $end = NULL) : array {
    $select = $this->database
      ->select('webprofiler', 'wp', [
      'fetch' => \PDO::FETCH_ASSOC,
    ]);
    if (NULL === $start) {
      $start = 0;
    }
    if (NULL === $end) {
      $end = time();
    }
    if ($ip = preg_replace('/[^\\d\\.]/', '', $ip)) {
      $select
        ->condition('ip', '%' . $this->database
        ->escapeLike($ip) . '%', 'LIKE');
    }
    if ($url) {
      $select
        ->condition('url', '%' . $this->database
        ->escapeLike(addcslashes($url, '%_\\')) . '%', 'LIKE');
    }
    if ($method) {
      $select
        ->condition('method', $method);
    }
    if (!empty($start)) {
      $select
        ->condition('time', $start, '>=');
    }
    if (!empty($end)) {
      $select
        ->condition('time', $end, '<=');
    }
    $select
      ->fields('wp', [
      'token',
      'ip',
      'method',
      'url',
      'time',
      'parent',
      'status_code',
    ]);
    $select
      ->orderBy('time', 'DESC');
    $select
      ->range(0, $limit);
    return $select
      ->execute()
      ->fetchAllAssoc('token');
  }

  /**
   * {@inheritdoc}
   */
  public function read($token) : ?Profile {
    $record = $this->database
      ->select('webprofiler', 'w')
      ->fields('w')
      ->condition('token', $token)
      ->execute()
      ->fetch();
    if (isset($record->data)) {
      return $this
        ->createProfileFromData($token, $record);
    }
  }

  /**
   * {@inheritdoc}
   */
  public function write(Profile $profile) : bool {
    $args = [
      '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 {
      $query = $this->database
        ->select('webprofiler', 'w')
        ->fields('w', [
        'token',
      ]);
      $query
        ->condition('token', $profile
        ->getToken());
      $count = $query
        ->countQuery()
        ->execute()
        ->fetchAssoc();
      if ($count['expression']) {
        $this->database
          ->update('webprofiler')
          ->fields($args)
          ->condition('token', $profile
          ->getToken())
          ->execute();
      }
      else {
        $this->database
          ->insert('webprofiler')
          ->fields($args)
          ->execute();
      }
      $status = TRUE;
    } catch (\Exception $e) {
      $status = FALSE;
    }
    return $status;
  }

  /**
   * {@inheritdoc}
   */
  public function purge() {
    $this->database
      ->truncate('webprofiler')
      ->execute();
  }

  /**
   * @param string $token
   * @param $data
   *
   * @return \Symfony\Component\HttpKernel\Profiler\Profile
   */
  private function createProfileFromData($token, $data) {
    $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)));
    $profile
      ->setStatusCode($data->status_code);
    return $profile;
  }

}

Classes

Name	Description
DatabaseProfilerStorage	Implements a profiler storage using the DBTNG query api.