Skip to content
Snippets Groups Projects
Commit ea0ead21 authored by Bo-Chun Chen's avatar Bo-Chun Chen
Browse files

Add debug message in RCRMQ class

parent 41552582
No related branches found
No related tags found
2 merge requests!23Feat resolve uid gid,!18Redesign RabbitMQ
...@@ -15,8 +15,9 @@ class RCRMQ(object): ...@@ -15,8 +15,9 @@ class RCRMQ(object):
QUEUE = None QUEUE = None
DURABLE = True DURABLE = True
ROUTING_KEY = None ROUTING_KEY = None
DEBUG = False
def __init__(self, config=None): def __init__(self, config=None, debug=False):
if config: if config:
if 'exchange' in config: if 'exchange' in config:
self.EXCHANGE = config['exchange'] self.EXCHANGE = config['exchange']
...@@ -30,6 +31,19 @@ class RCRMQ(object): ...@@ -30,6 +31,19 @@ class RCRMQ(object):
self.PASSWORD = rcfg.Password self.PASSWORD = rcfg.Password
self.VHOST = rcfg.VHost self.VHOST = rcfg.VHost
self.PORT = rcfg.Port self.PORT = rcfg.Port
self.DEBUG = debug
if self.DEBUG:
print("""
Created RabbitMQ instance with:
Exchange name: {},
Exchange type: {},
Host: {},
User: {},
VHost: {},
Port: {}
""".format(self.EXCHANGE, self.EXCHANGE_TYPE, self.HOST, self.USER, self.VHOST, self.PORT))
self._parameters = pika.ConnectionParameters( self._parameters = pika.ConnectionParameters(
self.HOST, self.HOST,
self.PORT, self.PORT,
...@@ -37,6 +51,9 @@ class RCRMQ(object): ...@@ -37,6 +51,9 @@ class RCRMQ(object):
pika.PlainCredentials(self.USER, self.PASSWORD)) pika.PlainCredentials(self.USER, self.PASSWORD))
def connect(self): def connect(self):
if self.DEBUG:
print("Connecting...\n" + "Exchange: " + self.EXCHANGE + " Exchange type: " + self.EXCHANGE_TYPE)
self._connection = pika.BlockingConnection(self._parameters) self._connection = pika.BlockingConnection(self._parameters)
self._channel = self._connection.channel() self._channel = self._connection.channel()
self._channel.exchange_declare( self._channel.exchange_declare(
...@@ -76,6 +93,9 @@ class RCRMQ(object): ...@@ -76,6 +93,9 @@ class RCRMQ(object):
if 'durable' in obj: if 'durable' in obj:
self.DURABLE = obj['durable'] self.DURABLE = obj['durable']
if self.DEBUG:
print("Queue: " + self.QUEUE + "\nRouting_key: " + self.ROUTING_KEY)
self.connect() self.connect()
self._consumer_tag = self._channel.basic_consume(self.QUEUE,obj['cb']) self._consumer_tag = self._channel.basic_consume(self.QUEUE,obj['cb'])
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment