mirror of
https://github.com/brianc/node-postgres.git
synced 2026-01-18 15:55:05 +00:00
db1b95e5f3
git-subtree-dir: packages/pg-query-stream git-subtree-mainline: cccf84e14b3281b753e1baab7bc194aaac5024a8 git-subtree-split: 9ced05e8aab65f3fdf1a67add87bfc9035e487e8
27 lines
685 B
JavaScript
27 lines
685 B
JavaScript
var helper = require('./helper')
|
|
var QueryStream = require('../')
|
|
var concat = require('concat-stream')
|
|
|
|
var Transform = require('stream').Transform
|
|
|
|
var mapper = new Transform({objectMode: true})
|
|
|
|
mapper._transform = function (obj, enc, cb) {
|
|
this.push(obj)
|
|
setTimeout(cb, 5)
|
|
}
|
|
|
|
helper('slow reader', function (client) {
|
|
it('works', function (done) {
|
|
this.timeout(50000)
|
|
var stream = new QueryStream('SELECT * FROM generate_series(0, 201) num', [], {highWaterMark: 100, batchSize: 50})
|
|
stream.on('end', function () {
|
|
// console.log('stream end')
|
|
})
|
|
client.query(stream)
|
|
stream.pipe(mapper).pipe(concat(function (res) {
|
|
done()
|
|
}))
|
|
})
|
|
})
|