feat: add server address tracing for database connections

src/sentry/src/ConfigProvider.php

@@ -33,6 +33,7 @@ public function __invoke(): array
                 Tracing\Aspect\CoordinatorAspect::class,
                 Tracing\Aspect\CoroutineAspect::class,
                 Tracing\Aspect\DbAspect::class,
+                Tracing\Aspect\DbConnectionAspect::class,
                 Tracing\Aspect\ElasticsearchAspect::class,
                 Tracing\Aspect\ElasticsearchRequestAspect::class,
                 Tracing\Aspect\FilesystemAspect::class,

src/sentry/src/Constants.php

@@ -15,6 +15,10 @@ class Constants
 {
     public const TRACE_CARRIER = 'sentry.tracing.trace_carrier';
 
+    public const TRACE_DB_SERVER_ADDRESS = 'sentry.tracing.db.server.address';
+
+    public const TRACE_DB_SERVER_PORT = 'sentry.tracing.db.server.port';
+
     public const TRACE_REDIS_SERVER_ADDRESS = 'sentry.tracing.redis.server.address';
 
     public const TRACE_REDIS_SERVER_PORT = 'sentry.tracing.redis.server.port';

src/sentry/src/Tracing/Aspect/DbConnectionAspect.php

@@ -0,0 +1,42 @@
+<?php
+
+declare(strict_types=1);
+/**
+ * This file is part of friendsofhyperf/components.
+ *
+ * @link     https://github.com/friendsofhyperf/components
+ * @document https://github.com/friendsofhyperf/components/blob/main/README.md
+ * @contact  huangdijia@gmail.com
+ */
+
+namespace FriendsOfHyperf\Sentry\Tracing\Aspect;
+
+use FriendsOfHyperf\Sentry\Constants;
+use Hyperf\Context\Context;
+use Hyperf\Di\Aop\AbstractAspect;
+use Hyperf\Di\Aop\ProceedingJoinPoint;
+use PDO;
+
+use function Hyperf\Tappable\tap;
+
+class DbConnectionAspect extends AbstractAspect
+{
+    public array $classes = [
+        'Hyperf\Database\Connection::getPdoForSelect',
+        'Hyperf\Database\Connection::getPdo',
+    ];
+
+    public function process(ProceedingJoinPoint $proceedingJoinPoint)
+    {
+        return tap($proceedingJoinPoint->process(), function ($pdo) {
+            Context::getOrSet(self::class, function () use ($pdo) {
+                $connectionStatus = $pdo->getAttribute(PDO::ATTR_CONNECTION_STATUS);
+                [$host] = explode(' ', $connectionStatus);
+
+                Context::set(Constants::TRACE_DB_SERVER_ADDRESS, $host);
+
+                return true;
+            });
+        });
+    }
+}

src/sentry/src/Tracing/Listener/EventHandleListener.php

@@ -193,6 +193,8 @@ protected function handleDbQueryExecuted(DbEvent\QueryExecuted $event): void
             'db.pool.max_idle_time' => $pool->getOption()->getMaxIdleTime(),
             'db.pool.idle' => $pool->getConnectionsInChannel(),
             'db.pool.using' => $pool->getCurrentConnections(),
+            'server.address' => (string) Context::get(Constants::TRACE_DB_SERVER_ADDRESS, 'localhost'),
+            'server.port' => (int) Context::get(Constants::TRACE_DB_SERVER_PORT, 3306),
         ];
 
         if ($this->feature->isTracingTagEnabled('db.sql.bindings', true)) {