json parser middleware, use freelist

tj · Sep 10, 2011 · e4fe18d · e4fe18d
1 parent 574497b
commit e4fe18d
Showing 1 changed file with 30 additions and 1 deletion.
diff --git a/lib/json.js b/lib/json.js
@@ -4,7 +4,8 @@
  */
 
 var EventEmitter = require('events').EventEmitter
-  , StringDecoder = require('string_decoder').StringDecoder;
+  , StringDecoder = require('string_decoder').StringDecoder
+  , FreeList = require('freelist').FreeList;
 
 /**
  * Parser
@@ -419,6 +420,34 @@ Parser.create = function(parser) {
 module.exports = exports = Parser.create;
 exports.Parser = Parser;
 
+exports.middleware = function() {
+  var parsers = new FreeList('parsers', 20, function() {
+    return Parser.create();
+  });
+  return function(req, res, next) {
+    if (req._json) return next();
+    req._json = true;
+
+    var type = req.headers['content-type'];
+    if (type && type.split(';')[0].trim() === 'application/json') {
+      var parser = parsers.alloc();
+
+      parser.on('error', function(err) {
+        req.socket.destroy();
+        next(err);
+      });
+
+      parser.on('end', function(data) {
+        parsers.free(parser);
+        req.body = data;
+        next();
+      });
+    } else {
+      next();
+    }
+  };
+};
+
 /**
  * Crude Test
  */