y-mongodb-provider

import http from 'http'; import { WebSocketServer } from 'ws'; import * as Y from 'yjs'; import { MongodbPersistence } from 'y-mongodb-provider'; import yUtils from 'y-websocket/bin/utils'; // Note: y-websocket/bin/utils is an internal utility, adapt as needed. const port = process.env.PORT || 1234; const mongoConnectionString = process.env.MONGODB_URL || 'mongodb://localhost:27017/yjstest'; const server = http.createServer((request, response) => { response.writeHead(200, { 'Content-Type': 'text/plain' }); response.end('okay'); }); // Initialize Yjs WebSocket server const wss = new WebSocketServer({ server }); wss.on('connection', yUtils.setupWSConnection); // Initialize MongoDB persistence for Yjs const mdb = new MongodbPersistence(mongoConnectionString, { collectionName: 'transactions', flushSize: 100, multipleCollections: false, // Default; set to true if each document needs its own collection }); // Set up persistence with y-websocket yUtils.setPersistence({ bindState: async (docName, ydoc) => { // Retrieve the persisted document state const persistedYdoc = await mdb.getYDoc(docName); // Apply the persisted state to the current ydoc Y.applyUpdate(ydoc, Y.encodeStateAsUpdate(persistedYdoc)); // Store initial state and listen for future updates mdb.storeUpdate(docName, Y.encodeStateAsUpdate(ydoc)); ydoc.on('update', async (update) => { mdb.storeUpdate(docName, update); }); }, writeState: async (docName, ydoc) => { // This function is called when all connections to a document are closed. // Ensure all pending updates are written to the database. // For y-mongodb-provider, `storeUpdate` handles this incrementally. // The promise resolution signals that the document can be destroyed. return new Promise(resolve => resolve()); }, }); server.listen(port, () => { console.log(`Yjs WebSocket server with MongoDB persistence listening on port: ${port}`); });