From 8eebc915a20c66b37c621c6442f95e1b1048979f Mon Sep 17 00:00:00 2001
From: AJ ONeal <coolaj86@gmail.com>
Date: Mon, 29 Nov 2010 00:44:27 -0700
Subject: [PATCH] using streaming ahr

---
 mock-census/mock-census.js | 33 ++++++++++++++++-----------------
 1 file changed, 16 insertions(+), 17 deletions(-)

diff --git a/mock-census/mock-census.js b/mock-census/mock-census.js
index 85a9ce3..8fa75d0 100644
--- a/mock-census/mock-census.js
+++ b/mock-census/mock-census.js
@@ -2,13 +2,11 @@
 "use strict";
 (function (undefined) {
   var fs = require('fs'),
-    url = require('url'),
-    ahr = require('ahr'),
+    request = require('ahr'),
     stream,
     node = process.argv[0],
     self = process.argv[1],
-    fulluri = process.argv[2],
-    uri;
+    fulluri = process.argv[2];
 
   self = self.substr(self.lastIndexOf('/') + 1);
 
@@ -167,7 +165,12 @@
     */
   }
 
-  function parseData(err, x, data) {
+  var data = '';
+  function parseData(err, x, chunk, end) {
+    if (!end) {
+      data += chunk.toString();
+      return;
+    }
     var lines = data.split('\n'),
       heads = [],
       curHead;
@@ -189,18 +192,14 @@
       return;
     }
       
-    uri = url.parse(fulluri, true);
-    if (!uri.protocol) {
-      uri.protocol = '';
-    }
-    if ('file:' === uri.protocol || '' === uri.protocol) {
-      readFile(uri.pathname).when(parseData);
-    } else if (uri.protocol.match(/^http[s]?:$/)) {
-      ahr.get(fulluri).when(parseData);
-    } else {
-      usage();
-      return;
-    }
+    request.get(fulluri, undefined, { stream: true}).when(parseData);
+
+    //if ('file' !== uri.protocol && '' === uri.protocol || uri.protocol.match(/^http[s]?:$/)) {
+    //  request.get(fulluri, undefined, { stream: true}).when(parseData);
+    //} else {
+    //  usage();
+    //  return;
+    //}
   }
 
   main();