vendor/doctrine/mongodb-odm-bundle/DataCollector/CommandDataCollector.php line 32

  1. <?php
  3. declare(strict_types=1);
  5. namespace Doctrine\Bundle\MongoDBBundle\DataCollector;
  7. use Doctrine\ODM\MongoDB\APM\Command;
  8. use Doctrine\ODM\MongoDB\APM\CommandLogger;
  9. use stdClass;
  10. use Symfony\Component\HttpFoundation\Request;
  11. use Symfony\Component\HttpFoundation\Response;
  12. use Symfony\Component\HttpKernel\DataCollector\DataCollector;
  13. use Throwable;
  15. use function array_map;
  16. use function array_reduce;
  17. use function count;
  18. use function json_decode;
  19. use function MongoDB\BSON\fromPHP;
  20. use function MongoDB\BSON\toCanonicalExtendedJSON;
  22. class CommandDataCollector extends DataCollector
  23. {
  24.     /** @var CommandLogger */
  25.     private $commandLogger;
  27.     public function __construct(CommandLogger $commandLogger)
  28.     {
  29.         $this->commandLogger = $commandLogger;
  30.     }
  32.     public function collect(Request $request, Response $response, ?Throwable $exception = null): void
  33.     {
  34.         $this->data = [
  35.             'num_commands' => count($this->commandLogger),
  36.             'commands' => array_map(
  37.                 static function (Command $command): array {
  38.                     $dbProperty = '$db';
  40.                     return [
  41.                         'database' => $command->getCommand()->$dbProperty ?? '',
  42.                         'command' => json_decode(toCanonicalExtendedJSON(fromPHP($command->getCommand()))),
  43.                         'durationMicros' => $command->getDurationMicros(),
  44.                     ];
  45.                 },
  46.                 $this->commandLogger->getAll()
  47.             ),
  48.             'time' => array_reduce(
  49.                 $this->commandLogger->getAll(),
  50.                 static function (int $total, Command $command): int {
  51.                     return $total + $command->getDurationMicros();
  52.                 },
  53.                 0
  54.             ),
  55.         ];
  56.     }
  58.     public function reset(): void
  59.     {
  60.         $this->commandLogger->clear();
  61.         $this->data = [
  62.             'num_commands' => 0,
  63.             'commands' => [],
  64.         ];
  65.     }
  67.     public function getCommandCount(): int
  68.     {
  69.         return $this->data['num_commands'];
  70.     }
  72.     public function getTime(): int
  73.     {
  74.         return $this->data['time'];
  75.     }
  77.     /**
  78.      * @return array<array{database: string, command: stdClass, durationMicros: int}>
  79.      */
  80.     public function getCommands(): array
  81.     {
  82.         return $this->data['commands'];
  83.     }
  85.     public function getName(): string
  86.     {
  87.         return 'mongodb';
  88.     }
  89. }