Wrap v2 firestore functions with trace context (#1440)

* Wrap v2 firestore functions with trace context * Add changelog
firebase · Jul 13, 2023 · 41d616f · 41d616f
1 parent ba368e8
commit 41d616f
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -1 +1,2 @@
 - Add features to task queue functions. (#1423)
+- Add traces to V2 Firestore trigger logs. (#1440)
diff --git a/src/v2/providers/firestore.ts b/src/v2/providers/firestore.ts
@@ -34,6 +34,7 @@ import {
   createSnapshotFromJson,
   createSnapshotFromProtobuf,
 } from "../../common/providers/firestore";
+import { wrapTraceContext } from "../trace";
 
 export { Change };
 
@@ -445,7 +446,7 @@ export function onOperation<Document extends string>(
     const event = raw as RawFirestoreEvent;
     const params = makeParams(event.document, documentPattern) as unknown as ParamsOf<Document>;
     const firestoreEvent = makeFirestoreEvent(eventType, event, params);
-    return handler(firestoreEvent);
+    return wrapTraceContext(handler)(firestoreEvent);
   };
 
   func.run = handler;