vendor/doctrine/mongodb-odm/lib/Doctrine/ODM/MongoDB/APM/CommandLogger.php line 52

Open in your IDE?
  1. <?php
  3. declare(strict_types=1);
  5. namespace Doctrine\ODM\MongoDB\APM;
  7. use Countable;
  8. use MongoDB\Driver\Monitoring\CommandFailedEvent;
  9. use MongoDB\Driver\Monitoring\CommandStartedEvent;
  10. use MongoDB\Driver\Monitoring\CommandSucceededEvent;
  12. use function count;
  13. use function MongoDB\Driver\Monitoring\addSubscriber;
  14. use function MongoDB\Driver\Monitoring\removeSubscriber;
  16. final class CommandLogger implements Countable, CommandLoggerInterface
  17. {
  18.     /** @var Command[] */
  19.     private $commands = [];
  21.     /** @var CommandStartedEvent[] */
  22.     private $startedCommands = [];
  24.     /** @var bool */
  25.     private $registered = false;
  27.     public function register(): void
  28.     {
  29.         if ($this->registered) {
  30.             return;
  31.         }
  33.         $this->registered = true;
  34.         addSubscriber($this);
  35.     }
  37.     public function unregister(): void
  38.     {
  39.         if (! $this->registered) {
  40.             return;
  41.         }
  43.         removeSubscriber($this);
  44.         $this->registered = false;
  45.     }
  47.     public function commandStarted(CommandStartedEvent $event)
  48.     {
  49.         $this->startedCommands[$event->getRequestId()] = $event;
  50.     }
  52.     public function commandSucceeded(CommandSucceededEvent $event)
  53.     {
  54.         $commandStartedEvent = $this->findAndRemoveCommandStartedEvent($event->getRequestId());
  55.         if (! $commandStartedEvent) {
  56.             return;
  57.         }
  59.         $this->logCommand(Command::createForSucceededCommand($commandStartedEvent, $event));
  60.     }
  62.     public function commandFailed(CommandFailedEvent $event)
  63.     {
  64.         $commandStartedEvent = $this->findAndRemoveCommandStartedEvent($event->getRequestId());
  65.         if (! $commandStartedEvent) {
  66.             return;
  67.         }
  69.         $this->logCommand(Command::createForFailedCommand($commandStartedEvent, $event));
  70.     }
  72.     public function clear(): void
  73.     {
  74.         $this->commands = [];
  75.     }
  77.     public function count(): int
  78.     {
  79.         return count($this->commands);
  80.     }
  82.     /**
  83.      * @return Command[]
  84.      */
  85.     public function getAll(): array
  86.     {
  87.         return $this->commands;
  88.     }
  90.     private function findAndRemoveCommandStartedEvent(string $requestId): ?CommandStartedEvent
  91.     {
  92.         $startedEvent = $this->startedCommands[$requestId] ?? null;
  93.         unset($this->startedCommands[$requestId]);
  95.         return $startedEvent;
  96.     }
  98.     private function logCommand(Command $command): void
  99.     {
  100.         $this->commands[] = $command;
  101.     }
  102. }