feat: support quit cluster gracefully. Close #315

redis · Jun 25, 2016 · f2ef3a8 · f2ef3a8
1 parent 0ac7eef
commit f2ef3a8
Showing 1 changed file with 21 additions and 0 deletions.
diff --git a/lib/cluster/index.js b/lib/cluster/index.js
@@ -208,6 +208,27 @@ Cluster.prototype.disconnect = function (reconnect) {
   this.connectionPool.reset([]);
 };
 
+/**
+ * Quit the cluster gracefully.
+ *
+ * @return {Promise}
+ * @public
+ */
+Cluster.prototype.quit = function (reconnect) {
+  this.setStatus('disconnecting');
+
+  if (!reconnect) {
+    this.manuallyClosing = true;
+  }
+  if (this.reconnectTimeout) {
+    clearTimeout(this.reconnectTimeout);
+    this.reconnectTimeout = null;
+  }
+  return this.nodes().map(function (node) {
+    return node.quit();
+  });
+};
+
 /**
  * Get nodes with the specified role
  *