Commit 032b6e96 authored by Jeremy Hylton's avatar Jeremy Hylton

Don't block forever is server is not up.

Refactor main() to report all errors and exit with non-zero status.
parent e6b721b8
......@@ -18,15 +18,35 @@ Options:
You must specify either -p and -h or -U.
"""
import getopt
import socket
import sys
import time
from ZEO.ClientStorage import ClientStorage
def main(addr, storage, days):
WAIT = 10 # wait no more than 10 seconds for client to connect
def connect(storage):
# The connect-on-startup logic that ZEO provides isn't too useful
# for this script. We'd like to client to attempt to startup, but
# fail if it can't get through to the server after a reasonable
# amount of time. There's no external support for this, so we'll
# expose the ZEO 1.0 internals. (consenting adults only)
t0 = time.time()
while t0 + WAIT > time.time():
storage._call.connect()
if storage._connected:
return
raise RuntimeError, "Unable to connect to ZEO server"
def pack(addr, storage, days):
cs = ClientStorage(addr, storage=storage, wait_for_server_on_startup=1)
# _startup() is an artifact of the way ZEO 1.0 works. The
# ClientStorage doesn't get fully initialized until registerDB()
# is called. The only thing we care about, though, is that
# registerDB() calls _startup().
cs._startup()
connect(cs)
cs.pack(wait=1, days=days)
def usage(exit=1):
......@@ -34,11 +54,7 @@ def usage(exit=1):
print " ".join(sys.argv)
sys.exit(exit)
if __name__ == "__main__":
import getopt
import socket
import sys
def main():
host = None
port = None
unix = None
......@@ -69,5 +85,12 @@ if __name__ == "__main__":
if port is None:
usage()
addr = host, port
main(addr, storage, days)
pack(addr, storage, days)
if __name__ == "__main__":
try:
main()
except Exception, err:
print err
sys.exit(1)
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment