locustEvents.py

from locust import events
import time
try:
   import cPickle as pickle
except:
   import pickle

metrics = {
			'req_success':[],
			'throughput':[],
			'downloads':[],
			'req_errors':[]
		  }
downloads=0

reqError = events.EventHook()
def req_error(**kw):
	metrics['req_errors'].append({'timestamp':time.time(), 'msg':kw.get('message','')})
	print "Req Error: %s"%kw.get("message","unknown")
reqError += req_error

def req_success(request_type, name, response_time, response_length, **kw):
	#print "hello"
	metrics['req_success'].append({'time':response_time, 'len':response_length, 'timestamp':time.time()})
events.request_success += req_success

downloadStart = events.EventHook()
def on_download_start(**kw):
	global downloads, metrics
	size = kw.get('size',-1)
	downloads += 1
	print "Downlod Started: %d"%downloads
	metrics['downloads'].append({ 'time':time.time(), 'downloads':downloads})
downloadStart += on_download_start

downloadEnd = events.EventHook()
def on_download_end(**kw):
	global downloads, metrics
	size = kw.get('size',-1)
	downloads -=1
	print "Download Ended: %d"%downloads
	metrics['downloads'].append({'time':time.time(), 'downloads':downloads})

downloadEnd += on_download_end

chunkHook = events.EventHook()
bytesRecvd = 0
timeSpent = 0.0
def on_chunk(**kw):
	global bytesRecvd, timeSpent, metrics, downloads
	bytesRecvd += kw.get('bytesRead',0)
	timeSpent += kw.get('seconds', 0.0)
	rate = 0
	if timeSpent > 0:
		rate= bytesRecvd/timeSpent
	metrics['throughput'].append({'time':time.time(), 'len':bytesRecvd, 'timeSpent':timeSpent, 'downloads':downloads, 'rate':rate})	
chunkHook += on_chunk

def exiting():
	# pass
	global metrics
	tfin = int(time.time())
	f = open('../results/%d_results.pkl'%tfin, 'wb')
	pickle.dump(metrics, f)
	f.flush()
	f.close()
	print "Req metrics captured: %d"%len(metrics)




events.quitting += exiting