From 54a82b8c62934131b94b5e112a90e84706c0a9cc Mon Sep 17 00:00:00 2001
From: Jakob Ackermann <jakob.ackermann@overleaf.com>
Date: Mon, 7 Sep 2020 09:44:19 +0100
Subject: [PATCH] [misc] mongodb: export a helper for adding new collections to
 db

Co-Authored-By: Eric Mc Sween <eric.mcsween@overleaf.com>
Co-Authored-By: Simon Detheridge <s@sd.ai>
---
 services/docstore/app/js/mongodb.js | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/services/docstore/app/js/mongodb.js b/services/docstore/app/js/mongodb.js
index bbecd6501d..e9c425ad95 100644
--- a/services/docstore/app/js/mongodb.js
+++ b/services/docstore/app/js/mongodb.js
@@ -18,9 +18,16 @@ async function setupDb() {
   db.docs = internalDb.collection('docs')
   db.docOps = internalDb.collection('docOps')
 }
+async function addCollection(name) {
+  await waitForDb()
+  const internalDb = (await clientPromise).db()
+
+  db[name] = internalDb.collection(name)
+}
 
 module.exports = {
   db,
   ObjectId,
+  addCollection,
   waitForDb
 }