@rocicorp/zero 1.6.0-canary.1 → 1.6.0-canary.11

package/out/zero-cache/src/services/change-source/pg/initial-sync.js.map

@@ -1 +1 @@
-{"version":3,"file":"initial-sync.js","names":[],"sources":["../../../../../../../zero-cache/src/services/change-source/pg/initial-sync.ts"],"sourcesContent":["import {mkdtemp, rm} from 'node:fs/promises';\nimport {platform, tmpdir} from 'node:os';\nimport {join} from 'node:path';\nimport {Writable} from 'node:stream';\nimport {pipeline} from 'node:stream/promises';\nimport type {LogContext} from '@rocicorp/logger';\nimport {resolver} from '@rocicorp/resolver';\nimport {assert} from '../../../../../shared/src/asserts.ts';\nimport type {JSONObject} from '../../../../../shared/src/bigint-json.ts';\nimport {must} from '../../../../../shared/src/must.ts';\nimport {equals} from '../../../../../shared/src/set-utils.ts';\nimport type {DownloadStatus} from '../../../../../zero-events/src/status.ts';\nimport {Database} from '../../../../../zqlite/src/db.ts';\nimport {\n  createLiteIndexStatement,\n  createLiteTableStatement,\n} from '../../../db/create.ts';\nimport {\n  computeZqlSpecs,\n  listIndexes,\n  listTables,\n} from '../../../db/lite-tables.ts';\nimport * as Mode from '../../../db/mode-enum.ts';\nimport {\n  BinaryCopyParser,\n  hasBinaryDecoder,\n  makeBinaryDecoder,\n  textCastDecoder,\n} from '../../../db/pg-copy-binary.ts';\nimport {TsvParser} from '../../../db/pg-copy.ts';\nimport {\n  mapPostgresToLite,\n  mapPostgresToLiteIndex,\n} from '../../../db/pg-to-lite.ts';\nimport {getTypeParsers} from '../../../db/pg-type-parser.ts';\nimport {runTx} from '../../../db/run-transaction.ts';\nimport type {IndexSpec, PublishedTableSpec} from '../../../db/specs.ts';\nimport {importSnapshot, TransactionPool} from '../../../db/transaction-pool.ts';\nimport {\n  JSON_STRINGIFIED,\n  liteValue,\n  type LiteValueType,\n} from '../../../types/lite.ts';\nimport {liteTableName} from '../../../types/names.ts';\nimport {PG_15, PG_17} from '../../../types/pg-versions.ts';\nimport {\n  pgClient,\n  type PostgresDB,\n  type PostgresTransaction,\n  type PostgresValueType,\n} from '../../../types/pg.ts';\nimport {CpuProfiler} from '../../../types/profiler.ts';\nimport type {ShardConfig} from '../../../types/shards.ts';\nimport {ALLOWED_APP_ID_CHARACTERS} from '../../../types/shards.ts';\nimport {id} from '../../../types/sql.ts';\nimport {ReplicationStatusPublisher} from '../../replicator/replication-status.ts';\nimport {ColumnMetadataStore} from '../../replicator/schema/column-metadata.ts';\nimport {initReplicationState} from '../../replicator/schema/replication-state.ts';\nimport {toStateVersionString} from './lsn.ts';\nimport {createReplicaAndSlot} from './replication-slots.ts';\nimport {ensureShardSchema} from './schema/init.ts';\nimport {getPublicationInfo} from './schema/published.ts';\nimport {\n  dropShard,\n  getInternalShardConfig,\n  initReplica,\n  validatePublications,\n} from './schema/shard.ts';\n\nexport type InitialSyncOptions = {\n  tableCopyWorkers: number;\n  profileCopy?: boolean | undefined;\n  textCopy?: boolean | undefined;\n  replicationSlotFailover?: boolean | undefined;\n  /**\n   * When set, run initial sync in \"shadow\" mode for verification: skip all\n   * upstream mutations (no replication slot, no addReplica, no dropShard, no\n   * slot drop on failure), suppress status events, and optionally sample\n   * rows from each table via TABLESAMPLE BERNOULLI + LIMIT. The caller is\n   * responsible for providing (and discarding) a throwaway SQLite `tx`.\n   */\n  shadow?:\n    | {\n        /** 0 < rate <= 1. When 1, no TABLESAMPLE clause is added. */\n        sampleRate: number;\n        /**\n         * LIMIT N cap appended after TABLESAMPLE. Required: shadow sync is\n         * for verification only, so every run must commit to a row budget.\n         */\n        maxRowsPerTable: number;\n      }\n    | undefined;\n};\n\n/** Server context to store with the initial sync metadata for debugging. */\nexport type ServerContext = JSONObject;\n\nexport async function initialSync(\n  lc: LogContext,\n  shard: ShardConfig,\n  tx: Database,\n  upstreamURI: string,\n  syncOptions: InitialSyncOptions,\n  context: ServerContext,\n) {\n  if (!ALLOWED_APP_ID_CHARACTERS.test(shard.appID)) {\n    throw new Error(\n      'The App ID may only consist of lower-case letters, numbers, and the underscore character',\n    );\n  }\n  const {\n    tableCopyWorkers,\n    profileCopy,\n    textCopy = false,\n    replicationSlotFailover = false,\n    shadow,\n  } = syncOptions;\n  const copyProfiler = profileCopy ? await CpuProfiler.connect() : null;\n  const sql = pgClient(lc, upstreamURI, 'initial-sync');\n  // Replication session is only needed to create a replication slot in the\n  // real path. In shadow mode we export a snapshot on a normal connection\n  // instead, so no replication session is opened.\n  const replicationSession = shadow\n    ? undefined\n    : pgClient(lc, upstreamURI, 'initial-sync-replication-session', {\n        ['fetch_types']: false, // Necessary for the streaming protocol\n        connection: {replication: 'database'}, // https://www.postgresql.org/docs/current/protocol-replication.html\n      });\n\n  const replicaID = Date.now().toString();\n  let slotName: string | undefined; // undefined === shadow\n  const statusPublisher = ReplicationStatusPublisher.forRunningTransaction(\n    tx,\n    shadow ? async () => {} : undefined,\n  ).publish(lc, 'Initializing');\n  let releaseShadowSnapshot: (() => Promise<void>) | undefined;\n  try {\n    const pgVersion = await checkUpstreamConfig(sql);\n\n    // In shadow mode we assume the shard is already initialized and just\n    // read back the existing publications. `ensurePublishedTables` would\n    // otherwise run DDL and potentially call `dropShard`, which must never\n    // happen during a shadow run.\n    const {publications} = shadow\n      ? await getInternalShardConfig(sql, shard)\n      : await ensurePublishedTables(lc, sql, shard);\n    lc.info?.(`Upstream is setup with publications [${publications}]`);\n\n    const {database, host} = sql.options;\n    lc.info?.(\n      shadow\n        ? `acquiring exported snapshot on ${database}@${host} (shadow mode)`\n        : `opening replication session to ${database}@${host}`,\n    );\n\n    let snapshot: string;\n    let lsn: string;\n\n    if (shadow) {\n      const acquired = await acquireExportedSnapshotForShadowSync(\n        lc,\n        upstreamURI,\n      );\n      snapshot = acquired.snapshot;\n      lsn = acquired.lsn;\n      releaseShadowSnapshot = acquired.release;\n    } else {\n      const slot = await createReplicaAndSlot(\n        lc,\n        sql,\n        must(replicationSession),\n        shard,\n        replicaID,\n        replicationSlotFailover && pgVersion >= PG_17,\n      );\n      snapshot = slot.snapshot_name;\n      lsn = slot.consistent_point;\n      slotName = slot.slot_name;\n    }\n\n    const initialVersion = toStateVersionString(lsn);\n\n    initReplicationState(tx, publications, initialVersion, context);\n\n    // Run up to MAX_WORKERS to copy of tables at the replication slot's snapshot.\n    const start = performance.now();\n    // Retrieve the published schema at the consistent_point.\n    const published = await runTx(\n      sql,\n      async tx => {\n        await tx.unsafe(/* sql*/ `SET TRANSACTION SNAPSHOT '${snapshot}'`);\n        return getPublicationInfo(tx, publications);\n      },\n      {mode: Mode.READONLY},\n    );\n    // Note: If this throws, initial-sync is aborted.\n    validatePublications(lc, published);\n\n    // Now that tables have been validated, kick off the copiers.\n    const {tables, indexes} = published;\n    const numTables = tables.length;\n    if (platform() === 'win32' && tableCopyWorkers < numTables) {\n      lc.warn?.(\n        `Increasing the number of copy workers from ${tableCopyWorkers} to ` +\n          `${numTables} to work around a Node/Postgres connection bug`,\n      );\n    }\n    const numWorkers =\n      platform() === 'win32'\n        ? numTables\n        : Math.min(tableCopyWorkers, numTables);\n\n    const copyPool = pgClient(lc, upstreamURI, 'initial-sync-copy-worker', {\n      max: numWorkers,\n      ['max_lifetime']: 120 * 60, // set a long (2h) limit for COPY streaming\n    });\n    const copiers = startTableCopyWorkers(\n      lc,\n      copyPool,\n      snapshot,\n      numWorkers,\n      numTables,\n    );\n    try {\n      createLiteTables(tx, tables, initialVersion);\n      const sampleRate = shadow?.sampleRate;\n      const maxRowsPerTable = shadow?.maxRowsPerTable;\n      const downloads = await Promise.all(\n        tables.map(spec =>\n          copiers.processReadTask((db, lc) =>\n            getInitialDownloadState(lc, db, spec, shadow !== undefined),\n          ),\n        ),\n      );\n      statusPublisher.publish(\n        lc,\n        'Initializing',\n        `Copying ${numTables} upstream tables at version ${initialVersion}`,\n        5000,\n        () => ({downloadStatus: downloads.map(({status}) => status)}),\n      );\n\n      void copyProfiler?.start();\n      const rowCounts = await Promise.all(\n        downloads.map(table =>\n          copiers.processReadTask((db, lc) =>\n            copy(\n              lc,\n              table,\n              copyPool,\n              db,\n              tx,\n              textCopy,\n              sampleRate,\n              maxRowsPerTable,\n            ),\n          ),\n        ),\n      );\n      void copyProfiler?.stopAndDispose(lc, 'initial-copy');\n      copiers.setDone();\n\n      const total = rowCounts.reduce(\n        (acc, curr) => ({\n          rows: acc.rows + curr.rows,\n          flushTime: acc.flushTime + curr.flushTime,\n        }),\n        {rows: 0, flushTime: 0},\n      );\n\n      statusPublisher.publish(\n        lc,\n        'Indexing',\n        `Creating ${indexes.length} indexes`,\n        5000,\n      );\n      const indexStart = performance.now();\n      createLiteIndices(tx, indexes);\n      const index = performance.now() - indexStart;\n      lc.info?.(`Created indexes (${index.toFixed(3)} ms)`);\n\n      if (slotName && replicaID) {\n        await initReplica(sql, shard, replicaID, published, context);\n      } else {\n        assert(shadow, 'expected to be in shadow sync if there is no slotName');\n        const rowsByTable = new Map<string, number>();\n        for (let i = 0; i < downloads.length; i++) {\n          rowsByTable.set(downloads[i].status.table, rowCounts[i].rows);\n        }\n        verifyShadowReplica(lc, tx, published, rowsByTable);\n      }\n\n      const elapsed = performance.now() - start;\n      lc.info?.(\n        `Synced ${total.rows.toLocaleString()} rows of ${numTables} tables in ${publications} up to ${lsn} ` +\n          `(flush: ${total.flushTime.toFixed(3)}, index: ${index.toFixed(3)}, total: ${elapsed.toFixed(3)} ms)`,\n      );\n    } finally {\n      // All meaningful errors are handled at the processReadTask() call site.\n      void copyPool.end().catch(e => lc.warn?.(`Error closing copyPool`, e));\n    }\n  } catch (e) {\n    if (slotName) {\n      // If initial-sync did not succeed, make a best effort to drop the\n      // orphaned replication slot to avoid running out of slots in\n      // pathological cases that result in repeated failures.\n      lc.warn?.(`dropping replication slot ${slotName}`, e);\n      await sql`\n        SELECT pg_drop_replication_slot(slot_name) FROM pg_replication_slots\n          WHERE slot_name = ${slotName};\n      `.catch(e => lc.warn?.(`Unable to drop replication slot ${slotName}`, e));\n    }\n    await statusPublisher.publishAndThrowError(lc, 'Initializing', e);\n  } finally {\n    statusPublisher.stop();\n    if (releaseShadowSnapshot) {\n      await releaseShadowSnapshot().catch(e =>\n        lc.warn?.(`Error releasing shadow snapshot`, e),\n      );\n    }\n    if (replicationSession) {\n      await replicationSession.end();\n    }\n    await sql.end();\n  }\n}\n\nexport type ShadowSyncOptions = {\n  sampleRate: number;\n  maxRowsPerTable: number;\n  /**\n   * Parent directory for the throwaway SQLite replica. Defaults to the OS\n   * tmpdir. Primarily for tests that need to isolate the scratch directory.\n   */\n  parentDir?: string | undefined;\n};\n\n/**\n * Exercises the initial-sync code path against a sample of rows from every\n * published table, writing into a throwaway SQLite database that is deleted\n * when the run ends. Produces zero upstream mutations: no replication slot,\n * no `addReplica`, no `dropShard`, no status events.\n *\n * Intended to be invoked periodically so that if a customer ever needs a\n * full reset, we have recent confidence that `initialSync` still works.\n * The shard must already be initialized upstream.\n */\nexport async function shadowInitialSync(\n  lc: LogContext,\n  shard: ShardConfig,\n  upstreamURI: string,\n  shadow: ShadowSyncOptions,\n  context: ServerContext,\n  syncOptions?: Pick<InitialSyncOptions, 'textCopy'>,\n): Promise<void> {\n  const dir = await mkdtemp(\n    join(shadow.parentDir ?? tmpdir(), 'zero-shadow-sync-'),\n  );\n  const dbPath = join(dir, 'shadow-replica.db');\n  const db = new Database(lc, dbPath);\n  try {\n    await initialSync(\n      lc,\n      shard,\n      db,\n      upstreamURI,\n      {\n        // Shadow sync copies small samples, so one worker is plenty —\n        // no reason to burn additional upstream connections.\n        tableCopyWorkers: 1,\n        textCopy: syncOptions?.textCopy,\n        shadow,\n      },\n      context,\n    );\n  } finally {\n    try {\n      db.close();\n    } catch (e) {\n      lc.warn?.(`Error closing shadow replica db`, e);\n    }\n    await rm(dir, {recursive: true, force: true}).catch(e =>\n      lc.warn?.(`Error cleaning up shadow replica dir ${dir}`, e),\n    );\n  }\n}\n\nasync function checkUpstreamConfig(sql: PostgresDB) {\n  const {walLevel, version} = (\n    await sql<{walLevel: string; version: number}[]>`\n      SELECT current_setting('wal_level') as \"walLevel\", \n             current_setting('server_version_num') as \"version\";\n  `\n  )[0];\n\n  if (walLevel !== 'logical') {\n    throw new Error(\n      `Postgres must be configured with \"wal_level = logical\" (currently: \"${walLevel})`,\n    );\n  }\n  if (version < PG_15) {\n    throw new Error(\n      `Must be running Postgres 15 or higher (currently: \"${version}\")`,\n    );\n  }\n  return version;\n}\n\nasync function ensurePublishedTables(\n  lc: LogContext,\n  sql: PostgresDB,\n  shard: ShardConfig,\n  validate = true,\n): Promise<{publications: string[]}> {\n  const {database, host} = sql.options;\n  lc.info?.(`Ensuring upstream PUBLICATION on ${database}@${host}`);\n\n  await ensureShardSchema(lc, sql, shard);\n  const {publications} = await getInternalShardConfig(sql, shard);\n\n  if (validate) {\n    let valid = false;\n    const nonInternalPublications = publications.filter(\n      p => !p.startsWith('_'),\n    );\n    const exists = await sql`\n      SELECT pubname FROM pg_publication WHERE pubname IN ${sql(publications)}\n      `.values();\n    if (exists.length !== publications.length) {\n      lc.warn?.(\n        `some configured publications [${publications}] are missing: ` +\n          `[${exists.flat()}]. resyncing`,\n      );\n    } else if (\n      !equals(new Set(shard.publications), new Set(nonInternalPublications))\n    ) {\n      lc.warn?.(\n        `requested publications [${shard.publications}] differ from previous` +\n          `publications [${nonInternalPublications}]. resyncing`,\n      );\n    } else {\n      valid = true;\n    }\n    if (!valid) {\n      await sql.unsafe(dropShard(shard.appID, shard.shardNum));\n      return ensurePublishedTables(lc, sql, shard, false);\n    }\n  }\n  return {publications};\n}\n\nfunction startTableCopyWorkers(\n  lc: LogContext,\n  db: PostgresDB,\n  snapshot: string,\n  numWorkers: number,\n  numTables: number,\n): TransactionPool {\n  const {init} = importSnapshot(snapshot);\n  const tableCopiers = new TransactionPool(lc, {\n    mode: Mode.READONLY,\n    init,\n    initialWorkers: numWorkers,\n  });\n  tableCopiers.run(db);\n\n  lc.info?.(`Started ${numWorkers} workers to copy ${numTables} tables`);\n\n  if (parseInt(process.versions.node) < 22) {\n    lc.warn?.(\n      `\\n\\n\\n` +\n        `Older versions of Node have a bug that results in an unresponsive\\n` +\n        `Postgres connection after running certain combinations of COPY commands.\\n` +\n        `If initial sync hangs, run zero-cache with Node v22+. This has the additional\\n` +\n        `benefit of being consistent with the Node version run in the production container image.` +\n        `\\n\\n\\n`,\n    );\n  }\n  return tableCopiers;\n}\n\n/**\n * Shadow-mode alternative to `createReplicationSlot`: opens a dedicated\n * READ ONLY REPEATABLE READ transaction on a normal connection, exports the\n * snapshot and captures the current WAL LSN, then holds the transaction\n * open until `release()` is called. The held transaction keeps the snapshot\n * importable by the table-copy workers for the duration of the COPY phase.\n *\n * Idle-in-transaction timeout is disabled locally so the exporter doesn't\n * get killed while workers are still importing.\n */\nasync function acquireExportedSnapshotForShadowSync(\n  lc: LogContext,\n  upstreamURI: string,\n): Promise<{\n  snapshot: string;\n  lsn: string;\n  release: () => Promise<void>;\n}> {\n  const holder = pgClient(lc, upstreamURI, 'shadow-initial-sync-snapshot', {\n    max: 1,\n  });\n  const ready = resolver<{snapshot: string; lsn: string}>();\n  const release = resolver<void>();\n  const held = holder\n    .begin(Mode.READONLY, async tx => {\n      await tx`SET LOCAL idle_in_transaction_session_timeout = 0`.execute();\n      const [row] = await tx<{snapshot: string; lsn: string}[]>`\n        SELECT pg_export_snapshot() AS snapshot,\n               pg_current_wal_lsn()::text AS lsn`;\n      ready.resolve(row);\n      await release.promise;\n    })\n    .catch(e => ready.reject(e));\n\n  let snapshot: string;\n  let lsn: string;\n  try {\n    ({snapshot, lsn} = await ready.promise);\n  } catch (e) {\n    await holder\n      .end()\n      .catch(err =>\n        lc.warn?.(`Error ending shadow snapshot holder after failure`, err),\n      );\n    throw e;\n  }\n  lc.info?.(\n    `Exported snapshot ${snapshot} at LSN ${lsn} (shadow initial sync)`,\n  );\n  return {\n    snapshot,\n    lsn,\n    release: async () => {\n      release.resolve();\n      try {\n        await held;\n      } catch (e) {\n        lc.warn?.(`snapshot holder transaction ended with error`, e);\n      }\n      await holder.end();\n    },\n  };\n}\n\nfunction createLiteTables(\n  tx: Database,\n  tables: PublishedTableSpec[],\n  initialVersion: string,\n) {\n  // TODO: Figure out how to reuse the ChangeProcessor here to avoid\n  //       duplicating the ColumnMetadata logic.\n  const columnMetadata = must(ColumnMetadataStore.getInstance(tx));\n  for (const t of tables) {\n    tx.exec(createLiteTableStatement(mapPostgresToLite(t, initialVersion)));\n    const tableName = liteTableName(t);\n    for (const [colName, colSpec] of Object.entries(t.columns)) {\n      columnMetadata.insert(tableName, colName, colSpec);\n    }\n  }\n}\n\nfunction createLiteIndices(tx: Database, indices: IndexSpec[]) {\n  for (const index of indices) {\n    tx.exec(createLiteIndexStatement(mapPostgresToLiteIndex(index)));\n  }\n}\n\n/**\n * Runs structural assertions over a just-synced replica and throws if any\n * fail. Only called in shadow mode — a successful return means the replica\n * is schema-complete, row-count consistent, ZQL-queryable, and its column\n * metadata is in sync with its lite schema.\n *\n * Exported for testing.\n */\nexport function verifyShadowReplica(\n  lc: LogContext,\n  db: Database,\n  published: {tables: PublishedTableSpec[]; indexes: IndexSpec[]},\n  rowsByTable: ReadonlyMap<string, number>,\n): void {\n  const issues: string[] = [];\n  let columnsChecked = 0;\n  let rowsChecked = 0;\n\n  // 1. Schema completeness: every published table exists in the replica\n  //    with at least the expected column set.\n  const liteTables = listTables(db);\n  const liteTableByName = new Map(liteTables.map(t => [t.name, t]));\n  for (const pt of published.tables) {\n    const name = liteTableName(pt);\n    const lite = liteTableByName.get(name);\n    if (!lite) {\n      issues.push(`missing table in replica: ${name}`);\n      continue;\n    }\n    for (const col of Object.keys(pt.columns)) {\n      columnsChecked++;\n      if (!(col in lite.columns)) {\n        issues.push(`column missing in replica table ${name}: ${col}`);\n      }\n    }\n  }\n\n  //    Every published index exists in the replica.\n  const liteIndexNames = new Set(listIndexes(db).map(i => i.name));\n  for (const ix of published.indexes) {\n    const mapped = mapPostgresToLiteIndex(ix);\n    if (!liteIndexNames.has(mapped.name)) {\n      issues.push(\n        `missing index in replica: ${mapped.name} on ${mapped.tableName}`,\n      );\n    }\n  }\n\n  // 2. Row counts: SQLite COUNT(*) matches the in-memory copy counter.\n  for (const [table, expected] of rowsByTable) {\n    try {\n      const [row] = db\n        .prepare(`SELECT COUNT(*) as count FROM \"${table}\"`)\n        .all<{count: number}>();\n      if (row.count !== expected) {\n        issues.push(\n          `row count mismatch for table ${table}: ` +\n            `copy counter reported ${expected}, replica has ${row.count}`,\n        );\n      } else {\n        rowsChecked += row.count;\n      }\n    } catch (e) {\n      issues.push(`could not count rows in table ${table}: ${String(e)}`);\n    }\n  }\n\n  // 3. ZQL-queryability: every published table survives computeZqlSpecs's\n  //    filtering (primary-key candidate, ZQL-typed columns, etc.).\n  const tableSpecs = computeZqlSpecs(lc, db, {\n    includeBackfillingColumns: false,\n  });\n  for (const pt of published.tables) {\n    const name = liteTableName(pt);\n    if (!tableSpecs.has(name)) {\n      issues.push(\n        `table not queryable via ZQL (dropped by computeZqlSpecs): ${name}`,\n      );\n    }\n  }\n\n  // 4. Column metadata: every published column has a _zero.column_metadata row.\n  const meta = must(ColumnMetadataStore.getInstance(db));\n  for (const pt of published.tables) {\n    const name = liteTableName(pt);\n    const rows = meta.getTable(name);\n    for (const col of Object.keys(pt.columns)) {\n      if (!rows.has(col)) {\n        issues.push(`missing column_metadata row for ${name}.${col}`);\n      }\n    }\n  }\n\n  if (issues.length) {\n    throw new Error(\n      `Shadow replica verification failed (${issues.length} issue(s)):\\n` +\n        issues.map(i => `  - ${i}`).join('\\n'),\n    );\n  }\n\n  lc.info?.(\n    `Shadow replica verification passed: ` +\n      `${published.tables.length} tables, ` +\n      `${published.indexes.length} indexes, ` +\n      `${columnsChecked} columns, ` +\n      `${rowsChecked.toLocaleString()} rows`,\n  );\n}\n\n// Verified empirically that batches of 50 seem to be the sweet spot,\n// similar to the report in https://sqlite.org/forum/forumpost/8878a512d3652655\n//\n// Exported for testing.\nexport const INSERT_BATCH_SIZE = 50;\n\nconst MB = 1024 * 1024;\nconst MAX_BUFFERED_ROWS = 10_000;\nconst BUFFERED_SIZE_THRESHOLD = 8 * MB;\n\nexport type DownloadStatements = {\n  select: string;\n  getTotalRows: string;\n  getTotalBytes: string;\n};\n\n/**\n * Produces ` TABLESAMPLE BERNOULLI(n)` when `sampleRate` is < 1, else `''`.\n * Row-level Bernoulli sampling is used (rather than SYSTEM) because it\n * produces a more uniform sample and, unlike SYSTEM, still returns rows\n * for small tables at low rates.\n */\nfunction tableSampleClause(sampleRate: number | undefined): string {\n  if (sampleRate === undefined || sampleRate >= 1) {\n    return '';\n  }\n  // Round away float noise (e.g. 0.3 * 100 = 30.000000000000004) while still\n  // preserving sub-integer rates like 0.001 (= 0.1%).\n  const pct = parseFloat((sampleRate * 100).toFixed(6));\n  return /*sql*/ ` TABLESAMPLE BERNOULLI(${pct})`;\n}\n\nfunction limitClause(maxRowsPerTable: number | undefined): string {\n  return maxRowsPerTable !== undefined\n    ? /*sql*/ ` LIMIT ${maxRowsPerTable}`\n    : '';\n}\n\n/**\n * Returns the SELECT column expressions for binary COPY, casting columns\n * without a known binary decoder to `::text`.\n */\nexport function makeBinarySelectExprs(\n  table: PublishedTableSpec,\n  cols: string[],\n): string[] {\n  return cols.map(col => {\n    const spec = table.columns[col];\n    return hasBinaryDecoder(spec) ? id(col) : `${id(col)}::text`;\n  });\n}\n\nexport function makeDownloadStatements(\n  table: PublishedTableSpec,\n  cols: string[],\n  sampleRate?: number | undefined,\n  maxRowsPerTable?: number | undefined,\n  selectExprs?: string[] | undefined,\n): DownloadStatements {\n  const filterConditions = Object.values(table.publications)\n    .map(({rowFilter}) => rowFilter)\n    .filter(f => !!f); // remove nulls\n  const where =\n    filterConditions.length === 0\n      ? ''\n      : /*sql*/ `WHERE ${filterConditions.join(' OR ')}`;\n  const sample = tableSampleClause(sampleRate);\n  const limit = limitClause(maxRowsPerTable);\n  const fromTable = /*sql*/ `FROM ${id(table.schema)}.${id(table.name)}${sample} ${where}`;\n  const select = /*sql*/ `SELECT ${(selectExprs ?? cols.map(id)).join(',')} ${fromTable}${limit}`;\n  if (limit) {\n    // With LIMIT, wrap counts/sums in a subquery so they reflect the\n    // capped rowset rather than the full (sampled) table.\n    const bytesExpr = cols\n      .map(col => `COALESCE(pg_column_size(${id(col)}), 0)`)\n      .join(' + ');\n    return {\n      select,\n      getTotalRows: /*sql*/ `SELECT COUNT(*)::bigint AS \"totalRows\" FROM (SELECT 1 AS _ ${fromTable}${limit}) s`,\n      getTotalBytes: /*sql*/ `SELECT COALESCE(SUM(b), 0)::bigint AS \"totalBytes\" FROM (SELECT (${bytesExpr}) AS b ${fromTable}${limit}) s`,\n    };\n  }\n  const totalBytes = `(${cols.map(col => `SUM(COALESCE(pg_column_size(${id(col)}), 0))`).join(' + ')})`;\n  return {\n    select,\n    getTotalRows: /*sql*/ `SELECT COUNT(*) AS \"totalRows\" ${fromTable}`,\n    getTotalBytes: /*sql*/ `SELECT ${totalBytes} AS \"totalBytes\" ${fromTable}`,\n  };\n}\n\ntype DownloadState = {\n  spec: PublishedTableSpec;\n  status: DownloadStatus;\n};\n\n// Exported for testing.\nexport async function getInitialDownloadState(\n  lc: LogContext,\n  sql: PostgresDB,\n  spec: PublishedTableSpec,\n  skipTotals: boolean,\n): Promise<DownloadState> {\n  const start = performance.now();\n  const table = liteTableName(spec);\n  const columns = Object.keys(spec.columns);\n  if (skipTotals) {\n    // Shadow sync suppresses status events, so the pg_class\n    // estimates would be computed and thrown away.\n    return {\n      spec,\n      status: {table, columns, rows: 0, totalRows: 0, totalBytes: 0},\n    };\n  }\n  // Use pg_class estimates instead of expensive COUNT(*) and\n  // SUM(pg_column_size(...)) full table scans. The exact values are only\n  // used for progress reporting, so estimates are sufficient.\n  const qualifiedName = `${id(spec.schema)}.${id(spec.name)}`;\n  const estimateResult = await sql<\n    {totalRows: number; totalBytes: number}[]\n  >`SELECT GREATEST(reltuples, 0)::float8 AS \"totalRows\",\n         pg_table_size(oid)::float8 AS \"totalBytes\"\n    FROM pg_class\n    WHERE oid = ${qualifiedName}::regclass`;\n\n  const {totalRows, totalBytes} = estimateResult[0] ?? {\n    totalRows: 0,\n    totalBytes: 0,\n  };\n\n  const state: DownloadState = {\n    spec,\n    status: {\n      table,\n      columns,\n      rows: 0,\n      totalRows,\n      totalBytes,\n    },\n  };\n  const elapsed = (performance.now() - start).toFixed(3);\n  lc.info?.(`Computed initial download state for ${table} (${elapsed} ms)`, {\n    state: state.status,\n  });\n  return state;\n}\n\nfunction copy(\n  lc: LogContext,\n  {spec: table, status}: DownloadState,\n  dbClient: PostgresDB,\n  from: PostgresTransaction,\n  to: Database,\n  textCopy: boolean,\n  sampleRate?: number | undefined,\n  maxRowsPerTable?: number | undefined,\n) {\n  if (textCopy) {\n    return copyText(\n      lc,\n      table,\n      status,\n      dbClient,\n      from,\n      to,\n      sampleRate,\n      maxRowsPerTable,\n    );\n  }\n  return copyBinary(lc, table, status, from, to, sampleRate, maxRowsPerTable);\n}\n\nasync function copyBinary(\n  lc: LogContext,\n  table: PublishedTableSpec,\n  status: DownloadStatus,\n  from: PostgresTransaction,\n  to: Database,\n  sampleRate?: number | undefined,\n  maxRowsPerTable?: number | undefined,\n) {\n  const start = performance.now();\n  let flushTime = 0;\n\n  const tableName = liteTableName(table);\n  const orderedColumns = Object.entries(table.columns);\n\n  const columnNames = orderedColumns.map(([c]) => c);\n  const columnSpecs = orderedColumns.map(([_name, spec]) => spec);\n  const insertColumnList = columnNames.map(c => id(c)).join(',');\n\n  const valuesSql =\n    columnNames.length > 0 ? `(${'?,'.repeat(columnNames.length - 1)}?)` : '()';\n  const insertSql = /*sql*/ `\n    INSERT INTO \"${tableName}\" (${insertColumnList}) VALUES ${valuesSql}`;\n  const insertStmt = to.prepare(insertSql);\n  const insertBatchStmt = to.prepare(\n    insertSql + `,${valuesSql}`.repeat(INSERT_BATCH_SIZE - 1),\n  );\n\n  // Build SELECT with ::text casts for columns without a known binary decoder.\n  const select = makeDownloadStatements(\n    table,\n    columnNames,\n    sampleRate,\n    maxRowsPerTable,\n    makeBinarySelectExprs(table, columnNames),\n  ).select;\n\n  const decoders = orderedColumns.map(([, spec]) =>\n    hasBinaryDecoder(spec) ? makeBinaryDecoder(spec) : textCastDecoder,\n  );\n\n  const valuesPerRow = columnSpecs.length;\n  const valuesPerBatch = valuesPerRow * INSERT_BATCH_SIZE;\n\n  const pendingValues: LiteValueType[] = Array.from({\n    length: MAX_BUFFERED_ROWS * valuesPerRow,\n  });\n  let pendingRows = 0;\n  let pendingSize = 0;\n\n  function flush() {\n    const start = performance.now();\n    const flushedRows = pendingRows;\n    const flushedSize = pendingSize;\n\n    let l = 0;\n    for (; pendingRows > INSERT_BATCH_SIZE; pendingRows -= INSERT_BATCH_SIZE) {\n      insertBatchStmt.run(pendingValues.slice(l, (l += valuesPerBatch)));\n    }\n    for (; pendingRows > 0; pendingRows--) {\n      insertStmt.run(pendingValues.slice(l, (l += valuesPerRow)));\n    }\n    const flushedValues = flushedRows * valuesPerRow;\n    for (let i = 0; i < flushedValues; i++) {\n      pendingValues[i] = undefined as unknown as LiteValueType;\n    }\n    pendingSize = 0;\n    status.rows += flushedRows;\n\n    const elapsed = performance.now() - start;\n    flushTime += elapsed;\n    lc.debug?.(\n      `flushed ${flushedRows} ${tableName} rows (${flushedSize} bytes) in ${elapsed.toFixed(3)} ms`,\n    );\n  }\n\n  const binaryParser = new BinaryCopyParser();\n  let col = 0;\n\n  lc.info?.(`Starting binary copy stream of ${tableName}:`, select);\n\n  await pipeline(\n    await from\n      .unsafe(`COPY (${select}) TO STDOUT WITH (FORMAT binary)`)\n      .readable(),\n    new Writable({\n      highWaterMark: BUFFERED_SIZE_THRESHOLD,\n\n      write(\n        chunk: Buffer,\n        _encoding: string,\n        callback: (error?: Error) => void,\n      ) {\n        try {\n          for (const fieldBuf of binaryParser.parse(chunk)) {\n            pendingSize += fieldBuf === null ? 4 : fieldBuf.length;\n            pendingValues[pendingRows * valuesPerRow + col] =\n              fieldBuf === null ? null : decoders[col](fieldBuf);\n\n            if (++col === decoders.length) {\n              col = 0;\n              if (\n                ++pendingRows >= MAX_BUFFERED_ROWS - valuesPerRow ||\n                pendingSize >= BUFFERED_SIZE_THRESHOLD\n              ) {\n                flush();\n              }\n            }\n          }\n          callback();\n        } catch (e) {\n          callback(e instanceof Error ? e : new Error(String(e)));\n        }\n      },\n\n      final: (callback: (error?: Error) => void) => {\n        try {\n          flush();\n          callback();\n        } catch (e) {\n          callback(e instanceof Error ? e : new Error(String(e)));\n        }\n      },\n    }),\n  );\n\n  const elapsed = performance.now() - start;\n  lc.info?.(\n    `Finished copying ${status.rows} rows into ${tableName} ` +\n      `(flush: ${flushTime.toFixed(3)} ms) (total: ${elapsed.toFixed(3)} ms) `,\n  );\n  return {rows: status.rows, flushTime};\n}\n\nasync function copyText(\n  lc: LogContext,\n  table: PublishedTableSpec,\n  status: DownloadStatus,\n  dbClient: PostgresDB,\n  from: PostgresTransaction,\n  to: Database,\n  sampleRate?: number | undefined,\n  maxRowsPerTable?: number | undefined,\n) {\n  const start = performance.now();\n  let flushTime = 0;\n\n  const tableName = liteTableName(table);\n  const orderedColumns = Object.entries(table.columns);\n\n  const columnNames = orderedColumns.map(([c]) => c);\n  const columnSpecs = orderedColumns.map(([_name, spec]) => spec);\n  const insertColumnList = columnNames.map(c => id(c)).join(',');\n\n  const valuesSql =\n    columnNames.length > 0 ? `(${'?,'.repeat(columnNames.length - 1)}?)` : '()';\n  const insertSql = /*sql*/ `\n    INSERT INTO \"${tableName}\" (${insertColumnList}) VALUES ${valuesSql}`;\n  const insertStmt = to.prepare(insertSql);\n  const insertBatchStmt = to.prepare(\n    insertSql + `,${valuesSql}`.repeat(INSERT_BATCH_SIZE - 1),\n  );\n\n  const {select} = makeDownloadStatements(\n    table,\n    columnNames,\n    sampleRate,\n    maxRowsPerTable,\n  );\n  const valuesPerRow = columnSpecs.length;\n  const valuesPerBatch = valuesPerRow * INSERT_BATCH_SIZE;\n\n  const pendingValues: LiteValueType[] = Array.from({\n    length: MAX_BUFFERED_ROWS * valuesPerRow,\n  });\n  let pendingRows = 0;\n  let pendingSize = 0;\n\n  function flush() {\n    const start = performance.now();\n    const flushedRows = pendingRows;\n    const flushedSize = pendingSize;\n\n    let l = 0;\n    for (; pendingRows > INSERT_BATCH_SIZE; pendingRows -= INSERT_BATCH_SIZE) {\n      insertBatchStmt.run(pendingValues.slice(l, (l += valuesPerBatch)));\n    }\n    for (; pendingRows > 0; pendingRows--) {\n      insertStmt.run(pendingValues.slice(l, (l += valuesPerRow)));\n    }\n    const flushedValues = flushedRows * valuesPerRow;\n    for (let i = 0; i < flushedValues; i++) {\n      pendingValues[i] = undefined as unknown as LiteValueType;\n    }\n    pendingSize = 0;\n    status.rows += flushedRows;\n\n    const elapsed = performance.now() - start;\n    flushTime += elapsed;\n    lc.debug?.(\n      `flushed ${flushedRows} ${tableName} rows (${flushedSize} bytes) in ${elapsed.toFixed(3)} ms`,\n    );\n  }\n\n  lc.info?.(`Starting text copy stream of ${tableName}:`, select);\n  const pgParsers = await getTypeParsers(dbClient, {returnJsonAsString: true});\n  const parsers = columnSpecs.map(c => {\n    const pgParse = pgParsers.getTypeParser(c.typeOID);\n    return (val: string) =>\n      liteValue(\n        pgParse(val) as PostgresValueType,\n        c.dataType,\n        JSON_STRINGIFIED,\n      );\n  });\n\n  const tsvParser = new TsvParser();\n  let col = 0;\n\n  await pipeline(\n    await from.unsafe(`COPY (${select}) TO STDOUT`).readable(),\n    new Writable({\n      highWaterMark: BUFFERED_SIZE_THRESHOLD,\n\n      write(\n        chunk: Buffer,\n        _encoding: string,\n        callback: (error?: Error) => void,\n      ) {\n        try {\n          for (const text of tsvParser.parse(chunk)) {\n            pendingSize += text === null ? 4 : text.length;\n            pendingValues[pendingRows * valuesPerRow + col] =\n              text === null ? null : parsers[col](text);\n\n            if (++col === parsers.length) {\n              col = 0;\n              if (\n                ++pendingRows >= MAX_BUFFERED_ROWS - valuesPerRow ||\n                pendingSize >= BUFFERED_SIZE_THRESHOLD\n              ) {\n                flush();\n              }\n            }\n          }\n          callback();\n        } catch (e) {\n          callback(e instanceof Error ? e : new Error(String(e)));\n        }\n      },\n\n      final: (callback: (error?: Error) => void) => {\n        try {\n          flush();\n          callback();\n        } catch (e) {\n          callback(e instanceof Error ? e : new Error(String(e)));\n        }\n      },\n    }),\n  );\n\n  const elapsed = performance.now() - start;\n  lc.info?.(\n    `Finished copying ${status.rows} rows into ${tableName} ` +\n      `(flush: ${flushTime.toFixed(3)} ms) (total: ${elapsed.toFixed(3)} ms) `,\n  );\n  return {rows: status.rows, flushTime};\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAiGA,eAAsB,YACpB,IACA,OACA,IACA,aACA,aACA,SACA;AACA,KAAI,CAAC,0BAA0B,KAAK,MAAM,MAAM,CAC9C,OAAM,IAAI,MACR,2FACD;CAEH,MAAM,EACJ,kBACA,aACA,WAAW,OACX,0BAA0B,OAC1B,WACE;CACJ,MAAM,eAAe,cAAc,MAAM,YAAY,SAAS,GAAG;CACjE,MAAM,MAAM,SAAS,IAAI,aAAa,eAAe;CAIrD,MAAM,qBAAqB,SACvB,KAAA,IACA,SAAS,IAAI,aAAa,oCAAoC;GAC3D,gBAAgB;EACjB,YAAY,EAAC,aAAa,YAAW;EACtC,CAAC;CAEN,MAAM,YAAY,KAAK,KAAK,CAAC,UAAU;CACvC,IAAI;CACJ,MAAM,kBAAkB,2BAA2B,sBACjD,IACA,SAAS,YAAY,KAAK,KAAA,EAC3B,CAAC,QAAQ,IAAI,eAAe;CAC7B,IAAI;AACJ,KAAI;EACF,MAAM,YAAY,MAAM,oBAAoB,IAAI;EAMhD,MAAM,EAAC,iBAAgB,SACnB,MAAM,uBAAuB,KAAK,MAAM,GACxC,MAAM,sBAAsB,IAAI,KAAK,MAAM;AAC/C,KAAG,OAAO,wCAAwC,aAAa,GAAG;EAElE,MAAM,EAAC,UAAU,SAAQ,IAAI;AAC7B,KAAG,OACD,SACI,kCAAkC,SAAS,GAAG,KAAK,kBACnD,kCAAkC,SAAS,GAAG,OACnD;EAED,IAAI;EACJ,IAAI;AAEJ,MAAI,QAAQ;GACV,MAAM,WAAW,MAAM,qCACrB,IACA,YACD;AACD,cAAW,SAAS;AACpB,SAAM,SAAS;AACf,2BAAwB,SAAS;SAC5B;GACL,MAAM,OAAO,MAAM,qBACjB,IACA,KACA,KAAK,mBAAmB,EACxB,OACA,WACA,2BAA2B,aAAA,KAC5B;AACD,cAAW,KAAK;AAChB,SAAM,KAAK;AACX,cAAW,KAAK;;EAGlB,MAAM,iBAAiB,qBAAqB,IAAI;AAEhD,uBAAqB,IAAI,cAAc,gBAAgB,QAAQ;EAG/D,MAAM,QAAQ,YAAY,KAAK;EAE/B,MAAM,YAAY,MAAM,MACtB,KACA,OAAM,OAAM;AACV,SAAM,GAAG,OAAgB,6BAA6B,SAAS,GAAG;AAClE,UAAO,mBAAmB,IAAI,aAAa;KAE7C,EAAC,MAAM,UAAc,CACtB;AAED,uBAAqB,IAAI,UAAU;EAGnC,MAAM,EAAC,QAAQ,YAAW;EAC1B,MAAM,YAAY,OAAO;AACzB,MAAI,UAAU,KAAK,WAAW,mBAAmB,UAC/C,IAAG,OACD,8CAA8C,iBAAiB,MAC1D,UAAU,gDAChB;EAEH,MAAM,aACJ,UAAU,KAAK,UACX,YACA,KAAK,IAAI,kBAAkB,UAAU;EAE3C,MAAM,WAAW,SAAS,IAAI,aAAa,4BAA4B;GACrE,KAAK;IACJ,iBAAiB;GACnB,CAAC;EACF,MAAM,UAAU,sBACd,IACA,UACA,UACA,YACA,UACD;AACD,MAAI;AACF,oBAAiB,IAAI,QAAQ,eAAe;GAC5C,MAAM,aAAa,QAAQ;GAC3B,MAAM,kBAAkB,QAAQ;GAChC,MAAM,YAAY,MAAM,QAAQ,IAC9B,OAAO,KAAI,SACT,QAAQ,iBAAiB,IAAI,OAC3B,wBAAwB,IAAI,IAAI,MAAM,WAAW,KAAA,EAAU,CAC5D,CACF,CACF;AACD,mBAAgB,QACd,IACA,gBACA,WAAW,UAAU,8BAA8B,kBACnD,YACO,EAAC,gBAAgB,UAAU,KAAK,EAAC,aAAY,OAAO,EAAC,EAC7D;AAEI,iBAAc,OAAO;GAC1B,MAAM,YAAY,MAAM,QAAQ,IAC9B,UAAU,KAAI,UACZ,QAAQ,iBAAiB,IAAI,OAC3B,KACE,IACA,OACA,UACA,IACA,IACA,UACA,YACA,gBACD,CACF,CACF,CACF;AACI,iBAAc,eAAe,IAAI,eAAe;AACrD,WAAQ,SAAS;GAEjB,MAAM,QAAQ,UAAU,QACrB,KAAK,UAAU;IACd,MAAM,IAAI,OAAO,KAAK;IACtB,WAAW,IAAI,YAAY,KAAK;IACjC,GACD;IAAC,MAAM;IAAG,WAAW;IAAE,CACxB;AAED,mBAAgB,QACd,IACA,YACA,YAAY,QAAQ,OAAO,WAC3B,IACD;GACD,MAAM,aAAa,YAAY,KAAK;AACpC,qBAAkB,IAAI,QAAQ;GAC9B,MAAM,QAAQ,YAAY,KAAK,GAAG;AAClC,MAAG,OAAO,oBAAoB,MAAM,QAAQ,EAAE,CAAC,MAAM;AAErD,OAAI,YAAY,UACd,OAAM,YAAY,KAAK,OAAO,WAAW,WAAW,QAAQ;QACvD;AACL,WAAO,QAAQ,wDAAwD;IACvE,MAAM,8BAAc,IAAI,KAAqB;AAC7C,SAAK,IAAI,IAAI,GAAG,IAAI,UAAU,QAAQ,IACpC,aAAY,IAAI,UAAU,GAAG,OAAO,OAAO,UAAU,GAAG,KAAK;AAE/D,wBAAoB,IAAI,IAAI,WAAW,YAAY;;GAGrD,MAAM,UAAU,YAAY,KAAK,GAAG;AACpC,MAAG,OACD,UAAU,MAAM,KAAK,gBAAgB,CAAC,WAAW,UAAU,aAAa,aAAa,SAAS,IAAI,WACrF,MAAM,UAAU,QAAQ,EAAE,CAAC,WAAW,MAAM,QAAQ,EAAE,CAAC,WAAW,QAAQ,QAAQ,EAAE,CAAC,MACnG;YACO;AAEH,YAAS,KAAK,CAAC,OAAM,MAAK,GAAG,OAAO,0BAA0B,EAAE,CAAC;;UAEjE,GAAG;AACV,MAAI,UAAU;AAIZ,MAAG,OAAO,6BAA6B,YAAY,EAAE;AACrD,SAAM,GAAG;;8BAEe,SAAS;QAC/B,OAAM,MAAK,GAAG,OAAO,mCAAmC,YAAY,EAAE,CAAC;;AAE3E,QAAM,gBAAgB,qBAAqB,IAAI,gBAAgB,EAAE;WACzD;AACR,kBAAgB,MAAM;AACtB,MAAI,sBACF,OAAM,uBAAuB,CAAC,OAAM,MAClC,GAAG,OAAO,mCAAmC,EAAE,CAChD;AAEH,MAAI,mBACF,OAAM,mBAAmB,KAAK;AAEhC,QAAM,IAAI,KAAK;;;;;;;;;;;;;AAwBnB,eAAsB,kBACpB,IACA,OACA,aACA,QACA,SACA,aACe;CACf,MAAM,MAAM,MAAM,QAChB,KAAK,OAAO,aAAa,QAAQ,EAAE,oBAAoB,CACxD;CAED,MAAM,KAAK,IAAI,SAAS,IADT,KAAK,KAAK,oBAAoB,CACV;AACnC,KAAI;AACF,QAAM,YACJ,IACA,OACA,IACA,aACA;GAGE,kBAAkB;GAClB,UAAU,aAAa;GACvB;GACD,EACD,QACD;WACO;AACR,MAAI;AACF,MAAG,OAAO;WACH,GAAG;AACV,MAAG,OAAO,mCAAmC,EAAE;;AAEjD,QAAM,GAAG,KAAK;GAAC,WAAW;GAAM,OAAO;GAAK,CAAC,CAAC,OAAM,MAClD,GAAG,OAAO,wCAAwC,OAAO,EAAE,CAC5D;;;AAIL,eAAe,oBAAoB,KAAiB;CAClD,MAAM,EAAC,UAAU,aACf,MAAM,GAA0C;;;KAIhD;AAEF,KAAI,aAAa,UACf,OAAM,IAAI,MACR,uEAAuE,SAAS,GACjF;AAEH,KAAI,UAAA,KACF,OAAM,IAAI,MACR,sDAAsD,QAAQ,IAC/D;AAEH,QAAO;;AAGT,eAAe,sBACb,IACA,KACA,OACA,WAAW,MACwB;CACnC,MAAM,EAAC,UAAU,SAAQ,IAAI;AAC7B,IAAG,OAAO,oCAAoC,SAAS,GAAG,OAAO;AAEjE,OAAM,kBAAkB,IAAI,KAAK,MAAM;CACvC,MAAM,EAAC,iBAAgB,MAAM,uBAAuB,KAAK,MAAM;AAE/D,KAAI,UAAU;EACZ,IAAI,QAAQ;EACZ,MAAM,0BAA0B,aAAa,QAC3C,MAAK,CAAC,EAAE,WAAW,IAAI,CACxB;EACD,MAAM,SAAS,MAAM,GAAG;4DACgC,IAAI,aAAa,CAAC;QACtE,QAAQ;AACZ,MAAI,OAAO,WAAW,aAAa,OACjC,IAAG,OACD,iCAAiC,aAAa,kBACxC,OAAO,MAAM,CAAC,cACrB;WAED,CAAC,OAAO,IAAI,IAAI,MAAM,aAAa,EAAE,IAAI,IAAI,wBAAwB,CAAC,CAEtE,IAAG,OACD,2BAA2B,MAAM,aAAa,sCAC3B,wBAAwB,cAC5C;MAED,SAAQ;AAEV,MAAI,CAAC,OAAO;AACV,SAAM,IAAI,OAAO,UAAU,MAAM,OAAO,MAAM,SAAS,CAAC;AACxD,UAAO,sBAAsB,IAAI,KAAK,OAAO,MAAM;;;AAGvD,QAAO,EAAC,cAAa;;AAGvB,SAAS,sBACP,IACA,IACA,UACA,YACA,WACiB;CACjB,MAAM,EAAC,SAAQ,eAAe,SAAS;CACvC,MAAM,eAAe,IAAI,gBAAgB,IAAI;EAC3C,MAAM;EACN;EACA,gBAAgB;EACjB,CAAC;AACF,cAAa,IAAI,GAAG;AAEpB,IAAG,OAAO,WAAW,WAAW,mBAAmB,UAAU,SAAS;AAEtE,KAAI,SAAS,QAAQ,SAAS,KAAK,GAAG,GACpC,IAAG,OACD,mUAMD;AAEH,QAAO;;;;;;;;;;;;AAaT,eAAe,qCACb,IACA,aAKC;CACD,MAAM,SAAS,SAAS,IAAI,aAAa,gCAAgC,EACvE,KAAK,GACN,CAAC;CACF,MAAM,QAAQ,UAA2C;CACzD,MAAM,UAAU,UAAgB;CAChC,MAAM,OAAO,OACV,MAAM,UAAe,OAAM,OAAM;AAChC,QAAM,EAAE,oDAAoD,SAAS;EACrE,MAAM,CAAC,OAAO,MAAM,EAAqC;;;AAGzD,QAAM,QAAQ,IAAI;AAClB,QAAM,QAAQ;GACd,CACD,OAAM,MAAK,MAAM,OAAO,EAAE,CAAC;CAE9B,IAAI;CACJ,IAAI;AACJ,KAAI;AACF,GAAC,CAAC,UAAU,OAAO,MAAM,MAAM;UACxB,GAAG;AACV,QAAM,OACH,KAAK,CACL,OAAM,QACL,GAAG,OAAO,qDAAqD,IAAI,CACpE;AACH,QAAM;;AAER,IAAG,OACD,qBAAqB,SAAS,UAAU,IAAI,wBAC7C;AACD,QAAO;EACL;EACA;EACA,SAAS,YAAY;AACnB,WAAQ,SAAS;AACjB,OAAI;AACF,UAAM;YACC,GAAG;AACV,OAAG,OAAO,gDAAgD,EAAE;;AAE9D,SAAM,OAAO,KAAK;;EAErB;;AAGH,SAAS,iBACP,IACA,QACA,gBACA;CAGA,MAAM,iBAAiB,KAAK,oBAAoB,YAAY,GAAG,CAAC;AAChE,MAAK,MAAM,KAAK,QAAQ;AACtB,KAAG,KAAK,yBAAyB,kBAAkB,GAAG,eAAe,CAAC,CAAC;EACvE,MAAM,YAAY,cAAc,EAAE;AAClC,OAAK,MAAM,CAAC,SAAS,YAAY,OAAO,QAAQ,EAAE,QAAQ,CACxD,gBAAe,OAAO,WAAW,SAAS,QAAQ;;;AAKxD,SAAS,kBAAkB,IAAc,SAAsB;AAC7D,MAAK,MAAM,SAAS,QAClB,IAAG,KAAK,yBAAyB,uBAAuB,MAAM,CAAC,CAAC;;;;;;;;;;AAYpE,SAAgB,oBACd,IACA,IACA,WACA,aACM;CACN,MAAM,SAAmB,EAAE;CAC3B,IAAI,iBAAiB;CACrB,IAAI,cAAc;CAIlB,MAAM,aAAa,WAAW,GAAG;CACjC,MAAM,kBAAkB,IAAI,IAAI,WAAW,KAAI,MAAK,CAAC,EAAE,MAAM,EAAE,CAAC,CAAC;AACjE,MAAK,MAAM,MAAM,UAAU,QAAQ;EACjC,MAAM,OAAO,cAAc,GAAG;EAC9B,MAAM,OAAO,gBAAgB,IAAI,KAAK;AACtC,MAAI,CAAC,MAAM;AACT,UAAO,KAAK,6BAA6B,OAAO;AAChD;;AAEF,OAAK,MAAM,OAAO,OAAO,KAAK,GAAG,QAAQ,EAAE;AACzC;AACA,OAAI,EAAE,OAAO,KAAK,SAChB,QAAO,KAAK,mCAAmC,KAAK,IAAI,MAAM;;;CAMpE,MAAM,iBAAiB,IAAI,IAAI,YAAY,GAAG,CAAC,KAAI,MAAK,EAAE,KAAK,CAAC;AAChE,MAAK,MAAM,MAAM,UAAU,SAAS;EAClC,MAAM,SAAS,uBAAuB,GAAG;AACzC,MAAI,CAAC,eAAe,IAAI,OAAO,KAAK,CAClC,QAAO,KACL,6BAA6B,OAAO,KAAK,MAAM,OAAO,YACvD;;AAKL,MAAK,MAAM,CAAC,OAAO,aAAa,YAC9B,KAAI;EACF,MAAM,CAAC,OAAO,GACX,QAAQ,kCAAkC,MAAM,GAAG,CACnD,KAAsB;AACzB,MAAI,IAAI,UAAU,SAChB,QAAO,KACL,gCAAgC,MAAM,0BACX,SAAS,gBAAgB,IAAI,QACzD;MAED,gBAAe,IAAI;UAEd,GAAG;AACV,SAAO,KAAK,iCAAiC,MAAM,IAAI,OAAO,EAAE,GAAG;;CAMvE,MAAM,aAAa,gBAAgB,IAAI,IAAI,EACzC,2BAA2B,OAC5B,CAAC;AACF,MAAK,MAAM,MAAM,UAAU,QAAQ;EACjC,MAAM,OAAO,cAAc,GAAG;AAC9B,MAAI,CAAC,WAAW,IAAI,KAAK,CACvB,QAAO,KACL,6DAA6D,OAC9D;;CAKL,MAAM,OAAO,KAAK,oBAAoB,YAAY,GAAG,CAAC;AACtD,MAAK,MAAM,MAAM,UAAU,QAAQ;EACjC,MAAM,OAAO,cAAc,GAAG;EAC9B,MAAM,OAAO,KAAK,SAAS,KAAK;AAChC,OAAK,MAAM,OAAO,OAAO,KAAK,GAAG,QAAQ,CACvC,KAAI,CAAC,KAAK,IAAI,IAAI,CAChB,QAAO,KAAK,mCAAmC,KAAK,GAAG,MAAM;;AAKnE,KAAI,OAAO,OACT,OAAM,IAAI,MACR,uCAAuC,OAAO,OAAO,iBACnD,OAAO,KAAI,MAAK,OAAO,IAAI,CAAC,KAAK,KAAK,CACzC;AAGH,IAAG,OACD,uCACK,UAAU,OAAO,OAAO,WACxB,UAAU,QAAQ,OAAO,YACzB,eAAe,YACf,YAAY,gBAAgB,CAAC,OACnC;;AASH,IAAM,KAAK,OAAO;AAClB,IAAM,oBAAoB;AAC1B,IAAM,0BAA0B,IAAI;;;;;;;AAcpC,SAAS,kBAAkB,YAAwC;AACjE,KAAI,eAAe,KAAA,KAAa,cAAc,EAC5C,QAAO;AAKT,QAAe,0BADH,YAAY,aAAa,KAAK,QAAQ,EAAE,CAAC,CACR;;AAG/C,SAAS,YAAY,iBAA6C;AAChE,QAAO,oBAAoB,KAAA,IACf,UAAU,oBAClB;;;;;;AAON,SAAgB,sBACd,OACA,MACU;AACV,QAAO,KAAK,KAAI,QAAO;EACrB,MAAM,OAAO,MAAM,QAAQ;AAC3B,SAAO,iBAAiB,KAAK,GAAG,GAAG,IAAI,GAAG,GAAG,GAAG,IAAI,CAAC;GACrD;;AAGJ,SAAgB,uBACd,OACA,MACA,YACA,iBACA,aACoB;CACpB,MAAM,mBAAmB,OAAO,OAAO,MAAM,aAAa,CACvD,KAAK,EAAC,gBAAe,UAAU,CAC/B,QAAO,MAAK,CAAC,CAAC,EAAE;CACnB,MAAM,QACJ,iBAAiB,WAAW,IACxB,KACQ,SAAS,iBAAiB,KAAK,OAAO;CACpD,MAAM,SAAS,kBAAkB,WAAW;CAC5C,MAAM,QAAQ,YAAY,gBAAgB;CAC1C,MAAM,YAAoB,QAAQ,GAAG,MAAM,OAAO,CAAC,GAAG,GAAG,MAAM,KAAK,GAAG,OAAO,GAAG;CACjF,MAAM,SAAiB,WAAW,eAAe,KAAK,IAAI,GAAG,EAAE,KAAK,IAAI,CAAC,GAAG,YAAY;AACxF,KAAI,OAAO;EAGT,MAAM,YAAY,KACf,KAAI,QAAO,2BAA2B,GAAG,IAAI,CAAC,OAAO,CACrD,KAAK,MAAM;AACd,SAAO;GACL;GACA,cAAsB,8DAA8D,YAAY,MAAM;GACtG,eAAuB,oEAAoE,UAAU,SAAS,YAAY,MAAM;GACjI;;CAEH,MAAM,aAAa,IAAI,KAAK,KAAI,QAAO,+BAA+B,GAAG,IAAI,CAAC,QAAQ,CAAC,KAAK,MAAM,CAAC;AACnG,QAAO;EACL;EACA,cAAsB,kCAAkC;EACxD,eAAuB,UAAU,WAAW,mBAAmB;EAChE;;AASH,eAAsB,wBACpB,IACA,KACA,MACA,YACwB;CACxB,MAAM,QAAQ,YAAY,KAAK;CAC/B,MAAM,QAAQ,cAAc,KAAK;CACjC,MAAM,UAAU,OAAO,KAAK,KAAK,QAAQ;AACzC,KAAI,WAGF,QAAO;EACL;EACA,QAAQ;GAAC;GAAO;GAAS,MAAM;GAAG,WAAW;GAAG,YAAY;GAAE;EAC/D;CAaH,MAAM,EAAC,WAAW,gBAPK,MAAM,GAE5B;;;kBAHqB,GAAG,GAAG,KAAK,OAAO,CAAC,GAAG,GAAG,KAAK,KAAK,GAM3B,aAEiB,MAAM;EACnD,WAAW;EACX,YAAY;EACb;CAED,MAAM,QAAuB;EAC3B;EACA,QAAQ;GACN;GACA;GACA,MAAM;GACN;GACA;GACD;EACF;CACD,MAAM,WAAW,YAAY,KAAK,GAAG,OAAO,QAAQ,EAAE;AACtD,IAAG,OAAO,uCAAuC,MAAM,IAAI,QAAQ,OAAO,EACxE,OAAO,MAAM,QACd,CAAC;AACF,QAAO;;AAGT,SAAS,KACP,IACA,EAAC,MAAM,OAAO,UACd,UACA,MACA,IACA,UACA,YACA,iBACA;AACA,KAAI,SACF,QAAO,SACL,IACA,OACA,QACA,UACA,MACA,IACA,YACA,gBACD;AAEH,QAAO,WAAW,IAAI,OAAO,QAAQ,MAAM,IAAI,YAAY,gBAAgB;;AAG7E,eAAe,WACb,IACA,OACA,QACA,MACA,IACA,YACA,iBACA;CACA,MAAM,QAAQ,YAAY,KAAK;CAC/B,IAAI,YAAY;CAEhB,MAAM,YAAY,cAAc,MAAM;CACtC,MAAM,iBAAiB,OAAO,QAAQ,MAAM,QAAQ;CAEpD,MAAM,cAAc,eAAe,KAAK,CAAC,OAAO,EAAE;CAClD,MAAM,cAAc,eAAe,KAAK,CAAC,OAAO,UAAU,KAAK;CAC/D,MAAM,mBAAmB,YAAY,KAAI,MAAK,GAAG,EAAE,CAAC,CAAC,KAAK,IAAI;CAE9D,MAAM,YACJ,YAAY,SAAS,IAAI,IAAI,KAAK,OAAO,YAAY,SAAS,EAAE,CAAC,MAAM;CACzE,MAAM,YAAoB;mBACT,UAAU,KAAK,iBAAiB,WAAW;CAC5D,MAAM,aAAa,GAAG,QAAQ,UAAU;CACxC,MAAM,kBAAkB,GAAG,QACzB,YAAY,IAAI,YAAY,OAAA,GAA6B,CAC1D;CAGD,MAAM,SAAS,uBACb,OACA,aACA,YACA,iBACA,sBAAsB,OAAO,YAAY,CAC1C,CAAC;CAEF,MAAM,WAAW,eAAe,KAAK,GAAG,UACtC,iBAAiB,KAAK,GAAG,kBAAkB,KAAK,GAAG,gBACpD;CAED,MAAM,eAAe,YAAY;CACjC,MAAM,iBAAiB,eAAA;CAEvB,MAAM,gBAAiC,MAAM,KAAK,EAChD,QAAQ,oBAAoB,cAC7B,CAAC;CACF,IAAI,cAAc;CAClB,IAAI,cAAc;CAElB,SAAS,QAAQ;EACf,MAAM,QAAQ,YAAY,KAAK;EAC/B,MAAM,cAAc;EACpB,MAAM,cAAc;EAEpB,IAAI,IAAI;AACR,SAAO,cAAA,IAAiC,eAAA,GACtC,iBAAgB,IAAI,cAAc,MAAM,GAAI,KAAK,eAAgB,CAAC;AAEpE,SAAO,cAAc,GAAG,cACtB,YAAW,IAAI,cAAc,MAAM,GAAI,KAAK,aAAc,CAAC;EAE7D,MAAM,gBAAgB,cAAc;AACpC,OAAK,IAAI,IAAI,GAAG,IAAI,eAAe,IACjC,eAAc,KAAK,KAAA;AAErB,gBAAc;AACd,SAAO,QAAQ;EAEf,MAAM,UAAU,YAAY,KAAK,GAAG;AACpC,eAAa;AACb,KAAG,QACD,WAAW,YAAY,GAAG,UAAU,SAAS,YAAY,aAAa,QAAQ,QAAQ,EAAE,CAAC,KAC1F;;CAGH,MAAM,eAAe,IAAI,kBAAkB;CAC3C,IAAI,MAAM;AAEV,IAAG,OAAO,kCAAkC,UAAU,IAAI,OAAO;AAEjE,OAAM,WACJ,MAAM,KACH,OAAO,SAAS,OAAO,kCAAkC,CACzD,UAAU,EACb,IAAI,SAAS;EACX,eAAe;EAEf,MACE,OACA,WACA,UACA;AACA,OAAI;AACF,SAAK,MAAM,YAAY,aAAa,MAAM,MAAM,EAAE;AAChD,oBAAe,aAAa,OAAO,IAAI,SAAS;AAChD,mBAAc,cAAc,eAAe,OACzC,aAAa,OAAO,OAAO,SAAS,KAAK,SAAS;AAEpD,SAAI,EAAE,QAAQ,SAAS,QAAQ;AAC7B,YAAM;AACN,UACE,EAAE,eAAe,oBAAoB,gBACrC,eAAe,wBAEf,QAAO;;;AAIb,cAAU;YACH,GAAG;AACV,aAAS,aAAa,QAAQ,IAAI,IAAI,MAAM,OAAO,EAAE,CAAC,CAAC;;;EAI3D,QAAQ,aAAsC;AAC5C,OAAI;AACF,WAAO;AACP,cAAU;YACH,GAAG;AACV,aAAS,aAAa,QAAQ,IAAI,IAAI,MAAM,OAAO,EAAE,CAAC,CAAC;;;EAG5D,CAAC,CACH;CAED,MAAM,UAAU,YAAY,KAAK,GAAG;AACpC,IAAG,OACD,oBAAoB,OAAO,KAAK,aAAa,UAAU,WAC1C,UAAU,QAAQ,EAAE,CAAC,eAAe,QAAQ,QAAQ,EAAE,CAAC,OACrE;AACD,QAAO;EAAC,MAAM,OAAO;EAAM;EAAU;;AAGvC,eAAe,SACb,IACA,OACA,QACA,UACA,MACA,IACA,YACA,iBACA;CACA,MAAM,QAAQ,YAAY,KAAK;CAC/B,IAAI,YAAY;CAEhB,MAAM,YAAY,cAAc,MAAM;CACtC,MAAM,iBAAiB,OAAO,QAAQ,MAAM,QAAQ;CAEpD,MAAM,cAAc,eAAe,KAAK,CAAC,OAAO,EAAE;CAClD,MAAM,cAAc,eAAe,KAAK,CAAC,OAAO,UAAU,KAAK;CAC/D,MAAM,mBAAmB,YAAY,KAAI,MAAK,GAAG,EAAE,CAAC,CAAC,KAAK,IAAI;CAE9D,MAAM,YACJ,YAAY,SAAS,IAAI,IAAI,KAAK,OAAO,YAAY,SAAS,EAAE,CAAC,MAAM;CACzE,MAAM,YAAoB;mBACT,UAAU,KAAK,iBAAiB,WAAW;CAC5D,MAAM,aAAa,GAAG,QAAQ,UAAU;CACxC,MAAM,kBAAkB,GAAG,QACzB,YAAY,IAAI,YAAY,OAAA,GAA6B,CAC1D;CAED,MAAM,EAAC,WAAU,uBACf,OACA,aACA,YACA,gBACD;CACD,MAAM,eAAe,YAAY;CACjC,MAAM,iBAAiB,eAAA;CAEvB,MAAM,gBAAiC,MAAM,KAAK,EAChD,QAAQ,oBAAoB,cAC7B,CAAC;CACF,IAAI,cAAc;CAClB,IAAI,cAAc;CAElB,SAAS,QAAQ;EACf,MAAM,QAAQ,YAAY,KAAK;EAC/B,MAAM,cAAc;EACpB,MAAM,cAAc;EAEpB,IAAI,IAAI;AACR,SAAO,cAAA,IAAiC,eAAA,GACtC,iBAAgB,IAAI,cAAc,MAAM,GAAI,KAAK,eAAgB,CAAC;AAEpE,SAAO,cAAc,GAAG,cACtB,YAAW,IAAI,cAAc,MAAM,GAAI,KAAK,aAAc,CAAC;EAE7D,MAAM,gBAAgB,cAAc;AACpC,OAAK,IAAI,IAAI,GAAG,IAAI,eAAe,IACjC,eAAc,KAAK,KAAA;AAErB,gBAAc;AACd,SAAO,QAAQ;EAEf,MAAM,UAAU,YAAY,KAAK,GAAG;AACpC,eAAa;AACb,KAAG,QACD,WAAW,YAAY,GAAG,UAAU,SAAS,YAAY,aAAa,QAAQ,QAAQ,EAAE,CAAC,KAC1F;;AAGH,IAAG,OAAO,gCAAgC,UAAU,IAAI,OAAO;CAC/D,MAAM,YAAY,MAAM,eAAe,UAAU,EAAC,oBAAoB,MAAK,CAAC;CAC5E,MAAM,UAAU,YAAY,KAAI,MAAK;EACnC,MAAM,UAAU,UAAU,cAAc,EAAE,QAAQ;AAClD,UAAQ,QACN,UACE,QAAQ,IAAI,EACZ,EAAE,UAAA,IAEH;GACH;CAEF,MAAM,YAAY,IAAI,WAAW;CACjC,IAAI,MAAM;AAEV,OAAM,WACJ,MAAM,KAAK,OAAO,SAAS,OAAO,aAAa,CAAC,UAAU,EAC1D,IAAI,SAAS;EACX,eAAe;EAEf,MACE,OACA,WACA,UACA;AACA,OAAI;AACF,SAAK,MAAM,QAAQ,UAAU,MAAM,MAAM,EAAE;AACzC,oBAAe,SAAS,OAAO,IAAI,KAAK;AACxC,mBAAc,cAAc,eAAe,OACzC,SAAS,OAAO,OAAO,QAAQ,KAAK,KAAK;AAE3C,SAAI,EAAE,QAAQ,QAAQ,QAAQ;AAC5B,YAAM;AACN,UACE,EAAE,eAAe,oBAAoB,gBACrC,eAAe,wBAEf,QAAO;;;AAIb,cAAU;YACH,GAAG;AACV,aAAS,aAAa,QAAQ,IAAI,IAAI,MAAM,OAAO,EAAE,CAAC,CAAC;;;EAI3D,QAAQ,aAAsC;AAC5C,OAAI;AACF,WAAO;AACP,cAAU;YACH,GAAG;AACV,aAAS,aAAa,QAAQ,IAAI,IAAI,MAAM,OAAO,EAAE,CAAC,CAAC;;;EAG5D,CAAC,CACH;CAED,MAAM,UAAU,YAAY,KAAK,GAAG;AACpC,IAAG,OACD,oBAAoB,OAAO,KAAK,aAAa,UAAU,WAC1C,UAAU,QAAQ,EAAE,CAAC,eAAe,QAAQ,QAAQ,EAAE,CAAC,OACrE;AACD,QAAO;EAAC,MAAM,OAAO;EAAM;EAAU"}
+{"version":3,"file":"initial-sync.js","names":[],"sources":["../../../../../../../zero-cache/src/services/change-source/pg/initial-sync.ts"],"sourcesContent":["import {mkdtemp, rm} from 'node:fs/promises';\nimport {platform, tmpdir} from 'node:os';\nimport {join} from 'node:path';\nimport {Writable} from 'node:stream';\nimport {pipeline} from 'node:stream/promises';\nimport type {LogContext} from '@rocicorp/logger';\nimport {resolver} from '@rocicorp/resolver';\nimport {assert} from '../../../../../shared/src/asserts.ts';\nimport type {JSONObject} from '../../../../../shared/src/bigint-json.ts';\nimport {must} from '../../../../../shared/src/must.ts';\nimport {equals} from '../../../../../shared/src/set-utils.ts';\nimport type {DownloadStatus} from '../../../../../zero-events/src/status.ts';\nimport {Database} from '../../../../../zqlite/src/db.ts';\nimport {\n  createLiteIndexStatement,\n  createLiteTableStatement,\n} from '../../../db/create.ts';\nimport {listIndexes, listTables} from '../../../db/lite-tables.ts';\nimport * as Mode from '../../../db/mode-enum.ts';\nimport {\n  BinaryCopyParser,\n  hasBinaryDecoder,\n  makeBinaryDecoder,\n  textCastDecoder,\n} from '../../../db/pg-copy-binary.ts';\nimport {TsvParser} from '../../../db/pg-copy.ts';\nimport {\n  mapPostgresToLite,\n  mapPostgresToLiteIndex,\n} from '../../../db/pg-to-lite.ts';\nimport {getTypeParsers} from '../../../db/pg-type-parser.ts';\nimport {runTx} from '../../../db/run-transaction.ts';\nimport type {IndexSpec, PublishedTableSpec} from '../../../db/specs.ts';\nimport {importSnapshot, TransactionPool} from '../../../db/transaction-pool.ts';\nimport {\n  JSON_STRINGIFIED,\n  liteValue,\n  type LiteValueType,\n} from '../../../types/lite.ts';\nimport {liteTableName} from '../../../types/names.ts';\nimport {PG_15, PG_17} from '../../../types/pg-versions.ts';\nimport {\n  connectPgClient,\n  pgClient,\n  type PostgresDB,\n  type PostgresTransaction,\n  type PostgresValueType,\n} from '../../../types/pg.ts';\nimport {CpuProfiler} from '../../../types/profiler.ts';\nimport type {ShardConfig} from '../../../types/shards.ts';\nimport {ALLOWED_APP_ID_CHARACTERS} from '../../../types/shards.ts';\nimport {id} from '../../../types/sql.ts';\nimport {ReplicationStatusPublisher} from '../../replicator/replication-status.ts';\nimport {ColumnMetadataStore} from '../../replicator/schema/column-metadata.ts';\nimport {initReplicationState} from '../../replicator/schema/replication-state.ts';\nimport {toStateVersionString} from './lsn.ts';\nimport {createReplicaAndSlot} from './replication-slots.ts';\nimport {ensureShardSchema} from './schema/init.ts';\nimport {getPublicationInfo} from './schema/published.ts';\nimport {\n  dropShard,\n  getInternalShardConfig,\n  initReplica,\n  validatePublications,\n} from './schema/shard.ts';\n\nexport type InitialSyncOptions = {\n  tableCopyWorkers: number;\n  profileCopy?: boolean | undefined;\n  textCopy?: boolean | undefined;\n  replicationSlotFailover?: boolean | undefined;\n  /**\n   * When set, run initial sync in \"shadow\" mode for verification: skip all\n   * upstream mutations (no replication slot, no addReplica, no dropShard, no\n   * slot drop on failure), suppress status events, and optionally sample\n   * rows from each table via TABLESAMPLE BERNOULLI + LIMIT. The caller is\n   * responsible for providing (and discarding) a throwaway SQLite `tx`.\n   */\n  shadow?:\n    | {\n        /** 0 < rate <= 1. When 1, no TABLESAMPLE clause is added. */\n        sampleRate: number;\n        /**\n         * LIMIT N cap appended after TABLESAMPLE. Required: shadow sync is\n         * for verification only, so every run must commit to a row budget.\n         */\n        maxRowsPerTable: number;\n      }\n    | undefined;\n};\n\n/** Server context to store with the initial sync metadata for debugging. */\nexport type ServerContext = JSONObject;\n\nexport async function initialSync(\n  lc: LogContext,\n  shard: ShardConfig,\n  tx: Database,\n  upstreamURI: string,\n  syncOptions: InitialSyncOptions,\n  context: ServerContext,\n) {\n  if (!ALLOWED_APP_ID_CHARACTERS.test(shard.appID)) {\n    throw new Error(\n      'The App ID may only consist of lower-case letters, numbers, and the underscore character',\n    );\n  }\n  const {\n    tableCopyWorkers,\n    profileCopy,\n    textCopy = false,\n    replicationSlotFailover = false,\n    shadow,\n  } = syncOptions;\n  const copyProfiler = profileCopy ? await CpuProfiler.connect() : null;\n  const sql = await connectPgClient(lc, upstreamURI, 'initial-sync');\n  // Replication session is only needed to create a replication slot in the\n  // real path. In shadow mode we export a snapshot on a normal connection\n  // instead, so no replication session is opened.\n  const replicationSession = shadow\n    ? undefined\n    : pgClient(lc, upstreamURI, 'initial-sync-replication-session', {\n        ['fetch_types']: false, // Necessary for the streaming protocol\n        connection: {replication: 'database'}, // https://www.postgresql.org/docs/current/protocol-replication.html\n      });\n\n  const replicaID = Date.now().toString();\n  let slotName: string | undefined; // undefined === shadow\n  const statusPublisher = ReplicationStatusPublisher.forRunningTransaction(\n    tx,\n    shadow ? async () => {} : undefined,\n  ).publish(lc, 'Initializing');\n  let releaseShadowSnapshot: (() => Promise<void>) | undefined;\n  try {\n    const pgVersion = await checkUpstreamConfig(sql);\n\n    // In shadow mode we assume the shard is already initialized and just\n    // read back the existing publications. `ensurePublishedTables` would\n    // otherwise run DDL and potentially call `dropShard`, which must never\n    // happen during a shadow run.\n    const {publications} = shadow\n      ? await getInternalShardConfig(sql, shard)\n      : await ensurePublishedTables(lc, sql, shard);\n    lc.info?.(`Upstream is setup with publications [${publications}]`);\n\n    const {database, host} = sql.options;\n    lc.info?.(\n      shadow\n        ? `acquiring exported snapshot on ${database}@${host} (shadow mode)`\n        : `opening replication session to ${database}@${host}`,\n    );\n\n    let snapshot: string;\n    let lsn: string;\n\n    if (shadow) {\n      const acquired = await acquireExportedSnapshotForShadowSync(\n        lc,\n        upstreamURI,\n      );\n      snapshot = acquired.snapshot;\n      lsn = acquired.lsn;\n      releaseShadowSnapshot = acquired.release;\n    } else {\n      const slot = await createReplicaAndSlot(\n        lc,\n        sql,\n        must(replicationSession),\n        shard,\n        replicaID,\n        replicationSlotFailover && pgVersion >= PG_17,\n      );\n      snapshot = slot.snapshot_name;\n      lsn = slot.consistent_point;\n      slotName = slot.slot_name;\n    }\n\n    const initialVersion = toStateVersionString(lsn);\n\n    initReplicationState(tx, publications, initialVersion, context);\n\n    // Run up to MAX_WORKERS to copy of tables at the replication slot's snapshot.\n    const start = performance.now();\n    // Retrieve the published schema at the consistent_point.\n    const published = await runTx(\n      sql,\n      async tx => {\n        await tx.unsafe(/* sql*/ `SET TRANSACTION SNAPSHOT '${snapshot}'`);\n        return getPublicationInfo(tx, publications);\n      },\n      {mode: Mode.READONLY},\n    );\n    // Note: If this throws, initial-sync is aborted.\n    validatePublications(lc, published);\n\n    // Now that tables have been validated, kick off the copiers.\n    const {tables, indexes} = published;\n    const numTables = tables.length;\n    if (platform() === 'win32' && tableCopyWorkers < numTables) {\n      lc.warn?.(\n        `Increasing the number of copy workers from ${tableCopyWorkers} to ` +\n          `${numTables} to work around a Node/Postgres connection bug`,\n      );\n    }\n    const numWorkers =\n      platform() === 'win32'\n        ? numTables\n        : Math.min(tableCopyWorkers, numTables);\n\n    const copyPool = await connectPgClient(\n      lc,\n      upstreamURI,\n      'initial-sync-copy-worker',\n      {\n        max: numWorkers,\n        ['max_lifetime']: 120 * 60, // set a long (2h) limit for COPY streaming\n      },\n    );\n    const copiers = startTableCopyWorkers(\n      lc,\n      copyPool,\n      snapshot,\n      numWorkers,\n      numTables,\n    );\n    try {\n      createLiteTables(tx, tables, initialVersion);\n      const sampleRate = shadow?.sampleRate;\n      const maxRowsPerTable = shadow?.maxRowsPerTable;\n      const downloads = await Promise.all(\n        tables.map(spec =>\n          copiers.processReadTask((db, lc) =>\n            getInitialDownloadState(lc, db, spec, shadow !== undefined),\n          ),\n        ),\n      );\n      statusPublisher.publish(\n        lc,\n        'Initializing',\n        `Copying ${numTables} upstream tables at version ${initialVersion}`,\n        5000,\n        () => ({downloadStatus: downloads.map(({status}) => status)}),\n      );\n\n      void copyProfiler?.start();\n      const rowCounts = await Promise.all(\n        downloads.map(table =>\n          copiers.processReadTask((db, lc) =>\n            copy(\n              lc,\n              table,\n              copyPool,\n              db,\n              tx,\n              textCopy,\n              sampleRate,\n              maxRowsPerTable,\n            ),\n          ),\n        ),\n      );\n      void copyProfiler?.stopAndDispose(lc, 'initial-copy');\n      copiers.setDone();\n\n      const total = rowCounts.reduce(\n        (acc, curr) => ({\n          rows: acc.rows + curr.rows,\n          flushTime: acc.flushTime + curr.flushTime,\n        }),\n        {rows: 0, flushTime: 0},\n      );\n\n      statusPublisher.publish(\n        lc,\n        'Indexing',\n        `Creating ${indexes.length} indexes`,\n        5000,\n      );\n      const indexStart = performance.now();\n      createLiteIndices(tx, indexes);\n      const index = performance.now() - indexStart;\n      lc.info?.(`Created indexes (${index.toFixed(3)} ms)`);\n\n      if (slotName && replicaID) {\n        await initReplica(sql, shard, replicaID, published, context);\n      } else {\n        assert(shadow, 'expected to be in shadow sync if there is no slotName');\n        const rowsByTable = new Map<string, number>();\n        for (let i = 0; i < downloads.length; i++) {\n          rowsByTable.set(downloads[i].status.table, rowCounts[i].rows);\n        }\n        verifyShadowReplica(lc, tx, published, rowsByTable);\n      }\n\n      const elapsed = performance.now() - start;\n      lc.info?.(\n        `Synced ${total.rows.toLocaleString()} rows of ${numTables} tables in ${publications} up to ${lsn} ` +\n          `(flush: ${total.flushTime.toFixed(3)}, index: ${index.toFixed(3)}, total: ${elapsed.toFixed(3)} ms)`,\n      );\n    } finally {\n      // All meaningful errors are handled at the processReadTask() call site.\n      void copyPool.end().catch(e => lc.warn?.(`Error closing copyPool`, e));\n    }\n  } catch (e) {\n    if (slotName) {\n      // If initial-sync did not succeed, make a best effort to drop the\n      // orphaned replication slot to avoid running out of slots in\n      // pathological cases that result in repeated failures.\n      lc.warn?.(`dropping replication slot ${slotName}`, e);\n      await sql`\n        SELECT pg_drop_replication_slot(slot_name) FROM pg_replication_slots\n          WHERE slot_name = ${slotName};\n      `.catch(e => lc.warn?.(`Unable to drop replication slot ${slotName}`, e));\n    }\n    await statusPublisher.publishAndThrowError(lc, 'Initializing', e);\n  } finally {\n    statusPublisher.stop();\n    if (releaseShadowSnapshot) {\n      await releaseShadowSnapshot().catch(e =>\n        lc.warn?.(`Error releasing shadow snapshot`, e),\n      );\n    }\n    if (replicationSession) {\n      await replicationSession.end();\n    }\n    await sql.end();\n  }\n}\n\nexport type ShadowSyncOptions = {\n  sampleRate: number;\n  maxRowsPerTable: number;\n  /**\n   * Parent directory for the throwaway SQLite replica. Defaults to the OS\n   * tmpdir. Primarily for tests that need to isolate the scratch directory.\n   */\n  parentDir?: string | undefined;\n};\n\n/**\n * Exercises the initial-sync code path against a sample of rows from every\n * published table, writing into a throwaway SQLite database that is deleted\n * when the run ends. Produces zero upstream mutations: no replication slot,\n * no `addReplica`, no `dropShard`, no status events.\n *\n * Intended to be invoked periodically so that if a customer ever needs a\n * full reset, we have recent confidence that `initialSync` still works.\n * The shard must already be initialized upstream.\n */\nexport async function shadowInitialSync(\n  lc: LogContext,\n  shard: ShardConfig,\n  upstreamURI: string,\n  shadow: ShadowSyncOptions,\n  context: ServerContext,\n  syncOptions?: Pick<InitialSyncOptions, 'textCopy'>,\n): Promise<void> {\n  const dir = await mkdtemp(\n    join(shadow.parentDir ?? tmpdir(), 'zero-shadow-sync-'),\n  );\n  const dbPath = join(dir, 'shadow-replica.db');\n  const db = new Database(lc, dbPath);\n  try {\n    await initialSync(\n      lc,\n      shard,\n      db,\n      upstreamURI,\n      {\n        // Shadow sync copies small samples, so one worker is plenty —\n        // no reason to burn additional upstream connections.\n        tableCopyWorkers: 1,\n        textCopy: syncOptions?.textCopy,\n        shadow,\n      },\n      context,\n    );\n  } finally {\n    try {\n      db.close();\n    } catch (e) {\n      lc.warn?.(`Error closing shadow replica db`, e);\n    }\n    await rm(dir, {recursive: true, force: true}).catch(e =>\n      lc.warn?.(`Error cleaning up shadow replica dir ${dir}`, e),\n    );\n  }\n}\n\nasync function checkUpstreamConfig(sql: PostgresDB) {\n  const {walLevel, version} = (\n    await sql<{walLevel: string; version: number}[]>`\n      SELECT current_setting('wal_level') as \"walLevel\", \n             current_setting('server_version_num') as \"version\";\n  `\n  )[0];\n\n  if (walLevel !== 'logical') {\n    throw new Error(\n      `Postgres must be configured with \"wal_level = logical\" (currently: \"${walLevel})`,\n    );\n  }\n  if (version < PG_15) {\n    throw new Error(\n      `Must be running Postgres 15 or higher (currently: \"${version}\")`,\n    );\n  }\n  return version;\n}\n\nasync function ensurePublishedTables(\n  lc: LogContext,\n  sql: PostgresDB,\n  shard: ShardConfig,\n  validate = true,\n): Promise<{publications: string[]}> {\n  const {database, host} = sql.options;\n  lc.info?.(`Ensuring upstream PUBLICATION on ${database}@${host}`);\n\n  await ensureShardSchema(lc, sql, shard);\n  const {publications} = await getInternalShardConfig(sql, shard);\n\n  if (validate) {\n    let valid = false;\n    const nonInternalPublications = publications.filter(\n      p => !p.startsWith('_'),\n    );\n    const exists = await sql`\n      SELECT pubname FROM pg_publication WHERE pubname IN ${sql(publications)}\n      `.values();\n    if (exists.length !== publications.length) {\n      lc.warn?.(\n        `some configured publications [${publications}] are missing: ` +\n          `[${exists.flat()}]. resyncing`,\n      );\n    } else if (\n      !equals(new Set(shard.publications), new Set(nonInternalPublications))\n    ) {\n      lc.warn?.(\n        `requested publications [${shard.publications}] differ from previous` +\n          `publications [${nonInternalPublications}]. resyncing`,\n      );\n    } else {\n      valid = true;\n    }\n    if (!valid) {\n      await sql.unsafe(dropShard(shard.appID, shard.shardNum));\n      return ensurePublishedTables(lc, sql, shard, false);\n    }\n  }\n  return {publications};\n}\n\nfunction startTableCopyWorkers(\n  lc: LogContext,\n  db: PostgresDB,\n  snapshot: string,\n  numWorkers: number,\n  numTables: number,\n): TransactionPool {\n  const {init} = importSnapshot(snapshot);\n  const tableCopiers = new TransactionPool(lc, {\n    mode: Mode.READONLY,\n    init,\n    initialWorkers: numWorkers,\n  });\n  tableCopiers.run(db);\n\n  lc.info?.(`Started ${numWorkers} workers to copy ${numTables} tables`);\n\n  if (parseInt(process.versions.node) < 22) {\n    lc.warn?.(\n      `\\n\\n\\n` +\n        `Older versions of Node have a bug that results in an unresponsive\\n` +\n        `Postgres connection after running certain combinations of COPY commands.\\n` +\n        `If initial sync hangs, run zero-cache with Node v22+. This has the additional\\n` +\n        `benefit of being consistent with the Node version run in the production container image.` +\n        `\\n\\n\\n`,\n    );\n  }\n  return tableCopiers;\n}\n\n/**\n * Shadow-mode alternative to `createReplicationSlot`: opens a dedicated\n * READ ONLY REPEATABLE READ transaction on a normal connection, exports the\n * snapshot and captures the current WAL LSN, then holds the transaction\n * open until `release()` is called. The held transaction keeps the snapshot\n * importable by the table-copy workers for the duration of the COPY phase.\n *\n * Idle-in-transaction timeout is disabled locally so the exporter doesn't\n * get killed while workers are still importing.\n */\nasync function acquireExportedSnapshotForShadowSync(\n  lc: LogContext,\n  upstreamURI: string,\n): Promise<{\n  snapshot: string;\n  lsn: string;\n  release: () => Promise<void>;\n}> {\n  const holder = await connectPgClient(\n    lc,\n    upstreamURI,\n    'shadow-initial-sync-snapshot',\n    {\n      max: 1,\n    },\n  );\n  const ready = resolver<{snapshot: string; lsn: string}>();\n  const release = resolver<void>();\n  const held = holder\n    .begin(Mode.READONLY, async tx => {\n      await tx`SET LOCAL idle_in_transaction_session_timeout = 0`.execute();\n      const [row] = await tx<{snapshot: string; lsn: string}[]>`\n        SELECT pg_export_snapshot() AS snapshot,\n               pg_current_wal_lsn()::text AS lsn`;\n      ready.resolve(row);\n      await release.promise;\n    })\n    .catch(e => ready.reject(e));\n\n  let snapshot: string;\n  let lsn: string;\n  try {\n    ({snapshot, lsn} = await ready.promise);\n  } catch (e) {\n    await holder\n      .end()\n      .catch(err =>\n        lc.warn?.(`Error ending shadow snapshot holder after failure`, err),\n      );\n    throw e;\n  }\n  lc.info?.(\n    `Exported snapshot ${snapshot} at LSN ${lsn} (shadow initial sync)`,\n  );\n  return {\n    snapshot,\n    lsn,\n    release: async () => {\n      release.resolve();\n      try {\n        await held;\n      } catch (e) {\n        lc.warn?.(`snapshot holder transaction ended with error`, e);\n      }\n      await holder.end();\n    },\n  };\n}\n\nfunction createLiteTables(\n  tx: Database,\n  tables: PublishedTableSpec[],\n  initialVersion: string,\n) {\n  // TODO: Figure out how to reuse the ChangeProcessor here to avoid\n  //       duplicating the ColumnMetadata logic.\n  const columnMetadata = must(ColumnMetadataStore.getInstance(tx));\n  for (const t of tables) {\n    tx.exec(createLiteTableStatement(mapPostgresToLite(t, initialVersion)));\n    const tableName = liteTableName(t);\n    for (const [colName, colSpec] of Object.entries(t.columns)) {\n      columnMetadata.insert(tableName, colName, colSpec);\n    }\n  }\n}\n\nfunction createLiteIndices(tx: Database, indices: IndexSpec[]) {\n  for (const index of indices) {\n    tx.exec(createLiteIndexStatement(mapPostgresToLiteIndex(index)));\n  }\n}\n\n/**\n * Runs structural assertions over a just-synced replica and throws if any\n * fail. Only called in shadow mode — a successful return means the replica\n * is schema-complete, row-count consistent, and its column metadata is in\n * sync with its lite schema.\n *\n * Note: this intentionally does NOT verify ZQL-queryability. Tables that\n * `computeZqlSpecs` drops (no PK / no all-NOT-NULL unique index, unsupported\n * column types, etc.) are silently skipped in production too — there's\n * nothing shadow-specific about them, so failing here would diverge from\n * prod over an upstream-schema condition prod accepts.\n *\n * Exported for testing.\n */\nexport function verifyShadowReplica(\n  lc: LogContext,\n  db: Database,\n  published: {tables: PublishedTableSpec[]; indexes: IndexSpec[]},\n  rowsByTable: ReadonlyMap<string, number>,\n): void {\n  const issues: string[] = [];\n  let columnsChecked = 0;\n  let rowsChecked = 0;\n\n  // 1. Schema completeness: every published table exists in the replica\n  //    with at least the expected column set.\n  const liteTables = listTables(db);\n  const liteTableByName = new Map(liteTables.map(t => [t.name, t]));\n  for (const pt of published.tables) {\n    const name = liteTableName(pt);\n    const lite = liteTableByName.get(name);\n    if (!lite) {\n      issues.push(`missing table in replica: ${name}`);\n      continue;\n    }\n    for (const col of Object.keys(pt.columns)) {\n      columnsChecked++;\n      if (!(col in lite.columns)) {\n        issues.push(`column missing in replica table ${name}: ${col}`);\n      }\n    }\n  }\n\n  //    Every published index exists in the replica.\n  const liteIndexNames = new Set(listIndexes(db).map(i => i.name));\n  for (const ix of published.indexes) {\n    const mapped = mapPostgresToLiteIndex(ix);\n    if (!liteIndexNames.has(mapped.name)) {\n      issues.push(\n        `missing index in replica: ${mapped.name} on ${mapped.tableName}`,\n      );\n    }\n  }\n\n  // 2. Row counts: SQLite COUNT(*) matches the in-memory copy counter.\n  for (const [table, expected] of rowsByTable) {\n    try {\n      const [row] = db\n        .prepare(`SELECT COUNT(*) as count FROM \"${table}\"`)\n        .all<{count: number}>();\n      if (row.count !== expected) {\n        issues.push(\n          `row count mismatch for table ${table}: ` +\n            `copy counter reported ${expected}, replica has ${row.count}`,\n        );\n      } else {\n        rowsChecked += row.count;\n      }\n    } catch (e) {\n      issues.push(`could not count rows in table ${table}: ${String(e)}`);\n    }\n  }\n\n  // 3. Column metadata: every published column has a _zero.column_metadata row.\n  const meta = must(ColumnMetadataStore.getInstance(db));\n  for (const pt of published.tables) {\n    const name = liteTableName(pt);\n    const rows = meta.getTable(name);\n    for (const col of Object.keys(pt.columns)) {\n      if (!rows.has(col)) {\n        issues.push(`missing column_metadata row for ${name}.${col}`);\n      }\n    }\n  }\n\n  if (issues.length) {\n    throw new Error(\n      `Shadow replica verification failed (${issues.length} issue(s)):\\n` +\n        issues.map(i => `  - ${i}`).join('\\n'),\n    );\n  }\n\n  lc.info?.(\n    `Shadow replica verification passed: ` +\n      `${published.tables.length} tables, ` +\n      `${published.indexes.length} indexes, ` +\n      `${columnsChecked} columns, ` +\n      `${rowsChecked.toLocaleString()} rows`,\n  );\n}\n\n// Verified empirically that batches of 50 seem to be the sweet spot,\n// similar to the report in https://sqlite.org/forum/forumpost/8878a512d3652655\n//\n// Exported for testing.\nexport const INSERT_BATCH_SIZE = 50;\n\nconst MB = 1024 * 1024;\nconst MAX_BUFFERED_ROWS = 10_000;\nconst BUFFERED_SIZE_THRESHOLD = 8 * MB;\n\nexport type DownloadStatements = {\n  select: string;\n  getTotalRows: string;\n  getTotalBytes: string;\n};\n\n/**\n * Produces ` TABLESAMPLE BERNOULLI(n)` when `sampleRate` is < 1, else `''`.\n * Row-level Bernoulli sampling is used (rather than SYSTEM) because it\n * produces a more uniform sample and, unlike SYSTEM, still returns rows\n * for small tables at low rates.\n */\nfunction tableSampleClause(sampleRate: number | undefined): string {\n  if (sampleRate === undefined || sampleRate >= 1) {\n    return '';\n  }\n  // Round away float noise (e.g. 0.3 * 100 = 30.000000000000004) while still\n  // preserving sub-integer rates like 0.001 (= 0.1%).\n  const pct = parseFloat((sampleRate * 100).toFixed(6));\n  return /*sql*/ ` TABLESAMPLE BERNOULLI(${pct})`;\n}\n\nfunction limitClause(maxRowsPerTable: number | undefined): string {\n  return maxRowsPerTable !== undefined\n    ? /*sql*/ ` LIMIT ${maxRowsPerTable}`\n    : '';\n}\n\n/**\n * Returns the SELECT column expressions for binary COPY, casting columns\n * without a known binary decoder to `::text`.\n */\nexport function makeBinarySelectExprs(\n  table: PublishedTableSpec,\n  cols: string[],\n): string[] {\n  return cols.map(col => {\n    const spec = table.columns[col];\n    return hasBinaryDecoder(spec) ? id(col) : `${id(col)}::text`;\n  });\n}\n\nexport function makeDownloadStatements(\n  table: PublishedTableSpec,\n  cols: string[],\n  sampleRate?: number | undefined,\n  maxRowsPerTable?: number | undefined,\n  selectExprs?: string[] | undefined,\n): DownloadStatements {\n  const filterConditions = Object.values(table.publications)\n    .map(({rowFilter}) => rowFilter)\n    .filter(f => !!f); // remove nulls\n  const where =\n    filterConditions.length === 0\n      ? ''\n      : /*sql*/ `WHERE ${filterConditions.join(' OR ')}`;\n  const sample = tableSampleClause(sampleRate);\n  const limit = limitClause(maxRowsPerTable);\n  const fromTable = /*sql*/ `FROM ${id(table.schema)}.${id(table.name)}${sample} ${where}`;\n  const select = /*sql*/ `SELECT ${(selectExprs ?? cols.map(id)).join(',')} ${fromTable}${limit}`;\n  if (limit) {\n    // With LIMIT, wrap counts/sums in a subquery so they reflect the\n    // capped rowset rather than the full (sampled) table.\n    const bytesExpr = cols\n      .map(col => `COALESCE(pg_column_size(${id(col)}), 0)`)\n      .join(' + ');\n    return {\n      select,\n      getTotalRows: /*sql*/ `SELECT COUNT(*)::bigint AS \"totalRows\" FROM (SELECT 1 AS _ ${fromTable}${limit}) s`,\n      getTotalBytes: /*sql*/ `SELECT COALESCE(SUM(b), 0)::bigint AS \"totalBytes\" FROM (SELECT (${bytesExpr}) AS b ${fromTable}${limit}) s`,\n    };\n  }\n  const totalBytes = `(${cols.map(col => `SUM(COALESCE(pg_column_size(${id(col)}), 0))`).join(' + ')})`;\n  return {\n    select,\n    getTotalRows: /*sql*/ `SELECT COUNT(*) AS \"totalRows\" ${fromTable}`,\n    getTotalBytes: /*sql*/ `SELECT ${totalBytes} AS \"totalBytes\" ${fromTable}`,\n  };\n}\n\ntype DownloadState = {\n  spec: PublishedTableSpec;\n  status: DownloadStatus;\n};\n\n// Exported for testing.\nexport async function getInitialDownloadState(\n  lc: LogContext,\n  sql: PostgresDB,\n  spec: PublishedTableSpec,\n  skipTotals: boolean,\n): Promise<DownloadState> {\n  const start = performance.now();\n  const table = liteTableName(spec);\n  const columns = Object.keys(spec.columns);\n  if (skipTotals) {\n    // Shadow sync suppresses status events, so the pg_class\n    // estimates would be computed and thrown away.\n    return {\n      spec,\n      status: {table, columns, rows: 0, totalRows: 0, totalBytes: 0},\n    };\n  }\n  // Use pg_class estimates instead of expensive COUNT(*) and\n  // SUM(pg_column_size(...)) full table scans. The exact values are only\n  // used for progress reporting, so estimates are sufficient.\n  const qualifiedName = `${id(spec.schema)}.${id(spec.name)}`;\n  const estimateResult = await sql<\n    {totalRows: number; totalBytes: number}[]\n  >`SELECT GREATEST(reltuples, 0)::float8 AS \"totalRows\",\n         pg_table_size(oid)::float8 AS \"totalBytes\"\n    FROM pg_class\n    WHERE oid = ${qualifiedName}::regclass`;\n\n  const {totalRows, totalBytes} = estimateResult[0] ?? {\n    totalRows: 0,\n    totalBytes: 0,\n  };\n\n  const state: DownloadState = {\n    spec,\n    status: {\n      table,\n      columns,\n      rows: 0,\n      totalRows,\n      totalBytes,\n    },\n  };\n  const elapsed = (performance.now() - start).toFixed(3);\n  lc.info?.(`Computed initial download state for ${table} (${elapsed} ms)`, {\n    state: state.status,\n  });\n  return state;\n}\n\nfunction copy(\n  lc: LogContext,\n  {spec: table, status}: DownloadState,\n  dbClient: PostgresDB,\n  from: PostgresTransaction,\n  to: Database,\n  textCopy: boolean,\n  sampleRate?: number | undefined,\n  maxRowsPerTable?: number | undefined,\n) {\n  if (textCopy) {\n    return copyText(\n      lc,\n      table,\n      status,\n      dbClient,\n      from,\n      to,\n      sampleRate,\n      maxRowsPerTable,\n    );\n  }\n  return copyBinary(lc, table, status, from, to, sampleRate, maxRowsPerTable);\n}\n\nasync function copyBinary(\n  lc: LogContext,\n  table: PublishedTableSpec,\n  status: DownloadStatus,\n  from: PostgresTransaction,\n  to: Database,\n  sampleRate?: number | undefined,\n  maxRowsPerTable?: number | undefined,\n) {\n  const start = performance.now();\n  let flushTime = 0;\n\n  const tableName = liteTableName(table);\n  const orderedColumns = Object.entries(table.columns);\n\n  const columnNames = orderedColumns.map(([c]) => c);\n  const columnSpecs = orderedColumns.map(([_name, spec]) => spec);\n  const insertColumnList = columnNames.map(c => id(c)).join(',');\n\n  const valuesSql =\n    columnNames.length > 0 ? `(${'?,'.repeat(columnNames.length - 1)}?)` : '()';\n  const insertSql = /*sql*/ `\n    INSERT INTO \"${tableName}\" (${insertColumnList}) VALUES ${valuesSql}`;\n  const insertStmt = to.prepare(insertSql);\n  const insertBatchStmt = to.prepare(\n    insertSql + `,${valuesSql}`.repeat(INSERT_BATCH_SIZE - 1),\n  );\n\n  // Build SELECT with ::text casts for columns without a known binary decoder.\n  const select = makeDownloadStatements(\n    table,\n    columnNames,\n    sampleRate,\n    maxRowsPerTable,\n    makeBinarySelectExprs(table, columnNames),\n  ).select;\n\n  const decoders = orderedColumns.map(([, spec]) =>\n    hasBinaryDecoder(spec) ? makeBinaryDecoder(spec) : textCastDecoder,\n  );\n\n  const valuesPerRow = columnSpecs.length;\n  const valuesPerBatch = valuesPerRow * INSERT_BATCH_SIZE;\n\n  const pendingValues: LiteValueType[] = Array.from({\n    length: MAX_BUFFERED_ROWS * valuesPerRow,\n  });\n  let pendingRows = 0;\n  let pendingSize = 0;\n\n  function flush() {\n    const start = performance.now();\n    const flushedRows = pendingRows;\n    const flushedSize = pendingSize;\n\n    let l = 0;\n    for (; pendingRows > INSERT_BATCH_SIZE; pendingRows -= INSERT_BATCH_SIZE) {\n      insertBatchStmt.run(pendingValues.slice(l, (l += valuesPerBatch)));\n    }\n    for (; pendingRows > 0; pendingRows--) {\n      insertStmt.run(pendingValues.slice(l, (l += valuesPerRow)));\n    }\n    const flushedValues = flushedRows * valuesPerRow;\n    for (let i = 0; i < flushedValues; i++) {\n      pendingValues[i] = undefined as unknown as LiteValueType;\n    }\n    pendingSize = 0;\n    status.rows += flushedRows;\n\n    const elapsed = performance.now() - start;\n    flushTime += elapsed;\n    lc.debug?.(\n      `flushed ${flushedRows} ${tableName} rows (${flushedSize} bytes) in ${elapsed.toFixed(3)} ms`,\n    );\n  }\n\n  const binaryParser = new BinaryCopyParser();\n  let col = 0;\n\n  lc.info?.(`Starting binary copy stream of ${tableName}:`, select);\n\n  await pipeline(\n    await from\n      .unsafe(`COPY (${select}) TO STDOUT WITH (FORMAT binary)`)\n      .readable(),\n    new Writable({\n      highWaterMark: BUFFERED_SIZE_THRESHOLD,\n\n      write(\n        chunk: Buffer,\n        _encoding: string,\n        callback: (error?: Error) => void,\n      ) {\n        try {\n          for (const fieldBuf of binaryParser.parse(chunk)) {\n            pendingSize += fieldBuf === null ? 4 : fieldBuf.length;\n            pendingValues[pendingRows * valuesPerRow + col] =\n              fieldBuf === null ? null : decoders[col](fieldBuf);\n\n            if (++col === decoders.length) {\n              col = 0;\n              if (\n                ++pendingRows >= MAX_BUFFERED_ROWS - valuesPerRow ||\n                pendingSize >= BUFFERED_SIZE_THRESHOLD\n              ) {\n                flush();\n              }\n            }\n          }\n          callback();\n        } catch (e) {\n          callback(e instanceof Error ? e : new Error(String(e)));\n        }\n      },\n\n      final: (callback: (error?: Error) => void) => {\n        try {\n          flush();\n          callback();\n        } catch (e) {\n          callback(e instanceof Error ? e : new Error(String(e)));\n        }\n      },\n    }),\n  );\n\n  const elapsed = performance.now() - start;\n  lc.info?.(\n    `Finished copying ${status.rows} rows into ${tableName} ` +\n      `(flush: ${flushTime.toFixed(3)} ms) (total: ${elapsed.toFixed(3)} ms) `,\n  );\n  return {rows: status.rows, flushTime};\n}\n\nasync function copyText(\n  lc: LogContext,\n  table: PublishedTableSpec,\n  status: DownloadStatus,\n  dbClient: PostgresDB,\n  from: PostgresTransaction,\n  to: Database,\n  sampleRate?: number | undefined,\n  maxRowsPerTable?: number | undefined,\n) {\n  const start = performance.now();\n  let flushTime = 0;\n\n  const tableName = liteTableName(table);\n  const orderedColumns = Object.entries(table.columns);\n\n  const columnNames = orderedColumns.map(([c]) => c);\n  const columnSpecs = orderedColumns.map(([_name, spec]) => spec);\n  const insertColumnList = columnNames.map(c => id(c)).join(',');\n\n  const valuesSql =\n    columnNames.length > 0 ? `(${'?,'.repeat(columnNames.length - 1)}?)` : '()';\n  const insertSql = /*sql*/ `\n    INSERT INTO \"${tableName}\" (${insertColumnList}) VALUES ${valuesSql}`;\n  const insertStmt = to.prepare(insertSql);\n  const insertBatchStmt = to.prepare(\n    insertSql + `,${valuesSql}`.repeat(INSERT_BATCH_SIZE - 1),\n  );\n\n  const {select} = makeDownloadStatements(\n    table,\n    columnNames,\n    sampleRate,\n    maxRowsPerTable,\n  );\n  const valuesPerRow = columnSpecs.length;\n  const valuesPerBatch = valuesPerRow * INSERT_BATCH_SIZE;\n\n  const pendingValues: LiteValueType[] = Array.from({\n    length: MAX_BUFFERED_ROWS * valuesPerRow,\n  });\n  let pendingRows = 0;\n  let pendingSize = 0;\n\n  function flush() {\n    const start = performance.now();\n    const flushedRows = pendingRows;\n    const flushedSize = pendingSize;\n\n    let l = 0;\n    for (; pendingRows > INSERT_BATCH_SIZE; pendingRows -= INSERT_BATCH_SIZE) {\n      insertBatchStmt.run(pendingValues.slice(l, (l += valuesPerBatch)));\n    }\n    for (; pendingRows > 0; pendingRows--) {\n      insertStmt.run(pendingValues.slice(l, (l += valuesPerRow)));\n    }\n    const flushedValues = flushedRows * valuesPerRow;\n    for (let i = 0; i < flushedValues; i++) {\n      pendingValues[i] = undefined as unknown as LiteValueType;\n    }\n    pendingSize = 0;\n    status.rows += flushedRows;\n\n    const elapsed = performance.now() - start;\n    flushTime += elapsed;\n    lc.debug?.(\n      `flushed ${flushedRows} ${tableName} rows (${flushedSize} bytes) in ${elapsed.toFixed(3)} ms`,\n    );\n  }\n\n  lc.info?.(`Starting text copy stream of ${tableName}:`, select);\n  const pgParsers = await getTypeParsers(dbClient, {returnJsonAsString: true});\n  const parsers = columnSpecs.map(c => {\n    const pgParse = pgParsers.getTypeParser(c.typeOID);\n    return (val: string) =>\n      liteValue(\n        pgParse(val) as PostgresValueType,\n        c.dataType,\n        JSON_STRINGIFIED,\n      );\n  });\n\n  const tsvParser = new TsvParser();\n  let col = 0;\n\n  await pipeline(\n    await from.unsafe(`COPY (${select}) TO STDOUT`).readable(),\n    new Writable({\n      highWaterMark: BUFFERED_SIZE_THRESHOLD,\n\n      write(\n        chunk: Buffer,\n        _encoding: string,\n        callback: (error?: Error) => void,\n      ) {\n        try {\n          for (const text of tsvParser.parse(chunk)) {\n            pendingSize += text === null ? 4 : text.length;\n            pendingValues[pendingRows * valuesPerRow + col] =\n              text === null ? null : parsers[col](text);\n\n            if (++col === parsers.length) {\n              col = 0;\n              if (\n                ++pendingRows >= MAX_BUFFERED_ROWS - valuesPerRow ||\n                pendingSize >= BUFFERED_SIZE_THRESHOLD\n              ) {\n                flush();\n              }\n            }\n          }\n          callback();\n        } catch (e) {\n          callback(e instanceof Error ? e : new Error(String(e)));\n        }\n      },\n\n      final: (callback: (error?: Error) => void) => {\n        try {\n          flush();\n          callback();\n        } catch (e) {\n          callback(e instanceof Error ? e : new Error(String(e)));\n        }\n      },\n    }),\n  );\n\n  const elapsed = performance.now() - start;\n  lc.info?.(\n    `Finished copying ${status.rows} rows into ${tableName} ` +\n      `(flush: ${flushTime.toFixed(3)} ms) (total: ${elapsed.toFixed(3)} ms) `,\n  );\n  return {rows: status.rows, flushTime};\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AA8FA,eAAsB,YACpB,IACA,OACA,IACA,aACA,aACA,SACA;CACA,IAAI,CAAC,0BAA0B,KAAK,MAAM,KAAK,GAC7C,MAAM,IAAI,MACR,0FACF;CAEF,MAAM,EACJ,kBACA,aACA,WAAW,OACX,0BAA0B,OAC1B,WACE;CACJ,MAAM,eAAe,cAAc,MAAM,YAAY,QAAQ,IAAI;CACjE,MAAM,MAAM,MAAM,gBAAgB,IAAI,aAAa,cAAc;CAIjE,MAAM,qBAAqB,SACvB,KAAA,IACA,SAAS,IAAI,aAAa,oCAAoC;GAC3D,gBAAgB;EACjB,YAAY,EAAC,aAAa,WAAU;CACtC,CAAC;CAEL,MAAM,YAAY,KAAK,IAAI,EAAE,SAAS;CACtC,IAAI;CACJ,MAAM,kBAAkB,2BAA2B,sBACjD,IACA,SAAS,YAAY,CAAC,IAAI,KAAA,CAC5B,EAAE,QAAQ,IAAI,cAAc;CAC5B,IAAI;CACJ,IAAI;EACF,MAAM,YAAY,MAAM,oBAAoB,GAAG;EAM/C,MAAM,EAAC,iBAAgB,SACnB,MAAM,uBAAuB,KAAK,KAAK,IACvC,MAAM,sBAAsB,IAAI,KAAK,KAAK;EAC9C,GAAG,OAAO,wCAAwC,aAAa,EAAE;EAEjE,MAAM,EAAC,UAAU,SAAQ,IAAI;EAC7B,GAAG,OACD,SACI,kCAAkC,SAAS,GAAG,KAAK,kBACnD,kCAAkC,SAAS,GAAG,MACpD;EAEA,IAAI;EACJ,IAAI;EAEJ,IAAI,QAAQ;GACV,MAAM,WAAW,MAAM,qCACrB,IACA,WACF;GACA,WAAW,SAAS;GACpB,MAAM,SAAS;GACf,wBAAwB,SAAS;EACnC,OAAO;GACL,MAAM,OAAO,MAAM,qBACjB,IACA,KACA,KAAK,kBAAkB,GACvB,OACA,WACA,2BAA2B,aAAA,IAC7B;GACA,WAAW,KAAK;GAChB,MAAM,KAAK;GACX,WAAW,KAAK;EAClB;EAEA,MAAM,iBAAiB,qBAAqB,GAAG;EAE/C,qBAAqB,IAAI,cAAc,gBAAgB,OAAO;EAG9D,MAAM,QAAQ,YAAY,IAAI;EAE9B,MAAM,YAAY,MAAM,MACtB,KACA,OAAM,OAAM;GACV,MAAM,GAAG,OAAgB,6BAA6B,SAAS,EAAE;GACjE,OAAO,mBAAmB,IAAI,YAAY;EAC5C,GACA,EAAC,MAAM,SAAa,CACtB;EAEA,qBAAqB,IAAI,SAAS;EAGlC,MAAM,EAAC,QAAQ,YAAW;EAC1B,MAAM,YAAY,OAAO;EACzB,IAAI,SAAS,MAAM,WAAW,mBAAmB,WAC/C,GAAG,OACD,8CAA8C,iBAAiB,MAC1D,UAAU,+CACjB;EAEF,MAAM,aACJ,SAAS,MAAM,UACX,YACA,KAAK,IAAI,kBAAkB,SAAS;EAE1C,MAAM,WAAW,MAAM,gBACrB,IACA,aACA,4BACA;GACE,KAAK;IACJ,iBAAiB;EACpB,CACF;EACA,MAAM,UAAU,sBACd,IACA,UACA,UACA,YACA,SACF;EACA,IAAI;GACF,iBAAiB,IAAI,QAAQ,cAAc;GAC3C,MAAM,aAAa,QAAQ;GAC3B,MAAM,kBAAkB,QAAQ;GAChC,MAAM,YAAY,MAAM,QAAQ,IAC9B,OAAO,KAAI,SACT,QAAQ,iBAAiB,IAAI,OAC3B,wBAAwB,IAAI,IAAI,MAAM,WAAW,KAAA,CAAS,CAC5D,CACF,CACF;GACA,gBAAgB,QACd,IACA,gBACA,WAAW,UAAU,8BAA8B,kBACnD,YACO,EAAC,gBAAgB,UAAU,KAAK,EAAC,aAAY,MAAM,EAAC,EAC7D;GAEA,cAAmB,MAAM;GACzB,MAAM,YAAY,MAAM,QAAQ,IAC9B,UAAU,KAAI,UACZ,QAAQ,iBAAiB,IAAI,OAC3B,KACE,IACA,OACA,UACA,IACA,IACA,UACA,YACA,eACF,CACF,CACF,CACF;GACA,cAAmB,eAAe,IAAI,cAAc;GACpD,QAAQ,QAAQ;GAEhB,MAAM,QAAQ,UAAU,QACrB,KAAK,UAAU;IACd,MAAM,IAAI,OAAO,KAAK;IACtB,WAAW,IAAI,YAAY,KAAK;GAClC,IACA;IAAC,MAAM;IAAG,WAAW;GAAC,CACxB;GAEA,gBAAgB,QACd,IACA,YACA,YAAY,QAAQ,OAAO,WAC3B,GACF;GACA,MAAM,aAAa,YAAY,IAAI;GACnC,kBAAkB,IAAI,OAAO;GAC7B,MAAM,QAAQ,YAAY,IAAI,IAAI;GAClC,GAAG,OAAO,oBAAoB,MAAM,QAAQ,CAAC,EAAE,KAAK;GAEpD,IAAI,YAAY,WACd,MAAM,YAAY,KAAK,OAAO,WAAW,WAAW,OAAO;QACtD;IACL,OAAO,QAAQ,uDAAuD;IACtE,MAAM,8BAAc,IAAI,IAAoB;IAC5C,KAAK,IAAI,IAAI,GAAG,IAAI,UAAU,QAAQ,KACpC,YAAY,IAAI,UAAU,GAAG,OAAO,OAAO,UAAU,GAAG,IAAI;IAE9D,oBAAoB,IAAI,IAAI,WAAW,WAAW;GACpD;GAEA,MAAM,UAAU,YAAY,IAAI,IAAI;GACpC,GAAG,OACD,UAAU,MAAM,KAAK,eAAe,EAAE,WAAW,UAAU,aAAa,aAAa,SAAS,IAAI,WACrF,MAAM,UAAU,QAAQ,CAAC,EAAE,WAAW,MAAM,QAAQ,CAAC,EAAE,WAAW,QAAQ,QAAQ,CAAC,EAAE,KACpG;EACF,UAAU;GAER,SAAc,IAAI,EAAE,OAAM,MAAK,GAAG,OAAO,0BAA0B,CAAC,CAAC;EACvE;CACF,SAAS,GAAG;EACV,IAAI,UAAU;GAIZ,GAAG,OAAO,6BAA6B,YAAY,CAAC;GACpD,MAAM,GAAG;;8BAEe,SAAS;QAC/B,OAAM,MAAK,GAAG,OAAO,mCAAmC,YAAY,CAAC,CAAC;EAC1E;EACA,MAAM,gBAAgB,qBAAqB,IAAI,gBAAgB,CAAC;CAClE,UAAU;EACR,gBAAgB,KAAK;EACrB,IAAI,uBACF,MAAM,sBAAsB,EAAE,OAAM,MAClC,GAAG,OAAO,mCAAmC,CAAC,CAChD;EAEF,IAAI,oBACF,MAAM,mBAAmB,IAAI;EAE/B,MAAM,IAAI,IAAI;CAChB;AACF;;;;;;;;;;;AAsBA,eAAsB,kBACpB,IACA,OACA,aACA,QACA,SACA,aACe;CACf,MAAM,MAAM,MAAM,QAChB,KAAK,OAAO,aAAa,OAAO,GAAG,mBAAmB,CACxD;CAEA,MAAM,KAAK,IAAI,SAAS,IADT,KAAK,KAAK,mBACG,CAAM;CAClC,IAAI;EACF,MAAM,YACJ,IACA,OACA,IACA,aACA;GAGE,kBAAkB;GAClB,UAAU,aAAa;GACvB;EACF,GACA,OACF;CACF,UAAU;EACR,IAAI;GACF,GAAG,MAAM;EACX,SAAS,GAAG;GACV,GAAG,OAAO,mCAAmC,CAAC;EAChD;EACA,MAAM,GAAG,KAAK;GAAC,WAAW;GAAM,OAAO;EAAI,CAAC,EAAE,OAAM,MAClD,GAAG,OAAO,wCAAwC,OAAO,CAAC,CAC5D;CACF;AACF;AAEA,eAAe,oBAAoB,KAAiB;CAClD,MAAM,EAAC,UAAU,aACf,MAAM,GAA0C;;;KAIhD;CAEF,IAAI,aAAa,WACf,MAAM,IAAI,MACR,uEAAuE,SAAS,EAClF;CAEF,IAAI,UAAA,MACF,MAAM,IAAI,MACR,sDAAsD,QAAQ,GAChE;CAEF,OAAO;AACT;AAEA,eAAe,sBACb,IACA,KACA,OACA,WAAW,MACwB;CACnC,MAAM,EAAC,UAAU,SAAQ,IAAI;CAC7B,GAAG,OAAO,oCAAoC,SAAS,GAAG,MAAM;CAEhE,MAAM,kBAAkB,IAAI,KAAK,KAAK;CACtC,MAAM,EAAC,iBAAgB,MAAM,uBAAuB,KAAK,KAAK;CAE9D,IAAI,UAAU;EACZ,IAAI,QAAQ;EACZ,MAAM,0BAA0B,aAAa,QAC3C,MAAK,CAAC,EAAE,WAAW,GAAG,CACxB;EACA,MAAM,SAAS,MAAM,GAAG;4DACgC,IAAI,YAAY,EAAE;QACtE,OAAO;EACX,IAAI,OAAO,WAAW,aAAa,QACjC,GAAG,OACD,iCAAiC,aAAa,kBACxC,OAAO,KAAK,EAAE,aACtB;OACK,IACL,CAAC,OAAO,IAAI,IAAI,MAAM,YAAY,GAAG,IAAI,IAAI,uBAAuB,CAAC,GAErE,GAAG,OACD,2BAA2B,MAAM,aAAa,sCAC3B,wBAAwB,aAC7C;OAEA,QAAQ;EAEV,IAAI,CAAC,OAAO;GACV,MAAM,IAAI,OAAO,UAAU,MAAM,OAAO,MAAM,QAAQ,CAAC;GACvD,OAAO,sBAAsB,IAAI,KAAK,OAAO,KAAK;EACpD;CACF;CACA,OAAO,EAAC,aAAY;AACtB;AAEA,SAAS,sBACP,IACA,IACA,UACA,YACA,WACiB;CACjB,MAAM,EAAC,SAAQ,eAAe,QAAQ;CACtC,MAAM,eAAe,IAAI,gBAAgB,IAAI;EAC3C,MAAM;EACN;EACA,gBAAgB;CAClB,CAAC;CACD,aAAa,IAAI,EAAE;CAEnB,GAAG,OAAO,WAAW,WAAW,mBAAmB,UAAU,QAAQ;CAErE,IAAI,SAAS,QAAQ,SAAS,IAAI,IAAI,IACpC,GAAG,OACD,kUAMF;CAEF,OAAO;AACT;;;;;;;;;;;AAYA,eAAe,qCACb,IACA,aAKC;CACD,MAAM,SAAS,MAAM,gBACnB,IACA,aACA,gCACA,EACE,KAAK,EACP,CACF;CACA,MAAM,QAAQ,SAA0C;CACxD,MAAM,UAAU,SAAe;CAC/B,MAAM,OAAO,OACV,MAAM,UAAe,OAAM,OAAM;EAChC,MAAM,EAAE,oDAAoD,QAAQ;EACpE,MAAM,CAAC,OAAO,MAAM,EAAqC;;;EAGzD,MAAM,QAAQ,GAAG;EACjB,MAAM,QAAQ;CAChB,CAAC,EACA,OAAM,MAAK,MAAM,OAAO,CAAC,CAAC;CAE7B,IAAI;CACJ,IAAI;CACJ,IAAI;EACF,CAAC,CAAC,UAAU,OAAO,MAAM,MAAM;CACjC,SAAS,GAAG;EACV,MAAM,OACH,IAAI,EACJ,OAAM,QACL,GAAG,OAAO,qDAAqD,GAAG,CACpE;EACF,MAAM;CACR;CACA,GAAG,OACD,qBAAqB,SAAS,UAAU,IAAI,uBAC9C;CACA,OAAO;EACL;EACA;EACA,SAAS,YAAY;GACnB,QAAQ,QAAQ;GAChB,IAAI;IACF,MAAM;GACR,SAAS,GAAG;IACV,GAAG,OAAO,gDAAgD,CAAC;GAC7D;GACA,MAAM,OAAO,IAAI;EACnB;CACF;AACF;AAEA,SAAS,iBACP,IACA,QACA,gBACA;CAGA,MAAM,iBAAiB,KAAK,oBAAoB,YAAY,EAAE,CAAC;CAC/D,KAAK,MAAM,KAAK,QAAQ;EACtB,GAAG,KAAK,yBAAyB,kBAAkB,GAAG,cAAc,CAAC,CAAC;EACtE,MAAM,YAAY,cAAc,CAAC;EACjC,KAAK,MAAM,CAAC,SAAS,YAAY,OAAO,QAAQ,EAAE,OAAO,GACvD,eAAe,OAAO,WAAW,SAAS,OAAO;CAErD;AACF;AAEA,SAAS,kBAAkB,IAAc,SAAsB;CAC7D,KAAK,MAAM,SAAS,SAClB,GAAG,KAAK,yBAAyB,uBAAuB,KAAK,CAAC,CAAC;AAEnE;;;;;;;;;;;;;;;AAgBA,SAAgB,oBACd,IACA,IACA,WACA,aACM;CACN,MAAM,SAAmB,CAAC;CAC1B,IAAI,iBAAiB;CACrB,IAAI,cAAc;CAIlB,MAAM,aAAa,WAAW,EAAE;CAChC,MAAM,kBAAkB,IAAI,IAAI,WAAW,KAAI,MAAK,CAAC,EAAE,MAAM,CAAC,CAAC,CAAC;CAChE,KAAK,MAAM,MAAM,UAAU,QAAQ;EACjC,MAAM,OAAO,cAAc,EAAE;EAC7B,MAAM,OAAO,gBAAgB,IAAI,IAAI;EACrC,IAAI,CAAC,MAAM;GACT,OAAO,KAAK,6BAA6B,MAAM;GAC/C;EACF;EACA,KAAK,MAAM,OAAO,OAAO,KAAK,GAAG,OAAO,GAAG;GACzC;GACA,IAAI,EAAE,OAAO,KAAK,UAChB,OAAO,KAAK,mCAAmC,KAAK,IAAI,KAAK;EAEjE;CACF;CAGA,MAAM,iBAAiB,IAAI,IAAI,YAAY,EAAE,EAAE,KAAI,MAAK,EAAE,IAAI,CAAC;CAC/D,KAAK,MAAM,MAAM,UAAU,SAAS;EAClC,MAAM,SAAS,uBAAuB,EAAE;EACxC,IAAI,CAAC,eAAe,IAAI,OAAO,IAAI,GACjC,OAAO,KACL,6BAA6B,OAAO,KAAK,MAAM,OAAO,WACxD;CAEJ;CAGA,KAAK,MAAM,CAAC,OAAO,aAAa,aAC9B,IAAI;EACF,MAAM,CAAC,OAAO,GACX,QAAQ,kCAAkC,MAAM,EAAE,EAClD,IAAqB;EACxB,IAAI,IAAI,UAAU,UAChB,OAAO,KACL,gCAAgC,MAAM,0BACX,SAAS,gBAAgB,IAAI,OAC1D;OAEA,eAAe,IAAI;CAEvB,SAAS,GAAG;EACV,OAAO,KAAK,iCAAiC,MAAM,IAAI,OAAO,CAAC,GAAG;CACpE;CAIF,MAAM,OAAO,KAAK,oBAAoB,YAAY,EAAE,CAAC;CACrD,KAAK,MAAM,MAAM,UAAU,QAAQ;EACjC,MAAM,OAAO,cAAc,EAAE;EAC7B,MAAM,OAAO,KAAK,SAAS,IAAI;EAC/B,KAAK,MAAM,OAAO,OAAO,KAAK,GAAG,OAAO,GACtC,IAAI,CAAC,KAAK,IAAI,GAAG,GACf,OAAO,KAAK,mCAAmC,KAAK,GAAG,KAAK;CAGlE;CAEA,IAAI,OAAO,QACT,MAAM,IAAI,MACR,uCAAuC,OAAO,OAAO,iBACnD,OAAO,KAAI,MAAK,OAAO,GAAG,EAAE,KAAK,IAAI,CACzC;CAGF,GAAG,OACD,uCACK,UAAU,OAAO,OAAO,WACxB,UAAU,QAAQ,OAAO,YACzB,eAAe,YACf,YAAY,eAAe,EAAE,MACpC;AACF;AAQA,IAAM,KAAK,OAAO;AAClB,IAAM,oBAAoB;AAC1B,IAAM,0BAA0B,IAAI;;;;;;;AAcpC,SAAS,kBAAkB,YAAwC;CACjE,IAAI,eAAe,KAAA,KAAa,cAAc,GAC5C,OAAO;CAKT,OAAe,0BADH,YAAY,aAAa,KAAK,QAAQ,CAAC,CACV,EAAI;AAC/C;AAEA,SAAS,YAAY,iBAA6C;CAChE,OAAO,oBAAoB,KAAA,IACf,UAAU,oBAClB;AACN;;;;;AAMA,SAAgB,sBACd,OACA,MACU;CACV,OAAO,KAAK,KAAI,QAAO;EACrB,MAAM,OAAO,MAAM,QAAQ;EAC3B,OAAO,iBAAiB,IAAI,IAAI,GAAG,GAAG,IAAI,GAAG,GAAG,GAAG,EAAE;CACvD,CAAC;AACH;AAEA,SAAgB,uBACd,OACA,MACA,YACA,iBACA,aACoB;CACpB,MAAM,mBAAmB,OAAO,OAAO,MAAM,YAAY,EACtD,KAAK,EAAC,gBAAe,SAAS,EAC9B,QAAO,MAAK,CAAC,CAAC,CAAC;CAClB,MAAM,QACJ,iBAAiB,WAAW,IACxB,KACQ,SAAS,iBAAiB,KAAK,MAAM;CACnD,MAAM,SAAS,kBAAkB,UAAU;CAC3C,MAAM,QAAQ,YAAY,eAAe;CACzC,MAAM,YAAoB,QAAQ,GAAG,MAAM,MAAM,EAAE,GAAG,GAAG,MAAM,IAAI,IAAI,OAAO,GAAG;CACjF,MAAM,SAAiB,WAAW,eAAe,KAAK,IAAI,EAAE,GAAG,KAAK,GAAG,EAAE,GAAG,YAAY;CACxF,IAAI,OAAO;EAGT,MAAM,YAAY,KACf,KAAI,QAAO,2BAA2B,GAAG,GAAG,EAAE,MAAM,EACpD,KAAK,KAAK;EACb,OAAO;GACL;GACA,cAAsB,8DAA8D,YAAY,MAAM;GACtG,eAAuB,oEAAoE,UAAU,SAAS,YAAY,MAAM;EAClI;CACF;CACA,MAAM,aAAa,IAAI,KAAK,KAAI,QAAO,+BAA+B,GAAG,GAAG,EAAE,OAAO,EAAE,KAAK,KAAK,EAAE;CACnG,OAAO;EACL;EACA,cAAsB,kCAAkC;EACxD,eAAuB,UAAU,WAAW,mBAAmB;CACjE;AACF;AAQA,eAAsB,wBACpB,IACA,KACA,MACA,YACwB;CACxB,MAAM,QAAQ,YAAY,IAAI;CAC9B,MAAM,QAAQ,cAAc,IAAI;CAChC,MAAM,UAAU,OAAO,KAAK,KAAK,OAAO;CACxC,IAAI,YAGF,OAAO;EACL;EACA,QAAQ;GAAC;GAAO;GAAS,MAAM;GAAG,WAAW;GAAG,YAAY;EAAC;CAC/D;CAaF,MAAM,EAAC,WAAW,gBAAc,MAPH,GAE5B;;;kBAGe,GANS,GAAG,KAAK,MAAM,EAAE,GAAG,GAAG,KAAK,IAAI,IAM1B,aAEiB,MAAM;EACnD,WAAW;EACX,YAAY;CACd;CAEA,MAAM,QAAuB;EAC3B;EACA,QAAQ;GACN;GACA;GACA,MAAM;GACN;GACA;EACF;CACF;CACA,MAAM,WAAW,YAAY,IAAI,IAAI,OAAO,QAAQ,CAAC;CACrD,GAAG,OAAO,uCAAuC,MAAM,IAAI,QAAQ,OAAO,EACxE,OAAO,MAAM,OACf,CAAC;CACD,OAAO;AACT;AAEA,SAAS,KACP,IACA,EAAC,MAAM,OAAO,UACd,UACA,MACA,IACA,UACA,YACA,iBACA;CACA,IAAI,UACF,OAAO,SACL,IACA,OACA,QACA,UACA,MACA,IACA,YACA,eACF;CAEF,OAAO,WAAW,IAAI,OAAO,QAAQ,MAAM,IAAI,YAAY,eAAe;AAC5E;AAEA,eAAe,WACb,IACA,OACA,QACA,MACA,IACA,YACA,iBACA;CACA,MAAM,QAAQ,YAAY,IAAI;CAC9B,IAAI,YAAY;CAEhB,MAAM,YAAY,cAAc,KAAK;CACrC,MAAM,iBAAiB,OAAO,QAAQ,MAAM,OAAO;CAEnD,MAAM,cAAc,eAAe,KAAK,CAAC,OAAO,CAAC;CACjD,MAAM,cAAc,eAAe,KAAK,CAAC,OAAO,UAAU,IAAI;CAC9D,MAAM,mBAAmB,YAAY,KAAI,MAAK,GAAG,CAAC,CAAC,EAAE,KAAK,GAAG;CAE7D,MAAM,YACJ,YAAY,SAAS,IAAI,IAAI,KAAK,OAAO,YAAY,SAAS,CAAC,EAAE,MAAM;CACzE,MAAM,YAAoB;mBACT,UAAU,KAAK,iBAAiB,WAAW;CAC5D,MAAM,aAAa,GAAG,QAAQ,SAAS;CACvC,MAAM,kBAAkB,GAAG,QACzB,YAAY,IAAI,YAAY,OAAA,EAA4B,CAC1D;CAGA,MAAM,SAAS,uBACb,OACA,aACA,YACA,iBACA,sBAAsB,OAAO,WAAW,CAC1C,EAAE;CAEF,MAAM,WAAW,eAAe,KAAK,GAAG,UACtC,iBAAiB,IAAI,IAAI,kBAAkB,IAAI,IAAI,eACrD;CAEA,MAAM,eAAe,YAAY;CACjC,MAAM,iBAAiB,eAAA;CAEvB,MAAM,gBAAiC,MAAM,KAAK,EAChD,QAAQ,oBAAoB,aAC9B,CAAC;CACD,IAAI,cAAc;CAClB,IAAI,cAAc;CAElB,SAAS,QAAQ;EACf,MAAM,QAAQ,YAAY,IAAI;EAC9B,MAAM,cAAc;EACpB,MAAM,cAAc;EAEpB,IAAI,IAAI;EACR,OAAO,cAAA,IAAiC,eAAA,IACtC,gBAAgB,IAAI,cAAc,MAAM,GAAI,KAAK,cAAe,CAAC;EAEnE,OAAO,cAAc,GAAG,eACtB,WAAW,IAAI,cAAc,MAAM,GAAI,KAAK,YAAa,CAAC;EAE5D,MAAM,gBAAgB,cAAc;EACpC,KAAK,IAAI,IAAI,GAAG,IAAI,eAAe,KACjC,cAAc,KAAK,KAAA;EAErB,cAAc;EACd,OAAO,QAAQ;EAEf,MAAM,UAAU,YAAY,IAAI,IAAI;EACpC,aAAa;EACb,GAAG,QACD,WAAW,YAAY,GAAG,UAAU,SAAS,YAAY,aAAa,QAAQ,QAAQ,CAAC,EAAE,IAC3F;CACF;CAEA,MAAM,eAAe,IAAI,iBAAiB;CAC1C,IAAI,MAAM;CAEV,GAAG,OAAO,kCAAkC,UAAU,IAAI,MAAM;CAEhE,MAAM,WACJ,MAAM,KACH,OAAO,SAAS,OAAO,iCAAiC,EACxD,SAAS,GACZ,IAAI,SAAS;EACX,eAAe;EAEf,MACE,OACA,WACA,UACA;GACA,IAAI;IACF,KAAK,MAAM,YAAY,aAAa,MAAM,KAAK,GAAG;KAChD,eAAe,aAAa,OAAO,IAAI,SAAS;KAChD,cAAc,cAAc,eAAe,OACzC,aAAa,OAAO,OAAO,SAAS,KAAK,QAAQ;KAEnD,IAAI,EAAE,QAAQ,SAAS,QAAQ;MAC7B,MAAM;MACN,IACE,EAAE,eAAe,oBAAoB,gBACrC,eAAe,yBAEf,MAAM;KAEV;IACF;IACA,SAAS;GACX,SAAS,GAAG;IACV,SAAS,aAAa,QAAQ,IAAI,IAAI,MAAM,OAAO,CAAC,CAAC,CAAC;GACxD;EACF;EAEA,QAAQ,aAAsC;GAC5C,IAAI;IACF,MAAM;IACN,SAAS;GACX,SAAS,GAAG;IACV,SAAS,aAAa,QAAQ,IAAI,IAAI,MAAM,OAAO,CAAC,CAAC,CAAC;GACxD;EACF;CACF,CAAC,CACH;CAEA,MAAM,UAAU,YAAY,IAAI,IAAI;CACpC,GAAG,OACD,oBAAoB,OAAO,KAAK,aAAa,UAAU,WAC1C,UAAU,QAAQ,CAAC,EAAE,eAAe,QAAQ,QAAQ,CAAC,EAAE,MACtE;CACA,OAAO;EAAC,MAAM,OAAO;EAAM;CAAS;AACtC;AAEA,eAAe,SACb,IACA,OACA,QACA,UACA,MACA,IACA,YACA,iBACA;CACA,MAAM,QAAQ,YAAY,IAAI;CAC9B,IAAI,YAAY;CAEhB,MAAM,YAAY,cAAc,KAAK;CACrC,MAAM,iBAAiB,OAAO,QAAQ,MAAM,OAAO;CAEnD,MAAM,cAAc,eAAe,KAAK,CAAC,OAAO,CAAC;CACjD,MAAM,cAAc,eAAe,KAAK,CAAC,OAAO,UAAU,IAAI;CAC9D,MAAM,mBAAmB,YAAY,KAAI,MAAK,GAAG,CAAC,CAAC,EAAE,KAAK,GAAG;CAE7D,MAAM,YACJ,YAAY,SAAS,IAAI,IAAI,KAAK,OAAO,YAAY,SAAS,CAAC,EAAE,MAAM;CACzE,MAAM,YAAoB;mBACT,UAAU,KAAK,iBAAiB,WAAW;CAC5D,MAAM,aAAa,GAAG,QAAQ,SAAS;CACvC,MAAM,kBAAkB,GAAG,QACzB,YAAY,IAAI,YAAY,OAAA,EAA4B,CAC1D;CAEA,MAAM,EAAC,WAAU,uBACf,OACA,aACA,YACA,eACF;CACA,MAAM,eAAe,YAAY;CACjC,MAAM,iBAAiB,eAAA;CAEvB,MAAM,gBAAiC,MAAM,KAAK,EAChD,QAAQ,oBAAoB,aAC9B,CAAC;CACD,IAAI,cAAc;CAClB,IAAI,cAAc;CAElB,SAAS,QAAQ;EACf,MAAM,QAAQ,YAAY,IAAI;EAC9B,MAAM,cAAc;EACpB,MAAM,cAAc;EAEpB,IAAI,IAAI;EACR,OAAO,cAAA,IAAiC,eAAA,IACtC,gBAAgB,IAAI,cAAc,MAAM,GAAI,KAAK,cAAe,CAAC;EAEnE,OAAO,cAAc,GAAG,eACtB,WAAW,IAAI,cAAc,MAAM,GAAI,KAAK,YAAa,CAAC;EAE5D,MAAM,gBAAgB,cAAc;EACpC,KAAK,IAAI,IAAI,GAAG,IAAI,eAAe,KACjC,cAAc,KAAK,KAAA;EAErB,cAAc;EACd,OAAO,QAAQ;EAEf,MAAM,UAAU,YAAY,IAAI,IAAI;EACpC,aAAa;EACb,GAAG,QACD,WAAW,YAAY,GAAG,UAAU,SAAS,YAAY,aAAa,QAAQ,QAAQ,CAAC,EAAE,IAC3F;CACF;CAEA,GAAG,OAAO,gCAAgC,UAAU,IAAI,MAAM;CAC9D,MAAM,YAAY,MAAM,eAAe,UAAU,EAAC,oBAAoB,KAAI,CAAC;CAC3E,MAAM,UAAU,YAAY,KAAI,MAAK;EACnC,MAAM,UAAU,UAAU,cAAc,EAAE,OAAO;EACjD,QAAQ,QACN,UACE,QAAQ,GAAG,GACX,EAAE,UAAA,GAEJ;CACJ,CAAC;CAED,MAAM,YAAY,IAAI,UAAU;CAChC,IAAI,MAAM;CAEV,MAAM,WACJ,MAAM,KAAK,OAAO,SAAS,OAAO,YAAY,EAAE,SAAS,GACzD,IAAI,SAAS;EACX,eAAe;EAEf,MACE,OACA,WACA,UACA;GACA,IAAI;IACF,KAAK,MAAM,QAAQ,UAAU,MAAM,KAAK,GAAG;KACzC,eAAe,SAAS,OAAO,IAAI,KAAK;KACxC,cAAc,cAAc,eAAe,OACzC,SAAS,OAAO,OAAO,QAAQ,KAAK,IAAI;KAE1C,IAAI,EAAE,QAAQ,QAAQ,QAAQ;MAC5B,MAAM;MACN,IACE,EAAE,eAAe,oBAAoB,gBACrC,eAAe,yBAEf,MAAM;KAEV;IACF;IACA,SAAS;GACX,SAAS,GAAG;IACV,SAAS,aAAa,QAAQ,IAAI,IAAI,MAAM,OAAO,CAAC,CAAC,CAAC;GACxD;EACF;EAEA,QAAQ,aAAsC;GAC5C,IAAI;IACF,MAAM;IACN,SAAS;GACX,SAAS,GAAG;IACV,SAAS,aAAa,QAAQ,IAAI,IAAI,MAAM,OAAO,CAAC,CAAC,CAAC;GACxD;EACF;CACF,CAAC,CACH;CAEA,MAAM,UAAU,YAAY,IAAI,IAAI;CACpC,GAAG,OACD,oBAAoB,OAAO,KAAK,aAAa,UAAU,WAC1C,UAAU,QAAQ,CAAC,EAAE,eAAe,QAAQ,QAAQ,CAAC,EAAE,MACtE;CACA,OAAO;EAAC,MAAM,OAAO;EAAM;CAAS;AACtC"}

package/out/zero-cache/src/services/change-source/pg/logical-replication/binary-reader.js.map

@@ -1 +1 @@
-{"version":3,"file":"binary-reader.js","names":["#b","#p"],"sources":["../../../../../../../../zero-cache/src/services/change-source/pg/logical-replication/binary-reader.ts"],"sourcesContent":["// Forked from https://github.com/kibae/pg-logical-replication/blob/c55abddc62eadd61bd38922037ecb7a1469fa8c3/src/output-plugins/pgoutput/binary-reader.ts\n\n// should not use { fatal: true } because ErrorResponse can use invalid utf8 chars\nconst textDecoder = new TextDecoder();\n\n// https://www.postgresql.org/docs/14/protocol-message-types.html\nexport class BinaryReader {\n  #p = 0;\n  readonly #b: Uint8Array;\n\n  constructor(b: Uint8Array) {\n    this.#b = b;\n  }\n\n  readUint8() {\n    this.checkSize(1);\n\n    return this.#b[this.#p++];\n  }\n\n  readInt16() {\n    this.checkSize(2);\n\n    return (this.#b[this.#p++] << 8) | this.#b[this.#p++];\n  }\n\n  readInt32() {\n    this.checkSize(4);\n\n    return (\n      (this.#b[this.#p++] << 24) |\n      (this.#b[this.#p++] << 16) |\n      (this.#b[this.#p++] << 8) |\n      this.#b[this.#p++]\n    );\n  }\n\n  readString() {\n    const endIdx = this.#b.indexOf(0x00, this.#p);\n\n    if (endIdx < 0) {\n      // TODO PgError.protocol_violation\n      throw Error('unexpected end of message');\n    }\n\n    const strBuf = this.#b.subarray(this.#p, endIdx);\n    this.#p = endIdx + 1;\n\n    return this.decodeText(strBuf);\n  }\n\n  decodeText(strBuf: Uint8Array) {\n    return textDecoder.decode(strBuf);\n  }\n\n  read(n: number) {\n    this.checkSize(n);\n\n    return this.#b.subarray(this.#p, (this.#p += n));\n  }\n\n  checkSize(n: number) {\n    if (this.#b.length < this.#p + n) {\n      // TODO PgError.protocol_violation\n      throw Error('unexpected end of message');\n    }\n  }\n\n  array<T>(length: number, fn: () => T): T[] {\n    return Array.from({length}, fn, this);\n  }\n\n  // replication helpers\n  readLsn() {\n    const h = this.readUint32();\n    const l = this.readUint32();\n\n    if (h === 0 && l === 0) {\n      return null;\n    }\n\n    return `${h.toString(16).padStart(8, '0')}/${l\n      .toString(16)\n      .padStart(8, '0')}`.toUpperCase();\n  }\n\n  readTime() {\n    // (POSTGRES_EPOCH_JDATE - UNIX_EPOCH_JDATE) * USECS_PER_DAY == 946684800000000\n    return this.readUint64() + BigInt('946684800000000');\n  }\n\n  readUint64() {\n    return (\n      (BigInt(this.readUint32()) << BigInt(32)) | BigInt(this.readUint32())\n    );\n  }\n\n  readUint32() {\n    return this.readInt32() >>> 0;\n  }\n}\n"],"mappings":";AAGA,IAAM,cAAc,IAAI,aAAa;AAGrC,IAAa,eAAb,MAA0B;CACxB,KAAK;CACL;CAEA,YAAY,GAAe;AACzB,QAAA,IAAU;;CAGZ,YAAY;AACV,OAAK,UAAU,EAAE;AAEjB,SAAO,MAAA,EAAQ,MAAA;;CAGjB,YAAY;AACV,OAAK,UAAU,EAAE;AAEjB,SAAQ,MAAA,EAAQ,MAAA,QAAc,IAAK,MAAA,EAAQ,MAAA;;CAG7C,YAAY;AACV,OAAK,UAAU,EAAE;AAEjB,SACG,MAAA,EAAQ,MAAA,QAAc,KACtB,MAAA,EAAQ,MAAA,QAAc,KACtB,MAAA,EAAQ,MAAA,QAAc,IACvB,MAAA,EAAQ,MAAA;;CAIZ,aAAa;EACX,MAAM,SAAS,MAAA,EAAQ,QAAQ,GAAM,MAAA,EAAQ;AAE7C,MAAI,SAAS,EAEX,OAAM,MAAM,4BAA4B;EAG1C,MAAM,SAAS,MAAA,EAAQ,SAAS,MAAA,GAAS,OAAO;AAChD,QAAA,IAAU,SAAS;AAEnB,SAAO,KAAK,WAAW,OAAO;;CAGhC,WAAW,QAAoB;AAC7B,SAAO,YAAY,OAAO,OAAO;;CAGnC,KAAK,GAAW;AACd,OAAK,UAAU,EAAE;AAEjB,SAAO,MAAA,EAAQ,SAAS,MAAA,GAAU,MAAA,KAAW,EAAG;;CAGlD,UAAU,GAAW;AACnB,MAAI,MAAA,EAAQ,SAAS,MAAA,IAAU,EAE7B,OAAM,MAAM,4BAA4B;;CAI5C,MAAS,QAAgB,IAAkB;AACzC,SAAO,MAAM,KAAK,EAAC,QAAO,EAAE,IAAI,KAAK;;CAIvC,UAAU;EACR,MAAM,IAAI,KAAK,YAAY;EAC3B,MAAM,IAAI,KAAK,YAAY;AAE3B,MAAI,MAAM,KAAK,MAAM,EACnB,QAAO;AAGT,SAAO,GAAG,EAAE,SAAS,GAAG,CAAC,SAAS,GAAG,IAAI,CAAC,GAAG,EAC1C,SAAS,GAAG,CACZ,SAAS,GAAG,IAAI,GAAG,aAAa;;CAGrC,WAAW;AAET,SAAO,KAAK,YAAY,GAAG,OAAO,kBAAkB;;CAGtD,aAAa;AACX,SACG,OAAO,KAAK,YAAY,CAAC,IAAI,OAAO,GAAG,GAAI,OAAO,KAAK,YAAY,CAAC;;CAIzE,aAAa;AACX,SAAO,KAAK,WAAW,KAAK"}
+{"version":3,"file":"binary-reader.js","names":["#b","#p"],"sources":["../../../../../../../../zero-cache/src/services/change-source/pg/logical-replication/binary-reader.ts"],"sourcesContent":["// Forked from https://github.com/kibae/pg-logical-replication/blob/c55abddc62eadd61bd38922037ecb7a1469fa8c3/src/output-plugins/pgoutput/binary-reader.ts\n\n// should not use { fatal: true } because ErrorResponse can use invalid utf8 chars\nconst textDecoder = new TextDecoder();\n\n// https://www.postgresql.org/docs/14/protocol-message-types.html\nexport class BinaryReader {\n  #p = 0;\n  readonly #b: Uint8Array;\n\n  constructor(b: Uint8Array) {\n    this.#b = b;\n  }\n\n  readUint8() {\n    this.checkSize(1);\n\n    return this.#b[this.#p++];\n  }\n\n  readInt16() {\n    this.checkSize(2);\n\n    return (this.#b[this.#p++] << 8) | this.#b[this.#p++];\n  }\n\n  readInt32() {\n    this.checkSize(4);\n\n    return (\n      (this.#b[this.#p++] << 24) |\n      (this.#b[this.#p++] << 16) |\n      (this.#b[this.#p++] << 8) |\n      this.#b[this.#p++]\n    );\n  }\n\n  readString() {\n    const endIdx = this.#b.indexOf(0x00, this.#p);\n\n    if (endIdx < 0) {\n      // TODO PgError.protocol_violation\n      throw Error('unexpected end of message');\n    }\n\n    const strBuf = this.#b.subarray(this.#p, endIdx);\n    this.#p = endIdx + 1;\n\n    return this.decodeText(strBuf);\n  }\n\n  decodeText(strBuf: Uint8Array) {\n    return textDecoder.decode(strBuf);\n  }\n\n  read(n: number) {\n    this.checkSize(n);\n\n    return this.#b.subarray(this.#p, (this.#p += n));\n  }\n\n  checkSize(n: number) {\n    if (this.#b.length < this.#p + n) {\n      // TODO PgError.protocol_violation\n      throw Error('unexpected end of message');\n    }\n  }\n\n  array<T>(length: number, fn: () => T): T[] {\n    return Array.from({length}, fn, this);\n  }\n\n  // replication helpers\n  readLsn() {\n    const h = this.readUint32();\n    const l = this.readUint32();\n\n    if (h === 0 && l === 0) {\n      return null;\n    }\n\n    return `${h.toString(16).padStart(8, '0')}/${l\n      .toString(16)\n      .padStart(8, '0')}`.toUpperCase();\n  }\n\n  readTime() {\n    // (POSTGRES_EPOCH_JDATE - UNIX_EPOCH_JDATE) * USECS_PER_DAY == 946684800000000\n    return this.readUint64() + BigInt('946684800000000');\n  }\n\n  readUint64() {\n    return (\n      (BigInt(this.readUint32()) << BigInt(32)) | BigInt(this.readUint32())\n    );\n  }\n\n  readUint32() {\n    return this.readInt32() >>> 0;\n  }\n}\n"],"mappings":";AAGA,IAAM,cAAc,IAAI,YAAY;AAGpC,IAAa,eAAb,MAA0B;CACxB,KAAK;CACL;CAEA,YAAY,GAAe;EACzB,KAAKA,KAAK;CACZ;CAEA,YAAY;EACV,KAAK,UAAU,CAAC;EAEhB,OAAO,KAAKA,GAAG,KAAKC;CACtB;CAEA,YAAY;EACV,KAAK,UAAU,CAAC;EAEhB,OAAQ,KAAKD,GAAG,KAAKC,SAAS,IAAK,KAAKD,GAAG,KAAKC;CAClD;CAEA,YAAY;EACV,KAAK,UAAU,CAAC;EAEhB,OACG,KAAKD,GAAG,KAAKC,SAAS,KACtB,KAAKD,GAAG,KAAKC,SAAS,KACtB,KAAKD,GAAG,KAAKC,SAAS,IACvB,KAAKD,GAAG,KAAKC;CAEjB;CAEA,aAAa;EACX,MAAM,SAAS,KAAKD,GAAG,QAAQ,GAAM,KAAKC,EAAE;EAE5C,IAAI,SAAS,GAEX,MAAM,MAAM,2BAA2B;EAGzC,MAAM,SAAS,KAAKD,GAAG,SAAS,KAAKC,IAAI,MAAM;EAC/C,KAAKA,KAAK,SAAS;EAEnB,OAAO,KAAK,WAAW,MAAM;CAC/B;CAEA,WAAW,QAAoB;EAC7B,OAAO,YAAY,OAAO,MAAM;CAClC;CAEA,KAAK,GAAW;EACd,KAAK,UAAU,CAAC;EAEhB,OAAO,KAAKD,GAAG,SAAS,KAAKC,IAAK,KAAKA,MAAM,CAAE;CACjD;CAEA,UAAU,GAAW;EACnB,IAAI,KAAKD,GAAG,SAAS,KAAKC,KAAK,GAE7B,MAAM,MAAM,2BAA2B;CAE3C;CAEA,MAAS,QAAgB,IAAkB;EACzC,OAAO,MAAM,KAAK,EAAC,OAAM,GAAG,IAAI,IAAI;CACtC;CAGA,UAAU;EACR,MAAM,IAAI,KAAK,WAAW;EAC1B,MAAM,IAAI,KAAK,WAAW;EAE1B,IAAI,MAAM,KAAK,MAAM,GACnB,OAAO;EAGT,OAAO,GAAG,EAAE,SAAS,EAAE,EAAE,SAAS,GAAG,GAAG,EAAE,GAAG,EAC1C,SAAS,EAAE,EACX,SAAS,GAAG,GAAG,IAAI,YAAY;CACpC;CAEA,WAAW;EAET,OAAO,KAAK,WAAW,IAAI,OAAO,iBAAiB;CACrD;CAEA,aAAa;EACX,OACG,OAAO,KAAK,WAAW,CAAC,KAAK,OAAO,EAAE,IAAK,OAAO,KAAK,WAAW,CAAC;CAExE;CAEA,aAAa;EACX,OAAO,KAAK,UAAU,MAAM;CAC9B;AACF"}

package/out/zero-cache/src/services/change-source/pg/logical-replication/pgoutput-parser.js.map

@@ -1 +1 @@
-{"version":3,"file":"pgoutput-parser.js","names":["#typeParsers","#getTypeParser"],"sources":["../../../../../../../../zero-cache/src/services/change-source/pg/logical-replication/pgoutput-parser.ts"],"sourcesContent":["// Forked from https://github.com/kibae/pg-logical-replication/blob/c55abddc62eadd61bd38922037ecb7a1469fa8c3/src/output-plugins/pgoutput/pgoutput-parser.ts\n/* oxlint-disable */\n\n// https://www.postgresql.org/docs/current/protocol-logicalrep-message-formats.html\n\n// Replaced by TypeParsers\n// import {types} from 'pg';\n\nexport type TypeParser = (val: string) => unknown;\n\nimport type {TypeParsers} from '../../../../db/pg-type-parser.ts';\nimport {BinaryReader} from './binary-reader.ts';\nimport type {\n  Message,\n  MessageBegin,\n  MessageCommit,\n  MessageDelete,\n  MessageInsert,\n  MessageMessage,\n  MessageOrigin,\n  MessageRelation,\n  MessageTruncate,\n  MessageType,\n  MessageUpdate,\n  RelationColumn,\n} from './pgoutput.types.ts';\n\nexport class PgoutputParser {\n  _typeCache = new Map<number, {typeSchema: string; typeName: string}>();\n  _relationCache = new Map<number, MessageRelation>();\n\n  // Replaces \"pg-types\" library.\n  #typeParsers: TypeParsers;\n  constructor(typeParsers: TypeParsers) {\n    this.#typeParsers = typeParsers;\n  }\n  #getTypeParser(typeOid: number): TypeParser {\n    return this.#typeParsers.getTypeParser(typeOid);\n  }\n  // Replaced \"pg-types\" library.\n\n  public parse(buf: Buffer): Message {\n    const reader = new BinaryReader(buf);\n    const tag = reader.readUint8();\n\n    switch (tag) {\n      case 0x42 /*B*/:\n        return this.msgBegin(reader);\n      case 0x4f /*O*/:\n        return this.msgOrigin(reader);\n      case 0x59 /*Y*/:\n        return this.msgType(reader);\n      case 0x52 /*R*/:\n        return this.msgRelation(reader);\n      case 0x49 /*I*/:\n        return this.msgInsert(reader);\n      case 0x55 /*U*/:\n        return this.msgUpdate(reader);\n      case 0x44 /*D*/:\n        return this.msgDelete(reader);\n      case 0x54 /*T*/:\n        return this.msgTruncate(reader);\n      case 0x4d /*M*/:\n        return this.msgMessage(reader);\n      case 0x43 /*C*/:\n        return this.msgCommit(reader);\n      default:\n        throw Error('unknown pgoutput message');\n    }\n  }\n\n  private msgBegin(reader: BinaryReader): MessageBegin {\n    // TODO lsn can be null if origin sended\n    // https://github.com/postgres/postgres/blob/85c61ba8920ba73500e1518c63795982ee455d14/src/backend/replication/pgoutput/pgoutput.c#L409\n    // https://github.com/postgres/postgres/blob/27b77ecf9f4d5be211900eda54d8155ada50d696/src/include/replication/reorderbuffer.h#L275\n\n    return {\n      tag: 'begin',\n      commitLsn: reader.readLsn(),\n      commitTime: reader.readTime(),\n      xid: reader.readInt32(),\n    };\n  }\n\n  private msgOrigin(reader: BinaryReader): MessageOrigin {\n    return {\n      tag: 'origin',\n      originLsn: reader.readLsn(),\n      originName: reader.readString(),\n    };\n  }\n\n  private msgType(reader: BinaryReader): MessageType {\n    const typeOid = reader.readInt32();\n    const typeSchema = reader.readString();\n    const typeName = reader.readString();\n\n    // mem leak not likely to happen because amount of types is usually small\n    this._typeCache.set(typeOid, {typeSchema, typeName});\n\n    return {tag: 'type', typeOid, typeSchema, typeName};\n  }\n\n  private msgRelation(reader: BinaryReader): MessageRelation {\n    // lsn expected to be null\n    // https://github.com/postgres/postgres/blob/27b77ecf9f4d5be211900eda54d8155ada50d696/src/backend/replication/walsender.c#L1342\n    const relationOid = reader.readInt32();\n    const schema = reader.readString();\n    const name = reader.readString();\n    const replicaIdentity = this.readRelationReplicaIdentity(reader);\n    const columns = reader.array(reader.readInt16(), () =>\n      this.readRelationColumn(reader),\n    );\n    const keyColumns = columns.filter(it => it.flags & 0b1).map(it => it.name);\n\n    const msg: MessageRelation = {\n      tag: 'relation',\n      relationOid,\n      schema,\n      name,\n      replicaIdentity,\n      columns,\n      keyColumns,\n    };\n\n    // mem leak not likely to happen because amount of relations is usually small\n    this._relationCache.set(relationOid, msg);\n\n    return msg;\n  }\n\n  private readRelationReplicaIdentity(reader: BinaryReader) {\n    // https://www.postgresql.org/docs/14/catalog-pg-class.html\n    const ident = reader.readUint8();\n\n    switch (ident) {\n      case 0x64 /*d*/:\n        return 'default';\n      case 0x6e /*n*/:\n        return 'nothing';\n      case 0x66 /*f*/:\n        return 'full';\n      case 0x69 /*i*/:\n        return 'index';\n      default:\n        throw Error(`unknown replica identity ${String.fromCharCode(ident)}`);\n    }\n  }\n\n  private readRelationColumn(reader: BinaryReader): RelationColumn {\n    const flags = reader.readUint8();\n    const name = reader.readString();\n    const typeOid = reader.readInt32();\n    const typeMod = reader.readInt32();\n\n    return {\n      flags,\n      name,\n      typeOid,\n      typeMod,\n      typeSchema: null,\n      typeName: null, // TODO resolve builtin type names?\n      ...this._typeCache.get(typeOid),\n      // parser: types.getTypeParser(typeOid),\n      parser: this.#getTypeParser(typeOid),\n    };\n  }\n\n  private msgInsert(reader: BinaryReader): MessageInsert {\n    const relation = this._relationCache.get(reader.readInt32());\n\n    if (!relation) {\n      throw Error('missing relation');\n    }\n\n    reader.readUint8(); // consume the 'N' key\n\n    return {\n      tag: 'insert',\n      relation,\n      new: this.readTuple(reader, relation),\n    };\n  }\n\n  private msgUpdate(reader: BinaryReader): MessageUpdate {\n    const relation = this._relationCache.get(reader.readInt32());\n\n    if (!relation) {\n      throw Error('missing relation');\n    }\n\n    let key: Record<string, any> | null = null;\n    let old: Record<string, any> | null = null;\n    let new_: Record<string, any> | null = null;\n    const subMsgKey = reader.readUint8();\n\n    if (subMsgKey === 0x4b /*K*/) {\n      key = this.readKeyTuple(reader, relation);\n      reader.readUint8(); // consume the 'N' key\n      new_ = this.readTuple(reader, relation);\n    } else if (subMsgKey === 0x4f /*O*/) {\n      old = this.readTuple(reader, relation);\n      reader.readUint8(); // consume the 'N' key\n      new_ = this.readTuple(reader, relation, old);\n    } else if (subMsgKey === 0x4e /*N*/) {\n      new_ = this.readTuple(reader, relation);\n    } else {\n      throw Error(`unknown submessage key ${String.fromCharCode(subMsgKey)}`);\n    }\n\n    return {tag: 'update', relation, key, old, new: new_};\n  }\n\n  private msgDelete(reader: BinaryReader): MessageDelete {\n    const relation = this._relationCache.get(reader.readInt32());\n\n    if (!relation) {\n      throw Error('missing relation');\n    }\n\n    let key: Record<string, any> | null = null;\n    let old: Record<string, any> | null = null;\n    const subMsgKey = reader.readUint8();\n\n    if (subMsgKey === 0x4b /*K*/) {\n      key = this.readKeyTuple(reader, relation);\n    } else if (subMsgKey === 0x4f /*O*/) {\n      old = this.readTuple(reader, relation);\n    } else {\n      throw Error(`unknown submessage key ${String.fromCharCode(subMsgKey)}`);\n    }\n\n    return {tag: 'delete', relation, key, old};\n  }\n\n  private readKeyTuple(\n    reader: BinaryReader,\n    relation: MessageRelation,\n  ): Record<string, any> {\n    const tuple = this.readTuple(reader, relation);\n    const key = Object.create(null);\n\n    for (const k of relation.keyColumns) {\n      // If value is `null`, then it is definitely not part of key,\n      // because key cannot have nulls by documentation.\n      // And if we got `null` while reading keyOnly tuple,\n      // then it means that `null` is not actual value\n      // but placeholder of non-key column.\n      key[k] = tuple[k] === null ? undefined : tuple[k];\n    }\n\n    return key;\n  }\n\n  private readTuple(\n    reader: BinaryReader,\n    {columns}: MessageRelation,\n    unchangedToastFallback?: Record<string, any> | null,\n  ): Record<string, any> {\n    const nfields = reader.readInt16();\n    const tuple = Object.create(null);\n\n    for (let i = 0; i < nfields; i++) {\n      const {name, parser} = columns[i];\n      const kind = reader.readUint8();\n\n      switch (kind) {\n        case 0x62: // 'b' binary\n          const bsize = reader.readInt32();\n          const bval = reader.read(bsize);\n          // dont need to .slice() because new buffer\n          // is created for each replication chunk\n          tuple[name] = bval;\n          break;\n        case 0x74: // 't' text\n          const valsize = reader.readInt32();\n          const valbuf = reader.read(valsize);\n          const valtext = reader.decodeText(valbuf);\n          tuple[name] = parser(valtext);\n          break;\n        case 0x6e: // 'n' null\n          tuple[name] = null;\n          break;\n        case 0x75: // 'u' unchanged toast datum\n          tuple[name] = unchangedToastFallback?.[name];\n          break;\n        default:\n          throw Error(`unknown attribute kind ${String.fromCharCode(kind)}`);\n      }\n    }\n\n    return tuple;\n  }\n\n  private msgTruncate(reader: BinaryReader): MessageTruncate {\n    const nrels = reader.readInt32();\n    const flags = reader.readUint8();\n\n    return {\n      tag: 'truncate',\n      cascade: Boolean(flags & 0b1),\n      restartIdentity: Boolean(flags & 0b10),\n      relations: reader.array(\n        nrels,\n        () => this._relationCache.get(reader.readInt32()) as MessageRelation,\n      ),\n    };\n  }\n\n  private msgMessage(reader: BinaryReader): MessageMessage {\n    const flags = reader.readUint8();\n\n    return {\n      tag: 'message',\n      flags,\n      transactional: Boolean(flags & 0b1),\n      messageLsn: reader.readLsn(),\n      prefix: reader.readString(),\n      content: reader.read(reader.readInt32()),\n    };\n  }\n\n  private msgCommit(reader: BinaryReader): MessageCommit {\n    return {\n      tag: 'commit',\n      flags: reader.readUint8(), // reserved unused\n      commitLsn: reader.readLsn(), // should be the same as begin.commitLsn\n      commitEndLsn: reader.readLsn(),\n      commitTime: reader.readTime(),\n    };\n  }\n}\n"],"mappings":";;AA2BA,IAAa,iBAAb,MAA4B;CAC1B,6BAAa,IAAI,KAAqD;CACtE,iCAAiB,IAAI,KAA8B;CAGnD;CACA,YAAY,aAA0B;AACpC,QAAA,cAAoB;;CAEtB,eAAe,SAA6B;AAC1C,SAAO,MAAA,YAAkB,cAAc,QAAQ;;CAIjD,MAAa,KAAsB;EACjC,MAAM,SAAS,IAAI,aAAa,IAAI;AAGpC,UAFY,OAAO,WAAW,EAE9B;GACE,KAAK,GACH,QAAO,KAAK,SAAS,OAAO;GAC9B,KAAK,GACH,QAAO,KAAK,UAAU,OAAO;GAC/B,KAAK,GACH,QAAO,KAAK,QAAQ,OAAO;GAC7B,KAAK,GACH,QAAO,KAAK,YAAY,OAAO;GACjC,KAAK,GACH,QAAO,KAAK,UAAU,OAAO;GAC/B,KAAK,GACH,QAAO,KAAK,UAAU,OAAO;GAC/B,KAAK,GACH,QAAO,KAAK,UAAU,OAAO;GAC/B,KAAK,GACH,QAAO,KAAK,YAAY,OAAO;GACjC,KAAK,GACH,QAAO,KAAK,WAAW,OAAO;GAChC,KAAK,GACH,QAAO,KAAK,UAAU,OAAO;GAC/B,QACE,OAAM,MAAM,2BAA2B;;;CAI7C,SAAiB,QAAoC;AAKnD,SAAO;GACL,KAAK;GACL,WAAW,OAAO,SAAS;GAC3B,YAAY,OAAO,UAAU;GAC7B,KAAK,OAAO,WAAW;GACxB;;CAGH,UAAkB,QAAqC;AACrD,SAAO;GACL,KAAK;GACL,WAAW,OAAO,SAAS;GAC3B,YAAY,OAAO,YAAY;GAChC;;CAGH,QAAgB,QAAmC;EACjD,MAAM,UAAU,OAAO,WAAW;EAClC,MAAM,aAAa,OAAO,YAAY;EACtC,MAAM,WAAW,OAAO,YAAY;AAGpC,OAAK,WAAW,IAAI,SAAS;GAAC;GAAY;GAAS,CAAC;AAEpD,SAAO;GAAC,KAAK;GAAQ;GAAS;GAAY;GAAS;;CAGrD,YAAoB,QAAuC;EAGzD,MAAM,cAAc,OAAO,WAAW;EACtC,MAAM,SAAS,OAAO,YAAY;EAClC,MAAM,OAAO,OAAO,YAAY;EAChC,MAAM,kBAAkB,KAAK,4BAA4B,OAAO;EAChE,MAAM,UAAU,OAAO,MAAM,OAAO,WAAW,QAC7C,KAAK,mBAAmB,OAAO,CAChC;EAGD,MAAM,MAAuB;GAC3B,KAAK;GACL;GACA;GACA;GACA;GACA;GACA,YATiB,QAAQ,QAAO,OAAM,GAAG,QAAQ,EAAI,CAAC,KAAI,OAAM,GAAG,KAAK;GAUzE;AAGD,OAAK,eAAe,IAAI,aAAa,IAAI;AAEzC,SAAO;;CAGT,4BAAoC,QAAsB;EAExD,MAAM,QAAQ,OAAO,WAAW;AAEhC,UAAQ,OAAR;GACE,KAAK,IACH,QAAO;GACT,KAAK,IACH,QAAO;GACT,KAAK,IACH,QAAO;GACT,KAAK,IACH,QAAO;GACT,QACE,OAAM,MAAM,4BAA4B,OAAO,aAAa,MAAM,GAAG;;;CAI3E,mBAA2B,QAAsC;EAC/D,MAAM,QAAQ,OAAO,WAAW;EAChC,MAAM,OAAO,OAAO,YAAY;EAChC,MAAM,UAAU,OAAO,WAAW;AAGlC,SAAO;GACL;GACA;GACA;GACA,SANc,OAAO,WAAW;GAOhC,YAAY;GACZ,UAAU;GACV,GAAG,KAAK,WAAW,IAAI,QAAQ;GAE/B,QAAQ,MAAA,cAAoB,QAAQ;GACrC;;CAGH,UAAkB,QAAqC;EACrD,MAAM,WAAW,KAAK,eAAe,IAAI,OAAO,WAAW,CAAC;AAE5D,MAAI,CAAC,SACH,OAAM,MAAM,mBAAmB;AAGjC,SAAO,WAAW;AAElB,SAAO;GACL,KAAK;GACL;GACA,KAAK,KAAK,UAAU,QAAQ,SAAS;GACtC;;CAGH,UAAkB,QAAqC;EACrD,MAAM,WAAW,KAAK,eAAe,IAAI,OAAO,WAAW,CAAC;AAE5D,MAAI,CAAC,SACH,OAAM,MAAM,mBAAmB;EAGjC,IAAI,MAAkC;EACtC,IAAI,MAAkC;EACtC,IAAI,OAAmC;EACvC,MAAM,YAAY,OAAO,WAAW;AAEpC,MAAI,cAAc,IAAY;AAC5B,SAAM,KAAK,aAAa,QAAQ,SAAS;AACzC,UAAO,WAAW;AAClB,UAAO,KAAK,UAAU,QAAQ,SAAS;aAC9B,cAAc,IAAY;AACnC,SAAM,KAAK,UAAU,QAAQ,SAAS;AACtC,UAAO,WAAW;AAClB,UAAO,KAAK,UAAU,QAAQ,UAAU,IAAI;aACnC,cAAc,GACvB,QAAO,KAAK,UAAU,QAAQ,SAAS;MAEvC,OAAM,MAAM,0BAA0B,OAAO,aAAa,UAAU,GAAG;AAGzE,SAAO;GAAC,KAAK;GAAU;GAAU;GAAK;GAAK,KAAK;GAAK;;CAGvD,UAAkB,QAAqC;EACrD,MAAM,WAAW,KAAK,eAAe,IAAI,OAAO,WAAW,CAAC;AAE5D,MAAI,CAAC,SACH,OAAM,MAAM,mBAAmB;EAGjC,IAAI,MAAkC;EACtC,IAAI,MAAkC;EACtC,MAAM,YAAY,OAAO,WAAW;AAEpC,MAAI,cAAc,GAChB,OAAM,KAAK,aAAa,QAAQ,SAAS;WAChC,cAAc,GACvB,OAAM,KAAK,UAAU,QAAQ,SAAS;MAEtC,OAAM,MAAM,0BAA0B,OAAO,aAAa,UAAU,GAAG;AAGzE,SAAO;GAAC,KAAK;GAAU;GAAU;GAAK;GAAI;;CAG5C,aACE,QACA,UACqB;EACrB,MAAM,QAAQ,KAAK,UAAU,QAAQ,SAAS;EAC9C,MAAM,MAAM,OAAO,OAAO,KAAK;AAE/B,OAAK,MAAM,KAAK,SAAS,WAMvB,KAAI,KAAK,MAAM,OAAO,OAAO,KAAA,IAAY,MAAM;AAGjD,SAAO;;CAGT,UACE,QACA,EAAC,WACD,wBACqB;EACrB,MAAM,UAAU,OAAO,WAAW;EAClC,MAAM,QAAQ,OAAO,OAAO,KAAK;AAEjC,OAAK,IAAI,IAAI,GAAG,IAAI,SAAS,KAAK;GAChC,MAAM,EAAC,MAAM,WAAU,QAAQ;GAC/B,MAAM,OAAO,OAAO,WAAW;AAE/B,WAAQ,MAAR;IACE,KAAK;KACH,MAAM,QAAQ,OAAO,WAAW;AAIhC,WAAM,QAHO,OAAO,KAAK,MAAM;AAI/B;IACF,KAAK;KACH,MAAM,UAAU,OAAO,WAAW;KAClC,MAAM,SAAS,OAAO,KAAK,QAAQ;AAEnC,WAAM,QAAQ,OADE,OAAO,WAAW,OAAO,CACZ;AAC7B;IACF,KAAK;AACH,WAAM,QAAQ;AACd;IACF,KAAK;AACH,WAAM,QAAQ,yBAAyB;AACvC;IACF,QACE,OAAM,MAAM,0BAA0B,OAAO,aAAa,KAAK,GAAG;;;AAIxE,SAAO;;CAGT,YAAoB,QAAuC;EACzD,MAAM,QAAQ,OAAO,WAAW;EAChC,MAAM,QAAQ,OAAO,WAAW;AAEhC,SAAO;GACL,KAAK;GACL,SAAS,QAAQ,QAAQ,EAAI;GAC7B,iBAAiB,QAAQ,QAAQ,EAAK;GACtC,WAAW,OAAO,MAChB,aACM,KAAK,eAAe,IAAI,OAAO,WAAW,CAAC,CAClD;GACF;;CAGH,WAAmB,QAAsC;EACvD,MAAM,QAAQ,OAAO,WAAW;AAEhC,SAAO;GACL,KAAK;GACL;GACA,eAAe,QAAQ,QAAQ,EAAI;GACnC,YAAY,OAAO,SAAS;GAC5B,QAAQ,OAAO,YAAY;GAC3B,SAAS,OAAO,KAAK,OAAO,WAAW,CAAC;GACzC;;CAGH,UAAkB,QAAqC;AACrD,SAAO;GACL,KAAK;GACL,OAAO,OAAO,WAAW;GACzB,WAAW,OAAO,SAAS;GAC3B,cAAc,OAAO,SAAS;GAC9B,YAAY,OAAO,UAAU;GAC9B"}
+{"version":3,"file":"pgoutput-parser.js","names":["#typeParsers","#getTypeParser"],"sources":["../../../../../../../../zero-cache/src/services/change-source/pg/logical-replication/pgoutput-parser.ts"],"sourcesContent":["// Forked from https://github.com/kibae/pg-logical-replication/blob/c55abddc62eadd61bd38922037ecb7a1469fa8c3/src/output-plugins/pgoutput/pgoutput-parser.ts\n/* oxlint-disable */\n\n// https://www.postgresql.org/docs/current/protocol-logicalrep-message-formats.html\n\n// Replaced by TypeParsers\n// import {types} from 'pg';\n\nexport type TypeParser = (val: string) => unknown;\n\nimport type {TypeParsers} from '../../../../db/pg-type-parser.ts';\nimport {BinaryReader} from './binary-reader.ts';\nimport type {\n  Message,\n  MessageBegin,\n  MessageCommit,\n  MessageDelete,\n  MessageInsert,\n  MessageMessage,\n  MessageOrigin,\n  MessageRelation,\n  MessageTruncate,\n  MessageType,\n  MessageUpdate,\n  RelationColumn,\n} from './pgoutput.types.ts';\n\nexport class PgoutputParser {\n  _typeCache = new Map<number, {typeSchema: string; typeName: string}>();\n  _relationCache = new Map<number, MessageRelation>();\n\n  // Replaces \"pg-types\" library.\n  #typeParsers: TypeParsers;\n  constructor(typeParsers: TypeParsers) {\n    this.#typeParsers = typeParsers;\n  }\n  #getTypeParser(typeOid: number): TypeParser {\n    return this.#typeParsers.getTypeParser(typeOid);\n  }\n  // Replaced \"pg-types\" library.\n\n  public parse(buf: Buffer): Message {\n    const reader = new BinaryReader(buf);\n    const tag = reader.readUint8();\n\n    switch (tag) {\n      case 0x42 /*B*/:\n        return this.msgBegin(reader);\n      case 0x4f /*O*/:\n        return this.msgOrigin(reader);\n      case 0x59 /*Y*/:\n        return this.msgType(reader);\n      case 0x52 /*R*/:\n        return this.msgRelation(reader);\n      case 0x49 /*I*/:\n        return this.msgInsert(reader);\n      case 0x55 /*U*/:\n        return this.msgUpdate(reader);\n      case 0x44 /*D*/:\n        return this.msgDelete(reader);\n      case 0x54 /*T*/:\n        return this.msgTruncate(reader);\n      case 0x4d /*M*/:\n        return this.msgMessage(reader);\n      case 0x43 /*C*/:\n        return this.msgCommit(reader);\n      default:\n        throw Error('unknown pgoutput message');\n    }\n  }\n\n  private msgBegin(reader: BinaryReader): MessageBegin {\n    // TODO lsn can be null if origin sended\n    // https://github.com/postgres/postgres/blob/85c61ba8920ba73500e1518c63795982ee455d14/src/backend/replication/pgoutput/pgoutput.c#L409\n    // https://github.com/postgres/postgres/blob/27b77ecf9f4d5be211900eda54d8155ada50d696/src/include/replication/reorderbuffer.h#L275\n\n    return {\n      tag: 'begin',\n      commitLsn: reader.readLsn(),\n      commitTime: reader.readTime(),\n      xid: reader.readInt32(),\n    };\n  }\n\n  private msgOrigin(reader: BinaryReader): MessageOrigin {\n    return {\n      tag: 'origin',\n      originLsn: reader.readLsn(),\n      originName: reader.readString(),\n    };\n  }\n\n  private msgType(reader: BinaryReader): MessageType {\n    const typeOid = reader.readInt32();\n    const typeSchema = reader.readString();\n    const typeName = reader.readString();\n\n    // mem leak not likely to happen because amount of types is usually small\n    this._typeCache.set(typeOid, {typeSchema, typeName});\n\n    return {tag: 'type', typeOid, typeSchema, typeName};\n  }\n\n  private msgRelation(reader: BinaryReader): MessageRelation {\n    // lsn expected to be null\n    // https://github.com/postgres/postgres/blob/27b77ecf9f4d5be211900eda54d8155ada50d696/src/backend/replication/walsender.c#L1342\n    const relationOid = reader.readInt32();\n    const schema = reader.readString();\n    const name = reader.readString();\n    const replicaIdentity = this.readRelationReplicaIdentity(reader);\n    const columns = reader.array(reader.readInt16(), () =>\n      this.readRelationColumn(reader),\n    );\n    const keyColumns = columns.filter(it => it.flags & 0b1).map(it => it.name);\n\n    const msg: MessageRelation = {\n      tag: 'relation',\n      relationOid,\n      schema,\n      name,\n      replicaIdentity,\n      columns,\n      keyColumns,\n    };\n\n    // mem leak not likely to happen because amount of relations is usually small\n    this._relationCache.set(relationOid, msg);\n\n    return msg;\n  }\n\n  private readRelationReplicaIdentity(reader: BinaryReader) {\n    // https://www.postgresql.org/docs/14/catalog-pg-class.html\n    const ident = reader.readUint8();\n\n    switch (ident) {\n      case 0x64 /*d*/:\n        return 'default';\n      case 0x6e /*n*/:\n        return 'nothing';\n      case 0x66 /*f*/:\n        return 'full';\n      case 0x69 /*i*/:\n        return 'index';\n      default:\n        throw Error(`unknown replica identity ${String.fromCharCode(ident)}`);\n    }\n  }\n\n  private readRelationColumn(reader: BinaryReader): RelationColumn {\n    const flags = reader.readUint8();\n    const name = reader.readString();\n    const typeOid = reader.readInt32();\n    const typeMod = reader.readInt32();\n\n    return {\n      flags,\n      name,\n      typeOid,\n      typeMod,\n      typeSchema: null,\n      typeName: null, // TODO resolve builtin type names?\n      ...this._typeCache.get(typeOid),\n      // parser: types.getTypeParser(typeOid),\n      parser: this.#getTypeParser(typeOid),\n    };\n  }\n\n  private msgInsert(reader: BinaryReader): MessageInsert {\n    const relation = this._relationCache.get(reader.readInt32());\n\n    if (!relation) {\n      throw Error('missing relation');\n    }\n\n    reader.readUint8(); // consume the 'N' key\n\n    return {\n      tag: 'insert',\n      relation,\n      new: this.readTuple(reader, relation),\n    };\n  }\n\n  private msgUpdate(reader: BinaryReader): MessageUpdate {\n    const relation = this._relationCache.get(reader.readInt32());\n\n    if (!relation) {\n      throw Error('missing relation');\n    }\n\n    let key: Record<string, any> | null = null;\n    let old: Record<string, any> | null = null;\n    let new_: Record<string, any> | null = null;\n    const subMsgKey = reader.readUint8();\n\n    if (subMsgKey === 0x4b /*K*/) {\n      key = this.readKeyTuple(reader, relation);\n      reader.readUint8(); // consume the 'N' key\n      new_ = this.readTuple(reader, relation);\n    } else if (subMsgKey === 0x4f /*O*/) {\n      old = this.readTuple(reader, relation);\n      reader.readUint8(); // consume the 'N' key\n      new_ = this.readTuple(reader, relation, old);\n    } else if (subMsgKey === 0x4e /*N*/) {\n      new_ = this.readTuple(reader, relation);\n    } else {\n      throw Error(`unknown submessage key ${String.fromCharCode(subMsgKey)}`);\n    }\n\n    return {tag: 'update', relation, key, old, new: new_};\n  }\n\n  private msgDelete(reader: BinaryReader): MessageDelete {\n    const relation = this._relationCache.get(reader.readInt32());\n\n    if (!relation) {\n      throw Error('missing relation');\n    }\n\n    let key: Record<string, any> | null = null;\n    let old: Record<string, any> | null = null;\n    const subMsgKey = reader.readUint8();\n\n    if (subMsgKey === 0x4b /*K*/) {\n      key = this.readKeyTuple(reader, relation);\n    } else if (subMsgKey === 0x4f /*O*/) {\n      old = this.readTuple(reader, relation);\n    } else {\n      throw Error(`unknown submessage key ${String.fromCharCode(subMsgKey)}`);\n    }\n\n    return {tag: 'delete', relation, key, old};\n  }\n\n  private readKeyTuple(\n    reader: BinaryReader,\n    relation: MessageRelation,\n  ): Record<string, any> {\n    const tuple = this.readTuple(reader, relation);\n    const key = Object.create(null);\n\n    for (const k of relation.keyColumns) {\n      // If value is `null`, then it is definitely not part of key,\n      // because key cannot have nulls by documentation.\n      // And if we got `null` while reading keyOnly tuple,\n      // then it means that `null` is not actual value\n      // but placeholder of non-key column.\n      key[k] = tuple[k] === null ? undefined : tuple[k];\n    }\n\n    return key;\n  }\n\n  private readTuple(\n    reader: BinaryReader,\n    {columns}: MessageRelation,\n    unchangedToastFallback?: Record<string, any> | null,\n  ): Record<string, any> {\n    const nfields = reader.readInt16();\n    const tuple = Object.create(null);\n\n    for (let i = 0; i < nfields; i++) {\n      const {name, parser} = columns[i];\n      const kind = reader.readUint8();\n\n      switch (kind) {\n        case 0x62: // 'b' binary\n          const bsize = reader.readInt32();\n          const bval = reader.read(bsize);\n          // dont need to .slice() because new buffer\n          // is created for each replication chunk\n          tuple[name] = bval;\n          break;\n        case 0x74: // 't' text\n          const valsize = reader.readInt32();\n          const valbuf = reader.read(valsize);\n          const valtext = reader.decodeText(valbuf);\n          tuple[name] = parser(valtext);\n          break;\n        case 0x6e: // 'n' null\n          tuple[name] = null;\n          break;\n        case 0x75: // 'u' unchanged toast datum\n          tuple[name] = unchangedToastFallback?.[name];\n          break;\n        default:\n          throw Error(`unknown attribute kind ${String.fromCharCode(kind)}`);\n      }\n    }\n\n    return tuple;\n  }\n\n  private msgTruncate(reader: BinaryReader): MessageTruncate {\n    const nrels = reader.readInt32();\n    const flags = reader.readUint8();\n\n    return {\n      tag: 'truncate',\n      cascade: Boolean(flags & 0b1),\n      restartIdentity: Boolean(flags & 0b10),\n      relations: reader.array(\n        nrels,\n        () => this._relationCache.get(reader.readInt32()) as MessageRelation,\n      ),\n    };\n  }\n\n  private msgMessage(reader: BinaryReader): MessageMessage {\n    const flags = reader.readUint8();\n\n    return {\n      tag: 'message',\n      flags,\n      transactional: Boolean(flags & 0b1),\n      messageLsn: reader.readLsn(),\n      prefix: reader.readString(),\n      content: reader.read(reader.readInt32()),\n    };\n  }\n\n  private msgCommit(reader: BinaryReader): MessageCommit {\n    return {\n      tag: 'commit',\n      flags: reader.readUint8(), // reserved unused\n      commitLsn: reader.readLsn(), // should be the same as begin.commitLsn\n      commitEndLsn: reader.readLsn(),\n      commitTime: reader.readTime(),\n    };\n  }\n}\n"],"mappings":";;AA2BA,IAAa,iBAAb,MAA4B;CAC1B,6BAAa,IAAI,IAAoD;CACrE,iCAAiB,IAAI,IAA6B;CAGlD;CACA,YAAY,aAA0B;EACpC,KAAKA,eAAe;CACtB;CACA,eAAe,SAA6B;EAC1C,OAAO,KAAKA,aAAa,cAAc,OAAO;CAChD;CAGA,MAAa,KAAsB;EACjC,MAAM,SAAS,IAAI,aAAa,GAAG;EAGnC,QAFY,OAAO,UAEX,GAAR;GACE,KAAK,IACH,OAAO,KAAK,SAAS,MAAM;GAC7B,KAAK,IACH,OAAO,KAAK,UAAU,MAAM;GAC9B,KAAK,IACH,OAAO,KAAK,QAAQ,MAAM;GAC5B,KAAK,IACH,OAAO,KAAK,YAAY,MAAM;GAChC,KAAK,IACH,OAAO,KAAK,UAAU,MAAM;GAC9B,KAAK,IACH,OAAO,KAAK,UAAU,MAAM;GAC9B,KAAK,IACH,OAAO,KAAK,UAAU,MAAM;GAC9B,KAAK,IACH,OAAO,KAAK,YAAY,MAAM;GAChC,KAAK,IACH,OAAO,KAAK,WAAW,MAAM;GAC/B,KAAK,IACH,OAAO,KAAK,UAAU,MAAM;GAC9B,SACE,MAAM,MAAM,0BAA0B;EAC1C;CACF;CAEA,SAAiB,QAAoC;EAKnD,OAAO;GACL,KAAK;GACL,WAAW,OAAO,QAAQ;GAC1B,YAAY,OAAO,SAAS;GAC5B,KAAK,OAAO,UAAU;EACxB;CACF;CAEA,UAAkB,QAAqC;EACrD,OAAO;GACL,KAAK;GACL,WAAW,OAAO,QAAQ;GAC1B,YAAY,OAAO,WAAW;EAChC;CACF;CAEA,QAAgB,QAAmC;EACjD,MAAM,UAAU,OAAO,UAAU;EACjC,MAAM,aAAa,OAAO,WAAW;EACrC,MAAM,WAAW,OAAO,WAAW;EAGnC,KAAK,WAAW,IAAI,SAAS;GAAC;GAAY;EAAQ,CAAC;EAEnD,OAAO;GAAC,KAAK;GAAQ;GAAS;GAAY;EAAQ;CACpD;CAEA,YAAoB,QAAuC;EAGzD,MAAM,cAAc,OAAO,UAAU;EACrC,MAAM,SAAS,OAAO,WAAW;EACjC,MAAM,OAAO,OAAO,WAAW;EAC/B,MAAM,kBAAkB,KAAK,4BAA4B,MAAM;EAC/D,MAAM,UAAU,OAAO,MAAM,OAAO,UAAU,SAC5C,KAAK,mBAAmB,MAAM,CAChC;EAGA,MAAM,MAAuB;GAC3B,KAAK;GACL;GACA;GACA;GACA;GACA;GACA,YATiB,QAAQ,QAAO,OAAM,GAAG,QAAQ,CAAG,EAAE,KAAI,OAAM,GAAG,IASnE;EACF;EAGA,KAAK,eAAe,IAAI,aAAa,GAAG;EAExC,OAAO;CACT;CAEA,4BAAoC,QAAsB;EAExD,MAAM,QAAQ,OAAO,UAAU;EAE/B,QAAQ,OAAR;GACE,KAAK,KACH,OAAO;GACT,KAAK,KACH,OAAO;GACT,KAAK,KACH,OAAO;GACT,KAAK,KACH,OAAO;GACT,SACE,MAAM,MAAM,4BAA4B,OAAO,aAAa,KAAK,GAAG;EACxE;CACF;CAEA,mBAA2B,QAAsC;EAC/D,MAAM,QAAQ,OAAO,UAAU;EAC/B,MAAM,OAAO,OAAO,WAAW;EAC/B,MAAM,UAAU,OAAO,UAAU;EAGjC,OAAO;GACL;GACA;GACA;GACA,SANc,OAAO,UAMrB;GACA,YAAY;GACZ,UAAU;GACV,GAAG,KAAK,WAAW,IAAI,OAAO;GAE9B,QAAQ,KAAKC,eAAe,OAAO;EACrC;CACF;CAEA,UAAkB,QAAqC;EACrD,MAAM,WAAW,KAAK,eAAe,IAAI,OAAO,UAAU,CAAC;EAE3D,IAAI,CAAC,UACH,MAAM,MAAM,kBAAkB;EAGhC,OAAO,UAAU;EAEjB,OAAO;GACL,KAAK;GACL;GACA,KAAK,KAAK,UAAU,QAAQ,QAAQ;EACtC;CACF;CAEA,UAAkB,QAAqC;EACrD,MAAM,WAAW,KAAK,eAAe,IAAI,OAAO,UAAU,CAAC;EAE3D,IAAI,CAAC,UACH,MAAM,MAAM,kBAAkB;EAGhC,IAAI,MAAkC;EACtC,IAAI,MAAkC;EACtC,IAAI,OAAmC;EACvC,MAAM,YAAY,OAAO,UAAU;EAEnC,IAAI,cAAc,IAAY;GAC5B,MAAM,KAAK,aAAa,QAAQ,QAAQ;GACxC,OAAO,UAAU;GACjB,OAAO,KAAK,UAAU,QAAQ,QAAQ;EACxC,OAAO,IAAI,cAAc,IAAY;GACnC,MAAM,KAAK,UAAU,QAAQ,QAAQ;GACrC,OAAO,UAAU;GACjB,OAAO,KAAK,UAAU,QAAQ,UAAU,GAAG;EAC7C,OAAO,IAAI,cAAc,IACvB,OAAO,KAAK,UAAU,QAAQ,QAAQ;OAEtC,MAAM,MAAM,0BAA0B,OAAO,aAAa,SAAS,GAAG;EAGxE,OAAO;GAAC,KAAK;GAAU;GAAU;GAAK;GAAK,KAAK;EAAI;CACtD;CAEA,UAAkB,QAAqC;EACrD,MAAM,WAAW,KAAK,eAAe,IAAI,OAAO,UAAU,CAAC;EAE3D,IAAI,CAAC,UACH,MAAM,MAAM,kBAAkB;EAGhC,IAAI,MAAkC;EACtC,IAAI,MAAkC;EACtC,MAAM,YAAY,OAAO,UAAU;EAEnC,IAAI,cAAc,IAChB,MAAM,KAAK,aAAa,QAAQ,QAAQ;OACnC,IAAI,cAAc,IACvB,MAAM,KAAK,UAAU,QAAQ,QAAQ;OAErC,MAAM,MAAM,0BAA0B,OAAO,aAAa,SAAS,GAAG;EAGxE,OAAO;GAAC,KAAK;GAAU;GAAU;GAAK;EAAG;CAC3C;CAEA,aACE,QACA,UACqB;EACrB,MAAM,QAAQ,KAAK,UAAU,QAAQ,QAAQ;EAC7C,MAAM,MAAM,OAAO,OAAO,IAAI;EAE9B,KAAK,MAAM,KAAK,SAAS,YAMvB,IAAI,KAAK,MAAM,OAAO,OAAO,KAAA,IAAY,MAAM;EAGjD,OAAO;CACT;CAEA,UACE,QACA,EAAC,WACD,wBACqB;EACrB,MAAM,UAAU,OAAO,UAAU;EACjC,MAAM,QAAQ,OAAO,OAAO,IAAI;EAEhC,KAAK,IAAI,IAAI,GAAG,IAAI,SAAS,KAAK;GAChC,MAAM,EAAC,MAAM,WAAU,QAAQ;GAC/B,MAAM,OAAO,OAAO,UAAU;GAE9B,QAAQ,MAAR;IACE,KAAK;KACH,MAAM,QAAQ,OAAO,UAAU;KAI/B,MAAM,QAHO,OAAO,KAAK,KAGX;KACd;IACF,KAAK;KACH,MAAM,UAAU,OAAO,UAAU;KACjC,MAAM,SAAS,OAAO,KAAK,OAAO;KAElC,MAAM,QAAQ,OADE,OAAO,WAAW,MACb,CAAO;KAC5B;IACF,KAAK;KACH,MAAM,QAAQ;KACd;IACF,KAAK;KACH,MAAM,QAAQ,yBAAyB;KACvC;IACF,SACE,MAAM,MAAM,0BAA0B,OAAO,aAAa,IAAI,GAAG;GACrE;EACF;EAEA,OAAO;CACT;CAEA,YAAoB,QAAuC;EACzD,MAAM,QAAQ,OAAO,UAAU;EAC/B,MAAM,QAAQ,OAAO,UAAU;EAE/B,OAAO;GACL,KAAK;GACL,SAAS,QAAQ,QAAQ,CAAG;GAC5B,iBAAiB,QAAQ,QAAQ,CAAI;GACrC,WAAW,OAAO,MAChB,aACM,KAAK,eAAe,IAAI,OAAO,UAAU,CAAC,CAClD;EACF;CACF;CAEA,WAAmB,QAAsC;EACvD,MAAM,QAAQ,OAAO,UAAU;EAE/B,OAAO;GACL,KAAK;GACL;GACA,eAAe,QAAQ,QAAQ,CAAG;GAClC,YAAY,OAAO,QAAQ;GAC3B,QAAQ,OAAO,WAAW;GAC1B,SAAS,OAAO,KAAK,OAAO,UAAU,CAAC;EACzC;CACF;CAEA,UAAkB,QAAqC;EACrD,OAAO;GACL,KAAK;GACL,OAAO,OAAO,UAAU;GACxB,WAAW,OAAO,QAAQ;GAC1B,cAAc,OAAO,QAAQ;GAC7B,YAAY,OAAO,SAAS;EAC9B;CACF;AACF"}

package/out/zero-cache/src/services/change-source/pg/logical-replication/stream.js.map

@@ -1 +1 @@
-{"version":3,"file":"stream.js","names":[],"sources":["../../../../../../../../zero-cache/src/services/change-source/pg/logical-replication/stream.ts"],"sourcesContent":["import {\n  PG_ADMIN_SHUTDOWN,\n  PG_OBJECT_IN_USE,\n  PG_OBJECT_NOT_IN_PREREQUISITE_STATE,\n} from '@drdgvhbh/postgres-error-codes';\nimport type {LogContext} from '@rocicorp/logger';\nimport {defu} from 'defu';\nimport postgres, {type Options, type PostgresType} from 'postgres';\nimport {sleep} from '../../../../../../shared/src/sleep.ts';\nimport {getTypeParsers} from '../../../../db/pg-type-parser.ts';\nimport {type PostgresDB} from '../../../../types/pg.ts';\nimport {id, lit} from '../../../../types/sql.ts';\nimport {pipe, type Sink, type Source} from '../../../../types/streams.ts';\nimport {Subscription} from '../../../../types/subscription.ts';\nimport {AutoResetSignal} from '../../../change-streamer/schema/tables.ts';\nimport {fromBigInt} from '../lsn.ts';\nimport {PgoutputParser} from './pgoutput-parser.ts';\nimport type {Message} from './pgoutput.types.ts';\n\nconst DEFAULT_RETRIES_IF_REPLICATION_SLOT_ACTIVE = 5;\n\nexport type StreamMessage = [\n  lsn: bigint,\n  Message | {tag: 'keepalive'; shouldRespond: boolean},\n];\n\n// Expose the `queued` variable of the Subscription to allow\n// the change-source to determine if there are more messages\n// immediately available.\ntype SourceWithPendingQueue<T> = Source<T> & {\n  queued: number;\n};\n\nexport async function subscribe(\n  lc: LogContext,\n  db: PostgresDB,\n  slot: string,\n  publications: string[],\n  lsn: bigint,\n  retriesIfReplicationSlotActive = DEFAULT_RETRIES_IF_REPLICATION_SLOT_ACTIVE,\n  applicationName = 'zero-replicator',\n): Promise<{\n  messages: SourceWithPendingQueue<StreamMessage>;\n  acks: Sink<bigint>;\n}> {\n  const session = postgres(\n    defu(\n      {\n        max: 1,\n        ['fetch_types']: false, // Necessary for the streaming protocol\n        ['idle_timeout']: null,\n        ['max_lifetime']: null as unknown as number,\n        connection: {\n          ['application_name']: applicationName,\n          replication: 'database', // https://www.postgresql.org/docs/current/protocol-replication.html\n        },\n      },\n      // ParsedOptions are technically compatible with Options, but happen\n      // to not be typed that way. The postgres.js author does an equivalent\n      // merge of ParsedOptions and Options here:\n      // https://github.com/porsager/postgres/blob/089214e85c23c90cf142d47fb30bd03f42874984/src/subscribe.js#L13\n      db.options as unknown as Options<Record<string, PostgresType>>,\n    ),\n  );\n\n  // Postgres will send keepalives before timing out a wal_sender. It is possible that\n  // these keepalives are not received if there is back-pressure in the replication\n  // stream. To keep the connection alive, explicitly send keepalives if none have been\n  // sent within the last 75% of the wal_sender_timeout.\n  //\n  // https://www.postgresql.org/docs/current/runtime-config-replication.html#GUC-WAL-SENDER-TIMEOUT\n  const [{walSenderTimeoutMs}] = await session<\n    {walSenderTimeoutMs: number}[]\n  >`SELECT EXTRACT(EPOCH FROM (setting || unit)::interval) * 1000 \n        AS \"walSenderTimeoutMs\" FROM pg_settings\n        WHERE name = 'wal_sender_timeout'`.simple();\n  const manualKeepaliveTimeout = Math.floor(walSenderTimeoutMs * 0.75);\n  lc.info?.(\n    `wal_sender_timeout: ${walSenderTimeoutMs}ms. ` +\n      `Ensuring manual keepalives at least every ${manualKeepaliveTimeout}ms`,\n  );\n\n  const [readable, writable] = await startReplicationStream(\n    lc,\n    session,\n    slot,\n    publications,\n    lsn,\n    retriesIfReplicationSlotActive + 1,\n  );\n\n  let lastAckTime = Date.now();\n  function sendAck(lsn: bigint) {\n    writable.write(makeAck(lsn));\n    lastAckTime = Date.now();\n  }\n\n  const livenessTimer = setInterval(() => {\n    const now = Date.now();\n    if (now - lastAckTime > manualKeepaliveTimeout) {\n      sendAck(0n);\n      lc.debug?.(`sent manual keepalive`);\n    }\n  }, manualKeepaliveTimeout / 5);\n\n  let destroyed = false;\n  const typeParsers = await getTypeParsers(db, {returnJsonAsString: true});\n  const parser = new PgoutputParser(typeParsers);\n  const messages = Subscription.create<StreamMessage>({\n    cleanup: () => {\n      destroyed = true;\n      readable.destroyed || readable.destroy();\n      clearInterval(livenessTimer);\n      return session.end();\n    },\n  });\n\n  readable.once(\n    'close',\n    () =>\n      // Only log a warning if the stream was not manually closed.\n      destroyed || lc.warn?.(`replication stream closed by ${db.options.host}`),\n  );\n  readable.once(\n    'error',\n    e =>\n      // Don't log the shutdown signal. This is the expected way for upstream\n      // to close the connection (and will be logged downstream).\n      (e instanceof postgres.PostgresError && e.code === PG_ADMIN_SHUTDOWN) ||\n      lc.warn?.(`error from ${db.options.host}`, e),\n  );\n\n  pipe({\n    source: readable,\n    sink: messages,\n    parse: buffer => parseStreamMessage(lc, buffer, parser),\n    // Allow a small buffer of messages to be queued in the subscription so\n    // that the change-source loop can check the queue to determine if more\n    // messages are immediately available.\n    bufferMessages: 5,\n  });\n\n  return {\n    messages,\n    acks: {push: sendAck},\n  };\n}\n\n/**\n * Formats publication names for the START_REPLICATION command.\n * The replication protocol expects format: publication_names 'pub1,pub2'\n * Each name is escaped to handle any quotes that may have passed validation.\n */\nfunction formatPublicationNames(publications: string[]): string {\n  // lit() returns 'escaped_name' with surrounding quotes\n  // We strip the quotes since the outer quotes are in the template\n  return publications.map(p => lit(p).slice(1, -1)).join(',');\n}\n\nasync function startReplicationStream(\n  lc: LogContext,\n  session: postgres.Sql,\n  slot: string,\n  publications: string[],\n  lsn: bigint,\n  maxAttempts: number,\n) {\n  for (let i = 0; i < maxAttempts; i++) {\n    try {\n      const stream = session\n        .unsafe(\n          `START_REPLICATION SLOT ${id(slot)} LOGICAL ${fromBigInt(lsn)} (\n        proto_version '1',\n        publication_names '${formatPublicationNames(publications)}',\n        messages 'true'\n      )`,\n        )\n        .execute();\n      return await Promise.all([stream.readable(), stream.writable()]);\n    } catch (e) {\n      if (e instanceof postgres.PostgresError) {\n        // error: replication slot \"zero_slot_change_source_test_id\" is active for PID 268\n        if (e.code === PG_OBJECT_IN_USE) {\n          // The freeing up of the replication slot is not transactional;\n          // sometimes it takes time for Postgres to consider the slot\n          // inactive.\n          lc.warn?.(`attempt ${i + 1}: ${String(e)}`, e);\n          await sleep(10);\n          continue;\n        }\n        // error: This slot has been invalidated because it exceeded the maximum reserved size.\n        // (This is a different manifestation of a slot being invalidated when\n        //  the wal exceeds the max_slot_wal_keep_size)\n        if (e.code === PG_OBJECT_NOT_IN_PREREQUISITE_STATE) {\n          lc.error?.(`error starting replication stream`, e);\n          throw new AutoResetSignal(`unable to start replication stream`, {\n            cause: e,\n          });\n        }\n      }\n      throw e;\n    }\n  }\n  throw new Error(\n    `exceeded max attempts (${maxAttempts}) to start the Postgres stream`,\n  );\n}\n\nfunction parseStreamMessage(\n  lc: LogContext,\n  buffer: Buffer,\n  parser: PgoutputParser,\n): StreamMessage | null {\n  // https://www.postgresql.org/docs/current/protocol-replication.html#PROTOCOL-REPLICATION-START-REPLICATION\n  if (buffer[0] !== 0x77 && buffer[0] !== 0x6b) {\n    lc.warn?.('Unknown message', buffer[0]);\n    return null;\n  }\n  const lsn = buffer.readBigUInt64BE(1);\n  if (buffer[0] === 0x77) {\n    // https://www.postgresql.org/docs/current/protocol-replication.html#PROTOCOL-REPLICATION-XLOGDATA\n    // (Byte 25 is where the WAL data begins)\n    return [lsn, parser.parse(buffer.subarray(25))];\n  }\n  // https://www.postgresql.org/docs/current/protocol-replication.html#PROTOCOL-REPLICATION-PRIMARY-KEEPALIVE-MESSAGE\n  // (Byte 17: shouldRespond)\n  const shouldRespond = buffer.readInt8(17) !== 0;\n  return [lsn, {tag: 'keepalive', shouldRespond}];\n}\n\n// https://www.postgresql.org/docs/current/protocol-replication.html#PROTOCOL-REPLICATION-STANDBY-STATUS-UPDATE\nfunction makeAck(lsn: bigint): Buffer {\n  const microNow = BigInt(Date.now() - Date.UTC(2000, 0, 1)) * BigInt(1000);\n\n  const x = Buffer.alloc(34);\n  x[0] = 'r'.charCodeAt(0);\n  x.writeBigInt64BE(lsn, 1);\n  x.writeBigInt64BE(lsn, 9);\n  x.writeBigInt64BE(lsn, 17);\n  x.writeBigInt64BE(microNow, 25);\n  return x;\n}\n"],"mappings":";;;;;;;;;;;;;AAmBA,IAAM,6CAA6C;AAcnD,eAAsB,UACpB,IACA,IACA,MACA,cACA,KACA,iCAAiC,4CACjC,kBAAkB,mBAIjB;CACD,MAAM,UAAU,SACd,KACE;EACE,KAAK;GACJ,gBAAgB;GAChB,iBAAiB;GACjB,iBAAiB;EAClB,YAAY;IACT,qBAAqB;GACtB,aAAa;GACd;EACF,EAKD,GAAG,QACJ,CACF;CAQD,MAAM,CAAC,EAAC,wBAAuB,MAAM,OAEpC;;2CAEwC,QAAQ;CACjD,MAAM,yBAAyB,KAAK,MAAM,qBAAqB,IAAK;AACpE,IAAG,OACD,uBAAuB,mBAAmB,gDACK,uBAAuB,IACvE;CAED,MAAM,CAAC,UAAU,YAAY,MAAM,uBACjC,IACA,SACA,MACA,cACA,KACA,iCAAiC,EAClC;CAED,IAAI,cAAc,KAAK,KAAK;CAC5B,SAAS,QAAQ,KAAa;AAC5B,WAAS,MAAM,QAAQ,IAAI,CAAC;AAC5B,gBAAc,KAAK,KAAK;;CAG1B,MAAM,gBAAgB,kBAAkB;AAEtC,MADY,KAAK,KAAK,GACZ,cAAc,wBAAwB;AAC9C,WAAQ,GAAG;AACX,MAAG,QAAQ,wBAAwB;;IAEpC,yBAAyB,EAAE;CAE9B,IAAI,YAAY;CAEhB,MAAM,SAAS,IAAI,eADC,MAAM,eAAe,IAAI,EAAC,oBAAoB,MAAK,CAAC,CAC1B;CAC9C,MAAM,WAAW,aAAa,OAAsB,EAClD,eAAe;AACb,cAAY;AACZ,WAAS,aAAa,SAAS,SAAS;AACxC,gBAAc,cAAc;AAC5B,SAAO,QAAQ,KAAK;IAEvB,CAAC;AAEF,UAAS,KACP,eAGE,aAAa,GAAG,OAAO,gCAAgC,GAAG,QAAQ,OAAO,CAC5E;AACD,UAAS,KACP,UACA,MAGG,aAAa,SAAS,iBAAiB,EAAE,SAAS,qBACnD,GAAG,OAAO,cAAc,GAAG,QAAQ,QAAQ,EAAE,CAChD;AAED,MAAK;EACH,QAAQ;EACR,MAAM;EACN,QAAO,WAAU,mBAAmB,IAAI,QAAQ,OAAO;EAIvD,gBAAgB;EACjB,CAAC;AAEF,QAAO;EACL;EACA,MAAM,EAAC,MAAM,SAAQ;EACtB;;;;;;;AAQH,SAAS,uBAAuB,cAAgC;AAG9D,QAAO,aAAa,KAAI,MAAK,IAAI,EAAE,CAAC,MAAM,GAAG,GAAG,CAAC,CAAC,KAAK,IAAI;;AAG7D,eAAe,uBACb,IACA,SACA,MACA,cACA,KACA,aACA;AACA,MAAK,IAAI,IAAI,GAAG,IAAI,aAAa,IAC/B,KAAI;EACF,MAAM,SAAS,QACZ,OACC,0BAA0B,GAAG,KAAK,CAAC,WAAW,WAAW,IAAI,CAAC;;6BAE3C,uBAAuB,aAAa,CAAC;;SAGzD,CACA,SAAS;AACZ,SAAO,MAAM,QAAQ,IAAI,CAAC,OAAO,UAAU,EAAE,OAAO,UAAU,CAAC,CAAC;UACzD,GAAG;AACV,MAAI,aAAa,SAAS,eAAe;AAEvC,OAAI,EAAE,SAAS,kBAAkB;AAI/B,OAAG,OAAO,WAAW,IAAI,EAAE,IAAI,OAAO,EAAE,IAAI,EAAE;AAC9C,UAAM,MAAM,GAAG;AACf;;AAKF,OAAI,EAAE,SAAS,qCAAqC;AAClD,OAAG,QAAQ,qCAAqC,EAAE;AAClD,UAAM,IAAI,gBAAgB,sCAAsC,EAC9D,OAAO,GACR,CAAC;;;AAGN,QAAM;;AAGV,OAAM,IAAI,MACR,0BAA0B,YAAY,gCACvC;;AAGH,SAAS,mBACP,IACA,QACA,QACsB;AAEtB,KAAI,OAAO,OAAO,OAAQ,OAAO,OAAO,KAAM;AAC5C,KAAG,OAAO,mBAAmB,OAAO,GAAG;AACvC,SAAO;;CAET,MAAM,MAAM,OAAO,gBAAgB,EAAE;AACrC,KAAI,OAAO,OAAO,IAGhB,QAAO,CAAC,KAAK,OAAO,MAAM,OAAO,SAAS,GAAG,CAAC,CAAC;AAKjD,QAAO,CAAC,KAAK;EAAC,KAAK;EAAa,eADV,OAAO,SAAS,GAAG,KAAK;EACA,CAAC;;AAIjD,SAAS,QAAQ,KAAqB;CACpC,MAAM,WAAW,OAAO,KAAK,KAAK,GAAG,KAAK,IAAI,KAAM,GAAG,EAAE,CAAC,GAAG,OAAO,IAAK;CAEzE,MAAM,IAAI,OAAO,MAAM,GAAG;AAC1B,GAAE,KAAK,IAAI,WAAW,EAAE;AACxB,GAAE,gBAAgB,KAAK,EAAE;AACzB,GAAE,gBAAgB,KAAK,EAAE;AACzB,GAAE,gBAAgB,KAAK,GAAG;AAC1B,GAAE,gBAAgB,UAAU,GAAG;AAC/B,QAAO"}
+{"version":3,"file":"stream.js","names":[],"sources":["../../../../../../../../zero-cache/src/services/change-source/pg/logical-replication/stream.ts"],"sourcesContent":["import {\n  PG_ADMIN_SHUTDOWN,\n  PG_OBJECT_IN_USE,\n  PG_OBJECT_NOT_IN_PREREQUISITE_STATE,\n} from '@drdgvhbh/postgres-error-codes';\nimport type {LogContext} from '@rocicorp/logger';\nimport {defu} from 'defu';\nimport postgres, {type Options, type PostgresType} from 'postgres';\nimport {sleep} from '../../../../../../shared/src/sleep.ts';\nimport {getTypeParsers} from '../../../../db/pg-type-parser.ts';\nimport {type PostgresDB} from '../../../../types/pg.ts';\nimport {id, lit} from '../../../../types/sql.ts';\nimport {pipe, type Sink, type Source} from '../../../../types/streams.ts';\nimport {Subscription} from '../../../../types/subscription.ts';\nimport {AutoResetSignal} from '../../../change-streamer/schema/tables.ts';\nimport {fromBigInt} from '../lsn.ts';\nimport {PgoutputParser} from './pgoutput-parser.ts';\nimport type {Message} from './pgoutput.types.ts';\n\nconst DEFAULT_RETRIES_IF_REPLICATION_SLOT_ACTIVE = 5;\n\nexport type StreamMessage = [\n  lsn: bigint,\n  Message | {tag: 'keepalive'; shouldRespond: boolean},\n];\n\n// Expose the `queued` variable of the Subscription to allow\n// the change-source to determine if there are more messages\n// immediately available.\ntype SourceWithPendingQueue<T> = Source<T> & {\n  queued: number;\n};\n\nexport async function subscribe(\n  lc: LogContext,\n  db: PostgresDB,\n  slot: string,\n  publications: string[],\n  lsn: bigint,\n  retriesIfReplicationSlotActive = DEFAULT_RETRIES_IF_REPLICATION_SLOT_ACTIVE,\n  applicationName = 'zero-replicator',\n): Promise<{\n  messages: SourceWithPendingQueue<StreamMessage>;\n  acks: Sink<bigint>;\n}> {\n  const session = postgres(\n    defu(\n      {\n        max: 1,\n        ['fetch_types']: false, // Necessary for the streaming protocol\n        ['idle_timeout']: null,\n        ['max_lifetime']: null as unknown as number,\n        connection: {\n          ['application_name']: applicationName,\n          replication: 'database', // https://www.postgresql.org/docs/current/protocol-replication.html\n        },\n      },\n      // ParsedOptions are technically compatible with Options, but happen\n      // to not be typed that way. The postgres.js author does an equivalent\n      // merge of ParsedOptions and Options here:\n      // https://github.com/porsager/postgres/blob/089214e85c23c90cf142d47fb30bd03f42874984/src/subscribe.js#L13\n      db.options as unknown as Options<Record<string, PostgresType>>,\n    ),\n  );\n\n  // Postgres will send keepalives before timing out a wal_sender. It is possible that\n  // these keepalives are not received if there is back-pressure in the replication\n  // stream. To keep the connection alive, explicitly send keepalives if none have been\n  // sent within the last 75% of the wal_sender_timeout.\n  //\n  // https://www.postgresql.org/docs/current/runtime-config-replication.html#GUC-WAL-SENDER-TIMEOUT\n  const [{walSenderTimeoutMs}] = await session<\n    {walSenderTimeoutMs: number}[]\n  >`SELECT EXTRACT(EPOCH FROM (setting || unit)::interval) * 1000 \n        AS \"walSenderTimeoutMs\" FROM pg_settings\n        WHERE name = 'wal_sender_timeout'`.simple();\n  const manualKeepaliveTimeout = Math.floor(walSenderTimeoutMs * 0.75);\n  lc.info?.(\n    `wal_sender_timeout: ${walSenderTimeoutMs}ms. ` +\n      `Ensuring manual keepalives at least every ${manualKeepaliveTimeout}ms`,\n  );\n\n  const [readable, writable] = await startReplicationStream(\n    lc,\n    session,\n    slot,\n    publications,\n    lsn,\n    retriesIfReplicationSlotActive + 1,\n  );\n\n  let lastAckTime = Date.now();\n  function sendAck(lsn: bigint) {\n    writable.write(makeAck(lsn));\n    lastAckTime = Date.now();\n  }\n\n  const livenessTimer = setInterval(() => {\n    const now = Date.now();\n    if (now - lastAckTime > manualKeepaliveTimeout) {\n      sendAck(0n);\n      lc.debug?.(`sent manual keepalive`);\n    }\n  }, manualKeepaliveTimeout / 5);\n\n  let destroyed = false;\n  const typeParsers = await getTypeParsers(db, {returnJsonAsString: true});\n  const parser = new PgoutputParser(typeParsers);\n  const messages = Subscription.create<StreamMessage>({\n    cleanup: () => {\n      destroyed = true;\n      readable.destroyed || readable.destroy();\n      clearInterval(livenessTimer);\n      return session.end();\n    },\n  });\n\n  readable.once(\n    'close',\n    () =>\n      // Only log a warning if the stream was not manually closed.\n      destroyed || lc.warn?.(`replication stream closed by ${db.options.host}`),\n  );\n  readable.once(\n    'error',\n    e =>\n      // Don't log the shutdown signal. This is the expected way for upstream\n      // to close the connection (and will be logged downstream).\n      (e instanceof postgres.PostgresError && e.code === PG_ADMIN_SHUTDOWN) ||\n      lc.warn?.(`error from ${db.options.host}`, e),\n  );\n\n  pipe({\n    source: readable,\n    sink: messages,\n    parse: buffer => parseStreamMessage(lc, buffer, parser),\n    // Allow a small buffer of messages to be queued in the subscription so\n    // that the change-source loop can check the queue to determine if more\n    // messages are immediately available.\n    bufferMessages: 5,\n  });\n\n  return {\n    messages,\n    acks: {push: sendAck},\n  };\n}\n\n/**\n * Formats publication names for the START_REPLICATION command.\n * The replication protocol expects format: publication_names 'pub1,pub2'\n * Each name is escaped to handle any quotes that may have passed validation.\n */\nfunction formatPublicationNames(publications: string[]): string {\n  // lit() returns 'escaped_name' with surrounding quotes\n  // We strip the quotes since the outer quotes are in the template\n  return publications.map(p => lit(p).slice(1, -1)).join(',');\n}\n\nasync function startReplicationStream(\n  lc: LogContext,\n  session: postgres.Sql,\n  slot: string,\n  publications: string[],\n  lsn: bigint,\n  maxAttempts: number,\n) {\n  for (let i = 0; i < maxAttempts; i++) {\n    try {\n      const stream = session\n        .unsafe(\n          `START_REPLICATION SLOT ${id(slot)} LOGICAL ${fromBigInt(lsn)} (\n        proto_version '1',\n        publication_names '${formatPublicationNames(publications)}',\n        messages 'true'\n      )`,\n        )\n        .execute();\n      return await Promise.all([stream.readable(), stream.writable()]);\n    } catch (e) {\n      if (e instanceof postgres.PostgresError) {\n        // error: replication slot \"zero_slot_change_source_test_id\" is active for PID 268\n        if (e.code === PG_OBJECT_IN_USE) {\n          // The freeing up of the replication slot is not transactional;\n          // sometimes it takes time for Postgres to consider the slot\n          // inactive.\n          lc.warn?.(`attempt ${i + 1}: ${String(e)}`, e);\n          await sleep(10);\n          continue;\n        }\n        // error: This slot has been invalidated because it exceeded the maximum reserved size.\n        // (This is a different manifestation of a slot being invalidated when\n        //  the wal exceeds the max_slot_wal_keep_size)\n        if (e.code === PG_OBJECT_NOT_IN_PREREQUISITE_STATE) {\n          lc.error?.(`error starting replication stream`, e);\n          throw new AutoResetSignal(`unable to start replication stream`, {\n            cause: e,\n          });\n        }\n      }\n      throw e;\n    }\n  }\n  throw new Error(\n    `exceeded max attempts (${maxAttempts}) to start the Postgres stream`,\n  );\n}\n\nfunction parseStreamMessage(\n  lc: LogContext,\n  buffer: Buffer,\n  parser: PgoutputParser,\n): StreamMessage | null {\n  // https://www.postgresql.org/docs/current/protocol-replication.html#PROTOCOL-REPLICATION-START-REPLICATION\n  if (buffer[0] !== 0x77 && buffer[0] !== 0x6b) {\n    lc.warn?.('Unknown message', buffer[0]);\n    return null;\n  }\n  const lsn = buffer.readBigUInt64BE(1);\n  if (buffer[0] === 0x77) {\n    // https://www.postgresql.org/docs/current/protocol-replication.html#PROTOCOL-REPLICATION-XLOGDATA\n    // (Byte 25 is where the WAL data begins)\n    return [lsn, parser.parse(buffer.subarray(25))];\n  }\n  // https://www.postgresql.org/docs/current/protocol-replication.html#PROTOCOL-REPLICATION-PRIMARY-KEEPALIVE-MESSAGE\n  // (Byte 17: shouldRespond)\n  const shouldRespond = buffer.readInt8(17) !== 0;\n  return [lsn, {tag: 'keepalive', shouldRespond}];\n}\n\n// https://www.postgresql.org/docs/current/protocol-replication.html#PROTOCOL-REPLICATION-STANDBY-STATUS-UPDATE\nfunction makeAck(lsn: bigint): Buffer {\n  const microNow = BigInt(Date.now() - Date.UTC(2000, 0, 1)) * BigInt(1000);\n\n  const x = Buffer.alloc(34);\n  x[0] = 'r'.charCodeAt(0);\n  x.writeBigInt64BE(lsn, 1);\n  x.writeBigInt64BE(lsn, 9);\n  x.writeBigInt64BE(lsn, 17);\n  x.writeBigInt64BE(microNow, 25);\n  return x;\n}\n"],"mappings":";;;;;;;;;;;;;AAmBA,IAAM,6CAA6C;AAcnD,eAAsB,UACpB,IACA,IACA,MACA,cACA,KACA,iCAAiC,4CACjC,kBAAkB,mBAIjB;CACD,MAAM,UAAU,SACd,KACE;EACE,KAAK;GACJ,gBAAgB;GAChB,iBAAiB;GACjB,iBAAiB;EAClB,YAAY;IACT,qBAAqB;GACtB,aAAa;EACf;CACF,GAKA,GAAG,OACL,CACF;CAQA,MAAM,CAAC,EAAC,wBAAuB,MAAM,OAEpC;;2CAEwC,OAAO;CAChD,MAAM,yBAAyB,KAAK,MAAM,qBAAqB,GAAI;CACnE,GAAG,OACD,uBAAuB,mBAAmB,gDACK,uBAAuB,GACxE;CAEA,MAAM,CAAC,UAAU,YAAY,MAAM,uBACjC,IACA,SACA,MACA,cACA,KACA,iCAAiC,CACnC;CAEA,IAAI,cAAc,KAAK,IAAI;CAC3B,SAAS,QAAQ,KAAa;EAC5B,SAAS,MAAM,QAAQ,GAAG,CAAC;EAC3B,cAAc,KAAK,IAAI;CACzB;CAEA,MAAM,gBAAgB,kBAAkB;EAEtC,IADY,KAAK,IACb,IAAM,cAAc,wBAAwB;GAC9C,QAAQ,EAAE;GACV,GAAG,QAAQ,uBAAuB;EACpC;CACF,GAAG,yBAAyB,CAAC;CAE7B,IAAI,YAAY;CAEhB,MAAM,SAAS,IAAI,eAAe,MADR,eAAe,IAAI,EAAC,oBAAoB,KAAI,CAAC,CAC1B;CAC7C,MAAM,WAAW,aAAa,OAAsB,EAClD,eAAe;EACb,YAAY;EACZ,SAAS,aAAa,SAAS,QAAQ;EACvC,cAAc,aAAa;EAC3B,OAAO,QAAQ,IAAI;CACrB,EACF,CAAC;CAED,SAAS,KACP,eAGE,aAAa,GAAG,OAAO,gCAAgC,GAAG,QAAQ,MAAM,CAC5E;CACA,SAAS,KACP,UACA,MAGG,aAAa,SAAS,iBAAiB,EAAE,SAAS,qBACnD,GAAG,OAAO,cAAc,GAAG,QAAQ,QAAQ,CAAC,CAChD;CAEA,KAAK;EACH,QAAQ;EACR,MAAM;EACN,QAAO,WAAU,mBAAmB,IAAI,QAAQ,MAAM;EAItD,gBAAgB;CAClB,CAAC;CAED,OAAO;EACL;EACA,MAAM,EAAC,MAAM,QAAO;CACtB;AACF;;;;;;AAOA,SAAS,uBAAuB,cAAgC;CAG9D,OAAO,aAAa,KAAI,MAAK,IAAI,CAAC,EAAE,MAAM,GAAG,EAAE,CAAC,EAAE,KAAK,GAAG;AAC5D;AAEA,eAAe,uBACb,IACA,SACA,MACA,cACA,KACA,aACA;CACA,KAAK,IAAI,IAAI,GAAG,IAAI,aAAa,KAC/B,IAAI;EACF,MAAM,SAAS,QACZ,OACC,0BAA0B,GAAG,IAAI,EAAE,WAAW,WAAW,GAAG,EAAE;;6BAE3C,uBAAuB,YAAY,EAAE;;QAG1D,EACC,QAAQ;EACX,OAAO,MAAM,QAAQ,IAAI,CAAC,OAAO,SAAS,GAAG,OAAO,SAAS,CAAC,CAAC;CACjE,SAAS,GAAG;EACV,IAAI,aAAa,SAAS,eAAe;GAEvC,IAAI,EAAE,SAAS,kBAAkB;IAI/B,GAAG,OAAO,WAAW,IAAI,EAAE,IAAI,OAAO,CAAC,KAAK,CAAC;IAC7C,MAAM,MAAM,EAAE;IACd;GACF;GAIA,IAAI,EAAE,SAAS,qCAAqC;IAClD,GAAG,QAAQ,qCAAqC,CAAC;IACjD,MAAM,IAAI,gBAAgB,sCAAsC,EAC9D,OAAO,EACT,CAAC;GACH;EACF;EACA,MAAM;CACR;CAEF,MAAM,IAAI,MACR,0BAA0B,YAAY,+BACxC;AACF;AAEA,SAAS,mBACP,IACA,QACA,QACsB;CAEtB,IAAI,OAAO,OAAO,OAAQ,OAAO,OAAO,KAAM;EAC5C,GAAG,OAAO,mBAAmB,OAAO,EAAE;EACtC,OAAO;CACT;CACA,MAAM,MAAM,OAAO,gBAAgB,CAAC;CACpC,IAAI,OAAO,OAAO,KAGhB,OAAO,CAAC,KAAK,OAAO,MAAM,OAAO,SAAS,EAAE,CAAC,CAAC;CAKhD,OAAO,CAAC,KAAK;EAAC,KAAK;EAAa,eADV,OAAO,SAAS,EAAE,MAAM;CACD,CAAC;AAChD;AAGA,SAAS,QAAQ,KAAqB;CACpC,MAAM,WAAW,OAAO,KAAK,IAAI,IAAI,KAAK,IAAI,KAAM,GAAG,CAAC,CAAC,IAAI,OAAO,GAAI;CAExE,MAAM,IAAI,OAAO,MAAM,EAAE;CACzB,EAAE,KAAK,IAAI,WAAW,CAAC;CACvB,EAAE,gBAAgB,KAAK,CAAC;CACxB,EAAE,gBAAgB,KAAK,CAAC;CACxB,EAAE,gBAAgB,KAAK,EAAE;CACzB,EAAE,gBAAgB,UAAU,EAAE;CAC9B,OAAO;AACT"}

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

@@ -1 +1 @@
-{"version":3,"file":"lsn.js","names":[],"sources":["../../../../../../../zero-cache/src/services/change-source/pg/lsn.ts"],"sourcesContent":["import {assert} from '../../../../../shared/src/asserts.ts';\nimport {\n  majorVersionToString,\n  stateVersionFromString,\n} from '../../../types/state-version.ts';\nimport type {Change} from '../protocol/current/data.ts';\n\n/**\n * Parsing and conversion utilities for the pg_lsn Type, which represents\n * the \"Log Sequence Number\" used as a monotonic progress marker for logical\n * replication from a PostgresSQL database.\n *\n * The LSN is a 64-bit integer represented in logical replication as two\n * hexadecimal numbers (up to 8 digits each) separated by a slash. This is\n * converted to a LexiVersion and used as DB-agnostic version in change log,\n * invalidation index, and row version in the tables of the sync replica.\n */\nexport type LSN = string;\n\nexport type RecordType = Change['tag'];\n\nexport function toBigInt(lsn: LSN): bigint {\n  const parts = lsn.split('/');\n  assert(parts.length === 2, `Malformed LSN: \"${lsn}\"`);\n  const high = BigInt(`0x${parts[0]}`);\n  const low = BigInt(`0x${parts[1]}`);\n  const val = (high << 32n) + low;\n  return val;\n}\n\nexport function toStateVersionString(lsn: LSN): string {\n  return majorVersionToString(toBigInt(lsn));\n}\n\n/** The LSN is tracked by the `major` component of the StateVersion. */\nexport function fromStateVersionString(ver: string): LSN {\n  const {major} = stateVersionFromString(ver);\n  return fromBigInt(major);\n}\n\nexport function fromBigInt(val: bigint): LSN {\n  const high = val >> 32n;\n  const low = val & 0xffffffffn;\n  return `${high.toString(16).toUpperCase()}/${low.toString(16).toUpperCase()}`;\n}\n"],"mappings":";;;AAqBA,SAAgB,SAAS,KAAkB;CACzC,MAAM,QAAQ,IAAI,MAAM,IAAI;AAC5B,QAAO,MAAM,WAAW,GAAG,mBAAmB,IAAI,GAAG;CACrD,MAAM,OAAO,OAAO,KAAK,MAAM,KAAK;CACpC,MAAM,MAAM,OAAO,KAAK,MAAM,KAAK;AAEnC,SADa,QAAQ,OAAO;;AAI9B,SAAgB,qBAAqB,KAAkB;AACrD,QAAO,qBAAqB,SAAS,IAAI,CAAC;;;AAI5C,SAAgB,uBAAuB,KAAkB;CACvD,MAAM,EAAC,UAAS,uBAAuB,IAAI;AAC3C,QAAO,WAAW,MAAM;;AAG1B,SAAgB,WAAW,KAAkB;CAC3C,MAAM,OAAO,OAAO;CACpB,MAAM,MAAM,MAAM;AAClB,QAAO,GAAG,KAAK,SAAS,GAAG,CAAC,aAAa,CAAC,GAAG,IAAI,SAAS,GAAG,CAAC,aAAa"}
+{"version":3,"file":"lsn.js","names":[],"sources":["../../../../../../../zero-cache/src/services/change-source/pg/lsn.ts"],"sourcesContent":["import {assert} from '../../../../../shared/src/asserts.ts';\nimport {\n  majorVersionToString,\n  stateVersionFromString,\n} from '../../../types/state-version.ts';\nimport type {Change} from '../protocol/current/data.ts';\n\n/**\n * Parsing and conversion utilities for the pg_lsn Type, which represents\n * the \"Log Sequence Number\" used as a monotonic progress marker for logical\n * replication from a PostgresSQL database.\n *\n * The LSN is a 64-bit integer represented in logical replication as two\n * hexadecimal numbers (up to 8 digits each) separated by a slash. This is\n * converted to a LexiVersion and used as DB-agnostic version in change log,\n * invalidation index, and row version in the tables of the sync replica.\n */\nexport type LSN = string;\n\nexport type RecordType = Change['tag'];\n\nexport function toBigInt(lsn: LSN): bigint {\n  const parts = lsn.split('/');\n  assert(parts.length === 2, `Malformed LSN: \"${lsn}\"`);\n  const high = BigInt(`0x${parts[0]}`);\n  const low = BigInt(`0x${parts[1]}`);\n  const val = (high << 32n) + low;\n  return val;\n}\n\nexport function toStateVersionString(lsn: LSN): string {\n  return majorVersionToString(toBigInt(lsn));\n}\n\n/** The LSN is tracked by the `major` component of the StateVersion. */\nexport function fromStateVersionString(ver: string): LSN {\n  const {major} = stateVersionFromString(ver);\n  return fromBigInt(major);\n}\n\nexport function fromBigInt(val: bigint): LSN {\n  const high = val >> 32n;\n  const low = val & 0xffffffffn;\n  return `${high.toString(16).toUpperCase()}/${low.toString(16).toUpperCase()}`;\n}\n"],"mappings":";;;AAqBA,SAAgB,SAAS,KAAkB;CACzC,MAAM,QAAQ,IAAI,MAAM,GAAG;CAC3B,OAAO,MAAM,WAAW,GAAG,mBAAmB,IAAI,EAAE;CACpD,MAAM,OAAO,OAAO,KAAK,MAAM,IAAI;CACnC,MAAM,MAAM,OAAO,KAAK,MAAM,IAAI;CAElC,QADa,QAAQ,OAAO;AAE9B;AAEA,SAAgB,qBAAqB,KAAkB;CACrD,OAAO,qBAAqB,SAAS,GAAG,CAAC;AAC3C;;AAGA,SAAgB,uBAAuB,KAAkB;CACvD,MAAM,EAAC,UAAS,uBAAuB,GAAG;CAC1C,OAAO,WAAW,KAAK;AACzB;AAEA,SAAgB,WAAW,KAAkB;CAC3C,MAAM,OAAO,OAAO;CACpB,MAAM,MAAM,MAAM;CAClB,OAAO,GAAG,KAAK,SAAS,EAAE,EAAE,YAAY,EAAE,GAAG,IAAI,SAAS,EAAE,EAAE,YAAY;AAC5E"}

package/out/zero-cache/src/services/change-source/pg/replication-slots.d.ts

@@ -1,5 +1,5 @@
 import type { LogContext } from '@rocicorp/logger';
-import postgres from 'postgres';
+import type postgres from 'postgres';
 import { type PostgresDB } from '../../../types/pg';
 import { type ShardID } from '../../../types/shards';
 export type ReplicationSlot = {

package/out/zero-cache/src/services/change-source/pg/replication-slots.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"replication-slots.d.ts","sourceRoot":"","sources":["../../../../../../../zero-cache/src/services/change-source/pg/replication-slots.ts"],"names":[],"mappings":"AAIA,OAAO,KAAK,EAAC,UAAU,EAAC,MAAM,kBAAkB,CAAC;AACjD,OAAO,QAAQ,MAAM,UAAU,CAAC;AAEhC,OAAO,EAAkB,KAAK,UAAU,EAAC,MAAM,mBAAmB,CAAC;AACnE,OAAO,EAAiB,KAAK,OAAO,EAAC,MAAM,uBAAuB,CAAC;AAUnE,MAAM,MAAM,eAAe,GAAG;IAC5B,SAAS,EAAE,MAAM,CAAC;IAClB,gBAAgB,EAAE,MAAM,CAAC;IACzB,aAAa,EAAE,MAAM,CAAC;IACtB,aAAa,EAAE,MAAM,CAAC;CACvB,CAAC;AAEF,MAAM,MAAM,cAAc,GAAG;IAC3B,QAAQ,EAAE,MAAM,CAAC;IAGjB,QAAQ,CAAC,EAAE,OAAO,CAAC;IAGnB,WAAW,CAAC,EAAE,MAAM,CAAC;CACtB,CAAC;AAoBF,wBAAsB,qBAAqB,CACzC,EAAE,EAAE,UAAU,EACd,OAAO,EAAE,QAAQ,CAAC,GAAG,EACrB,EAAC,QAAQ,EAAE,QAAQ,EAAE,WAAoC,EAAC,EAAE,cAAc,GACzE,OAAO,CAAC,eAAe,CAAC,CA0C1B;AAED;;;;;;;;;;;;;;;;;;;;;;GAsBG;AACH,wBAAsB,oBAAoB,CACxC,EAAE,EAAE,UAAU,EACd,GAAG,EAAE,UAAU,EACf,kBAAkB,EAAE,QAAQ,CAAC,GAAG,EAChC,KAAK,EAAE,OAAO,EACd,SAAS,EAAE,MAAM,EACjB,QAAQ,EAAE,OAAO,GAChB,OAAO,CAAC,eAAe,CAAC,CAqE1B;AAED;;;;;;;;;GASG;AACH,wBAAsB,uBAAuB,CAC3C,EAAE,EAAE,UAAU,EACd,GAAG,EAAE,UAAU,EACf,KAAK,EAAE,OAAO,EACd,UAAU,EAAE,MAAM,GACjB,OAAO,CAAC;IAAC,OAAO,EAAE,MAAM,CAAC;IAAC,MAAM,EAAE,MAAM,CAAC;IAAC,QAAQ,EAAE,MAAM,CAAA;CAAC,CAAC,CA2D9D;AAMD,wBAAgB,cAAc,CAAC,CAAC,EAAE,MAAM,UAUvC"}
+{"version":3,"file":"replication-slots.d.ts","sourceRoot":"","sources":["../../../../../../../zero-cache/src/services/change-source/pg/replication-slots.ts"],"names":[],"mappings":"AAIA,OAAO,KAAK,EAAC,UAAU,EAAC,MAAM,kBAAkB,CAAC;AACjD,OAAO,KAAK,QAAQ,MAAM,UAAU,CAAC;AAErC,OAAO,EAAkB,KAAK,UAAU,EAAC,MAAM,mBAAmB,CAAC;AACnE,OAAO,EAAiB,KAAK,OAAO,EAAC,MAAM,uBAAuB,CAAC;AAUnE,MAAM,MAAM,eAAe,GAAG;IAC5B,SAAS,EAAE,MAAM,CAAC;IAClB,gBAAgB,EAAE,MAAM,CAAC;IACzB,aAAa,EAAE,MAAM,CAAC;IACtB,aAAa,EAAE,MAAM,CAAC;CACvB,CAAC;AAEF,MAAM,MAAM,cAAc,GAAG;IAC3B,QAAQ,EAAE,MAAM,CAAC;IAGjB,QAAQ,CAAC,EAAE,OAAO,CAAC;IAGnB,WAAW,CAAC,EAAE,MAAM,CAAC;CACtB,CAAC;AAoBF,wBAAsB,qBAAqB,CACzC,EAAE,EAAE,UAAU,EACd,OAAO,EAAE,QAAQ,CAAC,GAAG,EACrB,EAAC,QAAQ,EAAE,QAAQ,EAAE,WAAoC,EAAC,EAAE,cAAc,GACzE,OAAO,CAAC,eAAe,CAAC,CA0C1B;AAED;;;;;;;;;;;;;;;;;;;;;;GAsBG;AACH,wBAAsB,oBAAoB,CACxC,EAAE,EAAE,UAAU,EACd,GAAG,EAAE,UAAU,EACf,kBAAkB,EAAE,QAAQ,CAAC,GAAG,EAChC,KAAK,EAAE,OAAO,EACd,SAAS,EAAE,MAAM,EACjB,QAAQ,EAAE,OAAO,GAChB,OAAO,CAAC,eAAe,CAAC,CAmE1B;AAED;;;;;;;;;GASG;AACH,wBAAsB,uBAAuB,CAC3C,EAAE,EAAE,UAAU,EACd,GAAG,EAAE,UAAU,EACf,KAAK,EAAE,OAAO,EACd,UAAU,EAAE,MAAM,GACjB,OAAO,CAAC;IAAC,OAAO,EAAE,MAAM,CAAC;IAAC,MAAM,EAAE,MAAM,CAAC;IAAC,QAAQ,EAAE,MAAM,CAAA;CAAC,CAAC,CAY9D;AA+DD,wBAAgB,cAAc,CAAC,CAAC,EAAE,MAAM,UAUvC"}

package/out/zero-cache/src/services/change-source/pg/replication-slots.js

@@ -4,7 +4,6 @@ import { toStateVersionString } from "./lsn.js";
 import { runTx } from "../../../db/run-transaction.js";
 import { createReplica, replicationSlotExpression, replicationSlotPrefix } from "./schema/shard.js";
 import { orTimeout } from "../../../types/timeout.js";
-import postgres from "postgres";
 import { PG_CONFIGURATION_LIMIT_EXCEEDED, PG_INSUFFICIENT_PRIVILEGE } from "@drdgvhbh/postgres-error-codes";
 //#region ../zero-cache/src/services/change-source/pg/replication-slots.ts
 var CREATE_REPLICATION_SLOT_TIMEOUT_MS = 3e4;
@@ -46,49 +45,46 @@ async function createReplicationSlot(lc, session, { slotName, failover, lockTime
 async function createReplicaAndSlot(lc, sql, replicationSession, shard, replicaID, failover) {
 	const lockName = replicationSlotManagementLock(shard);
 	const slotPoolPrefix = replicationSlotPrefix(shard);
-	for (let first = true;; first = false) try {
-		return runTx(sql, async (tx) => {
-			await tx`SELECT pg_advisory_xact_lock(hashtext(${lockName}))`;
-			let slotName;
-			const names = await tx`
+	for (let first = true;; first = false) {
+		await dropUnclaimedSlots(lc, sql, shard);
+		try {
+			return await runTx(sql, async (tx) => {
+				await tx`SELECT pg_advisory_xact_lock(hashtext(${lockName}))`;
+				let slotName;
+				const names = await tx`
           SELECT slot_name as name FROM pg_replication_slots
             WHERE slot_name LIKE ${slotPoolPrefix + "%"};
         `.values();
-			const inUse = new Set(names.flat());
-			for (let next = 0;; next++) {
-				const candidateName = `${slotPoolPrefix}${slotPoolSuffix(next)}`;
-				if (!inUse.has(candidateName)) {
-					slotName = candidateName;
-					break;
+				const inUse = new Set(names.flat());
+				for (let next = 0;; next++) {
+					const candidateName = `${slotPoolPrefix}${slotPoolSuffix(next)}`;
+					if (!inUse.has(candidateName)) {
+						slotName = candidateName;
+						break;
+					}
 				}
-			}
-			const slot = await createReplicationSlot(lc, replicationSession, {
-				slotName,
-				failover
+				const slot = await createReplicationSlot(lc, replicationSession, {
+					slotName,
+					failover
+				});
+				await createReplica(tx, shard, replicaID, slot.slot_name, toStateVersionString(slot.consistent_point));
+				return slot;
 			});
-			await createReplica(tx, shard, replicaID, slot.slot_name, toStateVersionString(slot.consistent_point));
-			return slot;
-		});
-	} catch (e) {
-		if (first && e instanceof postgres.PostgresError) {
-			if (isPostgresError(e, PG_INSUFFICIENT_PRIVILEGE)) {
+		} catch (e) {
+			if (first && isPostgresError(e, PG_INSUFFICIENT_PRIVILEGE)) {
 				await sql`ALTER ROLE current_user WITH REPLICATION`;
 				lc.info?.(`Added the REPLICATION role to database user`);
 				continue;
 			}
-			if (isPostgresError(e, PG_CONFIGURATION_LIMIT_EXCEEDED)) {
-				const dropped = await sql`
-                SELECT slot_name as slot, pg_drop_replication_slot(slot_name)
-                  FROM pg_replication_slots
-                  WHERE slot_name LIKE ${replicationSlotExpression(shard)} AND NOT active`;
-				if (dropped.length) {
-					lc.warn?.(`Dropped inactive replication slots: ${dropped.map(({ slot }) => slot)}`, e);
-					continue;
-				}
-				lc.error?.(`Unable to drop replication slots`, e);
+			if (first && isPostgresError(e, PG_CONFIGURATION_LIMIT_EXCEEDED)) {
+				lc.warn?.(`Reached max replication slots. Attempting to clean up unused slots`, e);
+				await sql`
+          DELETE FROM ${sql(`${upstreamSchema(shard)}.replicas`)} USING pg_replication_slots slots
+            WHERE replicas.slot = slots.slot_name AND NOT slots.active`;
+				continue;
 			}
+			throw e;
 		}
-		throw e;
 	}
 }
 /**
@@ -111,8 +107,12 @@ async function dropOldReplicasAndSlots(lc, sql, shard, beforeRank) {
 		lc.info?.(`Deleting ${oldReplicas.length} old replica(s)`, { oldReplicas });
 		await sql`DELETE FROM ${sql(replicasTable)} WHERE rank < ${beforeRank}`;
 	}
+	return dropUnclaimedSlots(lc, sql, shard);
+}
+function dropUnclaimedSlots(lc, sql, shard) {
 	const lockName = replicationSlotManagementLock(shard);
 	const slotExpression = replicationSlotExpression(shard);
+	const replicasTable = `${upstreamSchema(shard)}.replicas`;
 	return runTx(sql, async (tx) => {
 		await tx`SELECT pg_advisory_xact_lock(hashtext(${lockName}))`;
 		const dropped = await tx`