def unwrap_response(data):
resp = trp_pb2.Message()
resp.ParseFromString(data)
for x in resp.DESCRIPTOR.enum_types:
name = x.values_by_number.get(int(resp.trp_command)).name
return {
'COUNTER_GROUP_INFO_RESPONSE': resp.counter_group_info_response
}.get(name, resp)
def unwrap_response(data):
resp = trp_pb2.Message()
resp.ParseFromString(data)
for x in resp.DESCRIPTOR.enum_types:
name = x.values_by_number.get(int(resp.trp_command)).name
return {
'TIMESLICES_RESPONSE': resp.time_slices_response,
'COUNTER_GROUP_TOPPER_RESPONSE': resp.counter_group_topper_response
}.get(name, resp)
def get_timewindow():
req = trp_pb2.Message()
req.trp_command = req.TIMESLICES_REQUEST
req.time_slices_request.get_total_window = True
resp = get_response(sys.argv[1], req)
#construct time interval for last 1 hour
tm = trp_pb2.TimeInterval()
tm.to.tv_sec = resp.total_window.to.tv_sec
object = getattr(tm, 'from')
object.tv_sec = tm.to.tv_sec - 86400
return tm
def get_toppers(cgguid, name):
#construct counter group topper request for internal host
req = trp_pb2.Message()
req.trp_command = req.COUNTER_GROUP_TOPPER_REQUEST
req.counter_group_topper_request.counter_group = cgguid
req.counter_group_topper_request.maxitems = 100
req.counter_group_topper_request.meter = 0
tm = get_timewindow()
#assign time interval to counter group topper request
req.counter_group_topper_request.time_interval.MergeFrom(tm)
resp = get_response(sys.argv[1], req)
print "Top " + name + " hosts"
#parse and display
for idx, key in enumerate(resp.keys):
if key.key == "SYS:GROUP_TOTALS":
continue
if idx == 101:
break
print '{:<5}{:<30}{:<}'.format(idx, key.readable, key.label)
#zmq receive
data = socket.recv()
resp = unwrap_response(data)
return resp
socket.close
# helper
def unwrap_response(data):
resp = trp_pb2.Message()
resp.ParseFromString(data)
for x in resp.DESCRIPTOR.enum_types:
name = x.values_by_number.get(int(resp.trp_command)).name
return {
'COUNTER_GROUP_INFO_RESPONSE': resp.counter_group_info_response
}.get(name, resp)
#################
# actual code starts here
# to retrieve all counter groups send an empty COUNTER_GROUP_INFO_REQUEST
req = trp_pb2.Message()
req.trp_command = req.COUNTER_GROUP_INFO_REQUEST
resp = get_response(sys.argv[1], req)
# display the information
for cg in resp.group_details:
print cg.guid + "," + cg.name