Wrap zlib transforms

Author: creationix

bootstrap.js

@@ -121,6 +121,11 @@ if (fs.lchmod) wrap(fs, 'lchmod', activator);
 // only wrap ftruncate in versions of node that have it
 if (fs.ftruncate) wrap(fs, 'ftruncate', activator);
 
+// Wrap zlib streams
+var zProto = Object.getPrototypeOf(require('zlib').Deflate.prototype);
+wrap(zProto, "_transform", activator);
+
+
 ////////////////////////////////////////////////////////////////////////////////
 
 // Polyfilled version of process.addAsyncListener

test/zlib.tap.js

@@ -0,0 +1,27 @@
+'use strict';
+if (!process.addAsyncListener) require('../bootstrap.js');
+
+var tap             = require('tap')
+  , test            = tap.test
+  , createNamespace = require('continuation-local-storage').createNamespace
+  ;
+
+var zlib = require('zlib');
+
+test("continuation-local state with zlib", function (t) {
+  t.plan(1);
+
+  var namespace = createNamespace('namespace');
+  namespace.set('test', 0xabad1dea);
+
+  t.test("deflate", function (t) {
+    namespace.run(function () {
+      namespace.set('test', 42);
+      zlib.deflate(new Buffer("Goodbye World"), function (err, deflated) {
+        if (err) throw err;
+        t.equal(namespace.get('test'), 42, "mutated state was preserved");
+        t.end();
+      });
+    });
+  });
+});