@furystack/mongodb-store/src/mongo-client-factory.ts

MongoDB Store implementation for FuryStack

import { Injectable } from '@furystack/inject'
import type { MongoClientOptions } from 'mongodb'
import { MongoClient } from 'mongodb'

/**
 * Factory for instantiating MongoDb clients
 */
@Injectable({ lifetime: 'singleton' })
export class MongoClientFactory implements AsyncDisposable {
  private connections: Map<string, MongoClient> = new Map()

  public async [Symbol.asyncDispose]() {
    await Promise.all([...this.connections.values()].map((c) => c.close(true)))
    this.connections.clear()
  }

  public getClientFor(url: string, options?: MongoClientOptions) {
    const existing = this.connections.get(url)
    if (existing) {
      return existing
    }

    const client = new MongoClient(url, options)
    this.connections.set(url, client)
    return client
  }
}