vendor/doctrine/mongodb-odm-bundle/APM/StopwatchCommandLogger.php line 59

Open in your IDE?
  1. <?php
  3. declare(strict_types=1);
  5. namespace Doctrine\Bundle\MongoDBBundle\APM;
  7. use Doctrine\ODM\MongoDB\APM\CommandLoggerInterface;
  8. use MongoDB\Driver\Monitoring\CommandFailedEvent;
  9. use MongoDB\Driver\Monitoring\CommandStartedEvent;
  10. use MongoDB\Driver\Monitoring\CommandSucceededEvent;
  11. use Symfony\Component\Stopwatch\Stopwatch;
  13. use function MongoDB\Driver\Monitoring\addSubscriber;
  14. use function MongoDB\Driver\Monitoring\removeSubscriber;
  15. use function sprintf;
  17. final class StopwatchCommandLogger implements CommandLoggerInterface
  18. {
  19.     /** @var bool */
  20.     private $registered = false;
  22.     /** @var Stopwatch|null */
  23.     private $stopwatch;
  25.     public function __construct(?Stopwatch $stopwatch)
  26.     {
  27.         $this->stopwatch = $stopwatch;
  28.     }
  30.     public function register(): void
  31.     {
  32.         if ($this->stopwatch === null || $this->registered) {
  33.             return;
  34.         }
  36.         $this->registered = true;
  37.         addSubscriber($this);
  38.     }
  40.     public function unregister(): void
  41.     {
  42.         if (! $this->registered) {
  43.             return;
  44.         }
  46.         removeSubscriber($this);
  47.         $this->registered = false;
  48.     }
  50.     public function commandStarted(CommandStartedEvent $event)
  51.     {
  52.         if (! $this->stopwatch) {
  53.             return;
  54.         }
  56.         $this->stopwatch->start(sprintf('mongodb_%s', $event->getRequestId()), 'doctrine_mongodb');
  57.     }
  59.     public function commandSucceeded(CommandSucceededEvent $event)
  60.     {
  61.         if (! $this->stopwatch) {
  62.             return;
  63.         }
  65.         $this->stopwatch->stop(sprintf('mongodb_%s', $event->getRequestId()));
  66.     }
  68.     public function commandFailed(CommandFailedEvent $event)
  69.     {
  70.         if (! $this->stopwatch) {
  71.             return;
  72.         }
  74.         $this->stopwatch->stop(sprintf('mongodb_%s', $event->getRequestId()));
  75.     }
  76. }