Add sql.origin to SQL query spans

Author: stayallive

src/Sentry/Laravel/Tracing/BacktraceHelper.php

@@ -0,0 +1,110 @@
+<?php
+
+namespace Sentry\Laravel\Tracing;
+
+use Sentry\Frame;
+use Sentry\Options;
+use Sentry\FrameBuilder;
+use Illuminate\Support\Str;
+use Sentry\Serializer\RepresentationSerializerInterface;
+
+class BacktraceHelper
+{
+    /**
+     * @var Options The SDK client options
+     */
+    private $options;
+
+    /**
+     * @var FrameBuilder An instance of the builder of {@see Frame} objects
+     */
+    private $frameBuilder;
+
+    /**
+     * Constructor.
+     *
+     * @param Options                           $options                  The SDK client options
+     * @param RepresentationSerializerInterface $representationSerializer The representation serializer
+     */
+    public function __construct(Options $options, RepresentationSerializerInterface $representationSerializer)
+    {
+        $this->options = $options;
+        $this->frameBuilder = new FrameBuilder($options, $representationSerializer);
+    }
+
+    /**
+     * Find the first in app frame for a given backtrace.
+     *
+     * @param array<int, array<string, mixed>> $backtrace The backtrace
+     *
+     * @phpstan-param list<array{
+     *     line?: integer,
+     *     file?: string,
+     * }> $backtrace
+     */
+    public function findFirstInAppFrameForBacktrace(array $backtrace): ?Frame
+    {
+        $file = Frame::INTERNAL_FRAME_FILENAME;
+        $line = 0;
+
+        foreach ($backtrace as $backtraceFrame) {
+            $frame = $this->frameBuilder->buildFromBacktraceFrame($file, $line, $backtraceFrame);
+
+            if ($frame->isInApp()) {
+                return $frame;
+            }
+
+            $file = $backtraceFrame['file'] ?? Frame::INTERNAL_FRAME_FILENAME;
+            $line = $backtraceFrame['line'] ?? 0;
+        }
+
+        return null;
+    }
+
+    /**
+     * Takes a frame and if it's a compiled view path returns the original view path.
+     *
+     * @param \Sentry\Frame $frame
+     *
+     * @return string|null
+     */
+    public function getOriginalViewPathForFrameOfCompiledViewPath(Frame $frame): ?string
+    {
+        // Check if we are dealing with a frame for a cached view path
+        if (!Str::startsWith($frame->getFile(), '/storage/framework/views/')) {
+            return null;
+        }
+
+        // If for some reason the file does not exists, skip resolving
+        if (!file_exists($frame->getAbsoluteFilePath())) {
+            return null;
+        }
+
+        $viewFileContents = file_get_contents($frame->getAbsoluteFilePath());
+
+        preg_match('/PATH (?<originalPath>.*?) ENDPATH/', $viewFileContents, $matches);
+
+        // No path comment found in the file, must be a very old Laravel version
+        if (empty($matches['originalPath'])) {
+            return null;
+        }
+
+        return $this->stripPrefixFromFilePath($matches['originalPath']);
+    }
+
+    /**
+     * Removes from the given file path the specified prefixes.
+     *
+     * @param string $filePath The path to the file
+     */
+    private function stripPrefixFromFilePath(string $filePath): string
+    {
+        foreach ($this->options->getPrefixes() as $prefix) {
+            if (Str::startsWith($filePath, $prefix)) {
+                return mb_substr($filePath, mb_strlen($prefix));
+            }
+        }
+
+        return $filePath;
+    }
+}

src/Sentry/Laravel/Tracing/EventHandler.php

@@ -35,14 +35,22 @@ class EventHandler
      */
     private $events;
 
+    /**
+     * The backtrace helper.
+     *
+     * @var \Sentry\Laravel\Tracing\BacktraceHelper
+     */
+    private $backtraceHelper;
+
     /**
      * EventHandler constructor.
      *
      * @param \Illuminate\Contracts\Events\Dispatcher $events
      */
-    public function __construct(Dispatcher $events)
+    public function __construct(Dispatcher $events, BacktraceHelper $backtraceHelper)
     {
-        $this->events = $events;
+        $this->events          = $events;
+        $this->backtraceHelper = $backtraceHelper;
     }
 
     /**
@@ -152,6 +160,21 @@ private function recordQuerySpan($query, $time): void
         $context->setStartTimestamp(microtime(true) - $time / 1000);
         $context->setEndTimestamp($context->getStartTimestamp() + $time / 1000);
 
+        $this->resolveQuerySourceFromBacktrace($context);
+
         $parentSpan->startChild($context);
     }
+
+    private function resolveQuerySourceFromBacktrace(SpanContext $context): void
+    {
+        $firstAppFrame = $this->backtraceHelper->findFirstInAppFrameForBacktrace(debug_backtrace(DEBUG_BACKTRACE_IGNORE_ARGS));
+
+        if ($firstAppFrame !== null) {
+            $filePath = $this->backtraceHelper->getOriginalViewPathForFrameOfCompiledViewPath($firstAppFrame) ?? $firstAppFrame->getFile();
+
+            $context->setData([
+                'sql.origin' => "{$filePath}:{$firstAppFrame->getLine()}",
+            ]);
+        }
+    }
 }

src/Sentry/Laravel/Tracing/ServiceProvider.php

@@ -8,6 +8,7 @@
 use Illuminate\View\Engines\EngineResolver;
 use Illuminate\View\Factory as ViewFactory;
 use Sentry\Laravel\BaseServiceProvider;
+use Sentry\Serializer\RepresentationSerializer;
 
 class ServiceProvider extends BaseServiceProvider
 {
@@ -30,11 +31,23 @@ public function boot(): void
     public function register(): void
     {
         $this->app->singleton(Middleware::class);
+
+        $this->app->singleton(BacktraceHelper::class, function () {
+            /** @var \Sentry\State\Hub $sentry */
+            $sentry = $this->app->make(self::$abstract);
+
+            $options = $sentry->getClient()->getOptions();
+
+            return new BacktraceHelper($options, new RepresentationSerializer($options));
+        });
     }
 
     private function bindEvents(): void
     {
-        $handler = new EventHandler($this->app->events);
+        $handler = new EventHandler(
+            $this->app->events,
+            $this->app->make(BacktraceHelper::class)
+        );
 
         $handler->subscribe();
     }