From 3d4b90e67bb5d48bbedf382a8f95a61c63d31c47 Mon Sep 17 00:00:00 2001 From: Alexey Orlenko Date: Mon, 7 Oct 2024 17:04:26 +0200 Subject: [PATCH] fix(mongodb-query-connector): add missing span for aggregate Fixes: https://github.com/prisma/prisma/issues/24494 Refs: https://github.com/prisma/prisma/pull/25385 --- .../src/root_queries/aggregate.rs | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) diff --git a/query-engine/connectors/mongodb-query-connector/src/root_queries/aggregate.rs b/query-engine/connectors/mongodb-query-connector/src/root_queries/aggregate.rs index 05ff57053e95..1c0b79477dd4 100644 --- a/query-engine/connectors/mongodb-query-connector/src/root_queries/aggregate.rs +++ b/query-engine/connectors/mongodb-query-connector/src/root_queries/aggregate.rs @@ -3,6 +3,7 @@ use crate::{constants::*, output_meta, query_builder::MongoReadQueryBuilder, val use connector_interface::*; use mongodb::{bson::Document, ClientSession, Database}; use query_structure::{prelude::*, Filter, QueryArguments}; +use tracing::{info_span, Instrument}; pub async fn aggregate<'conn>( database: &Database, @@ -16,11 +17,17 @@ pub async fn aggregate<'conn>( let is_group_by = !group_by.is_empty(); let coll = database.collection(model.db_name()); + let span = info_span!( + "prisma:engine:db_query", + user_facing = true, + "db.statement" = &format_args!("db.{}.aggregate(*)", coll.name()) + ); + let query = MongoReadQueryBuilder::from_args(query_arguments)? .with_groupings(group_by, &selections, having)? .build()?; - let docs = query.execute(coll, session).await?; + let docs = query.execute(coll, session).instrument(span).await?; if is_group_by && docs.is_empty() { Ok(vec![])