diff options
| author | Rob Austein <sra@hactrn.net> | 2007-12-23 19:27:50 +0000 |
|---|---|---|
| committer | Rob Austein <sra@hactrn.net> | 2007-12-23 19:27:50 +0000 |
| commit | 5780246a4f7464cba603082e5f4997f17983833d (patch) | |
| tree | 8b8c7b3c12085bd10578a236a8d72ff8175013c1 /scripts/testdb.py | |
| parent | be257f3802d2d9da27805fb26504e439a3a497b3 (diff) | |
Add sleep and pause commands
svn path=/scripts/testdb.py; revision=1429
Diffstat (limited to 'scripts/testdb.py')
| -rw-r--r-- | scripts/testdb.py | 28 |
1 files changed, 27 insertions, 1 deletions
diff --git a/scripts/testdb.py b/scripts/testdb.py index aa217420..e60311a6 100644 --- a/scripts/testdb.py +++ b/scripts/testdb.py @@ -158,6 +158,28 @@ def main(): except Exception, data: rpki.log.warn("Couldn't clean up daemons (%s), continuing" % data) +# Signal handling and commands to make use of it + +def wakeup(signum, frame): + """Handler called when we receive a SIGALRM signal.""" + rpki.log.info("Wakeup call received, continuing") + +signal.signal(signal.SIGALRM, wakeup) + +def cmd_pause(): + """Do nothing until a signal arrives.""" + rpki.log.info("Pausing indefinitely, send a SIGALRM to wake me up") + signal.pause() + +def cmd_sleep(seconds): + """Set an alarm, then wait for it to go off.""" + rpki.log.info("Sleeping %s seconds" % seconds) + signal.alarm(int(seconds)) + signal.pause() + +cmds = { "pause" : cmd_pause, + "sleep" : cmd_sleep } + class allocation_db(list): def __init__(self, yaml): @@ -177,7 +199,11 @@ class allocation_db(list): def apply_delta(self, delta): for d in delta: - self.map[d["name"]].apply_delta(d) + if isinstance(d, str): + c = d.split() + cmds[c[0]](*c[1:]) + else: + self.map[d["name"]].apply_delta(d) self.root.closure() def dump(self): |