@checkstack/healthcheck-postgres-backend 0.1.13 → 0.2.0

package/CHANGELOG.md

@@ -1,5 +1,59 @@
 # @checkstack/healthcheck-postgres-backend
 
+## 0.2.0
+
+### Minor Changes
+
+- 3dd1914: Migrate health check strategies to VersionedAggregated with \_type discriminator
+
+  All 13 health check strategies now use `VersionedAggregated` for their `aggregatedResult` property, enabling automatic bucket merging with 100% mathematical fidelity.
+
+  **Key changes:**
+
+  - **`_type` discriminator**: All aggregated state objects now include a required `_type` field (`"average"`, `"rate"`, `"counter"`, `"minmax"`) for reliable type detection
+  - The `HealthCheckStrategy` interface now requires `aggregatedResult` to be a `VersionedAggregated<AggregatedResultShape>`
+  - Strategy/collector `mergeResult` methods return state objects with `_type` (e.g., `{ _type: "average", _sum, _count, avg }`)
+  - `mergeAggregatedBucketResults`, `combineBuckets`, and `reaggregateBuckets` now require `registry` and `strategyId` parameters
+  - `HealthCheckService` constructor now requires both `registry` and `collectorRegistry` parameters
+  - Frontend `extractComputedValue` now uses `_type` discriminator for robust type detection
+
+  **Breaking Change**: State objects now require `_type`. Merge functions automatically add `_type` to output. The bucket merging functions and `HealthCheckService` now require additional required parameters.
+
+### Patch Changes
+
+- Updated dependencies [3dd1914]
+  - @checkstack/backend-api@0.7.0
+
+## 0.1.14
+
+### Patch Changes
+
+- 48c2080: Migrate aggregation from batch to incremental (`mergeResult`)
+
+  ### Breaking Changes (Internal)
+
+  - Replaced `aggregateResult(runs[])` with `mergeResult(existing, run)` interface across all HealthCheckStrategy and CollectorStrategy implementations
+
+  ### New Features
+
+  - Added incremental aggregation utilities in `@checkstack/backend-api`:
+    - `mergeCounter()` - track occurrences
+    - `mergeAverage()` - track sum/count, compute avg
+    - `mergeRate()` - track success/total, compute %
+    - `mergeMinMax()` - track min/max values
+  - Exported Zod schemas for internal state: `averageStateSchema`, `rateStateSchema`, `minMaxStateSchema`, `counterStateSchema`
+
+  ### Improvements
+
+  - Enables O(1) storage overhead by maintaining incremental aggregation state
+  - Prepares for real-time hourly aggregation without batch accumulation
+
+- Updated dependencies [f676e11]
+- Updated dependencies [48c2080]
+  - @checkstack/common@0.6.2
+  - @checkstack/backend-api@0.6.0
+  - @checkstack/healthcheck-common@0.8.2
+
 ## 0.1.13
 
 ### Patch Changes

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@checkstack/healthcheck-postgres-backend",
-  "version": "0.1.13",
+  "version": "0.2.0",
   "type": "module",
   "main": "src/index.ts",
   "scripts": {
@@ -9,16 +9,16 @@
     "lint:code": "eslint . --max-warnings 0"
   },
   "dependencies": {
-    "@checkstack/backend-api": "0.5.1",
-    "@checkstack/common": "0.6.0",
-    "@checkstack/healthcheck-common": "0.7.0",
+    "@checkstack/backend-api": "0.5.2",
+    "@checkstack/common": "0.6.1",
+    "@checkstack/healthcheck-common": "0.8.1",
     "pg": "^8.11.0"
   },
   "devDependencies": {
     "@types/bun": "^1.0.0",
     "@types/pg": "^8.11.0",
     "typescript": "^5.0.0",
-    "@checkstack/tsconfig": "0.0.2",
-    "@checkstack/scripts": "0.1.0"
+    "@checkstack/tsconfig": "0.0.3",
+    "@checkstack/scripts": "0.1.1"
   }
 }

package/src/query-collector.test.ts

@@ -64,7 +64,7 @@ describe("QueryCollector", () => {
     });
   });
 
-  describe("aggregateResult", () => {
+  describe("mergeResult", () => {
     it("should calculate average execution time and success rate", () => {
       const collector = new QueryCollector();
       const runs = [
@@ -86,10 +86,11 @@ describe("QueryCollector", () => {
         },
       ];
 
-      const aggregated = collector.aggregateResult(runs);
+      let aggregated = collector.mergeResult(undefined, runs[0]);
+      aggregated = collector.mergeResult(aggregated, runs[1]);
 
-      expect(aggregated.avgExecutionTimeMs).toBe(75);
-      expect(aggregated.successRate).toBe(100);
+      expect(aggregated.avgExecutionTimeMs.avg).toBe(75);
+      expect(aggregated.successRate.rate).toBe(100);
     });
 
     it("should calculate success rate correctly", () => {
@@ -113,9 +114,10 @@ describe("QueryCollector", () => {
         },
       ];
 
-      const aggregated = collector.aggregateResult(runs);
+      let aggregated = collector.mergeResult(undefined, runs[0]);
+      aggregated = collector.mergeResult(aggregated, runs[1]);
 
-      expect(aggregated.successRate).toBe(50);
+      expect(aggregated.successRate.rate).toBe(50);
     });
   });
 

package/src/query-collector.ts

@@ -4,6 +4,12 @@ import {
   type HealthCheckRunForAggregation,
   type CollectorResult,
   type CollectorStrategy,
+  mergeAverage,
+  mergeRate,
+  VersionedAggregated,
+  aggregatedAverage,
+  aggregatedRate,
+  type InferAggregatedResult,
 } from "@checkstack/backend-api";
 import {
   healthResultNumber,
@@ -45,20 +51,24 @@ const queryResultSchema = healthResultSchema({
 
 export type QueryResult = z.infer<typeof queryResultSchema>;
 
-const queryAggregatedSchema = healthResultSchema({
-  avgExecutionTimeMs: healthResultNumber({
+// Aggregated result fields definition
+const queryAggregatedFields = {
+  avgExecutionTimeMs: aggregatedAverage({
     "x-chart-type": "line",
     "x-chart-label": "Avg Execution Time",
     "x-chart-unit": "ms",
   }),
-  successRate: healthResultNumber({
+  successRate: aggregatedRate({
     "x-chart-type": "gauge",
     "x-chart-label": "Success Rate",
     "x-chart-unit": "%",
   }),
-});
+};
 
-export type QueryAggregatedResult = z.infer<typeof queryAggregatedSchema>;
+// Type inferred from field definitions
+export type QueryAggregatedResult = InferAggregatedResult<
+  typeof queryAggregatedFields
+>;
 
 // ============================================================================
 // QUERY COLLECTOR
@@ -68,15 +78,12 @@ export type QueryAggregatedResult = z.infer<typeof queryAggregatedSchema>;
  * Built-in PostgreSQL query collector.
  * Executes SQL queries and checks results.
  */
-export class QueryCollector
-  implements
-    CollectorStrategy<
-      PostgresTransportClient,
-      QueryConfig,
-      QueryResult,
-      QueryAggregatedResult
-    >
-{
+export class QueryCollector implements CollectorStrategy<
+  PostgresTransportClient,
+  QueryConfig,
+  QueryResult,
+  QueryAggregatedResult
+> {
   id = "query";
   displayName = "SQL Query";
   description = "Execute a SQL query and check the result";
@@ -87,9 +94,9 @@ export class QueryCollector
 
   config = new Versioned({ version: 1, schema: queryConfigSchema });
   result = new Versioned({ version: 1, schema: queryResultSchema });
-  aggregatedResult = new Versioned({
+  aggregatedResult = new VersionedAggregated({
     version: 1,
-    schema: queryAggregatedSchema,
+    fields: queryAggregatedFields,
   });
 
   async execute({
@@ -115,28 +122,18 @@ export class QueryCollector
     };
   }
 
-  aggregateResult(
-    runs: HealthCheckRunForAggregation<QueryResult>[]
+  mergeResult(
+    existing: QueryAggregatedResult | undefined,
+    run: HealthCheckRunForAggregation<QueryResult>,
   ): QueryAggregatedResult {
-    const times = runs
-      .map((r) => r.metadata?.executionTimeMs)
-      .filter((v): v is number => typeof v === "number");
-
-    const successes = runs
-      .map((r) => r.metadata?.success)
-      .filter((v): v is boolean => typeof v === "boolean");
-
-    const successCount = successes.filter(Boolean).length;
+    const metadata = run.metadata;
 
     return {
-      avgExecutionTimeMs:
-        times.length > 0
-          ? Math.round(times.reduce((a, b) => a + b, 0) / times.length)
-          : 0,
-      successRate:
-        successes.length > 0
-          ? Math.round((successCount / successes.length) * 100)
-          : 0,
+      avgExecutionTimeMs: mergeAverage(
+        existing?.avgExecutionTimeMs,
+        metadata?.executionTimeMs,
+      ),
+      successRate: mergeRate(existing?.successRate, metadata?.success),
     };
   }
 }

package/src/strategy.test.ts

@@ -8,7 +8,7 @@ describe("PostgresHealthCheckStrategy", () => {
       rowCount?: number;
       queryError?: Error;
       connectError?: Error;
-    } = {}
+    } = {},
   ): DbClient => ({
     connect: mock(() =>
       config.connectError
@@ -17,10 +17,10 @@ describe("PostgresHealthCheckStrategy", () => {
             query: mock(() =>
               config.queryError
                 ? Promise.reject(config.queryError)
-                : Promise.resolve({ rowCount: config.rowCount ?? 1 })
+                : Promise.resolve({ rowCount: config.rowCount ?? 1 }),
             ),
             end: mock(() => Promise.resolve()),
-          })
+          }),
     ),
   });
 
@@ -46,7 +46,7 @@ describe("PostgresHealthCheckStrategy", () => {
 
     it("should throw for connection error", async () => {
       const strategy = new PostgresHealthCheckStrategy(
-        createMockClient({ connectError: new Error("Connection refused") })
+        createMockClient({ connectError: new Error("Connection refused") }),
       );
 
       await expect(
@@ -57,7 +57,7 @@ describe("PostgresHealthCheckStrategy", () => {
           user: "postgres",
           password: "secret",
           timeout: 5000,
-        })
+        }),
       ).rejects.toThrow("Connection refused");
     });
   });
@@ -85,7 +85,7 @@ describe("PostgresHealthCheckStrategy", () => {
 
     it("should return error for query error", async () => {
       const strategy = new PostgresHealthCheckStrategy(
-        createMockClient({ queryError: new Error("Syntax error") })
+        createMockClient({ queryError: new Error("Syntax error") }),
       );
       const connectedClient = await strategy.createClient({
         host: "localhost",
@@ -107,7 +107,7 @@ describe("PostgresHealthCheckStrategy", () => {
 
     it("should return custom row count", async () => {
       const strategy = new PostgresHealthCheckStrategy(
-        createMockClient({ rowCount: 5 })
+        createMockClient({ rowCount: 5 }),
       );
       const connectedClient = await strategy.createClient({
         host: "localhost",
@@ -128,8 +128,8 @@ describe("PostgresHealthCheckStrategy", () => {
     });
   });
 
-  describe("aggregateResult", () => {
-    it("should calculate averages correctly", () => {
+  describe("mergeResult", () => {
+    it("should calculate averages correctly through incremental merging", () => {
       const strategy = new PostgresHealthCheckStrategy();
       const runs = [
         {
@@ -158,34 +158,34 @@ describe("PostgresHealthCheckStrategy", () => {
         },
       ];
 
-      const aggregated = strategy.aggregateResult(runs);
+      // Merge incrementally
+      let aggregated = strategy.mergeResult(undefined, runs[0]);
+      aggregated = strategy.mergeResult(aggregated, runs[1]);
 
-      expect(aggregated.avgConnectionTime).toBe(75);
-      expect(aggregated.successRate).toBe(100);
-      expect(aggregated.errorCount).toBe(0);
+      expect(aggregated.avgConnectionTime.avg).toBe(75);
+      expect(aggregated.successRate.rate).toBe(100);
+      expect(aggregated.errorCount.count).toBe(0);
     });
 
     it("should count errors", () => {
       const strategy = new PostgresHealthCheckStrategy();
-      const runs = [
-        {
-          id: "1",
-          status: "unhealthy" as const,
-          latencyMs: 100,
-          checkId: "c1",
-          timestamp: new Date(),
-          metadata: {
-            connected: false,
-            connectionTimeMs: 100,
-            error: "Connection refused",
-          },
+      const run = {
+        id: "1",
+        status: "unhealthy" as const,
+        latencyMs: 100,
+        checkId: "c1",
+        timestamp: new Date(),
+        metadata: {
+          connected: false,
+          connectionTimeMs: 100,
+          error: "Connection refused",
         },
-      ];
+      };
 
-      const aggregated = strategy.aggregateResult(runs);
+      const aggregated = strategy.mergeResult(undefined, run);
 
-      expect(aggregated.errorCount).toBe(1);
-      expect(aggregated.successRate).toBe(0);
+      expect(aggregated.errorCount.count).toBe(1);
+      expect(aggregated.successRate.rate).toBe(0);
     });
   });
 });

package/src/strategy.ts

@@ -3,11 +3,21 @@ import {
   HealthCheckStrategy,
   HealthCheckRunForAggregation,
   Versioned,
+  VersionedAggregated,
+  aggregatedAverage,
+  aggregatedMinMax,
+  aggregatedRate,
+  aggregatedCounter,
+  mergeAverage,
+  mergeRate,
+  mergeCounter,
+  mergeMinMax,
   z,
   configString,
   configNumber,
   configBoolean,
   type ConnectedClient,
+  type InferAggregatedResult,
 } from "@checkstack/backend-api";
 import {
   healthResultBoolean,
@@ -70,32 +80,32 @@ const postgresResultSchema = healthResultSchema({
 
 type PostgresResult = z.infer<typeof postgresResultSchema>;
 
-/**
- * Aggregated metadata for buckets.
- */
-const postgresAggregatedSchema = healthResultSchema({
-  avgConnectionTime: healthResultNumber({
+/** Aggregated field definitions for bucket merging */
+const postgresAggregatedFields = {
+  avgConnectionTime: aggregatedAverage({
     "x-chart-type": "line",
     "x-chart-label": "Avg Connection Time",
     "x-chart-unit": "ms",
   }),
-  maxConnectionTime: healthResultNumber({
+  maxConnectionTime: aggregatedMinMax({
     "x-chart-type": "line",
     "x-chart-label": "Max Connection Time",
     "x-chart-unit": "ms",
   }),
-  successRate: healthResultNumber({
+  successRate: aggregatedRate({
     "x-chart-type": "gauge",
     "x-chart-label": "Success Rate",
     "x-chart-unit": "%",
   }),
-  errorCount: healthResultNumber({
+  errorCount: aggregatedCounter({
     "x-chart-type": "counter",
     "x-chart-label": "Errors",
   }),
-});
+};
 
-type PostgresAggregatedResult = z.infer<typeof postgresAggregatedSchema>;
+type PostgresAggregatedResult = InferAggregatedResult<
+  typeof postgresAggregatedFields
+>;
 
 // ============================================================================
 // DATABASE CLIENT INTERFACE (for testability)
@@ -133,15 +143,12 @@ const defaultDbClient: DbClient = {
 // STRATEGY
 // ============================================================================
 
-export class PostgresHealthCheckStrategy
-  implements
-    HealthCheckStrategy<
-      PostgresConfig,
-      PostgresTransportClient,
-      PostgresResult,
-      PostgresAggregatedResult
-    >
-{
+export class PostgresHealthCheckStrategy implements HealthCheckStrategy<
+  PostgresConfig,
+  PostgresTransportClient,
+  PostgresResult,
+  typeof postgresAggregatedFields
+> {
   id = "postgres";
   displayName = "PostgreSQL Health Check";
   description = "PostgreSQL database connectivity and query health check";
@@ -178,58 +185,38 @@ export class PostgresHealthCheckStrategy
     ],
   });
 
-  aggregatedResult: Versioned<PostgresAggregatedResult> = new Versioned({
+  aggregatedResult = new VersionedAggregated({
     version: 1,
-    schema: postgresAggregatedSchema,
+    fields: postgresAggregatedFields,
   });
 
-  aggregateResult(
-    runs: HealthCheckRunForAggregation<PostgresResult>[]
+  mergeResult(
+    existing: PostgresAggregatedResult | undefined,
+    run: HealthCheckRunForAggregation<PostgresResult>,
   ): PostgresAggregatedResult {
-    const validRuns = runs.filter((r) => r.metadata);
+    const metadata = run.metadata;
 
-    if (validRuns.length === 0) {
-      return {
-        avgConnectionTime: 0,
-        maxConnectionTime: 0,
-        successRate: 0,
-        errorCount: 0,
-      };
-    }
+    const avgConnectionTime = mergeAverage(
+      existing?.avgConnectionTime,
+      metadata?.connectionTimeMs,
+    );
 
-    const connectionTimes = validRuns
-      .map((r) => r.metadata?.connectionTimeMs)
-      .filter((t): t is number => typeof t === "number");
+    const maxConnectionTime = mergeMinMax(
+      existing?.maxConnectionTime,
+      metadata?.connectionTimeMs,
+    );
 
-    const avgConnectionTime =
-      connectionTimes.length > 0
-        ? Math.round(
-            connectionTimes.reduce((a, b) => a + b, 0) / connectionTimes.length
-          )
-        : 0;
+    const isSuccess = metadata?.connected ?? false;
+    const successRate = mergeRate(existing?.successRate, isSuccess);
 
-    const maxConnectionTime =
-      connectionTimes.length > 0 ? Math.max(...connectionTimes) : 0;
+    const hasError = metadata?.error !== undefined;
+    const errorCount = mergeCounter(existing?.errorCount, hasError);
 
-    const successCount = validRuns.filter(
-      (r) => r.metadata?.connected === true
-    ).length;
-    const successRate = Math.round((successCount / validRuns.length) * 100);
-
-    const errorCount = validRuns.filter(
-      (r) => r.metadata?.error !== undefined
-    ).length;
-
-    return {
-      avgConnectionTime,
-      maxConnectionTime,
-      successRate,
-      errorCount,
-    };
+    return { avgConnectionTime, maxConnectionTime, successRate, errorCount };
   }
 
   async createClient(
-    config: PostgresConfigInput
+    config: PostgresConfigInput,
   ): Promise<ConnectedClient<PostgresTransportClient>> {
     const validatedConfig = this.config.validate(config);