2014-10-23 10:52:21 -04:00
|
|
|
var Readable = require('stream').Readable;
|
|
|
|
|
|
|
|
// copy-paste from https://github.com/felixge/node-mysql/blob/master/lib/protocol/sequences/Query.js
|
|
|
|
module.exports = function (command, connectionStream) {
|
|
|
|
command.stream = function(options) {
|
2015-01-28 15:33:44 -05:00
|
|
|
var stream;
|
2014-10-23 10:52:21 -04:00
|
|
|
|
|
|
|
options = options || {};
|
|
|
|
options.objectMode = true;
|
|
|
|
stream = new Readable(options),
|
|
|
|
|
|
|
|
stream._read = function() {
|
|
|
|
connectionStream.resume();
|
|
|
|
};
|
|
|
|
|
|
|
|
this.on("result",function(row,i) {
|
|
|
|
if (!stream.push(row)) connectionStream.pause();
|
|
|
|
stream.emit("result",row,i); // replicate old emitter
|
|
|
|
});
|
|
|
|
|
|
|
|
this.on("error",function(err) {
|
2015-01-28 15:33:44 -05:00
|
|
|
stream.emit("error",err); // Pass on any errors
|
2014-10-23 10:52:21 -04:00
|
|
|
});
|
|
|
|
|
|
|
|
this.on("end", function() {
|
|
|
|
stream.push(null); // pushing null, indicating EOF
|
|
|
|
});
|
|
|
|
|
|
|
|
this.on("fields",function(fields,i) {
|
2015-01-28 15:33:44 -05:00
|
|
|
stream.emit("fields",fields,i); // replicate old emitter
|
2014-10-23 10:52:21 -04:00
|
|
|
});
|
|
|
|
|
|
|
|
return stream;
|
|
|
|
};
|
|
|
|
};
|