Implement DataCollector::reset()

Author: GromNaN

src/DataCollector/DriverEventSubscriber.php

@@ -25,8 +25,10 @@
 use MongoDB\Driver\Monitoring\CommandStartedEvent;
 use MongoDB\Driver\Monitoring\CommandSubscriber;
 use MongoDB\Driver\Monitoring\CommandSucceededEvent;
+use Symfony\Contracts\Service\ResetInterface;
 
-final class DriverEventSubscriber implements CommandSubscriber
+/** @internal */
+final class DriverEventSubscriber implements CommandSubscriber, ResetInterface
 {
     /**
      * @var list<CommandFailedEvent|CommandStartedEvent|CommandSucceededEvent>
@@ -60,4 +62,9 @@ public function commandSucceeded(CommandSucceededEvent $event): void
     {
         $this->events[] = $event;
     }
+
+    public function reset(): void
+    {
+        $this->events = [];
+    }
 }

src/DataCollector/MongoDBDataCollector.php

@@ -28,6 +28,7 @@
 use Symfony\Component\HttpFoundation\Response;
 use Symfony\Component\HttpKernel\DataCollector\DataCollector;
 
+/** @internal */
 final class MongoDBDataCollector extends DataCollector
 {
     /**
@@ -63,23 +64,20 @@ public function collect(Request $request, Response $response, ?\Throwable $excep
                         'startedAt' => hrtime(true),
                         'commandName' => $event->getCommandName(),
                         'command' => $command,
-                        // 'server' => $event->getServer()->getInfo(),
                         'operationId' => $event->getOperationId(),
                         'database' => $event->getDatabaseName(),
                         'serviceId' => $event->getServiceId(),
                     ];
                     ++$requestCount;
                 } elseif ($event instanceof CommandSucceededEvent) {
                     $requests[$requestId] += [
-                        // 'reply' => Document::fromPHP($event->getReply()),
                         'duration' => $event->getDurationMicros(),
                         'endedAt' => hrtime(true),
                         'success' => true,
                     ];
                     $totalTime += $event->getDurationMicros();
                 } elseif ($event instanceof CommandFailedEvent) {
                     $requests[$requestId] += [
-                        // 'reply' => Document::fromPHP($event->getReply()),
                         'duration' => $event->getDurationMicros(),
                         'error' => $event->getError(),
                         'success' => false,
@@ -127,6 +125,10 @@ public function getName(): string
 
     public function reset(): void
     {
-        // TODO: Implement reset() method.
+        $this->data = [];
+
+        foreach ($this->clients as ['subscriber' => $subscriber]) {
+            $subscriber->reset();
+        }
     }
 }