zlib: allow custom flush type for flush()

1 files changed, 7 insertions, 2 deletions

diff --git a/lib/zlib.js b/lib/zlib.js
index bbe09ef2c2..ef4edd6d7e 100644
--- a/lib/zlib.js
+++ b/lib/zlib.js
@@ -357,9 +357,14 @@ Zlib.prototype._flush = function(callback) {
   this._transform(new Buffer(0), '', callback);
 };
 
-Zlib.prototype.flush = function(callback) {
+Zlib.prototype.flush = function(kind, callback) {
   var ws = this._writableState;
 
+  if (typeof kind === 'function' || (kind === undefined && !callback)) {
+    callback = kind;
+    kind = binding.Z_FULL_FLUSH;
+  }
+
   if (ws.ended) {
     if (callback)
       process.nextTick(callback);
@@ -372,7 +377,7 @@ Zlib.prototype.flush = function(callback) {
       self.flush(callback);
     });
   } else {
-    this._flushFlag = binding.Z_FULL_FLUSH;
+    this._flushFlag = kind;
     this.write(new Buffer(0), '', callback);
   }
 };