feat(core): add mongodb-specific "db.collection.name" and "db.operati…

…on.name" attributes in spans
prisma · Oct 18, 2024 · 5ef57fc · 5ef57fc
1 parent 001a865
commit 5ef57fc
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/query-engine/connectors/mongodb-query-connector/src/root_queries/mod.rs b/query-engine/connectors/mongodb-query-connector/src/root_queries/mod.rs
@@ -73,6 +73,8 @@ where
         user_facing = true,
         "db.system" = SYSTEM_NAME,
         "db.statement" = %Arc::clone(&query_string),
+        "db.collection.name" = %builder.collection().unwrap_or_else(|| "raw"),
+        "db.operation.name" = builder.query_type(),
         "otel.kind" = "client"
     );
 

diff --git a/query-engine/core/src/telemetry/models.rs b/query-engine/core/src/telemetry/models.rs
@@ -7,7 +7,14 @@ use std::{
     time::{Duration, SystemTime},
 };
 
-const ACCEPT_ATTRIBUTES: &[&str] = &["db.system", "db.statement", "itx_id", "otel.kind"];
+const ACCEPT_ATTRIBUTES: &[&str] = &[
+    "db.system",
+    "db.statement",
+    "db.collection.name",
+    "db.operation.name",
+    "itx_id",
+    "otel.kind",
+];
 
 #[derive(Serialize, Debug, Clone, PartialEq, Eq)]
 pub enum OtelKind {