force draining pools

CartoDB · Aug 20, 2020 · e74f3d6 · e74f3d6
1 parent 8476670
commit e74f3d6
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/index.js b/index.js
@@ -97,6 +97,15 @@ module.exports = class RedisPool extends EventEmitter {
         }
     }
 
+    /**
+     * Closing all connection pools
+     * 
+     * https://github.com/coopernurse/node-pool/blob/v3.7.1/README.md#draining
+     */
+    async drain () {
+        await Promise.all(Object.values(this.pools).map(p => p.drain().then(() => p.clear())) )
+    }
+
     _addCommands () {
         if (this.options.commands.length) {
             this.options.commands.forEach(newCommand => redis.add_command(newCommand));