Skip to content

Commit 1b249e9

Browse files
committed
Add in proper error handling
1 parent aec85ce commit 1b249e9

File tree

2 files changed

+29
-0
lines changed

2 files changed

+29
-0
lines changed

index.js

+5
Original file line numberDiff line numberDiff line change
@@ -103,3 +103,8 @@ QueryStream.prototype.handleCommandComplete = function(msg) {
103103
QueryStream.prototype.handleReadyForQuery = function() {
104104
this.push(null)
105105
}
106+
107+
QueryStream.prototype.handleError = function(err) {
108+
this.connection.sync()
109+
this.emit('error', err)
110+
}

test/error.js

+24
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,24 @@
1+
var pg = require('pg')
2+
var assert = require('assert')
3+
var gonna = require('gonna')
4+
var _ = require('lodash')
5+
var concat = require('concat-stream')
6+
var through = require('through')
7+
8+
var QueryStream = require('../')
9+
10+
var client = new pg.Client()
11+
12+
var connected = gonna('connect', 100, function() {
13+
var stream = new QueryStream('SELECT * FROM asdf num', [])
14+
var query = client.query(stream)
15+
query.on('error', gonna('emit error', 100, function(err) {
16+
assert(err)
17+
assert.equal(err.code, '42P01')
18+
}))
19+
var done = gonna('keep connetion alive', 100)
20+
client.query('SELECT NOW()', done)
21+
})
22+
23+
client.connect(connected)
24+
client.on('drain', client.end.bind(client))

0 commit comments

Comments
 (0)