Feat push schema ci

package.json

@@ -21,7 +21,7 @@
     "@libpg-query/parser": "^17.6.3",
     "@opentelemetry/api": "^1.9.0",
     "@pgsql/types": "^17.6.2",
-    "@query-doctor/core": "^0.10.3",
+    "@query-doctor/core": "^0.10.4",
     "async-sema": "^3.1.1",
     "capnweb": "^0.7.0",
     "dedent": "^1.7.1",

src/main.ts

@@ -19,6 +19,7 @@ import { Remote } from "./remote/remote.ts";
 import { ConnectionManager } from "./sync/connection-manager.ts";
 import { PgbadgerSource } from "./sql/pgbadger.ts";
 import type { RecentQuerySource } from "./sql/recent-query.ts";
+import type { FullSchema } from "@query-doctor/core";
 
 async function runInCI(
   targetPostgresUrl: Connectable,
@@ -47,6 +48,17 @@ async function runInCI(
     log.warn(`API connection broken during CI run: ${err}`, "main");
   });
 
+  // `Runner.build` triggers `remote.syncFrom`, which emits `schemaSynced` with
+  // the schema dumped from the source DB. CI doesn't wire up `hookUpApiReporter`
+  // (that's the persistent-server path), so capture the schema here and push it
+  // explicitly. We await the push before `disposeApi` so the short-lived CI
+  // process doesn't exit before the WS write flushes.
+  let syncedSchema: FullSchema | undefined;
+  const onSchemaSynced = (schema: FullSchema) => {
+    syncedSchema = schema;
+  };
+  remote.on("schemaSynced", onSchemaSynced);
+
   try {
     const config = repo
       ? await api.getRepoConfig(repo, branch).catch(
@@ -81,6 +93,12 @@ async function runInCI(
       await runner.close();
     }
 
+    if (syncedSchema) {
+      await api.pushSchema(JSON.parse(JSON.stringify(syncedSchema))).catch((err) => {
+        log.warn(`Failed to push schema: ${err}`, "main");
+      });
+    }
+
     const queries = buildQueries(allResults, config);
 
     // POST to Site API first so we get the run ID (for baseline exclusion) and
@@ -198,6 +216,7 @@ async function runInCI(
       }
     }
   } finally {
+    remote.off("schemaSynced", onSchemaSynced);
     disposeApi();
   }
 }