Merge main into sweep/port-mongo-document-store

MongoDocumentStore.ts

@@ -0,0 +1,52 @@
+import { MongoClient } from 'mongodb';
+import { KVDocumentStore } from './KVDocumentStore';
+import { Document } from './types';
+import { handleMongoError } from './utils';
+
+export class MongoDocumentStore extends KVDocumentStore {
+  private client: MongoClient;
+  private dbName: string;
+
+  constructor(connectionString: string, dbName: string) {
+    super();
+    this.client = new MongoClient(connectionString);
+    this.dbName = dbName;
+  }
+
+  async get(key: string): Promise<Document | null> {
+    try {
+      const db = this.client.db(this.dbName);
+      return await db.collection('documents').findOne({ key });
+    } catch (error) {
+      handleMongoError(error);
+    }
+  }
+
+  async set(key: string, value: Document): Promise<void> {
+    try {
+      const db = this.client.db(this.dbName);
+      await db.collection('documents').updateOne({ key }, { $set: { value } }, { upsert: true });
+    } catch (error) {
+      handleMongoError(error);
+    }
+  }
+
+  async delete(key: string): Promise<void> {
+    try {
+      const db = this.client.db(this.dbName);
+      await db.collection('documents').deleteOne({ key });
+    } catch (error) {
+      handleMongoError(error);
+    }
+  }
+
+  async keys(): Promise<string[]> {
+    try {
+      const db = this.client.db(this.dbName);
+      const documents = await db.collection('documents').find({}, { projection: { key: 1 } }).toArray();
+      return documents.map(document => document.key);
+    } catch (error) {
+      handleMongoError(error);
+    }
+  }
+}

types.ts

@@ -0,0 +1,9 @@
+import { Document } from './types';
+
+export interface MongoDocumentStore {
+  constructor(connectionString: string, dbName: string): void;
+  get(key: string): Promise<Document | null>;
+  set(key: string, value: Document): Promise<void>;
+  delete(key: string): Promise<void>;
+  keys(): Promise<string[]>;
+}

utils.ts

@@ -0,0 +1,15 @@
+import { MongoClient } from 'mongodb';
+
+export function handleMongoError(error: Error): void {
+  console.error('An error occurred while interacting with MongoDB:', error);
+}
+
+export async function connectToMongoDB(connectionString: string): Promise<MongoClient> {
+  try {
+    const client = new MongoClient(connectionString);
+    await client.connect();
+    return client;
+  } catch (error) {
+    handleMongoError(error);
+  }
+}