feat(topology-base): support passing callbacks to close method

mongodb · Mar 11, 2019 · 7c111e0 · 7c111e0
1 parent 08f15ab
commit 7c111e0
Showing 1 changed file with 7 additions and 4 deletions.
diff --git a/lib/topologies/topology_base.js b/lib/topologies/topology_base.js
@@ -379,7 +379,7 @@ class TopologyBase extends EventEmitter {
     return this.s.coreTopology.connections();
   }
 
-  close(forceClosed) {
+  close(forceClosed, callback) {
     // If we have sessions, we want to individually move them to the session pool,
     // and then send a single endSessions call.
     if (this.s.sessions.length) {
@@ -390,9 +390,12 @@ class TopologyBase extends EventEmitter {
       this.s.sessionPool.endAllPooledSessions();
     }
 
-    this.s.coreTopology.destroy({
-      force: typeof forceClosed === 'boolean' ? forceClosed : false
-    });
+    this.s.coreTopology.destroy(
+      {
+        force: typeof forceClosed === 'boolean' ? forceClosed : false
+      },
+      callback
+    );
 
     // We need to wash out all stored processes
     if (forceClosed === true) {