From aa8680a7e662aebb335e1ac9f1e971251ecc04ea Mon Sep 17 00:00:00 2001 From: Gareth Coles Date: Wed, 6 Jun 2018 19:12:03 +0100 Subject: RabbitMQ service discovery --- pysite/service_discovery.py | 22 ++++++++++++++++++++++ 1 file changed, 22 insertions(+) create mode 100644 pysite/service_discovery.py (limited to 'pysite/service_discovery.py') diff --git a/pysite/service_discovery.py b/pysite/service_discovery.py new file mode 100644 index 00000000..2852c2fb --- /dev/null +++ b/pysite/service_discovery.py @@ -0,0 +1,22 @@ +import datetime +import socket +import time +from contextlib import closing + +from pysite.constants import RMQ_HOST, RMQ_PORT + +THIRTY_SECONDS = datetime.timedelta(seconds=30) + + +def wait_for_rmq(): + start = datetime.datetime.now() + + while True: + if datetime.datetime.now() - start > THIRTY_SECONDS: + return False + + with closing(socket.socket(socket.AF_INET, socket.SOCK_STREAM)) as sock: + if sock.connect_ex((RMQ_HOST, RMQ_PORT)) == 0: + return True + + time.sleep(0.5) -- cgit v1.2.3