Fix boundaries for Symfony HttpKernel auto instrumentation

Author: dciprian-petrisor

src/Instrumentation/Symfony/src/SymfonyInstrumentation.php

@@ -86,6 +86,30 @@ public static function register(): void
                 array $params,
                 ?Response $response,
                 ?\Throwable $exception
+            ): void {
+                $scope = Context::storage()->scope();
+                if (null === $scope || null === $exception) {
+                    return;
+                }
+
+                $span = Span::fromContext($scope->context());
+                $scope->detach();
+                $span->recordException($exception, [
+                    TraceAttributes::EXCEPTION_ESCAPED => true,
+                ]);
+                if (null !== $response && $response->getStatusCode() >= Response::HTTP_INTERNAL_SERVER_ERROR) {
+                    $span->setStatus(StatusCode::STATUS_ERROR, $exception->getMessage());
+                }
+            }
+        );
+
+        hook(
+            HttpKernel::class,
+            'terminate',
+            post: static function (
+                HttpKernel $kernel,
+                array $params,
+                ?\Throwable $exception
             ): void {
                 $scope = Context::storage()->scope();
                 if (null === $scope) {
@@ -95,6 +119,7 @@ public static function register(): void
                 $span = Span::fromContext($scope->context());
 
                 $request = ($params[0] instanceof Request) ? $params[0] : null;
+                $response = ($params[1] instanceof Response) ? $params[1] : null;
                 if (null !== $request) {
                     $routeName = $request->attributes->get('_route', '');
 

src/Instrumentation/Symfony/tests/Integration/SymfonyInstrumentationTest.php

@@ -33,6 +33,7 @@ public function test_http_kernel_handle_exception(): void
         $this->assertCount(0, $this->storage);
 
         $response = $kernel->handle(new Request());
+        $kernel->terminate(new Request(), $response);
 
         $this->assertArrayHasKey(
             TraceResponsePropagator::TRACERESPONSE,
@@ -50,6 +51,7 @@ public function test_http_kernel_marks_root_as_erroneous(): void
         $this->assertCount(0, $this->storage);
 
         $response = $kernel->handle(new Request(), HttpKernelInterface::MAIN_REQUEST, true);
+        $kernel->terminate(new Request(), $response);
 
         $this->assertCount(1, $this->storage);
         $this->assertSame(500, $this->storage[0]->getAttributes()->get(TraceAttributes::HTTP_RESPONSE_STATUS_CODE));
@@ -71,6 +73,7 @@ public function test_http_kernel_handle_attributes(): void
         $request->attributes->set('_route', 'test_route');
 
         $response = $kernel->handle($request);
+        $kernel->terminate($request, $response);
 
         $attributes = $this->storage[0]->getAttributes();
         $this->assertCount(1, $this->storage);
@@ -105,6 +108,8 @@ public function test_http_kernel_handle_stream_response(): void
         $this->assertCount(0, $this->storage);
 
         $response = $kernel->handle(new Request());
+        $kernel->terminate(new Request(), $response);
+
         $this->assertCount(1, $this->storage);
         $this->assertNull($this->storage[0]->getAttributes()->get(TraceAttributes::HTTP_RESPONSE_BODY_SIZE));
 
@@ -127,6 +132,8 @@ public function test_http_kernel_handle_binary_file_response(): void
         $this->assertCount(0, $this->storage);
 
         $response = $kernel->handle(new Request());
+        $kernel->terminate(new Request(), $response);
+
         $this->assertCount(1, $this->storage);
         $this->assertNull($this->storage[0]->getAttributes()->get(TraceAttributes::HTTP_RESPONSE_BODY_SIZE));
 
@@ -151,6 +158,8 @@ public function test_http_kernel_handle_with_empty_route(): void
         $request->attributes->set('_route', '');
 
         $response = $kernel->handle($request, HttpKernelInterface::MAIN_REQUEST, true);
+        $kernel->terminate(new Request(), $response);
+
         $this->assertCount(1, $this->storage);
         $this->assertFalse($this->storage[0]->getAttributes()->has(TraceAttributes::HTTP_ROUTE));
 
@@ -173,6 +182,8 @@ public function test_http_kernel_handle_without_route(): void
         $this->assertCount(0, $this->storage);
 
         $response = $kernel->handle(new Request(), HttpKernelInterface::MAIN_REQUEST, true);
+        $kernel->terminate(new Request(), $response);
+
         $this->assertCount(1, $this->storage);
         $this->assertFalse($this->storage[0]->getAttributes()->has(TraceAttributes::HTTP_ROUTE));
 
@@ -196,7 +207,9 @@ public function test_http_kernel_handle_subrequest(): void
         $request = new Request();
         $request->attributes->set('_controller', 'ErrorController');
 
-        $kernel->handle($request, HttpKernelInterface::SUB_REQUEST);
+        $response = $kernel->handle($request, HttpKernelInterface::SUB_REQUEST);
+        $kernel->terminate($request, $response);
+
         $this->assertCount(1, $this->storage);
 
         $span = $this->storage[0];