Skip to content

Instantly share code, notes, and snippets.

@lfittl lfittl/monitor.rb
Last active Mar 4, 2019

Embed
What would you like to do?
# Copyright (c) 2014 Lukas Fittl <lukas@pganalyze.com>
#
# Released in the public domain - fork as you wish.
require 'rubygems'
require 'mixlib/cli'
require 'pg'
require 'pg_query'
require 'curses'
class CLIHelper
include Mixlib::CLI
option :database,
short: "-d DATABASE",
long: "--database DATABASE",
description: "The database to be tracked",
required: true
option :table,
short: "-t TABLE",
long: "--table TABLE",
description: "Only show queries that use the specified table"
end
Curses.noecho
Curses.init_screen
cli = CLIHelper.new
cli.parse_options
conn = PG::Connection.open(dbname: cli.config[:database])
conn.exec('SELECT pg_stat_statements_reset()')
while true do
Curses.setpos(0, 0)
queries = conn.exec('SELECT query, calls, total_time FROM pg_stat_statements').to_a
queries = queries.sort_by {|q| (q["total_time"].to_f / q["calls"].to_f) }.reverse
Curses.addstr("AVG\t| QUERY\n")
Curses.addstr("-" * 80 + "\n")
queries.each do |query|
if cli.config[:table]
next unless PgQuery.parse(query["query"]).tables.include?(cli.config[:table])
end
Curses.addstr("%0.1fms\t" % (query["total_time"].to_f / query["calls"].to_f))
Curses.addstr("| " + query["query"].gsub(/\s+/, " ").strip + "\n")
end
Curses.refresh
sleep 1
end
Curses.close_screen
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
You can’t perform that action at this time.