Added basic implementation of MemoryDatabase

1 files changed, 4 insertions, 3 deletions

diff --git a/src/server/database.ts b/src/server/database.ts
index 6e0771c11..3039baedc 100644
--- a/src/server/database.ts
+++ b/src/server/database.ts
@@ -5,6 +5,7 @@ import { Utils, emptyFunction } from '../Utils';
 import { DashUploadUtils } from './DashUploadUtils';
 import { Credentials } from 'google-auth-library';
 import { GoogleApiServerUtils } from './apis/google/GoogleApiServerUtils';
+import { IDatabase } from './IDatabase';
 import * as mongoose from 'mongoose';
 
 export namespace Database {
@@ -44,7 +45,7 @@ export namespace Database {
         }
     }
 
-    class Database {
+    class Database implements IDatabase {
         public static DocumentsCollection = 'documents';
         private MongoClient = mongodb.MongoClient;
         private currentWrites: { [id: string]: Promise<void> } = {};
@@ -215,7 +216,7 @@ export namespace Database {
                     if (!fetchIds.length) {
                         continue;
                     }
-                    const docs = await new Promise<{ [key: string]: any }[]>(res => Instance.getDocuments(fetchIds, res, "newDocuments"));
+                    const docs = await new Promise<{ [key: string]: any }[]>(res => this.getDocuments(fetchIds, res, collectionName));
                     for (const doc of docs) {
                         const id = doc.id;
                         visited.add(id);
@@ -331,4 +332,4 @@ export namespace Database {
 
     }
 
-}
\ No newline at end of file
+}