@openhi/constructs 0.0.110 → 0.0.111

package/lib/rest-api-lambda.handler.mjs

@@ -1,6 +1,9 @@
 import {
   createRoleOperation
 } from "./chunk-AJ3G3THO.mjs";
+import {
+  buildSchemaBootstrapStatements
+} from "./chunk-2O3CXY2C.mjs";
 import {
   getRoleByIdOperation
 } from "./chunk-7FUAMZOF.mjs";
@@ -3727,6 +3730,7 @@ var DataApiPostgresQueryRunner = class {
     this.schema = options.schema;
   }
   async query(sql, params) {
+    await this.ensureSchemaBootstrapped();
     const out = await this.client.send(
       new ExecuteStatementCommand({
         resourceArn: this.clusterArn,
@@ -3752,6 +3756,42 @@ var DataApiPostgresQueryRunner = class {
     }
     return JSON.parse(out.formattedRecords);
   }
+  /**
+   * Run `CREATE SCHEMA / CREATE TABLE / CREATE INDEX` idempotent DDL once per
+   * runner instance. The replication Lambda's cold start does the same thing
+   * (data-store-postgres-replication.handler) but in a fresh environment with
+   * no DynamoDB writes yet the read path can be the first thing that touches
+   * the cluster — without this, every search returns
+   * `relation "<schema>.resources" does not exist` until something has been
+   * written. Memoized so subsequent queries on a warm container don't pay the
+   * DDL cost; resets the promise on failure so the next call retries.
+   *
+   * Data API's `ExecuteStatement` accepts only a single SQL statement per
+   * call, so each DDL statement from `buildSchemaBootstrapStatements` is
+   * issued individually. `IF NOT EXISTS` on every statement makes this safe
+   * to race across concurrent containers.
+   */
+  async ensureSchemaBootstrapped() {
+    if (!this.bootstrapPromise) {
+      this.bootstrapPromise = this.runBootstrap().catch((err) => {
+        this.bootstrapPromise = void 0;
+        throw err;
+      });
+    }
+    return this.bootstrapPromise;
+  }
+  async runBootstrap() {
+    for (const statement of buildSchemaBootstrapStatements(this.schema)) {
+      await this.client.send(
+        new ExecuteStatementCommand({
+          resourceArn: this.clusterArn,
+          secretArn: this.secretArn,
+          database: this.database,
+          sql: statement
+        })
+      );
+    }
+  }
 };
 function qualifyResourcesTable(sql, schema) {
   return sql.replace(