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.
     */
    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;
    }

}