print_stats.py

import datetime
import sys

import client
import stats
import utils

def print_samplers(samplers, current_epoch_sec):
  print "%64s  ||   %21s   ||   %21s   ||   %21s" % (
      "", "Last Minute", "Last Hour", "All Time"
  )
  print "%64s  ||   %10s %10s   ||   %10s %10s   ||   %10s %10s" % (
      "name", "median", "95th", "median", "95th", "median", "95th"
  )
  print "-" * 150
  for sampler_wrapper in samplers:
    sampler_stats = stats.SamplerStats(sampler_wrapper["value"], current_epoch_sec)
    minute_stats = sampler_stats.last_minute_stats()
    hour_stats = sampler_stats.last_hour_stats()
    all_time_stats = sampler_stats.all_time_stats()
    print "%64s  ||   %10d %10d   ||   %10d %10d   ||   %10d %10d" %  \
        (sampler_wrapper["name"],
         minute_stats["median"], minute_stats["percentile_95"],
         hour_stats["median"], hour_stats["percentile_95"],
         all_time_stats["median"], all_time_stats["percentile_95"])

def print_counters(counters, current_epoch_sec):
  print "%64s  ||  %10s  ||  %10s  ||  %10s" % (
    "name", "last min", "last hr", "all time")
  print "-" * 150
  for counter_wrapper in counters:
    counter_stats = stats.CounterStats(counter_wrapper["value"], current_epoch_sec)
    print "%64s  ||  %10d  ||  %10d  ||  %10d" % \
        (counter_wrapper["name"],
         counter_stats.last_minute_count(),
         counter_stats.last_hour_count(),
         counter_stats.all_time_count())

def sort_vars_by_name(vars_list):
  return sorted(vars_list, key=lambda x:x["name"])

def main(argv):
  current_epoch_sec = utils.datetime_to_sec_since_epoch(datetime.datetime.now())
  c = client.VARZClient()
  dump = c.all_dump()
  print "SAMPLERS"
  print_samplers(sort_vars_by_name(dump["mht_samplers"]), current_epoch_sec)
  print ""
  print "COUNTERS"
  print_counters(sort_vars_by_name(dump["mht_counters"]), current_epoch_sec)

if __name__ == "__main__":
  main(sys.argv)