universal-ai-brain/packages/core/src/real-time/ChangeStreamManager.ts

🧠 UNIVERSAL AI BRAIN 3.3 - The world's most advanced cognitive architecture with 24 specialized systems, MongoDB 8.1 $rankFusion hybrid search, latest Voyage 3.5 embeddings, and framework-agnostic design. Works with Mastra, Vercel AI, LangChain, OpenAI A

import { Collection, Db, ChangeStream, Document } from 'mongodb';

export type ChangeHandler = (change: Document) => Promise<void>;

export class ChangeStreamManager {
  private changeStream?: ChangeStream;
  private db: Db;
  private collectionName: string;
  private pipeline: Document[];
  private handler: ChangeHandler;

  constructor(db: Db, collectionName: string, pipeline: Document[], handler: ChangeHandler) {
    this.db = db;
    this.collectionName = collectionName;
    this.pipeline = pipeline;
    this.handler = handler;
  }

  public async start(): Promise<void> {
    const collection = this.db.collection(this.collectionName);
    this.changeStream = collection.watch(this.pipeline);

    this.changeStream.on('change', this.handler);

    this.changeStream.on('error', (error) => {
      console.error(`Change stream error on collection ${this.collectionName}:`, error);
      // In a production system, you would add reconnection logic here.
    });

    console.log(`Watching for changes on ${this.collectionName}...`);
  }

  public async stop(): Promise<void> {
    if (this.changeStream) {
      await this.changeStream.close();
      console.log(`Stopped watching for changes on ${this.collectionName}.`);
    }
  }
}