Have datadog.trace.db_client_split_by_instance also affect mysqli and sqlsrv (#2508)

Signed-off-by: Bob Weinand <[email protected]>

Author: bwoebi

src/Integrations/Integrations/Mysqli/MysqliIntegration.php

@@ -424,6 +424,12 @@ public function setConnectionInfo(SpanData $span, $mysqli)
         foreach ($hostInfo as $tagName => $value) {
             $span->meta[$tagName] = $value;
         }
+        if (\DDTrace\Util\Runtime::getBoolIni("datadog.trace.db_client_split_by_instance")) {
+            if (isset($hostInfo[Tag::TARGET_HOST])) {
+                $span->service .=
+                    '-' . \DDTrace\Util\Normalizer::normalizeHostUdsAsService($hostInfo[Tag::TARGET_HOST]);
+            }
+        }
         $dbName = ObjectKVStore::get($mysqli, MysqliIntegration::KEY_DATABASE_NAME);
         if ($dbName) {
             $span->meta[Tag::DB_NAME] = $dbName;

src/Integrations/Integrations/SQLSRV/SQLSRVIntegration.php

@@ -218,6 +218,13 @@ public static function setDefaultAttributes($source, SpanData $span, $name, $que
         foreach ($storedConnectionInfo as $tag => $value) {
             $span->meta[$tag] = $value;
         }
+
+        $targetName = $storedConnectionInfo[Tag::DB_INSTANCE] ?? $storedConnectionInfo[Tag::TARGET_HOST] ?? "<default>";
+        if (\DDTrace\Util\Runtime::getBoolIni("datadog.trace.db_client_split_by_instance")) {
+            if ($targetName !== "<default>") {
+                $span->service .= '-' . \DDTrace\Util\Normalizer::normalizeHostUdsAsService($targetName);
+            }
+        }
     }
 
     public static function detectError($SQLSRVRetval, SpanData $span)

tests/Integrations/Mysqli/MysqliTest.php

@@ -34,6 +34,7 @@ protected function envsToCleanUpAtTearDown()
             'DD_TRACE_PEER_SERVICE_DEFAULTS_ENABLED',
             'DD_TRACE_REMOVE_INTEGRATION_SERVICE_NAMES_ENABLED',
             'DD_SERVICE',
+            'DD_SERVICE_MAPPING',
         ];
     }
 
@@ -606,6 +607,19 @@ public function testNoFakeServices()
         ], true, false);
     }
 
+    public function testServiceMappedSplitByDomain()
+    {
+        self::putEnv('DD_TRACE_DB_CLIENT_SPLIT_BY_INSTANCE=true');
+        self::putEnv('DD_SERVICE_MAPPING=mysqli:my-mysqli');
+        $traces = $this->isolateTracer(function () {
+            new \mysqli(self::$host, self::$user, self::$password, self::$db);
+        });
+
+        $this->assertSpans($traces, [
+            SpanAssertion::build('mysqli.__construct', 'my-mysqli-mysql_integration', 'sql', 'mysqli.__construct', SpanAssertion::NOT_TESTED)
+        ]);
+    }
+
     private function baseTags($expectDbName = true, $expectPeerService = false)
     {
         $tags = [

tests/Integrations/PDO/PDOTest.php

@@ -50,6 +50,7 @@ protected function envsToCleanUpAtTearDown()
             'DD_TRACE_DB_CLIENT_SPLIT_BY_INSTANCE',
             'DD_TRACE_PEER_SERVICE_DEFAULTS_ENABLED',
             'DD_TRACE_REMOVE_INTEGRATION_SERVICE_NAMES_ENABLED',
+            'DD_SERVICE_MAPPING',
             'DD_SERVICE',
         ];
     }
@@ -166,6 +167,20 @@ public function testPDOSplitByDomain()
         ]);
     }
 
+    public function testPDOServiceMappedSplitByDomain()
+    {
+        self::putEnv('DD_TRACE_DB_CLIENT_SPLIT_BY_INSTANCE=true');
+        self::putEnv('DD_SERVICE_MAPPING=pdo:my-pdo');
+        $traces = $this->isolateTracer(function () {
+            $this->pdoInstance();
+        });
+
+        $this->assertSpans($traces, [
+            SpanAssertion::build('PDO.__construct', 'my-pdo-mysql_integration', 'sql', 'PDO.__construct')
+                ->withExactTags($this->baseTags()),
+        ]);
+    }
+
     public function testPDOConstructError()
     {
         $traces = $this->isolateTracer(function () {