def handler(event=None, context=None):
url = "https://raw.githubusercontent.com/Niraj-Fonseka/invokust-tinker/master/invokust/locustfile.py"
r = requests.get(url)
with open("temp_locustfile.py", 'wb') as f:
f.write(r.content)
logging.info("wrote file")
try:
if event:
settings = create_settings(**event)
else:
settings = create_settings(from_environment=True)
loadtest = LocustLoadTest(settings)
loadtest.run()
except Exception as e:
logging.error("Locust exception {0}".format(repr(e)))
else:
locust_stats = loadtest.stats()
lambda_runtime_info = get_lambda_runtime_info(context)
loadtest_results = locust_stats.copy()
loadtest_results.update(lambda_runtime_info)
json_results = json.dumps(loadtest_results)
logging.info(json_results)
return json_results
def handler(event=None, context=None):
try:
settings = create_settings(**event)
settings.no_reset_stats = True
loadtest = LocustLoadTest(settings)
loadtest.run(timeout=event.get("timeout", 260))
except Exception as e:
logging.error("Locust exception {0}".format(repr(e)))
else:
locust_stats = loadtest.stats()
lambda_runtime_info = get_lambda_runtime_info(context)
loadtest_results = locust_stats.copy()
loadtest_results.update(lambda_runtime_info)
json_results = json.dumps(loadtest_results)
logging.info(json_results)
return json_results
def handler(event=None, context=None):
try:
if event:
settings = create_settings(**event)
else:
settings = create_settings(from_environment=True)
loadtest = LocustLoadTest(settings)
loadtest.run()
except Exception as e:
logging.error("Locust exception {0}".format(repr(e)))
else:
locust_stats = loadtest.stats()
lambda_runtime_info = get_lambda_runtime_info(context)
loadtest_results = locust_stats.copy()
loadtest_results.update(lambda_runtime_info)
json_results = json.dumps(loadtest_results)
logging.info(json_results)
return json_results