Support for nested commands

Author: zoilomora

src/Symfony/Component/Console/EventSubscriber.php

@@ -3,23 +3,26 @@
 
 namespace PcComponentes\ElasticAPM\Symfony\Component\Console;
 
+use Symfony\Component\Console\Command\Command;
 use Symfony\Component\Console\ConsoleEvents;
 use Symfony\Component\Console\Event\ConsoleCommandEvent;
 use Symfony\Component\Console\Event\ConsoleErrorEvent;
 use Symfony\Component\Console\Event\ConsoleTerminateEvent;
 use Symfony\Component\EventDispatcher\EventSubscriberInterface;
 use ZoiloMora\ElasticAPM\ElasticApmTracer;
-use ZoiloMora\ElasticAPM\Events\Transaction\Transaction;
 
 final class EventSubscriber implements EventSubscriberInterface
 {
     private ElasticApmTracer $elasticApmTracer;
 
-    private Transaction $transaction;
+    private array $transactions;
+    private array $spans;
 
     public function __construct(ElasticApmTracer $elasticApmTracer)
     {
         $this->elasticApmTracer = $elasticApmTracer;
+        $this->transactions = [];
+        $this->spans = [];
     }
 
     public static function getSubscribedEvents(): array
@@ -34,19 +37,39 @@ public static function getSubscribedEvents(): array
     public function onConsoleCommandEvent(ConsoleCommandEvent $event): void
     {
         $command = $event->getCommand();
+        $key = $this->transactionKey($command);
 
-        $this->transaction = $this->elasticApmTracer->startTransaction(
+        if (0 !== \count($this->transactions)) {
+            $this->spans[$key] = $this->elasticApmTracer->startSpan(
+                $command->getName(),
+                'console',
+            );
+        }
+
+        $this->transactions[$key] = $this->elasticApmTracer->startTransaction(
             $command->getName(),
             'console',
         );
     }
 
     public function onConsoleTerminateEvent(ConsoleTerminateEvent $event): void
     {
-        $this->transaction->stop(
+        $key = $this->transactionKey(
+            $event->getCommand(),
+        );
+
+        $this->transactions[$key]->stop(
             (string) $event->getExitCode(),
         );
 
+        if (true === \array_key_exists($key, $this->spans)) {
+            $this->spans[$key]->stop();
+        }
+
+        if (\array_key_first($this->transactions) !== $key) {
+            return;
+        }
+
         $this->elasticApmTracer->flush();
     }
 
@@ -55,9 +78,10 @@ public function onConsoleErrorEvent(ConsoleErrorEvent $event): void
         $this->elasticApmTracer->captureException(
             $event->getError(),
         );
+    }
 
-        $this->transaction->stop(
-            (string) $event->getExitCode(),
-        );
+    private function transactionKey(Command $command): int
+    {
+        return \spl_object_id($command);
     }
 }