updating streaming support.

lib/index.js

@@ -225,8 +225,10 @@ function $extend(ctx, obj) {
         return obj.query(query, values, 'result');
     };
 
-    // TODO: Needs proper documentation;
-    // Has no expectation for the data;
+    // Has no expectation for the return data;
+    // Resolves when the streaming has finished;
+    // qs - stream-like object;
+    // init - stream initialization callback.
     obj.stream = function (qs, init) {
         return obj.query(qs, init, 'stream');
     };
@@ -303,9 +305,16 @@ function $stream(ctx, qs, init) {
         return $p.reject("Invalid stream initialization callback.");
     }
     // TODO: qs.text - the sql query to be reported;
-    var stream;
+    var stream, start = Date.now(), count = 0;
     try {
         stream = ctx.db.client.query(qs);
+        var oldCB = stream._fetch;
+        stream._fetch = function (size, func) {
+            oldCB.call(stream, size, function (err, rows) {
+                count += rows.length;
+                func(err, rows);
+            });
+        };
         init(stream); // initialize the stream;
     } catch (err) {
         return $p.reject(err);
@@ -318,7 +327,10 @@ function $stream(ctx, qs, init) {
         stream.once('end', function () {
             // TODO: Need to report the query?
             // TODO: Number of rows?
-            resolve();
+            resolve({
+                count: count,
+                duration: Date.now() - start
+            });
         });
         stream.once('error', function (error) {
             // TODO: Need to report the error?