From 642c239082561d6cc0dc7f61a2f44ba0e6b796c5 Mon Sep 17 00:00:00 2001
From: ankur12-1610 <anknerd.12@gmail.com>
Date: Mon, 29 Sep 2025 00:06:53 +0530
Subject: [PATCH] Add TTFB metric logic and report it to cloudflare analytics

---
 index.js | 63 ++++++++++++++++++++++++++++++++++++++++++++++++++++++--
 1 file changed, 61 insertions(+), 2 deletions(-)

diff --git a/index.js b/index.js
index c9099c3..d02cd51 100644
--- a/index.js
+++ b/index.js
@@ -11,6 +11,37 @@ const NO_ALERT_ON_RESPONSE_STATUS_CODES = [
   521,
 ]
 
+/**
+ * Simple function to report TTFB metrics to Cloudflare Analytics
+ * @param {object} metrics
+ * @param {string} metrics.url
+ * @param {string} metrics.botLocation
+ * @param {string} metrics.clientAddress
+ * @param {string} metrics.pieceCid
+ * @param {number} metrics.ttfb
+ * @param {number} metrics.statusCode
+ * @param {number} metrics.bytes
+ */
+async function reportToCloudflare(metrics) {
+  if (!process.env.CLOUDFLARE_ANALYTICS_URL) {
+    throw new Error('CLOUDFLARE_ANALYTICS_URL environment variable is not set')
+  }
+  const response = await fetch(process.env.CLOUDFLARE_ANALYTICS_URL, {
+    method: 'POST',
+    headers: { 'Content-Type': 'application/json' },
+    body: JSON.stringify({
+      blobs: [metrics.url, metrics.botLocation, metrics.clientAddress, metrics.pieceCid],
+      doubles: [metrics.ttfb, metrics.statusCode, metrics.bytes]
+    })
+  })
+  
+  if (response.ok) {
+    console.log('☁️ Sent to Cloudflare Analytics')
+  } else {
+    throw new Error(`HTTP ${response.status}`)
+  }
+}
+
 export const pdpVerifierAbi = [
   // Returns the next data set ID
   'function getNextDataSetId() public view returns (uint64)',
@@ -152,8 +183,13 @@ async function testRetrieval({
 }) {
   const url = `https://${clientAddress}.${CDN_HOSTNAME}/${pieceCid}`
   console.log('Fetching', url)
+  
+  // Measure TTFB (Time To First Byte)
+  const startTime = performance.now()
   const res = await fetch(url)
-  console.log('-> Status code:', res.status)
+  const ttfb = performance.now() - startTime
+  
+  console.log('-> Status code:', res.status, `TTFB: ${ttfb.toFixed(2)}ms`)
   if (!res.ok) {
     const reason = (await res.text()).trim()
     console.log(reason)
@@ -204,10 +240,33 @@ async function testRetrieval({
       reason,
     )
   } else if (res.body) {
+    const downloadStartTime = performance.now()
+    let totalBytes = 0
+    
     const reader = res.body.getReader()
     while (true) {
-      const { done } = await reader.read()
+      const { done, value } = await reader.read()
       if (done) break
+      if (value) totalBytes += value.length
+    }
+    
+    const downloadTime = performance.now() - downloadStartTime
+    const totalTime = performance.now() - startTime
+    
+    console.log(`✅ Successfully retrieved ${totalBytes} bytes in ${totalTime.toFixed(2)}ms total (download: ${downloadTime.toFixed(2)}ms)`)
+    console.log(`📊 TTFB: ${ttfb.toFixed(2)}ms, Status: ${res.status}, Bytes: ${totalBytes}, URL: ${url}`)
+    
+    // Optionally report to Cloudflare Analytics
+    if (process.env.CLOUDFLARE_ANALYTICS_URL) {
+      reportToCloudflare({
+        url,
+        ttfb: Math.round(ttfb),
+        statusCode: res.status,
+        bytes: totalBytes,
+        botLocation: botLocation || '<dev>',
+        clientAddress,
+        pieceCid
+      }).catch(err => console.warn('Analytics reporting failed:', err.message))
     }
   }
 }