Yiisoft\Db\Debug\DatabaseCollector

Complexity

Total Complexity

9

Size/Duplication

Total Lines	150
Duplicated Lines	0 %

Importance

Changes	1
Bugs	0	Features	0

9 Methods

Rating	Name	Duplication	Size	Complexity
A	collectQueryError()	0	9	1
A	collectTransactionRollback()	0	10	1
A	collectTransactionStart()	0	14	1
A	collectTransactionCommit()	0	10	1
A	reset()	0	3	1
A	getCollected()	0	7	1
A	getSummary()	0	18	1
A	collectQueryEnd()	0	9	1
A	collectQueryStart()	0	18	1

<?php

declare(strict_types=1);

namespace Yiisoft\Db\Debug;

use Throwable;
use Yiisoft\Yii\Debug\Collector\CollectorTrait;
use Yiisoft\Yii\Debug\Collector\SummaryCollectorInterface;

final class DatabaseCollector implements SummaryCollectorInterface
{
    use CollectorTrait;

    private const ACTION_QUERY_START = 'query.start';
    private const ACTION_QUERY_END = 'query.end';
    private const ACTION_QUERY_ERROR = 'query.error';

    private const ACTION_TRANSACTION_START = 'transaction.start';
    private const ACTION_TRANSACTION_ROLLBACK = 'transaction.rollback';
    private const ACTION_TRANSACTION_COMMIT = 'transaction.commit';

    private const TRANSACTION_STATUS_COMMIT = 'commit';
    private const TRANSACTION_STATUS_ROLLBACK = 'rollback';
    private const TRANSACTION_STATUS_START = 'start';

    private const QUERY_STATUS_INITIALIZED = 'initialized';
    private const QUERY_STATUS_ERROR = 'error';
    private const QUERY_STATUS_SUCCESS = 'success';

    private array $queries = [];
    private array $transactions = [];

    private int $position = 0;
    private int $currentTransactionId = 0;

    public function collectQueryStart(
        string $id,
        string $sql,
        string $rawSql,
        array $params,
        string $line,
    ): void {
        $this->queries[$id] = [
            'position' => $this->position++,
            'transactionId' => $this->currentTransactionId,
            'sql' => $sql,
            'rawSql' => $rawSql,
            'params' => $params,
            'line' => $line,
            'status' => self::QUERY_STATUS_INITIALIZED,
            'actions' => [
                'action' => self::ACTION_QUERY_START,
                'time' => microtime(true),
            ],
        ];
    }

    public function collectQueryEnd(
        string $id,
        int $rowsNumber,
    ): void {
        $this->queries[$id]['rowsNumber'] = $rowsNumber;
        $this->queries[$id]['status'] = self::QUERY_STATUS_SUCCESS;
        $this->queries[$id]['actions'][] = [
            'action' => self::ACTION_QUERY_END,
            'time' => microtime(true),
        ];
    }

    public function collectQueryError(
        string $id,
        Throwable $exception,
    ): void {
        $this->queries[$id]['exception'] = $exception;
        $this->queries[$id]['status'] = self::QUERY_STATUS_ERROR;
        $this->queries[$id]['actions'][] = [
            'action' => self::ACTION_QUERY_ERROR,
            'time' => microtime(true),
        ];
    }

    public function collectTransactionStart(
        ?string $isolationLevel,
        string $line,
    ): void {
        $id = ++$this->currentTransactionId;
        $this->transactions[$id] = [
            'id' => $id,
            'position' => $this->position++,
            'status' => self::TRANSACTION_STATUS_START,
            'line' => $line,
            'level' => $isolationLevel,
            'actions' => [
                'action' => self::ACTION_TRANSACTION_START,
                'time' => microtime(true),
            ],
        ];
    }

    public function collectTransactionRollback(
        string $line,
    ) {
        $this->transactions[$this->currentTransactionId]['status'] = self::TRANSACTION_STATUS_ROLLBACK;
        $this->transactions[$this->currentTransactionId]['actions'] = [
            'action' => self::ACTION_TRANSACTION_ROLLBACK,
            'line' => $line,
            'time' => microtime(true),
        ];
        ++$this->currentTransactionId;
    }

    public function collectTransactionCommit(
        string $line,
    ) {
        $this->transactions[$this->currentTransactionId]['status'] = self::TRANSACTION_STATUS_COMMIT;
        $this->transactions[$this->currentTransactionId]['actions'] = [
            'action' => self::ACTION_TRANSACTION_COMMIT,
            'line' => $line,
            'time' => microtime(true),
        ];
        ++$this->currentTransactionId;
    }

    public function getCollected(): array
    {
        usort($this->queries, fn (array $a, array $b) => $a['position'] <=> $b['position']);

        return [
            'queries' => array_values($this->queries),
            'transactions' => array_values($this->transactions),
        ];
    }

    public function getSummary(): array
    {
        return [
            'db' => [
                'queries' => [
                    'error' => count(
                        array_filter($this->queries, fn (array $query) => $query['status'] === self::QUERY_STATUS_ERROR)
                    ),
                    'total' => count($this->queries),
                ],
                'transactions' => [
                    'error' => count(
                        array_filter(
                            $this->transactions,
                            fn (array $query) => $query['status'] === self::TRANSACTION_STATUS_ROLLBACK
                        )
                    ),
                    'total' => count($this->transactions),
                ],
            ],
        ];
    }

    private function reset(): void

The method reset() is not used, and could be removed.

    {
        $this->queries = [];
    }
}