cartodb4/node-pg-copy-streams
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Do not emit row event

Browse Source
This commit is contained in:
Brian M. Carlson 2013-10-28 21:50:45 -05:00
parent 1f23523e61
commit 2f5b37e0ad
4 changed files with 6 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
copy-to.js
View File

@ -10,6 +10,7 @@ var CopyStreamQuery = function(text) {
this.text = text
this._listeners = null
this._copyOutResponse = null
this.rowCount = 0
}
util.inherits(CopyStreamQuery, Transform)
@ -78,7 +79,7 @@ CopyStreamQuery.prototype._transform = function(chunk, enc, cb) {
var slice = chunk.slice(offset, offset + length)
offset += length
this.push(slice)
this.emit('row')
this.rowCount++
} else {
break;
}

3
index.js
View File

@ -17,6 +17,7 @@ var CopyStreamQuery = function(text) {
this.text = text
this._listeners = null
this._copyOutResponse = null
this.rowCount = 0
}
util.inherits(CopyStreamQuery, Transform)
@ -39,7 +40,7 @@ CopyStreamQuery.prototype._transform = function(chunk, enc, cb) {
lenBuffer.writeUInt32BE(chunk.length + 4, 0)
this.push(lenBuffer)
this.push(chunk)
this.emit('row')
this.rowCount++
cb()
}

2
test/copy-from.js
View File

@ -36,7 +36,7 @@ var testRange = function(top) {
console.log('found ', res.rows.length, 'rows')
countDone()
var firstRowDone = gonna('have correct result')
assert.equal(rowEmitCount, top, 'should have emitted "row" event ' + top + ' times')
assert.equal(stream.rowCount, top, 'should have rowCount ' + top + ' ')
fromClient.query('SELECT (max(num)) AS num FROM numbers', function(err, res) {
assert.ifError(err)
assert.equal(res.rows[0].num, top-1)

6
test/copy-to.js
View File

@ -18,10 +18,6 @@ var testRange = function(top) {
var txt = 'COPY (SELECT * from generate_series(0, ' + (top - 1) + ')) TO STDOUT'
var stream = fromClient.query(copy(txt))
var rowEmitCount = 0
stream.on('row', function() {
rowEmitCount++
})
var done = gonna('finish piping out', 1000, function() {
fromClient.end()
})
@ -30,7 +26,7 @@ var testRange = function(top) {
var res = buf.toString('utf8')
var expected = _.range(0, top).join('\n') + '\n'
assert.equal(res, expected)
assert.equal(rowEmitCount, top, 'should have emitted "row" ' + top + ' times but got ' + rowEmitCount)
assert.equal(stream.rowCount, top, 'should have rowCount ' + top + ' but got ' + stream.rowCount)
done()
}))
}