29 lines
824 B
Python
29 lines
824 B
Python
import stomp, os, time, logging
|
|
|
|
logging.basicConfig(level=logging.INFO)
|
|
logger = logging.getLogger(__name__)
|
|
|
|
MQ_HOST = os.getenv("MQ_HOST", "activemq")
|
|
MQ_PORT = int(os.getenv("MQ_PORT", 61613))
|
|
MQ_USER = os.getenv("MQ_USER", "admin")
|
|
MQ_PASS = os.getenv("MQ_PASS", "admin")
|
|
|
|
def get_connection(listener=None):
|
|
"""
|
|
Vrací připojený STOMP Connection objekt.
|
|
Pokud listener je None, připojení bude bez listeneru.
|
|
"""
|
|
conn = stomp.Connection12([(MQ_HOST, MQ_PORT)])
|
|
if listener:
|
|
conn.set_listener("", listener)
|
|
|
|
while True:
|
|
try:
|
|
conn.connect(MQ_USER, MQ_PASS, wait=True)
|
|
print("Connected to ActiveMQ")
|
|
break
|
|
except Exception as e:
|
|
print("Waiting for ActiveMQ to be ready...", e)
|
|
time.sleep(3)
|
|
return conn
|