@rocicorp/zero 0.25.0 → 0.26.0-canary.0

package/out/zero-cache/src/services/change-source/pg/schema/ddl.js

@@ -1,78 +1,38 @@
-import { literal as lit } from 'pg-format';
+import { literal as literal$1 } from "pg-format";
 import { assert } from "../../../../../../shared/src/asserts.js";
-import * as v from "../../../../../../shared/src/valita.js";
+import "../../../../../../shared/src/valita.js";
 import { upstreamSchema } from "../../../../types/shards.js";
 import { id } from "../../../../types/sql.js";
-import { indexDefinitionsQuery, publishedSchema, publishedTableQuery, } from "./published.js";
-// Sent in the 'version' tag of "ddlStart" and "ddlUpdate" event messages.
-// This is used to ensure that the message constructed in the upstream
-// Trigger function is compatible with the code processing it in the zero-cache.
-//
-// Increment this when changing the format of the contents of the "ddl" events.
-// This will allow old / incompatible code to detect the change and abort.
-export const PROTOCOL_VERSION = 1;
-const triggerEvent = v.object({
-    context: v.object({ query: v.string() }).rest(v.string()),
+import { publishedSchema, publishedTableQuery, indexDefinitionsQuery } from "./published.js";
+import { object, string, literal, union } from "@badrap/valita";
+const PROTOCOL_VERSION = 1;
+const triggerEvent = object({
+  context: object({ query: string() }).rest(string())
 });
-// All DDL events contain a snapshot of the current tables and indexes that
-// are published / relevant to the shard.
-export const ddlEventSchema = triggerEvent.extend({
-    version: v.literal(PROTOCOL_VERSION),
-    schema: publishedSchema,
+const ddlEventSchema = triggerEvent.extend({
+  version: literal(PROTOCOL_VERSION),
+  schema: publishedSchema
 });
-// The `ddlStart` message is computed before every DDL event, regardless of
-// whether the subsequent event affects the shard. Downstream processing should
-// capture the contained schema information in order to determine the schema
-// changes necessary to apply a subsequent `ddlUpdate` message. Note that a
-// `ddlUpdate` message may not follow, as updates determined to be irrelevant
-// to the shard will not result in a message. However, all `ddlUpdate` messages
-// are guaranteed to be preceded by a `ddlStart` message.
-export const ddlStartEventSchema = ddlEventSchema.extend({
-    type: v.literal('ddlStart'),
+const ddlStartEventSchema = ddlEventSchema.extend({
+  type: literal("ddlStart")
 });
-/**
- * The {@link DdlUpdateEvent} contains an updated schema resulting from
- * a particular ddl event. The event type provides information
- * (i.e. constraints) on the difference from the schema of the preceding
- * {@link DdlStartEvent}.
- *
- * Note that in almost all cases (the exception being `CREATE` events),
- * it is possible that there is no relevant difference between the
- * ddl-start schema and the ddl-update schema, as many aspects of the
- * schema (e.g. column constraints) are not relevant to downstream
- * replication.
- */
-export const ddlUpdateEventSchema = ddlEventSchema.extend({
-    type: v.literal('ddlUpdate'),
-    event: v.object({ tag: v.string() }),
+const ddlUpdateEventSchema = ddlEventSchema.extend({
+  type: literal("ddlUpdate"),
+  event: object({ tag: string() })
 });
-export const replicationEventSchema = v.union(ddlStartEventSchema, ddlUpdateEventSchema);
-// Creates a function that appends `_{shard-num}` to the input and
-// quotes the result to be a valid identifier.
+const replicationEventSchema = union(
+  ddlStartEventSchema,
+  ddlUpdateEventSchema
+);
 function append(shardNum) {
-    return (name) => id(name + '_' + String(shardNum));
+  return (name) => id(name + "_" + String(shardNum));
 }
-/**
- * Event trigger functions contain the core logic that are invoked by triggers.
- *
- * Note that although many of these functions can theoretically be parameterized and
- * shared across shards, it is advantageous to keep the functions in each shard
- * isolated from each other in order to avoid the complexity of shared-function
- * versioning.
- *
- * In a sense, shards (and their triggers and functions) should be thought of as
- * execution environments that can be updated at different schedules. If per-shard
- * triggers called into shared functions, we would have to consider versioning the
- * functions when changing their behavior, backwards compatibility, removal of
- * unused versions, etc. (not unlike versioning of npm packages).
- *
- * Instead, we opt for the simplicity and isolation of having each shard
- * completely own (and maintain) the entirety of its trigger/function stack.
- */
 function createEventFunctionStatements(shard) {
-    const { appID, shardNum, publications } = shard;
-    const schema = id(upstreamSchema(shard)); // e.g. "{APP_ID}_{SHARD_ID}"
-    return /*sql*/ `
+  const { appID, shardNum, publications } = shard;
+  const schema = id(upstreamSchema(shard));
+  return (
+    /*sql*/
+    `
 CREATE SCHEMA IF NOT EXISTS ${schema};
 
 CREATE OR REPLACE FUNCTION ${schema}.get_trigger_context()
@@ -89,7 +49,7 @@ $$ LANGUAGE plpgsql;
 CREATE OR REPLACE FUNCTION ${schema}.notice_ignore(object_id TEXT)
 RETURNS void AS $$
 BEGIN
-  RAISE NOTICE 'zero(%) ignoring %', ${lit(shardNum)}, object_id;
+  RAISE NOTICE 'zero(%) ignoring %', ${literal$1(shardNum)}, object_id;
 END
 $$ LANGUAGE plpgsql;
 
@@ -125,7 +85,9 @@ BEGIN
     'context', ${schema}.get_trigger_context()
   ) INTO message;
 
-  PERFORM pg_logical_emit_message(true, ${lit(`${appID}/${shardNum}`)}, message);
+  PERFORM pg_logical_emit_message(true, ${literal$1(
+      `${appID}/${shardNum}`
+    )}, message);
 END
 $$ LANGUAGE plpgsql;
 
@@ -141,7 +103,7 @@ DECLARE
   message TEXT;
   event TEXT;
 BEGIN
-  publications := ARRAY[${lit(publications)}];
+  publications := ARRAY[${literal$1(publications)}];
 
   SELECT objid, object_type, object_identity 
     FROM pg_event_trigger_ddl_commands() 
@@ -238,74 +200,89 @@ BEGIN
     'context', ${schema}.get_trigger_context()
   ) INTO message;
 
-  PERFORM pg_logical_emit_message(true, ${lit(`${appID}/${shardNum}`)}, message);
+  PERFORM pg_logical_emit_message(true, ${literal$1(
+      `${appID}/${shardNum}`
+    )}, message);
 END
 $$ LANGUAGE plpgsql;
-`;
+`
+  );
 }
-// Exported for testing.
-export const TAGS = [
-    'CREATE TABLE',
-    'ALTER TABLE',
-    'CREATE INDEX',
-    'DROP TABLE',
-    'DROP INDEX',
-    'ALTER PUBLICATION',
-    'ALTER SCHEMA',
+const TAGS = [
+  "CREATE TABLE",
+  "ALTER TABLE",
+  "CREATE INDEX",
+  "DROP TABLE",
+  "DROP INDEX",
+  "ALTER PUBLICATION",
+  "ALTER SCHEMA"
 ];
-export function createEventTriggerStatements(shard) {
-    // Better to assert here than get a cryptic syntax error from Postgres.
-    assert(shard.publications.length, `shard publications must be non-empty`);
-    // Unlike functions, which are namespaced in shard-specific schemas,
-    // EVENT TRIGGER names are in the global namespace and thus must include
-    // the appID and shardNum.
-    const { appID, shardNum } = shard;
-    const sharded = append(shardNum);
-    const schema = id(upstreamSchema(shard));
-    const triggers = [
-        dropEventTriggerStatements(shard.appID, shard.shardNum),
-        createEventFunctionStatements(shard),
-    ];
-    // A single ddl_command_start trigger covering all relevant tags.
-    triggers.push(/*sql*/ `
+function createEventTriggerStatements(shard) {
+  assert(shard.publications.length, `shard publications must be non-empty`);
+  const { appID, shardNum } = shard;
+  const sharded = append(shardNum);
+  const schema = id(upstreamSchema(shard));
+  const triggers = [
+    dropEventTriggerStatements(shard.appID, shard.shardNum),
+    createEventFunctionStatements(shard)
+  ];
+  triggers.push(
+    /*sql*/
+    `
 CREATE EVENT TRIGGER ${sharded(`${appID}_ddl_start`)}
   ON ddl_command_start
-  WHEN TAG IN (${lit(TAGS)})
+  WHEN TAG IN (${literal$1(TAGS)})
   EXECUTE PROCEDURE ${schema}.emit_ddl_start();
-`);
-    // A per-tag ddl_command_end trigger that dispatches to ${schema}.emit_ddl_end(tag)
-    for (const tag of TAGS) {
-        const tagID = tag.toLowerCase().replace(' ', '_');
-        triggers.push(/*sql*/ `
+`
+  );
+  for (const tag of TAGS) {
+    const tagID = tag.toLowerCase().replace(" ", "_");
+    triggers.push(
+      /*sql*/
+      `
 CREATE OR REPLACE FUNCTION ${schema}.emit_${tagID}() 
 RETURNS event_trigger AS $$
 BEGIN
-  PERFORM ${schema}.emit_ddl_end(${lit(tag)});
+  PERFORM ${schema}.emit_ddl_end(${literal$1(tag)});
 END
 $$ LANGUAGE plpgsql;
 
 CREATE EVENT TRIGGER ${sharded(`${appID}_${tagID}`)}
   ON ddl_command_end
-  WHEN TAG IN (${lit(tag)})
+  WHEN TAG IN (${literal$1(tag)})
   EXECUTE PROCEDURE ${schema}.emit_${tagID}();
-`);
-    }
-    return triggers.join('');
+`
+    );
+  }
+  return triggers.join("");
 }
-// Exported for testing.
-export function dropEventTriggerStatements(appID, shardID) {
-    const stmts = [];
-    // A single ddl_command_start trigger covering all relevant tags.
-    stmts.push(/*sql*/ `
+function dropEventTriggerStatements(appID, shardID) {
+  const stmts = [];
+  stmts.push(
+    /*sql*/
+    `
     DROP EVENT TRIGGER IF EXISTS ${id(`${appID}_ddl_start_${shardID}`)};
-  `);
-    // A per-tag ddl_command_end trigger that dispatches to ${schema}.emit_ddl_end(tag)
-    for (const tag of TAGS) {
-        const tagID = tag.toLowerCase().replace(' ', '_');
-        stmts.push(/*sql*/ `
+  `
+  );
+  for (const tag of TAGS) {
+    const tagID = tag.toLowerCase().replace(" ", "_");
+    stmts.push(
+      /*sql*/
+      `
       DROP EVENT TRIGGER IF EXISTS ${id(`${appID}_${tagID}_${shardID}`)};
-    `);
-    }
-    return stmts.join('');
+    `
+    );
+  }
+  return stmts.join("");
 }
-//# sourceMappingURL=ddl.js.map
+export {
+  PROTOCOL_VERSION,
+  TAGS,
+  createEventTriggerStatements,
+  ddlEventSchema,
+  ddlStartEventSchema,
+  ddlUpdateEventSchema,
+  dropEventTriggerStatements,
+  replicationEventSchema
+};
+//# sourceMappingURL=ddl.js.map

package/out/zero-cache/src/services/change-source/pg/schema/ddl.js.map

@@ -1 +1 @@
-{"version":3,"file":"ddl.js","sourceRoot":"","sources":["../../../../../../../../zero-cache/src/services/change-source/pg/schema/ddl.ts"],"names":[],"mappings":"AAAA,OAAO,EAAC,OAAO,IAAI,GAAG,EAAC,MAAM,WAAW,CAAC;AACzC,OAAO,EAAC,MAAM,EAAC,MAAM,yCAAyC,CAAC;AAC/D,OAAO,KAAK,CAAC,MAAM,wCAAwC,CAAC;AAC5D,OAAO,EAAC,cAAc,EAAmB,MAAM,6BAA6B,CAAC;AAC7E,OAAO,EAAC,EAAE,EAAC,MAAM,0BAA0B,CAAC;AAC5C,OAAO,EACL,qBAAqB,EACrB,eAAe,EACf,mBAAmB,GACpB,MAAM,gBAAgB,CAAC;AAExB,0EAA0E;AAC1E,sEAAsE;AACtE,gFAAgF;AAChF,EAAE;AACF,+EAA+E;AAC/E,0EAA0E;AAC1E,MAAM,CAAC,MAAM,gBAAgB,GAAG,CAAC,CAAC;AAElC,MAAM,YAAY,GAAG,CAAC,CAAC,MAAM,CAAC;IAC5B,OAAO,EAAE,CAAC,CAAC,MAAM,CAAC,EAAC,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE,EAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;CACxD,CAAC,CAAC;AAEH,2EAA2E;AAC3E,yCAAyC;AACzC,MAAM,CAAC,MAAM,cAAc,GAAG,YAAY,CAAC,MAAM,CAAC;IAChD,OAAO,EAAE,CAAC,CAAC,OAAO,CAAC,gBAAgB,CAAC;IACpC,MAAM,EAAE,eAAe;CACxB,CAAC,CAAC;AAEH,2EAA2E;AAC3E,+EAA+E;AAC/E,4EAA4E;AAC5E,2EAA2E;AAC3E,6EAA6E;AAC7E,+EAA+E;AAC/E,yDAAyD;AACzD,MAAM,CAAC,MAAM,mBAAmB,GAAG,cAAc,CAAC,MAAM,CAAC;IACvD,IAAI,EAAE,CAAC,CAAC,OAAO,CAAC,UAAU,CAAC;CAC5B,CAAC,CAAC;AAIH;;;;;;;;;;;GAWG;AACH,MAAM,CAAC,MAAM,oBAAoB,GAAG,cAAc,CAAC,MAAM,CAAC;IACxD,IAAI,EAAE,CAAC,CAAC,OAAO,CAAC,WAAW,CAAC;IAC5B,KAAK,EAAE,CAAC,CAAC,MAAM,CAAC,EAAC,GAAG,EAAE,CAAC,CAAC,MAAM,EAAE,EAAC,CAAC;CACnC,CAAC,CAAC;AAIH,MAAM,CAAC,MAAM,sBAAsB,GAAG,CAAC,CAAC,KAAK,CAC3C,mBAAmB,EACnB,oBAAoB,CACrB,CAAC;AAIF,kEAAkE;AAClE,8CAA8C;AAC9C,SAAS,MAAM,CAAC,QAAgB;IAC9B,OAAO,CAAC,IAAY,EAAE,EAAE,CAAC,EAAE,CAAC,IAAI,GAAG,GAAG,GAAG,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC;AAC7D,CAAC;AAED;;;;;;;;;;;;;;;;GAgBG;AACH,SAAS,6BAA6B,CAAC,KAAkB;IACvD,MAAM,EAAC,KAAK,EAAE,QAAQ,EAAE,YAAY,EAAC,GAAG,KAAK,CAAC;IAC9C,MAAM,MAAM,GAAG,EAAE,CAAC,cAAc,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,6BAA6B;IACvE,OAAO,OAAO,CAAC;8BACa,MAAM;;6BAEP,MAAM;;;;;;;;;;;6BAWN,MAAM;;;uCAGI,GAAG,CAAC,QAAQ,CAAC;;;;;6BAKvB,MAAM;;;;;;IAM/B,mBAAmB,CAAC,YAAY,CAAC;IACjC,qBAAqB,CAAC,YAAY,CAAC;;;;;;;;;6BASV,MAAM;;;;;;WAMxB,MAAM;;;;iBAIA,gBAAgB;;iBAEhB,MAAM;;;0CAGmB,GAAG,CACzC,GAAG,KAAK,IAAI,QAAQ,EAAE,CACvB;;;;;6BAK0B,MAAM;;;;;;;;;;;0BAWT,GAAG,CAAC,YAAY,CAAC;;;;;;;;;;;;;;;;;;;;;;gBAsB3B,MAAM;;;;;;;;;;;;;;gBAcN,MAAM;;;;;;;;;;gBAUN,MAAM;;;;;;;;;;gBAUN,MAAM;;;;;;;;;;;gBAWN,MAAM;;;;;cAKR,MAAM;;;;;;;;;;;;;;;WAeT,MAAM;;;;iBAIA,gBAAgB;;;iBAGhB,MAAM;;;0CAGmB,GAAG,CACzC,GAAG,KAAK,IAAI,QAAQ,EAAE,CACvB;;;CAGF,CAAC;AACF,CAAC;AAED,wBAAwB;AACxB,MAAM,CAAC,MAAM,IAAI,GAAG;IAClB,cAAc;IACd,aAAa;IACb,cAAc;IACd,YAAY;IACZ,YAAY;IACZ,mBAAmB;IACnB,cAAc;CACN,CAAC;AAEX,MAAM,UAAU,4BAA4B,CAAC,KAAkB;IAC7D,uEAAuE;IACvE,MAAM,CAAC,KAAK,CAAC,YAAY,CAAC,MAAM,EAAE,sCAAsC,CAAC,CAAC;IAE1E,oEAAoE;IACpE,wEAAwE;IACxE,0BAA0B;IAC1B,MAAM,EAAC,KAAK,EAAE,QAAQ,EAAC,GAAG,KAAK,CAAC;IAChC,MAAM,OAAO,GAAG,MAAM,CAAC,QAAQ,CAAC,CAAC;IACjC,MAAM,MAAM,GAAG,EAAE,CAAC,cAAc,CAAC,KAAK,CAAC,CAAC,CAAC;IAEzC,MAAM,QAAQ,GAAG;QACf,0BAA0B,CAAC,KAAK,CAAC,KAAK,EAAE,KAAK,CAAC,QAAQ,CAAC;QACvD,6BAA6B,CAAC,KAAK,CAAC;KACrC,CAAC;IAEF,iEAAiE;IACjE,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC;uBACD,OAAO,CAAC,GAAG,KAAK,YAAY,CAAC;;iBAEnC,GAAG,CAAC,IAAI,CAAC;sBACJ,MAAM;CAC3B,CAAC,CAAC;IAED,mFAAmF;IACnF,KAAK,MAAM,GAAG,IAAI,IAAI,EAAE,CAAC;QACvB,MAAM,KAAK,GAAG,GAAG,CAAC,WAAW,EAAE,CAAC,OAAO,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;QAClD,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC;6BACG,MAAM,SAAS,KAAK;;;YAGrC,MAAM,iBAAiB,GAAG,CAAC,GAAG,CAAC;;;;uBAIpB,OAAO,CAAC,GAAG,KAAK,IAAI,KAAK,EAAE,CAAC;;iBAElC,GAAG,CAAC,GAAG,CAAC;sBACH,MAAM,SAAS,KAAK;CACzC,CAAC,CAAC;IACD,CAAC;IACD,OAAO,QAAQ,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;AAC3B,CAAC;AAED,wBAAwB;AACxB,MAAM,UAAU,0BAA0B,CACxC,KAAa,EACb,OAAwB;IAExB,MAAM,KAAK,GAAa,EAAE,CAAC;IAC3B,iEAAiE;IACjE,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC;mCACc,EAAE,CAAC,GAAG,KAAK,cAAc,OAAO,EAAE,CAAC;GACnE,CAAC,CAAC;IAEH,mFAAmF;IACnF,KAAK,MAAM,GAAG,IAAI,IAAI,EAAE,CAAC;QACvB,MAAM,KAAK,GAAG,GAAG,CAAC,WAAW,EAAE,CAAC,OAAO,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;QAClD,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC;qCACc,EAAE,CAAC,GAAG,KAAK,IAAI,KAAK,IAAI,OAAO,EAAE,CAAC;KAClE,CAAC,CAAC;IACL,CAAC;IACD,OAAO,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;AACxB,CAAC"}
+{"version":3,"file":"ddl.js","sources":["../../../../../../../../zero-cache/src/services/change-source/pg/schema/ddl.ts"],"sourcesContent":["import {literal as lit} from 'pg-format';\nimport {assert} from '../../../../../../shared/src/asserts.ts';\nimport * as v from '../../../../../../shared/src/valita.ts';\nimport {upstreamSchema, type ShardConfig} from '../../../../types/shards.ts';\nimport {id} from '../../../../types/sql.ts';\nimport {\n  indexDefinitionsQuery,\n  publishedSchema,\n  publishedTableQuery,\n} from './published.ts';\n\n// Sent in the 'version' tag of \"ddlStart\" and \"ddlUpdate\" event messages.\n// This is used to ensure that the message constructed in the upstream\n// Trigger function is compatible with the code processing it in the zero-cache.\n//\n// Increment this when changing the format of the contents of the \"ddl\" events.\n// This will allow old / incompatible code to detect the change and abort.\nexport const PROTOCOL_VERSION = 1;\n\nconst triggerEvent = v.object({\n  context: v.object({query: v.string()}).rest(v.string()),\n});\n\n// All DDL events contain a snapshot of the current tables and indexes that\n// are published / relevant to the shard.\nexport const ddlEventSchema = triggerEvent.extend({\n  version: v.literal(PROTOCOL_VERSION),\n  schema: publishedSchema,\n});\n\n// The `ddlStart` message is computed before every DDL event, regardless of\n// whether the subsequent event affects the shard. Downstream processing should\n// capture the contained schema information in order to determine the schema\n// changes necessary to apply a subsequent `ddlUpdate` message. Note that a\n// `ddlUpdate` message may not follow, as updates determined to be irrelevant\n// to the shard will not result in a message. However, all `ddlUpdate` messages\n// are guaranteed to be preceded by a `ddlStart` message.\nexport const ddlStartEventSchema = ddlEventSchema.extend({\n  type: v.literal('ddlStart'),\n});\n\nexport type DdlStartEvent = v.Infer<typeof ddlStartEventSchema>;\n\n/**\n * The {@link DdlUpdateEvent} contains an updated schema resulting from\n * a particular ddl event. The event type provides information\n * (i.e. constraints) on the difference from the schema of the preceding\n * {@link DdlStartEvent}.\n *\n * Note that in almost all cases (the exception being `CREATE` events),\n * it is possible that there is no relevant difference between the\n * ddl-start schema and the ddl-update schema, as many aspects of the\n * schema (e.g. column constraints) are not relevant to downstream\n * replication.\n */\nexport const ddlUpdateEventSchema = ddlEventSchema.extend({\n  type: v.literal('ddlUpdate'),\n  event: v.object({tag: v.string()}),\n});\n\nexport type DdlUpdateEvent = v.Infer<typeof ddlUpdateEventSchema>;\n\nexport const replicationEventSchema = v.union(\n  ddlStartEventSchema,\n  ddlUpdateEventSchema,\n);\n\nexport type ReplicationEvent = v.Infer<typeof replicationEventSchema>;\n\n// Creates a function that appends `_{shard-num}` to the input and\n// quotes the result to be a valid identifier.\nfunction append(shardNum: number) {\n  return (name: string) => id(name + '_' + String(shardNum));\n}\n\n/**\n * Event trigger functions contain the core logic that are invoked by triggers.\n *\n * Note that although many of these functions can theoretically be parameterized and\n * shared across shards, it is advantageous to keep the functions in each shard\n * isolated from each other in order to avoid the complexity of shared-function\n * versioning.\n *\n * In a sense, shards (and their triggers and functions) should be thought of as\n * execution environments that can be updated at different schedules. If per-shard\n * triggers called into shared functions, we would have to consider versioning the\n * functions when changing their behavior, backwards compatibility, removal of\n * unused versions, etc. (not unlike versioning of npm packages).\n *\n * Instead, we opt for the simplicity and isolation of having each shard\n * completely own (and maintain) the entirety of its trigger/function stack.\n */\nfunction createEventFunctionStatements(shard: ShardConfig) {\n  const {appID, shardNum, publications} = shard;\n  const schema = id(upstreamSchema(shard)); // e.g. \"{APP_ID}_{SHARD_ID}\"\n  return /*sql*/ `\nCREATE SCHEMA IF NOT EXISTS ${schema};\n\nCREATE OR REPLACE FUNCTION ${schema}.get_trigger_context()\nRETURNS record AS $$\nDECLARE\n  result record;\nBEGIN\n  SELECT current_query() AS \"query\" into result;\n  RETURN result;\nEND\n$$ LANGUAGE plpgsql;\n\n\nCREATE OR REPLACE FUNCTION ${schema}.notice_ignore(object_id TEXT)\nRETURNS void AS $$\nBEGIN\n  RAISE NOTICE 'zero(%) ignoring %', ${lit(shardNum)}, object_id;\nEND\n$$ LANGUAGE plpgsql;\n\n\nCREATE OR REPLACE FUNCTION ${schema}.schema_specs()\nRETURNS TEXT AS $$\nDECLARE\n  tables record;\n  indexes record;\nBEGIN\n  ${publishedTableQuery(publications)} INTO tables;\n  ${indexDefinitionsQuery(publications)} INTO indexes;\n  RETURN json_build_object(\n    'tables', tables.tables,\n    'indexes', indexes.indexes\n  );\nEND\n$$ LANGUAGE plpgsql;\n\n\nCREATE OR REPLACE FUNCTION ${schema}.emit_ddl_start()\nRETURNS event_trigger AS $$\nDECLARE\n  schema_specs TEXT;\n  message TEXT;\nBEGIN\n  SELECT ${schema}.schema_specs() INTO schema_specs;\n\n  SELECT json_build_object(\n    'type', 'ddlStart',\n    'version', ${PROTOCOL_VERSION},\n    'schema', schema_specs::json,\n    'context', ${schema}.get_trigger_context()\n  ) INTO message;\n\n  PERFORM pg_logical_emit_message(true, ${lit(\n    `${appID}/${shardNum}`,\n  )}, message);\nEND\n$$ LANGUAGE plpgsql;\n\n\nCREATE OR REPLACE FUNCTION ${schema}.emit_ddl_end(tag TEXT)\nRETURNS void AS $$\nDECLARE\n  publications TEXT[];\n  cmd RECORD;\n  relevant RECORD;\n  deprecated RECORD;\n  schema_specs TEXT;\n  message TEXT;\n  event TEXT;\nBEGIN\n  publications := ARRAY[${lit(publications)}];\n\n  SELECT objid, object_type, object_identity \n    FROM pg_event_trigger_ddl_commands() \n    WHERE object_type IN (\n      'table',\n      'table column',\n      'index',\n      'publication relation',\n      'publication namespace',\n      'schema')\n    LIMIT 1 INTO cmd;\n\n  -- Filter DDL updates that are not relevant to the shard (i.e. publications) when possible.\n\n  IF cmd.object_type = 'table' OR cmd.object_type = 'table column' THEN\n    SELECT ns.nspname AS \"schema\", c.relname AS \"name\" FROM pg_class AS c\n      JOIN pg_namespace AS ns ON c.relnamespace = ns.oid\n      JOIN pg_publication_tables AS pb ON pb.schemaname = ns.nspname AND pb.tablename = c.relname\n      WHERE c.oid = cmd.objid AND pb.pubname = ANY (publications)\n      INTO relevant;\n    IF relevant IS NULL THEN\n      PERFORM ${schema}.notice_ignore(cmd.object_identity);\n      RETURN;\n    END IF;\n\n    cmd.object_type := 'table';  -- normalize the 'table column' target to 'table'\n\n  ELSIF cmd.object_type = 'index' THEN\n    SELECT ns.nspname AS \"schema\", c.relname AS \"name\" FROM pg_class AS c\n      JOIN pg_namespace AS ns ON c.relnamespace = ns.oid\n      JOIN pg_indexes as ind ON ind.schemaname = ns.nspname AND ind.indexname = c.relname\n      JOIN pg_publication_tables AS pb ON pb.schemaname = ns.nspname AND pb.tablename = ind.tablename\n      WHERE c.oid = cmd.objid AND pb.pubname = ANY (publications)\n      INTO relevant;\n    IF relevant IS NULL THEN\n      PERFORM ${schema}.notice_ignore(cmd.object_identity);\n      RETURN;\n    END IF;\n\n  ELSIF cmd.object_type = 'publication relation' THEN\n    SELECT pb.pubname FROM pg_publication_rel AS rel\n      JOIN pg_publication AS pb ON pb.oid = rel.prpubid\n      WHERE rel.oid = cmd.objid AND pb.pubname = ANY (publications) \n      INTO relevant;\n    IF relevant IS NULL THEN\n      PERFORM ${schema}.notice_ignore(cmd.object_identity);\n      RETURN;\n    END IF;\n\n  ELSIF cmd.object_type = 'publication namespace' THEN\n    SELECT pb.pubname FROM pg_publication_namespace AS ns\n      JOIN pg_publication AS pb ON pb.oid = ns.pnpubid\n      WHERE ns.oid = cmd.objid AND pb.pubname = ANY (publications) \n      INTO relevant;\n    IF relevant IS NULL THEN\n      PERFORM ${schema}.notice_ignore(cmd.object_identity);\n      RETURN;\n    END IF;\n\n  ELSIF cmd.object_type = 'schema' THEN\n    SELECT ns.nspname AS \"schema\", c.relname AS \"name\" FROM pg_class AS c\n      JOIN pg_namespace AS ns ON c.relnamespace = ns.oid\n      JOIN pg_publication_tables AS pb ON pb.schemaname = ns.nspname AND pb.tablename = c.relname\n      WHERE ns.oid = cmd.objid AND pb.pubname = ANY (publications)\n      INTO relevant;\n    IF relevant IS NULL THEN\n      PERFORM ${schema}.notice_ignore(cmd.object_identity);\n      RETURN;\n    END IF;\n\n  ELSIF tag LIKE 'CREATE %' THEN\n    PERFORM ${schema}.notice_ignore('noop ' || tag);\n    RETURN;\n  END IF;\n\n  -- Construct and emit the DdlUpdateEvent message.\n\n  -- TODO: Remove backwards-compatibility fields after a few releases.\n  SELECT 'deprecated' as \"schema\", 'deprecated' as \"name\" INTO deprecated;\n\n  SELECT json_build_object(\n    'tag', tag,\n    'table', deprecated,\n    'index', deprecated\n  ) INTO event;\n  \n  SELECT ${schema}.schema_specs() INTO schema_specs;\n\n  SELECT json_build_object(\n    'type', 'ddlUpdate',\n    'version', ${PROTOCOL_VERSION},\n    'schema', schema_specs::json,\n    'event', event::json,\n    'context', ${schema}.get_trigger_context()\n  ) INTO message;\n\n  PERFORM pg_logical_emit_message(true, ${lit(\n    `${appID}/${shardNum}`,\n  )}, message);\nEND\n$$ LANGUAGE plpgsql;\n`;\n}\n\n// Exported for testing.\nexport const TAGS = [\n  'CREATE TABLE',\n  'ALTER TABLE',\n  'CREATE INDEX',\n  'DROP TABLE',\n  'DROP INDEX',\n  'ALTER PUBLICATION',\n  'ALTER SCHEMA',\n] as const;\n\nexport function createEventTriggerStatements(shard: ShardConfig) {\n  // Better to assert here than get a cryptic syntax error from Postgres.\n  assert(shard.publications.length, `shard publications must be non-empty`);\n\n  // Unlike functions, which are namespaced in shard-specific schemas,\n  // EVENT TRIGGER names are in the global namespace and thus must include\n  // the appID and shardNum.\n  const {appID, shardNum} = shard;\n  const sharded = append(shardNum);\n  const schema = id(upstreamSchema(shard));\n\n  const triggers = [\n    dropEventTriggerStatements(shard.appID, shard.shardNum),\n    createEventFunctionStatements(shard),\n  ];\n\n  // A single ddl_command_start trigger covering all relevant tags.\n  triggers.push(/*sql*/ `\nCREATE EVENT TRIGGER ${sharded(`${appID}_ddl_start`)}\n  ON ddl_command_start\n  WHEN TAG IN (${lit(TAGS)})\n  EXECUTE PROCEDURE ${schema}.emit_ddl_start();\n`);\n\n  // A per-tag ddl_command_end trigger that dispatches to ${schema}.emit_ddl_end(tag)\n  for (const tag of TAGS) {\n    const tagID = tag.toLowerCase().replace(' ', '_');\n    triggers.push(/*sql*/ `\nCREATE OR REPLACE FUNCTION ${schema}.emit_${tagID}() \nRETURNS event_trigger AS $$\nBEGIN\n  PERFORM ${schema}.emit_ddl_end(${lit(tag)});\nEND\n$$ LANGUAGE plpgsql;\n\nCREATE EVENT TRIGGER ${sharded(`${appID}_${tagID}`)}\n  ON ddl_command_end\n  WHEN TAG IN (${lit(tag)})\n  EXECUTE PROCEDURE ${schema}.emit_${tagID}();\n`);\n  }\n  return triggers.join('');\n}\n\n// Exported for testing.\nexport function dropEventTriggerStatements(\n  appID: string,\n  shardID: string | number,\n) {\n  const stmts: string[] = [];\n  // A single ddl_command_start trigger covering all relevant tags.\n  stmts.push(/*sql*/ `\n    DROP EVENT TRIGGER IF EXISTS ${id(`${appID}_ddl_start_${shardID}`)};\n  `);\n\n  // A per-tag ddl_command_end trigger that dispatches to ${schema}.emit_ddl_end(tag)\n  for (const tag of TAGS) {\n    const tagID = tag.toLowerCase().replace(' ', '_');\n    stmts.push(/*sql*/ `\n      DROP EVENT TRIGGER IF EXISTS ${id(`${appID}_${tagID}_${shardID}`)};\n    `);\n  }\n  return stmts.join('');\n}\n"],"names":["v.object","v.string","v.literal","v.union","lit"],"mappings":";;;;;;;AAiBO,MAAM,mBAAmB;AAEhC,MAAM,eAAeA,OAAS;AAAA,EAC5B,SAASA,OAAS,EAAC,OAAOC,OAAE,GAAS,EAAE,KAAKA,QAAU;AACxD,CAAC;AAIM,MAAM,iBAAiB,aAAa,OAAO;AAAA,EAChD,SAASC,QAAU,gBAAgB;AAAA,EACnC,QAAQ;AACV,CAAC;AASM,MAAM,sBAAsB,eAAe,OAAO;AAAA,EACvD,MAAMA,QAAU,UAAU;AAC5B,CAAC;AAgBM,MAAM,uBAAuB,eAAe,OAAO;AAAA,EACxD,MAAMA,QAAU,WAAW;AAAA,EAC3B,OAAOF,OAAS,EAAC,KAAKC,OAAE,GAAS;AACnC,CAAC;AAIM,MAAM,yBAAyBE;AAAAA,EACpC;AAAA,EACA;AACF;AAMA,SAAS,OAAO,UAAkB;AAChC,SAAO,CAAC,SAAiB,GAAG,OAAO,MAAM,OAAO,QAAQ,CAAC;AAC3D;AAmBA,SAAS,8BAA8B,OAAoB;AACzD,QAAM,EAAC,OAAO,UAAU,aAAA,IAAgB;AACxC,QAAM,SAAS,GAAG,eAAe,KAAK,CAAC;AACvC;AAAA;AAAA,IAAe;AAAA,8BACa,MAAM;AAAA;AAAA,6BAEP,MAAM;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,6BAWN,MAAM;AAAA;AAAA;AAAA,uCAGIC,UAAI,QAAQ,CAAC;AAAA;AAAA;AAAA;AAAA;AAAA,6BAKvB,MAAM;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,IAM/B,oBAAoB,YAAY,CAAC;AAAA,IACjC,sBAAsB,YAAY,CAAC;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,6BASV,MAAM;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,WAMxB,MAAM;AAAA;AAAA;AAAA;AAAA,iBAIA,gBAAgB;AAAA;AAAA,iBAEhB,MAAM;AAAA;AAAA;AAAA,0CAGmBA;AAAAA,MACtC,GAAG,KAAK,IAAI,QAAQ;AAAA,IAAA,CACrB;AAAA;AAAA;AAAA;AAAA;AAAA,6BAK0B,MAAM;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,0BAWTA,UAAI,YAAY,CAAC;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,gBAsB3B,MAAM;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,gBAcN,MAAM;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,gBAUN,MAAM;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,gBAUN,MAAM;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,gBAWN,MAAM;AAAA;AAAA;AAAA;AAAA;AAAA,cAKR,MAAM;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,WAeT,MAAM;AAAA;AAAA;AAAA;AAAA,iBAIA,gBAAgB;AAAA;AAAA;AAAA,iBAGhB,MAAM;AAAA;AAAA;AAAA,0CAGmBA;AAAAA,MACtC,GAAG,KAAK,IAAI,QAAQ;AAAA,IAAA,CACrB;AAAA;AAAA;AAAA;AAAA;AAIH;AAGO,MAAM,OAAO;AAAA,EAClB;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AACF;AAEO,SAAS,6BAA6B,OAAoB;AAE/D,SAAO,MAAM,aAAa,QAAQ,sCAAsC;AAKxE,QAAM,EAAC,OAAO,SAAA,IAAY;AAC1B,QAAM,UAAU,OAAO,QAAQ;AAC/B,QAAM,SAAS,GAAG,eAAe,KAAK,CAAC;AAEvC,QAAM,WAAW;AAAA,IACf,2BAA2B,MAAM,OAAO,MAAM,QAAQ;AAAA,IACtD,8BAA8B,KAAK;AAAA,EAAA;AAIrC,WAAS;AAAA;AAAA,IAAa;AAAA,uBACD,QAAQ,GAAG,KAAK,YAAY,CAAC;AAAA;AAAA,iBAEnCA,UAAI,IAAI,CAAC;AAAA,sBACJ,MAAM;AAAA;AAAA,EAAA;AAI1B,aAAW,OAAO,MAAM;AACtB,UAAM,QAAQ,IAAI,YAAA,EAAc,QAAQ,KAAK,GAAG;AAChD,aAAS;AAAA;AAAA,MAAa;AAAA,6BACG,MAAM,SAAS,KAAK;AAAA;AAAA;AAAA,YAGrC,MAAM,iBAAiBA,UAAI,GAAG,CAAC;AAAA;AAAA;AAAA;AAAA,uBAIpB,QAAQ,GAAG,KAAK,IAAI,KAAK,EAAE,CAAC;AAAA;AAAA,iBAElCA,UAAI,GAAG,CAAC;AAAA,sBACH,MAAM,SAAS,KAAK;AAAA;AAAA,IAAA;AAAA,EAExC;AACA,SAAO,SAAS,KAAK,EAAE;AACzB;AAGO,SAAS,2BACd,OACA,SACA;AACA,QAAM,QAAkB,CAAA;AAExB,QAAM;AAAA;AAAA,IAAa;AAAA,mCACc,GAAG,GAAG,KAAK,cAAc,OAAO,EAAE,CAAC;AAAA;AAAA,EAAA;AAIpE,aAAW,OAAO,MAAM;AACtB,UAAM,QAAQ,IAAI,YAAA,EAAc,QAAQ,KAAK,GAAG;AAChD,UAAM;AAAA;AAAA,MAAa;AAAA,qCACc,GAAG,GAAG,KAAK,IAAI,KAAK,IAAI,OAAO,EAAE,CAAC;AAAA;AAAA,IAAA;AAAA,EAErE;AACA,SAAO,MAAM,KAAK,EAAE;AACtB;"}

package/out/zero-cache/src/services/change-source/pg/schema/init.js

@@ -1,147 +1,168 @@
 import { assert } from "../../../../../../shared/src/asserts.js";
-import * as v from "../../../../../../shared/src/valita.js";
-import { getVersionHistory, runSchemaMigrations, } from "../../../../db/migration.js";
+import { parse } from "../../../../../../shared/src/valita.js";
+import { runSchemaMigrations, getVersionHistory } from "../../../../db/migration.js";
 import { upstreamSchema } from "../../../../types/shards.js";
 import { AutoResetSignal } from "../../../change-streamer/schema/tables.js";
 import { decommissionShard } from "../decommission.js";
 import { publishedSchema } from "./published.js";
-import { getMutationsTableDefinition, legacyReplicationSlot, metadataPublicationName, setupTablesAndReplication, setupTriggers, } from "./shard.js";
+import { getMutationsTableDefinition, metadataPublicationName, setupTriggers, legacyReplicationSlot, setupTablesAndReplication } from "./shard.js";
 import { id } from "../../../../types/sql.js";
-/**
- * Ensures that a shard is set up for initial sync.
- */
-export async function ensureShardSchema(lc, db, shard) {
-    const initialSetup = {
-        migrateSchema: (lc, tx) => setupTablesAndReplication(lc, tx, shard),
-        minSafeVersion: 1,
-    };
-    await runSchemaMigrations(lc, `upstream-shard-${shard.appID}`, upstreamSchema(shard), db, initialSetup, 
+import { object, string } from "@badrap/valita";
+async function ensureShardSchema(lc, db, shard) {
+  const initialSetup = {
+    migrateSchema: (lc2, tx) => setupTablesAndReplication(lc2, tx, shard),
+    minSafeVersion: 1
+  };
+  await runSchemaMigrations(
+    lc,
+    `upstream-shard-${shard.appID}`,
+    upstreamSchema(shard),
+    db,
+    initialSetup,
     // The incremental migration of any existing replicas will be replaced by
     // the incoming replica being synced, so the replicaVersion here is
     // unnecessary.
-    getIncrementalMigrations(shard, 'obsolete'));
+    getIncrementalMigrations(shard, "obsolete")
+  );
 }
-/**
- * Updates the schema for an existing shard.
- */
-export async function updateShardSchema(lc, db, shard, replicaVersion) {
-    await runSchemaMigrations(lc, `upstream-shard-${shard.appID}`, upstreamSchema(shard), db, {
-        // If the expected existing shard is absent, throw an
-        // AutoResetSignal to backtrack and initial sync.
-        migrateSchema: () => {
-            throw new AutoResetSignal(`upstream shard ${upstreamSchema(shard)} is not initialized`);
-        },
-    }, getIncrementalMigrations(shard, replicaVersion));
-    // The decommission check is run in updateShardSchema so that it happens
-    // after initial sync, and not when the shard schema is initially set up.
-    await decommissionLegacyShard(lc, db, shard);
+async function updateShardSchema(lc, db, shard, replicaVersion) {
+  await runSchemaMigrations(
+    lc,
+    `upstream-shard-${shard.appID}`,
+    upstreamSchema(shard),
+    db,
+    {
+      // If the expected existing shard is absent, throw an
+      // AutoResetSignal to backtrack and initial sync.
+      migrateSchema: () => {
+        throw new AutoResetSignal(
+          `upstream shard ${upstreamSchema(shard)} is not initialized`
+        );
+      }
+    },
+    getIncrementalMigrations(shard, replicaVersion)
+  );
+  await decommissionLegacyShard(lc, db, shard);
 }
 function getIncrementalMigrations(shard, replicaVersion) {
-    const shardConfigTable = `${upstreamSchema(shard)}.shardConfig`;
-    return {
-        4: {
-            migrateSchema: () => {
-                throw new AutoResetSignal('resetting to upgrade shard schema');
-            },
-            minSafeVersion: 3,
-        },
-        // v5 changes the upstream schema organization from "zero_{SHARD_ID}" to
-        // the "{APP_ID}_0". An incremental migration indicates that the previous
-        // SHARD_ID was "0" and the new APP_ID is "zero" (i.e. the default values
-        // for those options). In this case, the upstream format is identical, and
-        // no migration is necessary. However, the version is bumped to v5 to
-        // indicate that it was created with the {APP_ID} configuration and should
-        // not be decommissioned as a legacy shard.
-        5: {},
-        6: {
-            migrateSchema: async (lc, sql) => {
-                assert(replicaVersion, `replicaVersion is always passed for incremental migrations`);
-                await Promise.all([
-                    sql `
+  const shardConfigTable = `${upstreamSchema(shard)}.shardConfig`;
+  return {
+    4: {
+      migrateSchema: () => {
+        throw new AutoResetSignal("resetting to upgrade shard schema");
+      },
+      minSafeVersion: 3
+    },
+    // v5 changes the upstream schema organization from "zero_{SHARD_ID}" to
+    // the "{APP_ID}_0". An incremental migration indicates that the previous
+    // SHARD_ID was "0" and the new APP_ID is "zero" (i.e. the default values
+    // for those options). In this case, the upstream format is identical, and
+    // no migration is necessary. However, the version is bumped to v5 to
+    // indicate that it was created with the {APP_ID} configuration and should
+    // not be decommissioned as a legacy shard.
+    5: {},
+    6: {
+      migrateSchema: async (lc, sql) => {
+        assert(
+          replicaVersion,
+          `replicaVersion is always passed for incremental migrations`
+        );
+        await Promise.all([
+          sql`
           ALTER TABLE ${sql(shardConfigTable)} ADD "replicaVersion" TEXT`,
-                    sql `
-          UPDATE ${sql(shardConfigTable)} SET ${sql({ replicaVersion })}`,
-                ]);
-                lc.info?.(`Recorded replicaVersion ${replicaVersion} in upstream shardConfig`);
-            },
-        },
-        // Updates the DDL event trigger protocol to v2, and adds support for
-        // ALTER SCHEMA x RENAME TO y
-        7: {
-            migrateSchema: async (lc, sql) => {
-                const [{ publications }] = await sql `
+          sql`
+          UPDATE ${sql(shardConfigTable)} SET ${sql({ replicaVersion })}`
+        ]);
+        lc.info?.(
+          `Recorded replicaVersion ${replicaVersion} in upstream shardConfig`
+        );
+      }
+    },
+    // Updates the DDL event trigger protocol to v2, and adds support for
+    // ALTER SCHEMA x RENAME TO y
+    7: {
+      migrateSchema: async (lc, sql) => {
+        const [{ publications }] = await sql`
           SELECT publications FROM ${sql(shardConfigTable)}`;
-                await setupTriggers(lc, sql, { ...shard, publications });
-                lc.info?.(`Upgraded to v2 event triggers`);
-            },
-        },
-        // Adds support for non-disruptive resyncs, which tracks multiple
-        // replicas with different slot names.
-        8: {
-            migrateSchema: async (lc, sql) => {
-                const legacyShardConfigSchema = v.object({
-                    replicaVersion: v.string().nullable(),
-                    initialSchema: publishedSchema.nullable(),
-                });
-                const result = await sql `
+        await setupTriggers(lc, sql, { ...shard, publications });
+        lc.info?.(`Upgraded to v2 event triggers`);
+      }
+    },
+    // Adds support for non-disruptive resyncs, which tracks multiple
+    // replicas with different slot names.
+    8: {
+      migrateSchema: async (lc, sql) => {
+        const legacyShardConfigSchema = object({
+          replicaVersion: string().nullable(),
+          initialSchema: publishedSchema.nullable()
+        });
+        const result = await sql`
           SELECT "replicaVersion", "initialSchema" FROM ${sql(shardConfigTable)}`;
-                assert(result.length === 1);
-                const { replicaVersion, initialSchema } = v.parse(result[0], legacyShardConfigSchema, 'passthrough');
-                await Promise.all([
-                    sql `
+        assert(result.length === 1);
+        const { replicaVersion: replicaVersion2, initialSchema } = parse(
+          result[0],
+          legacyShardConfigSchema,
+          "passthrough"
+        );
+        await Promise.all([
+          sql`
           CREATE TABLE ${sql(upstreamSchema(shard))}.replicas (
             "slot"          TEXT PRIMARY KEY,
             "version"       TEXT NOT NULL,
             "initialSchema" JSON NOT NULL
           );
           `,
-                    sql `
+          sql`
           INSERT INTO ${sql(upstreamSchema(shard))}.replicas ${sql({
-                        slot: legacyReplicationSlot(shard),
-                        version: replicaVersion,
-                        initialSchema,
-                    })}
+            slot: legacyReplicationSlot(shard),
+            version: replicaVersion2,
+            initialSchema
+          })}
           `,
-                    sql `
+          sql`
           ALTER TABLE ${sql(shardConfigTable)} DROP "replicaVersion", DROP "initialSchema"
-          `,
-                ]);
-                lc.info?.(`Upgraded schema to support non-disruptive resyncs`);
-            },
-        },
-        // Fixes field ordering of compound indexes. This incremental migration
-        // only fixes indexes resulting from new schema changes. A full resync is
-        // required to fix existing indexes.
-        9: {
-            migrateSchema: async (lc, sql) => {
-                const [{ publications }] = await sql `
+          `
+        ]);
+        lc.info?.(`Upgraded schema to support non-disruptive resyncs`);
+      }
+    },
+    // Fixes field ordering of compound indexes. This incremental migration
+    // only fixes indexes resulting from new schema changes. A full resync is
+    // required to fix existing indexes.
+    9: {
+      migrateSchema: async (lc, sql) => {
+        const [{ publications }] = await sql`
           SELECT publications FROM ${sql(shardConfigTable)}`;
-                await setupTriggers(lc, sql, { ...shard, publications });
-                lc.info?.(`Upgraded DDL event triggers`);
-            },
-        },
-        // Adds the `mutations` table used to track mutation results.
-        10: {
-            migrateSchema: async (lc, sql) => {
-                await sql.unsafe(/*sql*/ `
+        await setupTriggers(lc, sql, { ...shard, publications });
+        lc.info?.(`Upgraded DDL event triggers`);
+      }
+    },
+    // Adds the `mutations` table used to track mutation results.
+    10: {
+      migrateSchema: async (lc, sql) => {
+        await sql.unsafe(
+          /*sql*/
+          `
           ${getMutationsTableDefinition(upstreamSchema(shard))}
           ALTER PUBLICATION ${id(metadataPublicationName(shard.appID, shard.shardNum))} ADD TABLE ${id(upstreamSchema(shard))}."mutations";
-        `);
-                lc.info?.('Upgraded schema with new mutations table');
-            },
-        },
-    };
+        `
+        );
+        lc.info?.("Upgraded schema with new mutations table");
+      }
+    }
+  };
 }
-export async function decommissionLegacyShard(lc, db, shard) {
-    if (shard.appID !== 'zero') {
-        // When migration from non-default shard ids, e.g. "zero_prod" => "prod_0",
-        // clean up the old "zero_prod" shard if it is pre-v5. Note that the v5
-        // check is important to guard against cleaning up a **new** "zero_0" app
-        // that coexists with the current App (with app-id === "0").
-        const versionHistory = await getVersionHistory(db, `zero_${shard.appID}`);
-        if (versionHistory !== null && versionHistory.schemaVersion < 5) {
-            await decommissionShard(lc, db, 'zero', shard.appID);
-        }
+async function decommissionLegacyShard(lc, db, shard) {
+  if (shard.appID !== "zero") {
+    const versionHistory = await getVersionHistory(db, `zero_${shard.appID}`);
+    if (versionHistory !== null && versionHistory.schemaVersion < 5) {
+      await decommissionShard(lc, db, "zero", shard.appID);
     }
+  }
 }
-//# sourceMappingURL=init.js.map
+export {
+  decommissionLegacyShard,
+  ensureShardSchema,
+  updateShardSchema
+};
+//# sourceMappingURL=init.js.map