@powersync/service-module-mongodb 0.0.0-dev-20250102111825 → 0.0.0-dev-20250108073049

package/src/storage/MongoBucketStorage.ts

@@ -1,531 +0,0 @@
-import { SqlSyncRules } from '@powersync/service-sync-rules';
-import { wrapWithAbort } from 'ix/asynciterable/operators/withabort.js';
-import { LRUCache } from 'lru-cache/min';
-import * as mongo from 'mongodb';
-import * as timers from 'timers/promises';
-
-import { storage, sync, utils } from '@powersync/service-core';
-
-import { DisposableObserver, logger } from '@powersync/lib-services-framework';
-import { v4 as uuid } from 'uuid';
-
-import { MongoLockManager } from '../locks/MongoLockManager.js';
-import { PowerSyncMongo } from './implementation/db.js';
-import { SyncRuleDocument } from './implementation/models.js';
-import { MongoPersistedSyncRulesContent } from './implementation/MongoPersistedSyncRulesContent.js';
-import { MongoSyncBucketStorage } from './implementation/MongoSyncBucketStorage.js';
-import { generateSlotName } from './implementation/util.js';
-
-export class MongoBucketStorage
-  extends DisposableObserver<storage.BucketStorageFactoryListener>
-  implements storage.BucketStorageFactory
-{
-  private readonly client: mongo.MongoClient;
-  private readonly session: mongo.ClientSession;
-  // TODO: This is still Postgres specific and needs to be reworked
-  public readonly slot_name_prefix: string;
-
-  private readonly storageCache = new LRUCache<number, MongoSyncBucketStorage>({
-    max: 3,
-    fetchMethod: async (id) => {
-      const doc2 = await this.db.sync_rules.findOne(
-        {
-          _id: id
-        },
-        { limit: 1 }
-      );
-      if (doc2 == null) {
-        // Deleted in the meantime?
-        return undefined;
-      }
-      const rules = new MongoPersistedSyncRulesContent(this.db, doc2);
-      return this.getInstance(rules);
-    },
-    dispose: (storage) => {
-      storage[Symbol.dispose]();
-    }
-  });
-
-  public readonly db: PowerSyncMongo;
-
-  constructor(
-    db: PowerSyncMongo,
-    options: {
-      slot_name_prefix: string;
-    }
-  ) {
-    super();
-    this.client = db.client;
-    this.db = db;
-    this.session = this.client.startSession();
-    this.slot_name_prefix = options.slot_name_prefix;
-  }
-
-  getInstance(options: storage.PersistedSyncRulesContent): MongoSyncBucketStorage {
-    let { id, slot_name } = options;
-    if ((typeof id as any) == 'bigint') {
-      id = Number(id);
-    }
-    const storage = new MongoSyncBucketStorage(this, id, options, slot_name);
-    this.iterateListeners((cb) => cb.syncStorageCreated?.(storage));
-    storage.registerListener({
-      batchStarted: (batch) => {
-        // This nested listener will be automatically disposed when the storage is disposed
-        batch.registerManagedListener(storage, {
-          replicationEvent: (payload) => this.iterateListeners((cb) => cb.replicationEvent?.(payload))
-        });
-      }
-    });
-    return storage;
-  }
-
-  async configureSyncRules(sync_rules: string, options?: { lock?: boolean }) {
-    const next = await this.getNextSyncRulesContent();
-    const active = await this.getActiveSyncRulesContent();
-
-    if (next?.sync_rules_content == sync_rules) {
-      logger.info('Sync rules from configuration unchanged');
-      return { updated: false };
-    } else if (next == null && active?.sync_rules_content == sync_rules) {
-      logger.info('Sync rules from configuration unchanged');
-      return { updated: false };
-    } else {
-      logger.info('Sync rules updated from configuration');
-      const persisted_sync_rules = await this.updateSyncRules({
-        content: sync_rules,
-        lock: options?.lock
-      });
-      return { updated: true, persisted_sync_rules, lock: persisted_sync_rules.current_lock ?? undefined };
-    }
-  }
-
-  async slotRemoved(slot_name: string) {
-    const next = await this.getNextSyncRulesContent();
-    const active = await this.getActiveSyncRulesContent();
-
-    // In both the below cases, we create a new sync rules instance.
-    // The current one will continue erroring until the next one has finished processing.
-    // TODO: Update
-    if (next != null && next.slot_name == slot_name) {
-      // We need to redo the "next" sync rules
-      await this.updateSyncRules({
-        content: next.sync_rules_content
-      });
-      // Pro-actively stop replicating
-      await this.db.sync_rules.updateOne(
-        {
-          _id: next.id,
-          state: storage.SyncRuleState.PROCESSING
-        },
-        {
-          $set: {
-            state: storage.SyncRuleState.STOP
-          }
-        }
-      );
-    } else if (next == null && active?.slot_name == slot_name) {
-      // Slot removed for "active" sync rules, while there is no "next" one.
-      await this.updateSyncRules({
-        content: active.sync_rules_content
-      });
-
-      // Pro-actively stop replicating
-      await this.db.sync_rules.updateOne(
-        {
-          _id: active.id,
-          state: storage.SyncRuleState.ACTIVE
-        },
-        {
-          $set: {
-            state: storage.SyncRuleState.STOP
-          }
-        }
-      );
-    }
-  }
-
-  async updateSyncRules(options: storage.UpdateSyncRulesOptions): Promise<MongoPersistedSyncRulesContent> {
-    // Parse and validate before applying any changes
-    const parsed = SqlSyncRules.fromYaml(options.content, {
-      // No schema-based validation at this point
-      schema: undefined,
-      defaultSchema: 'not_applicable', // Not needed for validation
-      throwOnError: true
-    });
-
-    let rules: MongoPersistedSyncRulesContent | undefined = undefined;
-
-    await this.session.withTransaction(async () => {
-      // Only have a single set of sync rules with PROCESSING.
-      await this.db.sync_rules.updateMany(
-        {
-          state: storage.SyncRuleState.PROCESSING
-        },
-        { $set: { state: storage.SyncRuleState.STOP } }
-      );
-
-      const id_doc = await this.db.op_id_sequence.findOneAndUpdate(
-        {
-          _id: 'sync_rules'
-        },
-        {
-          $inc: {
-            op_id: 1n
-          }
-        },
-        {
-          upsert: true,
-          returnDocument: 'after'
-        }
-      );
-
-      const id = Number(id_doc!.op_id);
-      const slot_name = generateSlotName(this.slot_name_prefix, id);
-
-      const doc: SyncRuleDocument = {
-        _id: id,
-        content: options.content,
-        last_checkpoint: null,
-        last_checkpoint_lsn: null,
-        no_checkpoint_before: null,
-        keepalive_op: null,
-        snapshot_done: false,
-        state: storage.SyncRuleState.PROCESSING,
-        slot_name: slot_name,
-        last_checkpoint_ts: null,
-        last_fatal_error: null,
-        last_keepalive_ts: null
-      };
-      await this.db.sync_rules.insertOne(doc);
-      rules = new MongoPersistedSyncRulesContent(this.db, doc);
-      if (options.lock) {
-        const lock = await rules.lock();
-      }
-    });
-
-    return rules!;
-  }
-
-  async getActiveSyncRulesContent(): Promise<MongoPersistedSyncRulesContent | null> {
-    const doc = await this.db.sync_rules.findOne(
-      {
-        state: storage.SyncRuleState.ACTIVE
-      },
-      { sort: { _id: -1 }, limit: 1 }
-    );
-    if (doc == null) {
-      return null;
-    }
-
-    return new MongoPersistedSyncRulesContent(this.db, doc);
-  }
-
-  async getActiveSyncRules(options: storage.ParseSyncRulesOptions): Promise<storage.PersistedSyncRules | null> {
-    const content = await this.getActiveSyncRulesContent();
-    return content?.parsed(options) ?? null;
-  }
-
-  async getNextSyncRulesContent(): Promise<MongoPersistedSyncRulesContent | null> {
-    const doc = await this.db.sync_rules.findOne(
-      {
-        state: storage.SyncRuleState.PROCESSING
-      },
-      { sort: { _id: -1 }, limit: 1 }
-    );
-    if (doc == null) {
-      return null;
-    }
-
-    return new MongoPersistedSyncRulesContent(this.db, doc);
-  }
-
-  async getNextSyncRules(options: storage.ParseSyncRulesOptions): Promise<storage.PersistedSyncRules | null> {
-    const content = await this.getNextSyncRulesContent();
-    return content?.parsed(options) ?? null;
-  }
-
-  async getReplicatingSyncRules(): Promise<storage.PersistedSyncRulesContent[]> {
-    const docs = await this.db.sync_rules
-      .find({
-        $or: [{ state: storage.SyncRuleState.ACTIVE }, { state: storage.SyncRuleState.PROCESSING }]
-      })
-      .toArray();
-
-    return docs.map((doc) => {
-      return new MongoPersistedSyncRulesContent(this.db, doc);
-    });
-  }
-
-  async getStoppedSyncRules(): Promise<storage.PersistedSyncRulesContent[]> {
-    const docs = await this.db.sync_rules
-      .find({
-        state: storage.SyncRuleState.STOP
-      })
-      .toArray();
-
-    return docs.map((doc) => {
-      return new MongoPersistedSyncRulesContent(this.db, doc);
-    });
-  }
-
-  async getActiveCheckpoint(): Promise<storage.ActiveCheckpoint> {
-    const doc = await this.db.sync_rules.findOne(
-      {
-        state: storage.SyncRuleState.ACTIVE
-      },
-      {
-        sort: { _id: -1 },
-        limit: 1,
-        projection: { _id: 1, last_checkpoint: 1, last_checkpoint_lsn: 1 }
-      }
-    );
-
-    return this.makeActiveCheckpoint(doc);
-  }
-
-  async getStorageMetrics(): Promise<storage.StorageMetrics> {
-    const ignoreNotExiting = (e: unknown) => {
-      if (e instanceof mongo.MongoServerError && e.codeName == 'NamespaceNotFound') {
-        // Collection doesn't exist - return 0
-        return [{ storageStats: { size: 0 } }];
-      } else {
-        return Promise.reject(e);
-      }
-    };
-
-    const active_sync_rules = await this.getActiveSyncRules({ defaultSchema: 'public' });
-    if (active_sync_rules == null) {
-      return {
-        operations_size_bytes: 0,
-        parameters_size_bytes: 0,
-        replication_size_bytes: 0
-      };
-    }
-    const operations_aggregate = await this.db.bucket_data
-
-      .aggregate([
-        {
-          $collStats: {
-            storageStats: {}
-          }
-        }
-      ])
-      .toArray()
-      .catch(ignoreNotExiting);
-
-    const parameters_aggregate = await this.db.bucket_parameters
-      .aggregate([
-        {
-          $collStats: {
-            storageStats: {}
-          }
-        }
-      ])
-      .toArray()
-      .catch(ignoreNotExiting);
-
-    const replication_aggregate = await this.db.current_data
-      .aggregate([
-        {
-          $collStats: {
-            storageStats: {}
-          }
-        }
-      ])
-      .toArray()
-      .catch(ignoreNotExiting);
-
-    return {
-      operations_size_bytes: Number(operations_aggregate[0].storageStats.size),
-      parameters_size_bytes: Number(parameters_aggregate[0].storageStats.size),
-      replication_size_bytes: Number(replication_aggregate[0].storageStats.size)
-    };
-  }
-
-  async getPowerSyncInstanceId(): Promise<string> {
-    let instance = await this.db.instance.findOne({
-      _id: { $exists: true }
-    });
-
-    if (!instance) {
-      const manager = new MongoLockManager({
-        collection: this.db.locks,
-        name: `instance-id-insertion-lock`
-      });
-
-      await manager.lock(async () => {
-        await this.db.instance.insertOne({
-          _id: uuid()
-        });
-      });
-
-      instance = await this.db.instance.findOne({
-        _id: { $exists: true }
-      });
-    }
-
-    return instance!._id;
-  }
-
-  private makeActiveCheckpoint(doc: SyncRuleDocument | null) {
-    return {
-      checkpoint: utils.timestampToOpId(doc?.last_checkpoint ?? 0n),
-      lsn: doc?.last_checkpoint_lsn ?? null,
-      hasSyncRules() {
-        return doc != null;
-      },
-      getBucketStorage: async () => {
-        if (doc == null) {
-          return null;
-        }
-        return (await this.storageCache.fetch(doc._id)) ?? null;
-      }
-    } satisfies storage.ActiveCheckpoint;
-  }
-
-  /**
-   * Instance-wide watch on the latest available checkpoint (op_id + lsn).
-   */
-  private async *watchActiveCheckpoint(signal: AbortSignal): AsyncIterable<storage.ActiveCheckpoint> {
-    const pipeline: mongo.Document[] = [
-      {
-        $match: {
-          'fullDocument.state': 'ACTIVE',
-          operationType: { $in: ['insert', 'update'] }
-        }
-      },
-      {
-        $project: {
-          operationType: 1,
-          'fullDocument._id': 1,
-          'fullDocument.last_checkpoint': 1,
-          'fullDocument.last_checkpoint_lsn': 1
-        }
-      }
-    ];
-
-    // Use this form instead of (doc: SyncRuleDocument | null = null),
-    // otherwise we get weird "doc: never" issues.
-    let doc = null as SyncRuleDocument | null;
-    let clusterTime = null as mongo.Timestamp | null;
-
-    await this.client.withSession(async (session) => {
-      doc = await this.db.sync_rules.findOne(
-        {
-          state: storage.SyncRuleState.ACTIVE
-        },
-        {
-          session,
-          sort: { _id: -1 },
-          limit: 1,
-          projection: {
-            _id: 1,
-            last_checkpoint: 1,
-            last_checkpoint_lsn: 1
-          }
-        }
-      );
-      const time = session.clusterTime?.clusterTime ?? null;
-      clusterTime = time;
-    });
-    if (clusterTime == null) {
-      throw new Error('Could not get clusterTime');
-    }
-
-    if (signal.aborted) {
-      return;
-    }
-
-    if (doc) {
-      yield this.makeActiveCheckpoint(doc);
-    }
-
-    const stream = this.db.sync_rules.watch(pipeline, {
-      fullDocument: 'updateLookup',
-      // Start at the cluster time where we got the initial doc, to make sure
-      // we don't skip any updates.
-      // This may result in the first operation being a duplicate, but we filter
-      // it out anyway.
-      startAtOperationTime: clusterTime
-    });
-
-    signal.addEventListener(
-      'abort',
-      () => {
-        stream.close();
-      },
-      { once: true }
-    );
-
-    let lastOp: storage.ActiveCheckpoint | null = null;
-
-    for await (const update of stream.stream()) {
-      if (signal.aborted) {
-        break;
-      }
-      if (update.operationType != 'insert' && update.operationType != 'update') {
-        continue;
-      }
-      const doc = update.fullDocument!;
-      if (doc == null) {
-        continue;
-      }
-
-      const op = this.makeActiveCheckpoint(doc);
-      // Check for LSN / checkpoint changes - ignore other metadata changes
-      if (lastOp == null || op.lsn != lastOp.lsn || op.checkpoint != lastOp.checkpoint) {
-        lastOp = op;
-        yield op;
-      }
-    }
-  }
-
-  // Nothing is done here until a subscriber starts to iterate
-  private readonly sharedIter = new sync.BroadcastIterable((signal) => {
-    return this.watchActiveCheckpoint(signal);
-  });
-
-  /**
-   * User-specific watch on the latest checkpoint and/or write checkpoint.
-   */
-  async *watchWriteCheckpoint(user_id: string, signal: AbortSignal): AsyncIterable<storage.WriteCheckpoint> {
-    let lastCheckpoint: utils.OpId | null = null;
-    let lastWriteCheckpoint: bigint | null = null;
-
-    const iter = wrapWithAbort(this.sharedIter, signal);
-    for await (const cp of iter) {
-      const { checkpoint, lsn } = cp;
-
-      // lsn changes are not important by itself.
-      // What is important is:
-      // 1. checkpoint (op_id) changes.
-      // 2. write checkpoint changes for the specific user
-      const bucketStorage = await cp.getBucketStorage();
-      if (!bucketStorage) {
-        continue;
-      }
-
-      const lsnFilters: Record<string, string> = lsn ? { 1: lsn } : {};
-
-      const currentWriteCheckpoint = await bucketStorage.lastWriteCheckpoint({
-        user_id,
-        heads: {
-          ...lsnFilters
-        }
-      });
-
-      if (currentWriteCheckpoint == lastWriteCheckpoint && checkpoint == lastCheckpoint) {
-        // No change - wait for next one
-        // In some cases, many LSNs may be produced in a short time.
-        // Add a delay to throttle the write checkpoint lookup a bit.
-        await timers.setTimeout(20 + 10 * Math.random());
-        continue;
-      }
-
-      lastWriteCheckpoint = currentWriteCheckpoint;
-      lastCheckpoint = checkpoint;
-
-      yield { base: cp, writeCheckpoint: currentWriteCheckpoint };
-    }
-  }
-}