Wrap 2nd gen onCall functions with trace context (#1491)

* wrap oncall functions with trace context
firebase · Nov 28, 2023 · a64fd48 · a64fd48
1 parent 414ff0a
commit a64fd48
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -0,0 +1 @@
+- Wrap 2nd gen onCall functions with trace context. (#1491)
diff --git a/src/v2/providers/https.ts b/src/v2/providers/https.ts
@@ -366,7 +366,7 @@ export function onCall<T = any, Return = any | Promise<any>>(
   // onCallHandler sniffs the function length to determine which API to present.
   // fix the length to prevent api versions from being mismatched.
   const fixedLen = (req: CallableRequest<T>) => handler(req);
-  const func: any = onCallHandler(
+  let func: any = onCallHandler(
     {
       cors: { origin, methods: "POST" },
       enforceAppCheck: opts.enforceAppCheck ?? options.getGlobalOptions().enforceAppCheck,
@@ -375,6 +375,8 @@ export function onCall<T = any, Return = any | Promise<any>>(
     fixedLen
   );
 
+  func = wrapTraceContext(func);
+
   Object.defineProperty(func, "__trigger", {
     get: () => {
       const baseOpts = options.optionsToTriggerAnnotations(options.getGlobalOptions());