Fix getting server

huggingface · Apr 3, 2019 · 737fbf3 · 737fbf3
1 parent b6b0a07
commit 737fbf3
Showing 1 changed file with 1 addition and 7 deletions.
diff --git a/lib/MongoManager.ts b/lib/MongoManager.ts
@@ -22,9 +22,6 @@ export class MongoManager {
       : `mongodb://${host.path}`;
     const url = URL.parse(urlStr);
     let hostname = url.host || host.path;
-    if (hostname.indexOf(':') === -1) {
-      hostname = `${hostname}:27017`;
-    }
 
     if (this._servers[hostname] instanceof Server) {
       // Already connected
@@ -46,10 +43,7 @@ export class MongoManager {
   }
 
   private getServer(name: string) {
-    if (name.indexOf(':') === -1) {
-      name = `${name}:27017`;
-    }
-    const server = this._servers[name];
+    const server = this._servers[name] || this._servers[`${name}:27017`];
     if (!server) {
       throw new Error('Server does not exist');
     }