2 Commits

Author SHA1 Message Date
Brian M. Carlson
824936176c Bump version 2013-10-28 21:50:54 -05:00
Brian M. Carlson
2f5b37e0ad Do not emit row event 2013-10-28 21:50:45 -05:00
5 changed files with 7 additions and 9 deletions

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;
}

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()
}

View File

@@ -1,6 +1,6 @@
{
"name": "pg-copy-streams",
"version": "0.1.0",
"version": "0.2.0",
"description": "Low-Level COPY TO and COPY FROM streams for PostgreSQL in JavaScript using",
"main": "index.js",
"scripts": {

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)

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()
}))
}