OutboundPlugin#tagPeerService()

DataDog · Jul 3, 2023 · 67debdb · 67debdb
1 parent db4ec88
commit 67debdb
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/packages/datadog-plugin-http2/src/client.js b/packages/datadog-plugin-http2/src/client.js
@@ -110,6 +110,7 @@ class Http2ClientPlugin extends ClientPlugin {
   }
 
   _onClose ({ span }) {
+    this.tagPeerService(span)
     span.finish()
   }
 }

diff --git a/packages/dd-trace/src/plugins/outbound.js b/packages/dd-trace/src/plugins/outbound.js
@@ -61,13 +61,17 @@ class OutboundPlugin extends TracingPlugin {
 
   finish () {
     const span = this.activeSpan
+    this.tagPeerService(span)
+    super.finish(...arguments)
+  }
+
+  tagPeerService (span) {
     if (this.tracer._computePeerService) {
       const peerData = this.getPeerService(span.context()._tags)
       if (peerData) {
         span.addTags(peerData)
       }
     }
-    super.finish(...arguments)
   }
 
   connect (url) {