From d52f84e88215b6dfc7d9f721992368a9aa632e63 Mon Sep 17 00:00:00 2001
From: alphaeusmote <41258468-alphaeusmote@users.noreply.replit.com>
Date: Thu, 10 Apr 2025 02:22:44 +0000
Subject: [PATCH] Add DNS metrics tracking to the database and API

Replit-Commit-Author: Agent
Replit-Commit-Session-Id: 9111ef36-26c8-4085-84ca-a35dc1fec1b5
Replit-Commit-Screenshot-Url: https://storage.googleapis.com/screenshot-production-us-central1/7083d608-d6d3-4a6a-9a27-6286c5109627/33606e50-05ad-4026-abd3-a13c56e34ca4.jpg
---
 server/database-storage.ts | 80 +++++++++++++++++++++++++++++++++++++-
 server/routes.ts           |  1 +
 2 files changed, 79 insertions(+), 2 deletions(-)

diff --git a/server/database-storage.ts b/server/database-storage.ts
index ba0926f..16c4087 100644
--- a/server/database-storage.ts
+++ b/server/database-storage.ts
@@ -1,6 +1,6 @@
 import { 
   users, organizations, domains, dnsRecords, 
-  providers, dnsHistory, apiTokens, webhooks, webhookDeliveryLogs,
+  providers, dnsHistory, apiTokens, webhooks, webhookDeliveryLogs, dnsMetrics,
   type User, type InsertUser, 
   type Organization, type InsertOrganization,
   type Domain, type InsertDomain,
@@ -8,7 +8,8 @@ import {
   type Provider, type InsertProvider,
   type ApiToken, type InsertApiToken,
   type DnsHistory, type Webhook, type InsertWebhook,
-  type WebhookDeliveryLog, type InsertWebhookDeliveryLog
+  type WebhookDeliveryLog, type InsertWebhookDeliveryLog,
+  type DnsMetric, type InsertDnsMetric
 } from "@shared/schema";
 import { db } from "./db";
 import { eq, and, desc, or, inArray, sql } from "drizzle-orm";
@@ -374,4 +375,79 @@ export class DatabaseStorage implements IStorage {
       .where(eq(webhookDeliveryLogs.webhookId, webhookId))
       .orderBy(desc(webhookDeliveryLogs.createdAt));
   }
+
+  // DNS Metrics
+  async addDnsMetric(metric: InsertDnsMetric): Promise<DnsMetric> {
+    const [newMetric] = await db.insert(dnsMetrics).values(metric).returning();
+    return newMetric;
+  }
+  
+  async getDnsMetric(id: string): Promise<DnsMetric | undefined> {
+    const [metric] = await db.select().from(dnsMetrics).where(eq(dnsMetrics.id, id));
+    return metric;
+  }
+  
+  async getDnsMetricsByDomain(
+    domainId: string, 
+    metricType?: string, 
+    startDate?: Date, 
+    endDate?: Date
+  ): Promise<DnsMetric[]> {
+    let query = db.select().from(dnsMetrics).where(eq(dnsMetrics.domainId, domainId));
+    
+    if (metricType) {
+      query = query.where(eq(dnsMetrics.metricType, metricType));
+    }
+    
+    if (startDate) {
+      query = query.where(sql`${dnsMetrics.timestamp} >= ${startDate}`);
+    }
+    
+    if (endDate) {
+      query = query.where(sql`${dnsMetrics.timestamp} <= ${endDate}`);
+    }
+    
+    return await query.orderBy(desc(dnsMetrics.timestamp));
+  }
+  
+  async getDnsMetricsByRecord(
+    recordId: string, 
+    metricType?: string, 
+    startDate?: Date, 
+    endDate?: Date
+  ): Promise<DnsMetric[]> {
+    let query = db.select().from(dnsMetrics).where(eq(dnsMetrics.recordId, recordId));
+    
+    if (metricType) {
+      query = query.where(eq(dnsMetrics.metricType, metricType));
+    }
+    
+    if (startDate) {
+      query = query.where(sql`${dnsMetrics.timestamp} >= ${startDate}`);
+    }
+    
+    if (endDate) {
+      query = query.where(sql`${dnsMetrics.timestamp} <= ${endDate}`);
+    }
+    
+    return await query.orderBy(desc(dnsMetrics.timestamp));
+  }
+  
+  async getDnsMetricsByType(
+    metricType: string, 
+    startDate?: Date, 
+    endDate?: Date
+  ): Promise<DnsMetric[]> {
+    let query = db.select().from(dnsMetrics).where(eq(dnsMetrics.metricType, metricType));
+    
+    if (startDate) {
+      query = query.where(sql`${dnsMetrics.timestamp} >= ${startDate}`);
+    }
+    
+    if (endDate) {
+      query = query.where(sql`${dnsMetrics.timestamp} <= ${endDate}`);
+    }
+    
+    return await query.orderBy(desc(dnsMetrics.timestamp));
+  }
 }
\ No newline at end of file
diff --git a/server/routes.ts b/server/routes.ts
index 58b37e9..a121e65 100644
--- a/server/routes.ts
+++ b/server/routes.ts
@@ -10,6 +10,7 @@ import {
   insertApiTokenSchema,
   insertOrganizationSchema,
   insertWebhookSchema,
+  insertDnsMetricSchema,
   recordTypes,
   providerTypes
 } from "@shared/schema";