mirror of
https://github.com/brianc/node-postgres.git
synced 2026-01-18 15:55:05 +00:00
9870c86fde
PostgreSQL 10 reports its version as only `major.minor`, so it can’t be parsed with semver. The `server_version_num` setting is a major version followed by a four-digit minor version since version 10, and was a three-digit major version followed by a two-digit minor version before that.
99 lines
2.8 KiB
JavaScript
99 lines
2.8 KiB
JavaScript
'use strict'
|
|
var helper = require('./test-helper')
|
|
var Query = helper.pg.Query
|
|
var util = require('util')
|
|
|
|
var suite = new helper.Suite()
|
|
|
|
suite.test('client end during query execution of prepared statement', function (done) {
|
|
var client = new Client()
|
|
client.connect(assert.success(function () {
|
|
var sleepQuery = 'select pg_sleep($1)'
|
|
|
|
var queryConfig = {
|
|
name: 'sleep query',
|
|
text: sleepQuery,
|
|
values: [5]
|
|
}
|
|
|
|
var queryInstance = new Query(queryConfig, assert.calls(function (err, result) {
|
|
assert.equal(err.message, 'Connection terminated')
|
|
done()
|
|
}))
|
|
|
|
var query1 = client.query(queryInstance)
|
|
|
|
query1.on('error', function (err) {
|
|
assert.fail('Prepared statement should not emit error')
|
|
})
|
|
|
|
query1.on('row', function (row) {
|
|
assert.fail('Prepared statement should not emit row')
|
|
})
|
|
|
|
query1.on('end', function (err) {
|
|
assert.fail('Prepared statement when executed should not return before being killed')
|
|
})
|
|
|
|
client.end()
|
|
}))
|
|
})
|
|
|
|
function killIdleQuery (targetQuery, cb) {
|
|
var client2 = new Client(helper.args)
|
|
var pidColName = 'procpid'
|
|
var queryColName = 'current_query'
|
|
client2.connect(assert.success(function () {
|
|
helper.versionGTE(client2, 90200, assert.success(function (isGreater) {
|
|
if (isGreater) {
|
|
pidColName = 'pid'
|
|
queryColName = 'query'
|
|
}
|
|
var killIdleQuery = 'SELECT ' + pidColName + ', (SELECT pg_terminate_backend(' + pidColName + ')) AS killed FROM pg_stat_activity WHERE ' + queryColName + ' = $1'
|
|
client2.query(killIdleQuery, [targetQuery], assert.calls(function (err, res) {
|
|
assert.ifError(err)
|
|
assert.equal(res.rows.length, 1)
|
|
client2.end(cb)
|
|
assert.emits(client2, 'end')
|
|
}))
|
|
}))
|
|
}))
|
|
}
|
|
|
|
suite.test('query killed during query execution of prepared statement', function (done) {
|
|
if (helper.args.native) {
|
|
return done()
|
|
}
|
|
var client = new Client(helper.args)
|
|
client.connect(assert.success(function () {
|
|
var sleepQuery = 'select pg_sleep($1)'
|
|
|
|
const queryConfig = {
|
|
name: 'sleep query',
|
|
text: sleepQuery,
|
|
values: [5]
|
|
}
|
|
|
|
// client should emit an error because it is unexpectedly disconnected
|
|
assert.emits(client, 'error')
|
|
|
|
var query1 = client.query(new Query(queryConfig), assert.calls(function (err, result) {
|
|
assert.equal(err.message, 'terminating connection due to administrator command')
|
|
}))
|
|
|
|
query1.on('error', function (err) {
|
|
assert.fail('Prepared statement should not emit error')
|
|
})
|
|
|
|
query1.on('row', function (row) {
|
|
assert.fail('Prepared statement should not emit row')
|
|
})
|
|
|
|
query1.on('end', function (err) {
|
|
assert.fail('Prepared statement when executed should not return before being killed')
|
|
})
|
|
|
|
killIdleQuery(sleepQuery, done)
|
|
}))
|
|
})
|