Update teardown method

balderdashy · Apr 29, 2014 · 545f61a · 545f61a
1 parent aa86451
commit 545f61a
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/lib/adapter.js b/lib/adapter.js
@@ -41,13 +41,21 @@ module.exports = (function () {
       connections[connection.identity].initialize(cb);
     },
 
-    // Teardown a Connection
     teardown: function (conn, cb) {
+
+      if (typeof conn == 'function') {
+        cb = conn;
+        conn = null;
+      }
+      if (conn == null) {
+        connections = {};
+        return cb();
+      }
       if(!connections[conn]) return cb();
       delete connections[conn];
       cb();
     },
-
+    
     // Return attributes
     describe: function (conn, coll, cb) {
       grabConnection(conn).describe(coll, cb);