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

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. /** @internal since version 4.7.0 */
  23. class CommandDataCollector extends DataCollector
  24. {
  25.     public function __construct(private CommandLogger $commandLogger)
  26.     {
  27.     }
  29.     public function collect(Request $request, Response $response, ?Throwable $exception = null): void
  30.     {
  31.         $this->data = [
  32.             'num_commands' => count($this->commandLogger),
  33.             'commands' => array_map(
  34.                 static function (Command $command): array {
  35.                     $dbProperty = '$db';
  37.                     return [
  38.                         'database' => $command->getCommand()->$dbProperty ?? '',
  39.                         'command' => json_decode(toCanonicalExtendedJSON(fromPHP($command->getCommand()))),
  40.                         'durationMicros' => $command->getDurationMicros(),
  41.                     ];
  42.                 },
  43.                 $this->commandLogger->getAll(),
  44.             ),
  45.             'time' => array_reduce(
  46.                 $this->commandLogger->getAll(),
  47.                 static fn (int $total, Command $command): int => $total + $command->getDurationMicros(),
  48.                 0,
  49.             ),
  50.         ];
  51.     }
  53.     public function reset(): void
  54.     {
  55.         $this->commandLogger->clear();
  56.         $this->data = [
  57.             'num_commands' => 0,
  58.             'commands' => [],
  59.         ];
  60.     }
  62.     public function getCommandCount(): int
  63.     {
  64.         return $this->data['num_commands'];
  65.     }
  67.     public function getTime(): int
  68.     {
  69.         return $this->data['time'];
  70.     }
  72.     /** @return array<array{database: string, command: stdClass, durationMicros: int}> */
  73.     public function getCommands(): array
  74.     {
  75.         return $this->data['commands'];
  76.     }
  78.     public function getName(): string
  79.     {
  80.         return 'mongodb';
  81.     }
  82. }