CommandLogger.php

<?php

declare(strict_types=1);

namespace Doctrine\ODM\MongoDB\APM;

use Countable;
use MongoDB\Driver\Monitoring\CommandFailedEvent;
use MongoDB\Driver\Monitoring\CommandStartedEvent;
use MongoDB\Driver\Monitoring\CommandSubscriber;
use MongoDB\Driver\Monitoring\CommandSucceededEvent;
use function count;
use function MongoDB\Driver\Monitoring\addSubscriber;
use function MongoDB\Driver\Monitoring\removeSubscriber;

final class CommandLogger implements Countable, CommandSubscriber
{
    /** @var Command[] */
    private $commands = [];

    /** @var CommandStartedEvent[] */
    private $startedCommands = [];

    /** @var bool */
    private $registered = false;

    public function register() : void
    {
        if ($this->registered) {
            return;
        }

        $this->registered = true;
        addSubscriber($this);
    }

    public function unregister() : void
    {
        if (! $this->registered) {
            return;
        }

        removeSubscriber($this);
        $this->registered = false;
    }

    public function commandStarted(CommandStartedEvent $event)
    {
        $this->startedCommands[$event->getRequestId()] = $event;
    }

    public function commandSucceeded(CommandSucceededEvent $event)
    {
        $commandStartedEvent = $this->findAndRemoveCommandStartedEvent($event->getRequestId());
        if (! $commandStartedEvent) {
            return;
        }

        $this->logCommand(Command::createForSucceededCommand($commandStartedEvent, $event));
    }

    public function commandFailed(CommandFailedEvent $event)
    {
        $commandStartedEvent = $this->findAndRemoveCommandStartedEvent($event->getRequestId());
        if (! $commandStartedEvent) {
            return;
        }

        $this->logCommand(Command::createForFailedCommand($commandStartedEvent, $event));
    }

    public function clear() : void
    {
        $this->commands = [];
    }

    public function count() : int
    {
        return count($this->commands);
    }

    /**
     * @return Command[]
     */
    public function getAll() : array
    {
        return $this->commands;
    }

    private function findAndRemoveCommandStartedEvent(string $requestId) : ?CommandStartedEvent
    {
        $startedEvent = $this->startedCommands[$requestId] ?? null;
        unset($this->startedCommands[$requestId]);

        return $startedEvent;
    }

    private function logCommand(Command $command) : void
    {
        $this->commands[] = $command;
    }
}