devpush 0.0.0-stage → 1.0.0-beta.3

package/src/server/servers/metric-history.ts

@@ -0,0 +1,55 @@
+import type { AgentMetricSample } from './metric-history-schema'
+import { createHash } from 'node:crypto'
+import { sql } from 'drizzle-orm'
+
+import { config } from '#devpush/server/config'
+import { db } from '#devpush/server/db'
+import type { AtomicStore } from '#devpush/server/db/atomic'
+
+// Samples have stable random identities, so retries and clock changes cannot
+// collide. Namespace them by the authenticated server using the existing PK.
+export function metricRowId(serverId: string, sampleId: string) {
+  return createHash('sha256').update(`${serverId}\0${sampleId}`).digest('hex')
+}
+
+export async function recordAgentMetricSamples(
+  serverId: string,
+  samples: Array<AgentMetricSample>,
+  now = new Date(),
+  store: Pick<AtomicStore, 'run'> = db,
+) {
+  if (samples.length === 0) return undefined
+  const cutoff =
+    now.getTime() - config.DEVPUSH_MONITORING_RETENTION_DAYS * 86_400_000
+  const rows = samples
+    .filter((sample) => Date.parse(sample.observedAt) >= cutoff)
+    .map((sample) => ({
+      id: metricRowId(serverId, sample.id),
+      observedAt: Date.parse(sample.observedAt),
+      ...sample.metrics,
+    }))
+  if (rows.length > 0) {
+    // One JSON parameter avoids D1's 100-parameter limit. A single insert is
+    // atomic; the response acknowledges only after persistence completes.
+    await store.run(sql`
+      insert into server_metrics (
+        id, server_id, observed_at, cpu_percent, memory_used_bytes,
+        memory_total_bytes, disk_used_bytes, disk_total_bytes,
+        network_receive_bytes_per_second, network_transmit_bytes_per_second,
+        disk_read_bytes_per_second, disk_write_bytes_per_second
+      )
+      select
+        json_extract(value, '$.id'), ${serverId},
+        json_extract(value, '$.observedAt'), json_extract(value, '$.cpuPercent'),
+        json_extract(value, '$.memoryUsedBytes'), json_extract(value, '$.memoryTotalBytes'),
+        json_extract(value, '$.diskUsedBytes'), json_extract(value, '$.diskTotalBytes'),
+        json_extract(value, '$.networkReceiveBytesPerSecond'),
+        json_extract(value, '$.networkTransmitBytesPerSecond'),
+        json_extract(value, '$.diskReadBytesPerSecond'), json_extract(value, '$.diskWriteBytesPerSecond')
+      from json_each(${JSON.stringify(rows)}) where true
+      on conflict (id) do nothing
+    `)
+  }
+  // Older-than-retention records are deliberately expired, not retried forever.
+  return samples.at(-1)?.id
+}

package/src/server/servers/monitoring.ts

@@ -0,0 +1,592 @@
+import { randomUUID } from 'node:crypto'
+
+import {
+  and,
+  desc,
+  eq,
+  gt,
+  gte,
+  inArray,
+  isNull,
+  lt,
+  or,
+  sql,
+} from 'drizzle-orm'
+
+import { db } from '#devpush/server/db'
+import { config } from '#devpush/server/config'
+import {
+  serverAvailabilityWindows,
+  serverHeartbeats,
+  serverMetrics,
+  servers as serverTable,
+} from '#devpush/server/db/schema'
+import { serverDisconnectedAfterMs } from '#devpush/server/servers/status'
+import type { ServerStatus } from '#devpush/server/servers/status'
+
+const availabilityWindowMaxMs = 24 * 60 * 60 * 1000
+const metricsWindowMs = 24 * 60 * 60 * 1000
+const defaultMetricsLimit = 3_000
+const metricGapAfterMs = 75_000
+
+export async function pruneServerMonitoring(now = new Date()) {
+  const cutoff = new Date(
+    now.getTime() -
+      config.DEVPUSH_MONITORING_RETENTION_DAYS * 24 * 60 * 60 * 1000,
+  )
+  const counts = { heartbeats: 0, metrics: 0 }
+  for (const [name, table] of [
+    ['heartbeats', serverHeartbeats],
+    ['metrics', serverMetrics],
+  ] as const) {
+    // Bound each write and the entire maintenance run, including after an outage.
+    for (let batch = 0; batch < 10; batch++) {
+      const rows = await db
+        .delete(table)
+        .where(
+          inArray(
+            table.id,
+            db
+              .select({ id: table.id })
+              .from(table)
+              .where(lt(table.observedAt, cutoff))
+              .orderBy(table.observedAt, table.id)
+              .limit(1000),
+          ),
+        )
+        .returning({ id: table.id })
+      counts[name] += rows.length
+      if (rows.length < 1000) break
+    }
+  }
+  return counts
+}
+
+export type ServerMonitoringSummary = {
+  availabilityPercent: number | null
+  availabilityWindowMs: number
+  latestMetrics: ServerMetricsSample | null
+}
+
+export type ServerMetricsSample = {
+  observedAt: string
+  cpuPercent: number | null
+  load1: number | null
+  memoryUsedBytes: number | null
+  memoryTotalBytes: number | null
+  diskUsedBytes: number | null
+  diskTotalBytes: number | null
+  networkReceiveBytesPerSecond: number | null
+  networkTransmitBytesPerSecond: number | null
+  diskReadBytesPerSecond: number | null
+  diskWriteBytesPerSecond: number | null
+}
+
+type ServerWindow = {
+  id: string
+  createdAt: Date
+  status?: ServerStatus
+}
+
+type HeartbeatRow = {
+  serverId: string
+  observedAt: Date
+}
+
+type AvailabilityWindow = {
+  startedAt: Date
+  endedAt: Date | null
+}
+
+export async function recordServerHeartbeat({
+  serverId,
+  observedAt,
+  agentVersion,
+  publicAddress,
+}: {
+  serverId: string
+  observedAt: Date
+  agentVersion: string | null
+  publicAddress: string | null
+}) {
+  await db.insert(serverHeartbeats).values({
+    id: randomUUID(),
+    serverId,
+    observedAt,
+    agentVersion,
+    publicAddress,
+  })
+}
+
+export async function listServerMonitoringSummaries(
+  servers: Array<ServerWindow>,
+  now = new Date(),
+): Promise<Map<string, ServerMonitoringSummary>> {
+  const serverIds = servers.map((server) => server.id)
+  const summaries = new Map<string, ServerMonitoringSummary>()
+
+  for (const server of servers) {
+    summaries.set(server.id, {
+      availabilityPercent: null,
+      availabilityWindowMs: 0,
+      latestMetrics: null,
+    })
+  }
+
+  if (serverIds.length === 0) {
+    return summaries
+  }
+
+  const [heartbeatRows, availabilityRows, latestMetricRows] = await Promise.all(
+    [
+      db
+        .select({
+          serverId: serverHeartbeats.serverId,
+          observedAt: serverHeartbeats.observedAt,
+        })
+        .from(serverHeartbeats)
+        .where(
+          and(
+            sql`${serverHeartbeats.serverId} in (select value from json_each(${JSON.stringify(serverIds)}))`,
+            gte(
+              serverHeartbeats.observedAt,
+              new Date(
+                now.getTime() -
+                  availabilityWindowMaxMs -
+                  serverDisconnectedAfterMs,
+              ),
+            ),
+          ),
+        ),
+      db
+        .select({
+          serverId: serverAvailabilityWindows.serverId,
+          startedAt: serverAvailabilityWindows.startedAt,
+          endedAt: serverAvailabilityWindows.endedAt,
+        })
+        .from(serverAvailabilityWindows)
+        .where(
+          and(
+            sql`${serverAvailabilityWindows.serverId} in (select value from json_each(${JSON.stringify(serverIds)}))`,
+            lt(serverAvailabilityWindows.startedAt, now),
+            or(
+              isNull(serverAvailabilityWindows.endedAt),
+              gt(
+                serverAvailabilityWindows.endedAt,
+                new Date(now.getTime() - availabilityWindowMaxMs),
+              ),
+            ),
+          ),
+        ),
+      db
+        .select({
+          serverId: serverMetrics.serverId,
+          observedAt: serverMetrics.observedAt,
+          cpuPercent: serverMetrics.cpuPercent,
+          load1: serverMetrics.load1,
+          memoryUsedBytes: serverMetrics.memoryUsedBytes,
+          memoryTotalBytes: serverMetrics.memoryTotalBytes,
+          diskUsedBytes: serverMetrics.diskUsedBytes,
+          diskTotalBytes: serverMetrics.diskTotalBytes,
+          networkReceiveBytesPerSecond:
+            serverMetrics.networkReceiveBytesPerSecond,
+          networkTransmitBytesPerSecond:
+            serverMetrics.networkTransmitBytesPerSecond,
+          diskReadBytesPerSecond: serverMetrics.diskReadBytesPerSecond,
+          diskWriteBytesPerSecond: serverMetrics.diskWriteBytesPerSecond,
+        })
+        .from(serverTable)
+        .innerJoin(
+          serverMetrics,
+          sql`${serverMetrics.id} = (
+          select latest.id from server_metrics as latest
+          where latest.server_id = ${serverTable.id}
+          order by latest.observed_at desc, latest.id desc limit 1
+        )`,
+        )
+        .where(
+          sql`${serverTable.id} in (select value from json_each(${JSON.stringify(serverIds)}))`,
+        )
+        .orderBy(serverMetrics.serverId, desc(serverMetrics.observedAt)),
+    ],
+  )
+
+  const heartbeatsByServer = new Map<string, Array<HeartbeatRow>>()
+  for (const row of heartbeatRows) {
+    const rows = heartbeatsByServer.get(row.serverId) ?? []
+    rows.push(row)
+    heartbeatsByServer.set(row.serverId, rows)
+  }
+
+  const availabilityByServer = new Map<string, Array<AvailabilityWindow>>()
+  for (const row of availabilityRows) {
+    const rows = availabilityByServer.get(row.serverId) ?? []
+    rows.push({ startedAt: row.startedAt, endedAt: row.endedAt })
+    availabilityByServer.set(row.serverId, rows)
+  }
+  const latestMetrics = new Map(
+    latestMetricRows.map((row) => [row.serverId, formatMetricSample(row)]),
+  )
+
+  for (const server of servers) {
+    const heartbeats = (heartbeatsByServer.get(server.id) ?? []).sort(
+      (a, b) => a.observedAt.getTime() - b.observedAt.getTime(),
+    )
+    const persistedWindows = availabilityByServer.get(server.id) ?? []
+    const availabilityWindows =
+      persistedWindows.length > 0
+        ? persistedWindows
+        : server.status === 'disabled'
+          ? []
+          : [{ startedAt: server.createdAt, endedAt: null }]
+
+    const availability = calculateHeartbeatAvailability({
+      availabilityWindows,
+      heartbeats,
+      now,
+    })
+
+    summaries.set(server.id, {
+      availabilityPercent: availability.percent,
+      availabilityWindowMs: availability.windowMs,
+      latestMetrics: latestMetrics.get(server.id) ?? null,
+    })
+  }
+
+  return summaries
+}
+
+export function calculateHeartbeatAvailability({
+  availabilityWindows,
+  heartbeats,
+  now,
+}: {
+  availabilityWindows: Array<AvailabilityWindow>
+  heartbeats: Array<Pick<HeartbeatRow, 'observedAt'>>
+  now: Date
+}) {
+  const windowStart = now.getTime() - availabilityWindowMaxMs
+  const eligibleIntervals = mergeIntervals(
+    availabilityWindows
+      .map((window) => ({
+        start: Math.max(window.startedAt.getTime(), windowStart),
+        end: Math.min(
+          window.endedAt?.getTime() ?? now.getTime(),
+          now.getTime(),
+        ),
+      }))
+      .filter((interval) => interval.end > interval.start),
+  )
+  const eligibleMs = sumIntervals(eligibleIntervals)
+
+  if (eligibleMs <= 0) {
+    return { percent: null, windowMs: 0 }
+  }
+
+  const heartbeatIntervals = mergeIntervals(
+    heartbeats
+      .map((heartbeat) => ({
+        start: Math.max(heartbeat.observedAt.getTime(), windowStart),
+        end: Math.min(
+          heartbeat.observedAt.getTime() + serverDisconnectedAfterMs,
+          now.getTime(),
+        ),
+      }))
+      .filter((interval) => interval.end > interval.start),
+  )
+  const coveredMs = sumIntervals(
+    intersectIntervals(heartbeatIntervals, eligibleIntervals),
+  )
+
+  return {
+    percent: Math.min(100, Math.max(0, (coveredMs / eligibleMs) * 100)),
+    windowMs: eligibleMs,
+  }
+}
+
+export async function listServerMetricSeries({
+  serverId,
+  now = new Date(),
+  startAt,
+  limit = defaultMetricsLimit,
+}: {
+  serverId: string
+  now?: Date
+  startAt?: Date
+  limit?: number
+}): Promise<Array<ServerMetricsSample>> {
+  const seriesStart = new Date(
+    Math.max(
+      startAt?.getTime() ?? Number.NEGATIVE_INFINITY,
+      now.getTime() - metricsWindowMs,
+    ),
+  )
+  const [rows, availabilityWindows] = await Promise.all([
+    db
+      .select({
+        observedAt: serverMetrics.observedAt,
+        cpuPercent: serverMetrics.cpuPercent,
+        load1: serverMetrics.load1,
+        memoryUsedBytes: serverMetrics.memoryUsedBytes,
+        memoryTotalBytes: serverMetrics.memoryTotalBytes,
+        diskUsedBytes: serverMetrics.diskUsedBytes,
+        diskTotalBytes: serverMetrics.diskTotalBytes,
+        networkReceiveBytesPerSecond:
+          serverMetrics.networkReceiveBytesPerSecond,
+        networkTransmitBytesPerSecond:
+          serverMetrics.networkTransmitBytesPerSecond,
+        diskReadBytesPerSecond: serverMetrics.diskReadBytesPerSecond,
+        diskWriteBytesPerSecond: serverMetrics.diskWriteBytesPerSecond,
+      })
+      .from(serverMetrics)
+      .where(
+        and(
+          eq(serverMetrics.serverId, serverId),
+          gte(serverMetrics.observedAt, seriesStart),
+        ),
+      )
+      .orderBy(desc(serverMetrics.observedAt))
+      .limit(limit),
+    db
+      .select({
+        startedAt: serverAvailabilityWindows.startedAt,
+        endedAt: serverAvailabilityWindows.endedAt,
+      })
+      .from(serverAvailabilityWindows)
+      .where(
+        and(
+          eq(serverAvailabilityWindows.serverId, serverId),
+          lt(serverAvailabilityWindows.startedAt, now),
+          or(
+            isNull(serverAvailabilityWindows.endedAt),
+            gt(serverAvailabilityWindows.endedAt, seriesStart),
+          ),
+        ),
+      ),
+  ])
+
+  return insertMetricGaps(
+    rows.reverse().map(formatMetricSample),
+    metricGapAfterMs,
+    availabilityWindows,
+  )
+}
+
+export function insertMetricGaps(
+  samples: Array<ServerMetricsSample>,
+  gapAfterMs = metricGapAfterMs,
+  availabilityWindows: Array<AvailabilityWindow> = [],
+) {
+  const points: Array<ServerMetricsSample> = []
+  const windows = [...availabilityWindows].sort(
+    (left, right) => left.startedAt.getTime() - right.startedAt.getTime(),
+  )
+  const visibleSamples =
+    windows.length === 0
+      ? samples
+      : samples.filter((sample) => {
+          const observedAt = new Date(sample.observedAt).getTime()
+          return windows.some(
+            (window) =>
+              observedAt >= window.startedAt.getTime() &&
+              observedAt <=
+                (window.endedAt?.getTime() ?? Number.POSITIVE_INFINITY),
+          )
+        })
+
+  for (const sample of visibleSamples) {
+    const previous = points.at(-1)
+    if (
+      previous &&
+      new Date(sample.observedAt).getTime() -
+        new Date(previous.observedAt).getTime() >
+        gapAfterMs
+    ) {
+      points.push({
+        observedAt: new Date(
+          (new Date(previous.observedAt).getTime() +
+            new Date(sample.observedAt).getTime()) /
+            2,
+        ).toISOString(),
+        cpuPercent: null,
+        load1: null,
+        memoryUsedBytes: null,
+        memoryTotalBytes: null,
+        diskUsedBytes: null,
+        diskTotalBytes: null,
+        networkReceiveBytesPerSecond: null,
+        networkTransmitBytesPerSecond: null,
+        diskReadBytesPerSecond: null,
+        diskWriteBytesPerSecond: null,
+      })
+    }
+
+    points.push(sample)
+  }
+
+  for (const [index, window] of windows.entries()) {
+    if (index === 0) continue
+    const disabledAt = windows[index - 1]?.endedAt?.getTime()
+    if (disabledAt === undefined) {
+      continue
+    }
+
+    const enabledAt = window.startedAt.getTime()
+    points.push(emptyMetricSample(new Date((disabledAt + enabledAt) / 2)))
+  }
+
+  return points.sort(
+    (left, right) =>
+      new Date(left.observedAt).getTime() -
+      new Date(right.observedAt).getTime(),
+  )
+}
+
+function emptyMetricSample(observedAt: Date): ServerMetricsSample {
+  return {
+    observedAt: observedAt.toISOString(),
+    cpuPercent: null,
+    load1: null,
+    memoryUsedBytes: null,
+    memoryTotalBytes: null,
+    diskUsedBytes: null,
+    diskTotalBytes: null,
+    networkReceiveBytesPerSecond: null,
+    networkTransmitBytesPerSecond: null,
+    diskReadBytesPerSecond: null,
+    diskWriteBytesPerSecond: null,
+  }
+}
+
+function mergeIntervals(intervals: Array<{ start: number; end: number }>) {
+  const sorted = [...intervals].sort((a, b) => a.start - b.start)
+  const merged: Array<{ start: number; end: number }> = []
+
+  for (const interval of sorted) {
+    const current = merged.at(-1)
+    if (!current || interval.start > current.end) {
+      merged.push({ ...interval })
+      continue
+    }
+
+    current.end = Math.max(current.end, interval.end)
+  }
+
+  return merged
+}
+
+function intersectIntervals(
+  left: Array<{ start: number; end: number }>,
+  right: Array<{ start: number; end: number }>,
+) {
+  const intersections: Array<{ start: number; end: number }> = []
+  let leftIndex = 0
+  let rightIndex = 0
+
+  while (leftIndex < left.length && rightIndex < right.length) {
+    const leftInterval = left[leftIndex]
+    const rightInterval = right[rightIndex]
+    if (!leftInterval || !rightInterval) break
+    const start = Math.max(leftInterval.start, rightInterval.start)
+    const end = Math.min(leftInterval.end, rightInterval.end)
+
+    if (end > start) {
+      intersections.push({ start, end })
+    }
+
+    if (leftInterval.end < rightInterval.end) {
+      leftIndex += 1
+    } else {
+      rightIndex += 1
+    }
+  }
+
+  return intersections
+}
+
+function sumIntervals(intervals: Array<{ start: number; end: number }>) {
+  return intervals.reduce(
+    (total, interval) => total + interval.end - interval.start,
+    0,
+  )
+}
+
+function formatMetricSample(row: {
+  observedAt: Date
+  cpuPercent: number | null
+  load1: number | null
+  memoryUsedBytes: number | null
+  memoryTotalBytes: number | null
+  diskUsedBytes: number | null
+  diskTotalBytes: number | null
+  networkReceiveBytesPerSecond: number | null
+  networkTransmitBytesPerSecond: number | null
+  diskReadBytesPerSecond: number | null
+  diskWriteBytesPerSecond: number | null
+}): ServerMetricsSample {
+  return {
+    observedAt: row.observedAt.toISOString(),
+    cpuPercent: row.cpuPercent,
+    load1: row.load1,
+    memoryUsedBytes: row.memoryUsedBytes,
+    memoryTotalBytes: row.memoryTotalBytes,
+    diskUsedBytes: row.diskUsedBytes,
+    diskTotalBytes: row.diskTotalBytes,
+    networkReceiveBytesPerSecond: row.networkReceiveBytesPerSecond,
+    networkTransmitBytesPerSecond: row.networkTransmitBytesPerSecond,
+    diskReadBytesPerSecond: row.diskReadBytesPerSecond,
+    diskWriteBytesPerSecond: row.diskWriteBytesPerSecond,
+  }
+}
+
+// Keep extrema for every displayed series, plus bucket boundaries. Preserve
+// null transitions separately so downsampling never joins across an outage or
+// a period where an individual metric was unavailable. Raw rows remain stored.
+export function downsampleServerMetricSeries(
+  samples: Array<ServerMetricsSample>,
+  targetPoints = 600,
+) {
+  if (samples.length <= targetPoints) return samples
+  const ratio = (used: number | null, total: number | null) =>
+    used === null || total === null || total <= 0 ? null : used / total
+  const values = samples.map((sample) => [
+    sample.cpuPercent,
+    ratio(sample.memoryUsedBytes, sample.memoryTotalBytes),
+    ratio(sample.diskUsedBytes, sample.diskTotalBytes),
+    sample.networkReceiveBytesPerSecond,
+    sample.networkTransmitBytesPerSecond,
+  ])
+  const selected = new Set<number>()
+  const bucketSize = Math.ceil(
+    samples.length / Math.max(1, Math.floor(targetPoints / 12)),
+  )
+  for (let start = 0; start < samples.length; start += bucketSize) {
+    const end = Math.min(start + bucketSize, samples.length)
+    selected.add(start)
+    selected.add(end - 1)
+    for (let series = 0; series < 5; series++) {
+      let minimum = -1
+      let maximum = -1
+      for (let index = start; index < end; index++) {
+        const value = values[index]![series]
+        if (value === null || value === undefined) continue
+        if (minimum === -1 || value < values[minimum]![series]!) minimum = index
+        if (maximum === -1 || value > values[maximum]![series]!) maximum = index
+      }
+      if (minimum !== -1) selected.add(minimum)
+      if (maximum !== -1) selected.add(maximum)
+    }
+  }
+  for (let index = 1; index < values.length; index++) {
+    if (
+      values[index]!.some(
+        (value, series) =>
+          (value === null) !== (values[index - 1]![series] === null),
+      )
+    ) {
+      selected.add(index - 1)
+      selected.add(index)
+    }
+  }
+  return [...selected].sort((a, b) => a - b).map((index) => samples[index]!)
+}