Skip to content

Commit 08072a9

Browse files
authored
Pass options to cursor (#65)
* Bump version of pg-cursor This includes fixes in [email protected]. I've relaxed semver a touch so I don't have to release a new version here just for patch changes to pg-cursor. * Pass options to pg-cursor fixes #55
1 parent 05b4c57 commit 08072a9

File tree

2 files changed

+39
-1
lines changed

2 files changed

+39
-1
lines changed

Diff for: index.js

+1-1
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@ var Readable = require('stream').Readable
55
class PgQueryStream extends Readable {
66
constructor (text, values, options) {
77
super(Object.assign({ objectMode: true }, options))
8-
this.cursor = new Cursor(text, values)
8+
this.cursor = new Cursor(text, values, options)
99
this._reading = false
1010
this._closed = false
1111
this.batchSize = (options || {}).batchSize || 100

Diff for: test/passing-options.js

+38
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,38 @@
1+
var assert = require('assert')
2+
var helper = require('./helper')
3+
var QueryStream = require('../')
4+
5+
helper('passing options', function(client) {
6+
it('passes row mode array', function(done) {
7+
var stream = new QueryStream('SELECT * FROM generate_series(0, 10) num', [], { rowMode: 'array' })
8+
var query = client.query(stream)
9+
var result = []
10+
query.on('data', datum => {
11+
result.push(datum)
12+
})
13+
query.on('end', () => {
14+
const expected = new Array(11).fill(0).map((_, i) => [i])
15+
assert.deepEqual(result, expected)
16+
done()
17+
})
18+
})
19+
20+
it('passes custom types', function(done) {
21+
const types = {
22+
getTypeParser: () => string => string,
23+
}
24+
var stream = new QueryStream('SELECT * FROM generate_series(0, 10) num', [], { types })
25+
var query = client.query(stream)
26+
var result = []
27+
query.on('data', datum => {
28+
result.push(datum)
29+
})
30+
query.on('end', () => {
31+
const expected = new Array(11).fill(0).map((_, i) => ({
32+
num: i.toString(),
33+
}))
34+
assert.deepEqual(result, expected)
35+
done()
36+
})
37+
})
38+
})

0 commit comments

Comments
 (0)