monkeytype/backend/init/db.js

const { MongoClient } = require("mongodb");

class DatabaseClient {
  static mongoClient = null;
  static db = null;
  static collections = {};
  static connected = false;

  static async connect() {
    const {
      DB_USERNAME,
      DB_PASSWORD,
      DB_AUTH_MECHANISM,
      DB_AUTH_SOURCE,
      DB_URI,
      DB_NAME,
    } = process.env;

    const connectionOptions = {
      useNewUrlParser: true,
      useUnifiedTopology: true,
      connectTimeoutMS: 2000,
      serverSelectionTimeoutMS: 2000,
    };

    if (DB_USERNAME && DB_PASSWORD) {
      connectionOptions.auth = {
        username: DB_USERNAME,
        password: DB_PASSWORD,
      };
    }

    if (DB_AUTH_MECHANISM) {
      connectionOptions.authMechanism = DB_AUTH_MECHANISM;
    }

    if (DB_AUTH_SOURCE) {
      connectionOptions.authSource = DB_AUTH_SOURCE;
    }

    this.mongoClient = new MongoClient(DB_URI, connectionOptions);

    try {
      await this.mongoClient.connect();
      this.db = this.mongoClient.db(DB_NAME);
      this.connected = true;
    } catch (error) {
      console.error(error.message);
      console.error(
        "Failed to connect to database. Exiting with exit status code 1."
      );
      process.exit(1);
    }
  }

  static async close() {
    if (this.connected) {
      await this.mongoClient.close();
    }
  }

  static collection(collectionName) {
    if (!this.connected) {
      return null;
    }

    if (!(collectionName in this.collections)) {
      this.collections[collectionName] = this.db.collection(collectionName);
    }

    return this.collections[collectionName];
  }
}

module.exports = DatabaseClient;