Move all influx db related code to lib/influx.js

Author: pyropy

bin/worker.js

@@ -1,4 +1,4 @@
-import { reportMetric } from "../lib/metrics";
+import { reportMetric } from "../lib/influx.js";
 
 export const handleRequest = async (request, env) => {
   const url = new URL(request.url);

lib/influx.js

@@ -0,0 +1,38 @@
+import zlib from 'node:zlib';
+import { promisify } from 'node:util';
+
+/*
+* Sends compressed request metrics to InfluxDB
+* https://docs.influxdata.com/enterprise_influxdb/v1/guides/write_data/
+*/
+export const reportMetric = async (request, response, env) => {
+  const compress = promisify(zlib.gzip);
+
+  // Define API endpoint and headers
+  const url = `${env.INFLUX_URL}/api/v2/write?&bucket=${env.INFLUX_DATABASE}&precision=ms`;
+
+  return fetch(url, {
+    method: 'POST',
+    headers: {
+      'Authorization': `Token ${env.INFLUX_TOKEN}`,
+      'Content-Encoding': 'gzip',
+      'Content-Type': 'text/plain; charset=utf-8',
+    },
+    body: await compress(createMetricsFromRequest(request, response, env)),
+  })
+}
+
+/*
+* Returns request metrics in InfluxDB line protocol format
+* https://docs.influxdata.com/influxdb/cloud/reference/syntax/line-protocol/ 
+*/
+export const createMetricsFromRequest = (request, response = {}, env) => {
+  const url = new URL(request.url);
+  const timestamp = Date.now();
+  const apiKey = request.headers.get('api-key') || url.searchParams.get('api-key') || (request.headers.get('Authorization')?.startsWith('Bearer ') ? request.headers.get('Authorization').substring(7) : 'unknown');
+  const cfCacheStatus = response?.headers.get('CF-Cache-Status') ?? 'miss'
+  const formattedUrl = url.toString().replaceAll('=', '\\=');
+
+  // We're setting field value to 1 to count the number of requests
+  return `${env.INFLUX_METRIC_NAME},status_code=${response.status},url=${formattedUrl},hostname=${url.hostname},pathname="${url.pathname}",method=${request.method},cf_cache_status=${cfCacheStatus},api_key=${apiKey} value=1 ${timestamp}`
+}

lib/metrics.js

@@ -1,5 +1,5 @@
 import { describe, it, expect, vi } from 'vitest';
-import { reportMetric } from '../lib/metrics.js';
+import { reportMetric, createMetricsFromRequest } from '../lib/influx.js';
 
 describe('reportMetric', () => {
   it('should report metrics correctly', async () => {
@@ -40,4 +40,36 @@ describe('reportMetric', () => {
       })
     );
   });
+});
+
+describe('createMetricsFromRequest', () => {
+  it('should format request data correctly', () => {
+    const request = {
+      url: 'https://example.com/path?api-key=test-key',
+      method: 'GET',
+      headers: new Map([
+        ['api-key', 'test-key'],
+        ['Authorization', 'Bearer test-token']
+      ])
+    };
+    const response = {
+      status: 200,
+      headers: new Map([
+        ['CF-Cache-Status', 'HIT']
+      ])
+    };
+    const env = {
+      INFLUX_METRIC: 'test_metric'
+    };
+
+    const result = createMetricsFromRequest(request, response, env);
+    expect(result).toContain('test_metric');
+    expect(result).toContain('status_code=200');
+    expect(result).toContain('url=https://example.com/path?api-key\\=test-key');
+    expect(result).toContain('hostname=example.com');
+    expect(result).toContain('pathname="/path"');
+    expect(result).toContain('method=GET');
+    expect(result).toContain('cf_cache_status=HIT');
+    expect(result).toContain('api_key=test-key');
+  });
 });

lib/request.js

@@ -1,6 +1,6 @@
 import { describe, it, expect, vi } from 'vitest';
 import worker, { handleRequest } from '../bin/worker.js';
-import { reportMetric } from '../lib/metrics.js';
+import { reportMetric } from '../lib/influx.js';
 
 describe('handleRequest', () => {
   it('should handle request correctly', async () => {

test/metrics.test.js renamed to test/influx.test.js

@@ -5,7 +5,7 @@ compatibility_flags = ["nodejs_compat"]
 
 [vars]
 REQUEST_URL = "https://example.com"
-INFLUX_METRIC = "example"
+INFLUX_METRIC_NAME = "example"
 INFLUX_URL = "http://localhost:8086"
 INFLUX_TOKEN = "example"
 INFLUX_DATABASE = "example"