diff --git a/bin/collector_loadtest.py b/bin/collector_loadtest.py --- a/bin/collector_loadtest.py +++ b/bin/collector_loadtest.py @@ -7,6 +7,8 @@ from twisted.internet import reactor import time import logging log.setLevel(logging.DEBUG) + + def loadTest(): print "scheduling loadtest" n = 2500 @@ -14,23 +16,27 @@ def loadTest(): session = "loadtest%s" % time.time() offset = 0 for i in range(n): + def send(i): if i % 100 == 0: log.info('sendToCollector %s', i) d = sendToCollector("http://localhost:999999/", session, - [[DEV["backlight1"], L9["color"], "#ffffff"], - [DEV["backlight2"], L9["color"], "#ffffff"], - [DEV["backlight3"], L9["color"], "#ffffff"], - [DEV["backlight4"], L9["color"], "#ffffff"], - [DEV["backlight5"], L9["color"], "#ffffff"], - [DEV["down2"], L9["color"], "#ffffff"], - [DEV["down3"], L9["color"], "#ffffff"], - [DEV["down4"], L9["color"], "#ffffff"], - [DEV["houseSide"], L9["level"], .8], - [DEV["backlight5"], L9["uv"], 0.011]]) + [[DEV["backlight1"], L9["color"], "#ffffff"], + [DEV["backlight2"], L9["color"], "#ffffff"], + [DEV["backlight3"], L9["color"], "#ffffff"], + [DEV["backlight4"], L9["color"], "#ffffff"], + [DEV["backlight5"], L9["color"], "#ffffff"], + [DEV["down2"], L9["color"], "#ffffff"], + [DEV["down3"], L9["color"], "#ffffff"], + [DEV["down4"], L9["color"], "#ffffff"], + [DEV["houseSide"], L9["level"], .8], + [DEV["backlight5"], L9["uv"], 0.011]]) + def ontime(dt, i=i): times[i] = dt + d.addCallback(ontime) + reactor.callLater(offset, send, i) offset += .002 @@ -39,8 +45,10 @@ def loadTest(): with open('/tmp/times', 'w') as f: f.write(''.join('%s\n' % t for t in times)) reactor.stop() - reactor.callLater(offset+.5, done) + + reactor.callLater(offset + .5, done) reactor.run() + if __name__ == '__main__': loadTest()