| 1 | 'use strict'
|
| 2 |
|
| 3 | const cli = require('heroku-cli-util')
|
| 4 |
|
| 5 | async function run(context, heroku) {
|
| 6 | const fetcher = require('../lib/fetcher')
|
| 7 | const psql = require('../lib/psql')
|
| 8 |
|
| 9 | const { app, args, flags } = context
|
| 10 | const { pid, database } = args
|
| 11 | const { force } = flags
|
| 12 |
|
| 13 | let db = await fetcher(heroku).database(app, database)
|
| 14 |
|
| 15 | let query = `
|
| 16 | SELECT ${force ? 'pg_terminate_backend' : 'pg_cancel_backend'}(${parseInt(pid)});`
|
| 17 |
|
| 18 | let output = await psql.exec(db, query)
|
| 19 | process.stdout.write(output)
|
| 20 | }
|
| 21 |
|
| 22 | module.exports = {
|
| 23 | topic: 'pg',
|
| 24 | command: 'kill',
|
| 25 | description: 'kill a query',
|
| 26 | needsApp: true,
|
| 27 | needsAuth: true,
|
| 28 | flags: [{ name: 'force', char: 'f' }],
|
| 29 | args: [
|
| 30 | { name: 'pid' },
|
| 31 | { name: 'database', optional: true }
|
| 32 | ],
|
| 33 | run: cli.command({ preauth: true }, run)
|
| 34 | }
|