fix: createSpanMetadata function

MapColonies · Aug 22, 2024 · 5a8c27c · 5a8c27c
1 parent 1f73b03
commit 5a8c27c
Showing 1 changed file with 5 additions and 13 deletions.
diff --git a/src/common/utils.ts b/src/common/utils.ts
@@ -87,27 +87,19 @@ export const generateGeoIdentifier = (geo: FeatureCollection): string => {
 export function createSpanMetadata(
   functionName?: string,
   spanKind?: SpanKind,
-  parentContext?: { traceId: string; spanId: string }
+  context?: { traceId: string; spanId: string }
 ): { traceContext: SpanContext | undefined; spanOptions: SpanOptions | undefined } {
-  if (!parentContext) {
-    return { spanOptions: undefined, traceContext: undefined };
-  }
-  const traceContext: SpanContext = {
-    ...parentContext,
-    traceFlags: FLAG_SAMPLED,
-  };
+  const traceContext = context ? { ...context, traceFlags: FLAG_SAMPLED } : undefined;
+
   const spanOptions: SpanOptions = {
     kind: spanKind,
-    links: [
-      {
-        context: traceContext,
-      },
-    ],
+    ...(traceContext && { links: [{ context: traceContext }] }),
     attributes: {
       // eslint-disable-next-line @typescript-eslint/naming-convention
       'code.function': functionName,
     },
   };
+
   return { traceContext, spanOptions };
 }