summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorKali Kaneko <kali@leap.se>2015-08-09 15:22:29 -0400
committerKali Kaneko <kali@leap.se>2015-08-11 15:22:23 -0400
commit75c833df2a14a754789976d60bbd98e3ca0cd257 (patch)
treea87c2dccecdaba06e1a953bfb93089b56e96fffd
parentd41aad86c8ec1b071b09feaa61a04bf3dc3d5b90 (diff)
[feat] add support for twisted-aware theseus profiler
-rwxr-xr-xscripts/profiling/sync/profile-sync.py30
1 files changed, 26 insertions, 4 deletions
diff --git a/scripts/profiling/sync/profile-sync.py b/scripts/profiling/sync/profile-sync.py
index fb887588..5f01f1f5 100755
--- a/scripts/profiling/sync/profile-sync.py
+++ b/scripts/profiling/sync/profile-sync.py
@@ -31,6 +31,12 @@ def get_and_run_plop_collector():
return collector
+def get_and_run_theseus_tracer():
+ from theseus import Tracer
+ t = Tracer()
+ t.install()
+ return t
+
# main program
if __name__ == '__main__':
@@ -57,7 +63,11 @@ if __name__ == '__main__':
parser.add_argument(
'--plop', dest='do_plop', action='store_true',
help='run sync script under plop profiler')
- parser.set_defaults(do_stats=True, do_plot=False, do_plop=False)
+ parser.add_argument(
+ '--theseus', dest='do_theseus', action='store_true',
+ help='run sync script under theseus profiler')
+ parser.set_defaults(
+ do_stats=True, do_plot=False, do_plop=False, do_theseus=False)
args = parser.parse_args()
# get the password
@@ -79,8 +89,11 @@ if __name__ == '__main__':
s = _get_soledad_instance(
uuid, passphrase, basedir, server_url, cert_file, token)
+ # TODO Profile this with more realistic payloads
+ # TODO Add option to disable sending new docs. If we're profiling
+ # receiving against a fixed account, this will alter each run's results.
+
for i in xrange(10):
- # XXX Profile this with more realistic payloads
s.create_doc({})
def start_sync():
@@ -96,11 +109,16 @@ if __name__ == '__main__':
else:
plop_collector = None
+ if args.do_theseus:
+ theseus = get_and_run_theseus_tracer()
+ else:
+ theseus = None
+
t0 = datetime.now()
d = s.sync()
- d.addCallback(onSyncDone, sl, t0, plop_collector)
+ d.addCallback(onSyncDone, sl, t0, plop_collector, theseus)
- def onSyncDone(sync_result, sl, t0, plop_collector):
+ def onSyncDone(sync_result, sl, t0, plop_collector, theseus):
print "GOT SYNC RESULT: ", sync_result
t1 = datetime.now()
if sl:
@@ -112,6 +130,10 @@ if __name__ == '__main__':
# XXX mkdir profiles dir if not exist
with open('profiles/plop-sync-%s' % GITVER, 'w') as f:
f.write(formatter.format(plop_collector))
+ if theseus:
+ with open('callgrind.theseus', 'wb') as outfile:
+ theseus.write_data(outfile)
+ theseus.uninstall()
delta = (t1 - t0).total_seconds()
print "[+] Sync took %s seconds." % delta