custom data collection stream

Author: polotek

collector.js

@@ -0,0 +1,35 @@
+var Stream = require('stream').Stream
+  , util = require('util');
+
+var Collector = function() {
+  this._bytesWritten = 0;
+  this._data = '';
+  this.readable = true;
+  this.writable = true;
+};
+util.inherits(Collector, Stream);
+Collector.prototype.write = function(data) {
+  if(typeof data == 'string') {
+    this._bytesWritten += Buffer.byteLength(data);
+  } else {
+    this._bytesWritten += data.length;
+  }
+
+  this._data += data.toString();
+
+  this.emit('data', data);
+};
+Collector.prototype.end = function(data) {
+  if(data) { this.write(data); }
+
+  this.readable = false;
+  this.writable = false;
+  this.emit('end');
+};
+Collector.prototype.getData = function() {
+  return this._data;
+};
+Collector.prototype.getBytesWritten = function() {
+  return this._bytesWritten;
+};
+module.exports = Collector;

logstream.js

@@ -11,7 +11,7 @@ LogStream.prototype.write = function(data) {
   this.emit('data', data);
 };
 LogStream.prototype.end = function(data) {
-  if(data) { this.emit('data', data); }
+  if(data) { this.write(data); }
 
   this.emit('end');
 };

streams.js

@@ -1,11 +1,18 @@
 var Stream = require('stream'),
   fs = require('fs'),
   http = require('http'),
-  LogStream = require('./logstream');
+  Collector = require('./collector');
 
 http.createServer(function(request, response) {
-  var logstream = new LogStream();
-  fs.createReadStream('./large-file.txt').pipe(logstream).pipe(response);
+  var collector = new Collector();
+  fs.createReadStream('./large-file.txt').pipe(collector);
+  collector.on('end', function() {
+    response.writeHead(200, {
+      'Content-Type': 'text/plain',
+      'Content-Length': collector.getBytesWritten()
+    });
+    response.end(collector.getData());
+  });
 }).listen(8080);
 
 console.log('Server listening on port 8080\n');