From 6b59adacf96d3135e8f9c1810fb420ffdf17f939 Mon Sep 17 00:00:00 2001
From: Fine0830 <fanxue0830@gmail.com>
Date: Tue, 25 Jun 2024 15:37:42 +0800
Subject: [PATCH] fix request header

---
 src/trace/interceptors/fetch.ts | 25 ++++++++++++++-----------
 1 file changed, 14 insertions(+), 11 deletions(-)

diff --git a/src/trace/interceptors/fetch.ts b/src/trace/interceptors/fetch.ts
index f62f975..b6746e0 100644
--- a/src/trace/interceptors/fetch.ts
+++ b/src/trace/interceptors/fetch.ts
@@ -49,8 +49,7 @@ export default function windowFetch(options: CustomOptionsType, segments: Segmen
         url = new URL(window.location.href);
         url.pathname = args[0];
       }
-    }  
-
+    }
 
     const noTraceOrigins = customConfig.noTraceOrigins.some((rule: string | RegExp) => {
       if (typeof rule === 'string') {
@@ -79,16 +78,20 @@ export default function windowFetch(options: CustomOptionsType, segments: Segmen
       const index = segment.spans.length;
       const values = `${1}-${traceIdStr}-${segmentId}-${index}-${service}-${instance}-${endpoint}-${peer}`;
 
-      if (!args[1]) {
-        args[1] = {};
-      }
-      if (!args[1].headers) {
-        args[1].headers = new Headers();
-      }
-      if (args[1].headers instanceof Headers) {
-        args[1].headers.append('sw8', values);
+      if (args[0] instanceof Request) {
+        args[0].headers.append('sw8', values);
       } else {
-        args[1].headers['sw8'] = values;
+        if (!args[1]) {
+          args[1] = {};
+        }
+        if (!args[1].headers) {
+          args[1].headers = new Headers();
+        }
+        if (args[1].headers instanceof Headers) {
+          args[1].headers.append('sw8', values);
+        } else {
+          args[1].headers['sw8'] = values;
+        }
       }
     }