@wtfalch/reporting 0.5.1 → 0.8.1

package/dist/requests.js

@@ -0,0 +1,72 @@
+import { sql } from 'drizzle-orm';
+import { deployTags } from './deploy.js';
+import { reportingRequests } from './tables.js';
+/**
+ * Counts responses in memory and writes one upsert per minute bucket, so a
+ * request costs a map increment and the database sees a handful of rows a
+ * minute. `errors` is status >= 500. A failed write puts its counts back for
+ * the next one; the process dying loses at most `flushEveryMs` of counts.
+ */
+export class RequestCounter {
+    o;
+    buckets = new Map();
+    timer = null;
+    constructor(o) {
+        this.o = o;
+    }
+    record(status) {
+        const minute = Math.floor(this.o.now().getTime() / 60_000) * 60_000;
+        const b = this.buckets.get(minute) ?? { requests: 0, errors: 0 };
+        b.requests += 1;
+        if (status >= 500)
+            b.errors += 1;
+        this.buckets.set(minute, b);
+        if (!this.timer) {
+            this.timer = setTimeout(() => {
+                this.timer = null;
+                void this.flush();
+            }, this.o.flushEveryMs);
+            this.timer.unref?.();
+        }
+    }
+    async flush() {
+        if (this.buckets.size === 0)
+            return;
+        const taken = [...this.buckets];
+        this.buckets.clear();
+        const { color, release } = deployTags();
+        try {
+            await this.o.db
+                .insert(reportingRequests)
+                .values(taken.map(([minute, c]) => ({
+                bucket: new Date(minute),
+                site: this.o.site,
+                color: color ?? '',
+                release: release ?? '',
+                requests: c.requests,
+                errors: c.errors,
+            })))
+                .onConflictDoUpdate({
+                target: [
+                    reportingRequests.bucket,
+                    reportingRequests.site,
+                    reportingRequests.color,
+                    reportingRequests.release,
+                ],
+                set: {
+                    requests: sql `${reportingRequests.requests} + excluded.requests`,
+                    errors: sql `${reportingRequests.errors} + excluded.errors`,
+                },
+            });
+        }
+        catch (error) {
+            for (const [minute, c] of taken) {
+                const b = this.buckets.get(minute) ?? { requests: 0, errors: 0 };
+                b.requests += c.requests;
+                b.errors += c.errors;
+                this.buckets.set(minute, b);
+            }
+            this.o.log.error({ err: error instanceof Error ? error.name : 'unknown' }, 'reporting: request counter flush failed');
+        }
+    }
+}