Add support for query loggers and command loggers

Author: alcaeus

APM/PSRCommandLogger.php

@@ -0,0 +1,69 @@
+<?php
+
+declare(strict_types = 1);
+
+namespace Doctrine\Bundle\MongoDBBundle\APM;
+
+use MongoDB\Driver\Monitoring\CommandFailedEvent;
+use MongoDB\Driver\Monitoring\CommandStartedEvent;
+use MongoDB\Driver\Monitoring\CommandSubscriber;
+use MongoDB\Driver\Monitoring\CommandSucceededEvent;
+use Psr\Log\LoggerInterface;
+use function json_encode;
+use function MongoDB\Driver\Monitoring\addSubscriber;
+use function MongoDB\Driver\Monitoring\removeSubscriber;
+
+final class PSRCommandLogger implements CommandSubscriber
+{
+    /** @var bool */
+    private $registered = false;
+
+    /** @var LoggerInterface|null */
+    private $logger;
+
+    /** @var string */
+    private $prefix;
+
+    public function __construct(?LoggerInterface $logger, string $prefix = 'MongoDB command: ')
+    {
+        $this->logger = $logger;
+        $this->prefix = $prefix;
+    }
+
+    public function register(): void
+    {
+        if ($this->logger === null || $this->registered) {
+            return;
+        }
+
+        $this->registered = true;
+        addSubscriber($this);
+    }
+
+    public function unregister(): void
+    {
+        if (! $this->registered) {
+            return;
+        }
+
+        removeSubscriber($this);
+        $this->registered = false;
+    }
+
+    public function commandStarted(CommandStartedEvent $event)
+    {
+        if (!$this->logger) {
+            return;
+        }
+
+        $this->logger->debug($this->prefix . json_encode($event->getCommand()));
+    }
+
+    public function commandSucceeded(CommandSucceededEvent $event)
+    {
+    }
+
+    public function commandFailed(CommandFailedEvent $event)
+    {
+    }
+}

DataCollector/CommandDataCollector.php

@@ -0,0 +1,59 @@
+<?php
+
+namespace Doctrine\Bundle\MongoDBBundle\DataCollector;
+
+use Doctrine\ODM\MongoDB\APM\Command;
+use Doctrine\ODM\MongoDB\APM\CommandLogger;
+use Symfony\Component\HttpFoundation\Request;
+use Symfony\Component\HttpFoundation\Response;
+use Symfony\Component\HttpKernel\DataCollector\DataCollector;
+
+class CommandDataCollector extends DataCollector
+{
+    private $commandLogger;
+
+    public function __construct(CommandLogger $commandLogger)
+    {
+        $this->commandLogger = $commandLogger;
+    }
+
+    public function collect(Request $request, Response $response, \Exception $exception = null)
+    {
+        $this->data = [
+            'num_commands' => count($this->commandLogger),
+            'commands' => array_map(
+                function (Command $command): string {
+                    return json_encode($command->getCommand());
+                },
+                $this->commandLogger->getAll()
+            ),
+        ];
+    }
+
+    public function reset()
+    {
+        $this->commandLogger->clear();
+        $this->data = [
+            'num_commands' => 0,
+            'commands' => [],
+        ];
+    }
+
+    public function getCommandCount(): int
+    {
+        return $this->data['num_commands'];
+    }
+
+    /**
+     * @return string[]
+     */
+    public function getCommands(): array
+    {
+        return $this->data['commands'];
+    }
+
+    public function getName()
+    {
+        return 'mongodb';
+    }
+}