Prepare streaming mapreduce.

roidrage · roidrage · commit 14d937f88516 · 2013-04-10T11:04:09.000+02:00
diff --git a/lib/protocol-buffers-client.js b/lib/protocol-buffers-client.js
@@ -11,7 +11,7 @@ var ProtocolBuffersClient = function ProtocolBuffersClient(options) {
   this._defaults = options || {};
   this._connection = riakpbc.createClient(options);
   var execute = function(operation, meta) {
-    this._execute(operation, meta);
+    return this._execute(operation, meta);
   }.bind(this);
   this.search = new ProtocolBuffersSearchClient(this._defaults, execute);
   this.mapreduce = new ProtocolBuffersMapReduceClient(this._defaults, execute);
@@ -88,9 +88,11 @@ ProtocolBuffersClient.prototype._execute = function(operation, meta, includePara
   }
 
   if (includeParameters === false) {
-    this._connection[operation](callback);
+    return this._connection[operation](callback);
+  } else if (meta.chunked) {
+    return this._connection[operation](meta.requestParameters(), true, callback);
   } else {
-    this._connection[operation](meta.requestParameters(), callback);
+    return this._connection[operation](meta.requestParameters(), callback);
   }
 }
 
diff --git a/lib/protocol-buffers-mapreduce-client.js b/lib/protocol-buffers-mapreduce-client.js
@@ -13,7 +13,16 @@ ProtocolBuffersMapReduceClient.prototype.add = function(inputs) {
 ProtocolBuffersMapReduceClient.prototype._run = function(job, options, callback) {
   var meta = new ProtocolBuffersMapReduceMeta(this._defaults, options, {callback: callback});
   meta.loadData(job.data);
-  this._execute('mapred', meta);
+  if (meta.chunked) {
+    var emitter = this._execute('mapred', meta);
+    emitter.on('data', function(data) {
+      emitter.emit('keys', data);
+    });
+    emitter.start = function() {};
+    return emitter;
+  } else {
+    this._execute('mapred', meta);
+  }
 }
 
 module.exports = ProtocolBuffersMapReduceClient;
