From 3e4616f4ad2e25d2850ff6dfa09a8cb053594dd8 Mon Sep 17 00:00:00 2001 From: Zooko O'Whielacronx Date: Fri, 21 Sep 2007 15:03:47 -0700 Subject: [PATCH] cli: add --profile flag --- src/allmydata/scripts/startstop_node.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/src/allmydata/scripts/startstop_node.py b/src/allmydata/scripts/startstop_node.py index c2d7df48..3988ec3e 100644 --- a/src/allmydata/scripts/startstop_node.py +++ b/src/allmydata/scripts/startstop_node.py @@ -9,6 +9,9 @@ class StartOptions(BasedirMixin, usage.Options): optParameters = [ ["basedir", "C", None, "which directory to start the node in"], ] + optFlags = [ + ["profile", "p", "whether to run under the Python profiler, putting results in \"profiling_results.prof\""], + ] class StopOptions(BasedirMixin, usage.Options): optParameters = [ @@ -25,7 +28,7 @@ class RestartOptions(BasedirMixin, usage.Options): "of 'restart'"], ] -def do_start(basedir, out=sys.stdout, err=sys.stderr): +def do_start(basedir, profile=False, out=sys.stdout, err=sys.stderr): print >>out, "STARTING", basedir if os.path.exists(os.path.join(basedir, "client.tac")): tac = "client.tac" @@ -53,6 +56,8 @@ def do_start(basedir, out=sys.stdout, err=sys.stderr): fileutil.make_dirs(os.path.join(basedir, "logs")) cmd.extend(["-y", tac, "--logfile", os.path.join("logs", "twistd.log")]) + if profile: + cmd.extend(["--profile=profiling_results.prof", "--savestats",]) curdir = os.getcwd() try: os.chdir(basedir) @@ -100,7 +105,7 @@ def do_stop(basedir, out=sys.stdout, err=sys.stderr): def start(config, stdout, stderr): rc = 0 for basedir in config['basedirs']: - rc = do_start(basedir, stdout, stderr) or rc + rc = do_start(basedir, config['profile'], stdout, stderr) or rc return rc def stop(config, stdout, stderr): -- 2.45.2