Version 1

[yaffs-website] / web / modules / contrib / devel / webprofiler / src / Profiler / DatabaseProfilerStorage.php

diff --git a/web/modules/contrib/devel/webprofiler/src/Profiler/DatabaseProfilerStorage.php b/web/modules/contrib/devel/webprofiler/src/Profiler/DatabaseProfilerStorage.php
new file mode 100644 (file)
index 0000000..d86045d
--- /dev/null
+++ b/web/modules/contrib/devel/webprofiler/src/Profiler/DatabaseProfilerStorage.php
@@ -0,0 +1,158 @@
+<?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.
+   */
+  function __construct(Connection $database) {
+    $this->database = $database;
+  }
+
+  /**
+   * {@inheritdoc}
+   */
+  public function find($ip, $url, $limit, $method, $start = NULL, $end = NULL) {
+    $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) {
+    $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) {
+    $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 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;
+  }
+}