Merge pull request #10 from axiomhq/arne/0.3.0

Release 0.3.0

src/worker.js

@@ -1,8 +1,11 @@
 const axiomDataset = 'my-dataset' // Your Axiom dataset
 const axiomToken = 'xapt-xxx' // Your Axiom API token
 
+const requestHeadersToCapture = ['user-agent'];
+const responseHeadersToCapture = ['cf-cache-status', 'cf-ray'];
+
 // 8< ----------- snip ------------
-const Version = '0.2.0'
+const Version = '0.3.0'
 const axiomEndpoint = 'https://api.axiom.co'
 let workerTimestamp
 let batch = []
@@ -31,8 +34,8 @@ const throttle = (fn, wait, maxLen) => {
         setTimeout(() => {
           timeoutInProgress = false
           fn.apply(context, args).then(resolve).catch(resolve);
-        })
-      }, wait)
+        }, wait)
+      })
     }
   }
 }
@@ -48,8 +51,7 @@ async function sendLogs () {
   return fetch(url, {
     signal: AbortSignal.timeout(30_000),
     method: 'POST',
-    body: logs.map(JSON.stringify).join('\n'),
-    keepalive: true,
+    body: logs.map(log => JSON.stringify(log)).join('\n'),
     headers: {
       'Content-Type': 'application/x-ndjson',
       Authorization: `Bearer ${axiomToken}`,
@@ -61,6 +63,20 @@ async function sendLogs () {
 // This will send logs every 10 seconds or every 1000 logs
 const throttledSendLogs = throttle(sendLogs, 10_000, 1000)
 
+function getHeaderMap(headers, allowlist) {
+  if (!allowlist.length) {
+    return {};
+  }
+
+  return [...headers].reduce((acc, [headerKey, headerValue]) => {
+    if (allowlist.includes(headerKey)) {
+      acc[headerKey] = headerValue;
+    }
+
+    return acc;
+  }, {});
+}
+
 async function handleRequest (request, context) {
   const start = Date.now()
 
@@ -81,13 +97,13 @@ async function handleRequest (request, context) {
     _time: Date.now(),
     request: {
       url: request.url,
-      headers: request.headers,
+      headers: getHeaderMap(request.headers, requestHeadersToCapture),
       method: request.method,
       ...cf
     },
     response: {
       duration,
-      headers: response.headers,
+      headers: getHeaderMap(response.headers, responseHeadersToCapture),
       status: response.status
     },
     worker: {