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

Open in your IDE?
  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.     private CommandLogger $commandLogger;
  26.     public function __construct(CommandLogger $commandLogger)
  27.     {
  28.         $this->commandLogger = $commandLogger;
  29.     }
  31.     public function collect(Request $request, Response $response, ?Throwable $exception = null): void
  32.     {
  33.         $this->data = [
  34.             'num_commands' => count($this->commandLogger),
  35.             'commands' => array_map(
  36.                 static function (Command $command): array {
  37.                     $dbProperty = '$db';
  39.                     return [
  40.                         'database' => $command->getCommand()->$dbProperty ?? '',
  41.                         'command' => json_decode(toCanonicalExtendedJSON(fromPHP($command->getCommand()))),
  42.                         'durationMicros' => $command->getDurationMicros(),
  43.                     ];
  44.                 },
  45.                 $this->commandLogger->getAll(),
  46.             ),
  47.             'time' => array_reduce(
  48.                 $this->commandLogger->getAll(),
  49.                 static fn (int $total, Command $command): int => $total + $command->getDurationMicros(),
  50.                 0,
  51.             ),
  52.         ];
  53.     }
  55.     public function reset(): void
  56.     {
  57.         $this->commandLogger->clear();
  58.         $this->data = [
  59.             'num_commands' => 0,
  60.             'commands' => [],
  61.         ];
  62.     }
  64.     public function getCommandCount(): int
  65.     {
  66.         return $this->data['num_commands'];
  67.     }
  69.     public function getTime(): int
  70.     {
  71.         return $this->data['time'];
  72.     }
  74.     /** @return array<array{database: string, command: stdClass, durationMicros: int}> */
  75.     public function getCommands(): array
  76.     {
  77.         return $this->data['commands'];
  78.     }
  80.     public function getName(): string
  81.     {
  82.         return 'mongodb';
  83.     }
  84. }