birbnetes/cnn-classification-service
birbnetes/cnn-classification-service
4
0
Fork 0
Code Pull Requests Releases Wiki Activity

merged stuff
All checks were successful
continuous-integration/drone/push Build is passing
Details

Browse Source
This commit is contained in:
Pünkösd Marcell 2021-06-14 03:16:56 +02:00
commit f752668cb5
1 changed files with 7 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
cnn_classification_service/main.py
View File

@ -28,6 +28,7 @@ def message_callback(channel, method, properties, body):
routing_key='classification-result', routing_key='classification-result',
body=json.dumps(results).encode("utf-8") body=json.dumps(results).encode("utf-8")
) )
channel.basic_ack(delivery_tag=method.delivery_tag)
def main(): def main():
@ -56,14 +57,16 @@ def main():
logging.info("Connecting to MQ service...") logging.info("Connecting to MQ service...")
connection = pika.BlockingConnection(pika.connection.URLParameters(os.environ['PIKA_URL'])) connection = pika.BlockingConnection(pika.connection.URLParameters(os.environ['PIKA_URL']))
channel = connection.channel() channel = connection.channel()
channel.exchange_declare(exchange=os.environ['PIKA_INPUT_EXCHANGE'], exchange_type='fanout') channel.exchange_declare(exchange=os.environ['PIKA_INPUT_EXCHANGE'], exchange_type='direct')
queue_declare_result = channel.queue_declare(queue='', exclusive=True) queue_declare_result = channel.queue_declare(queue='cnnqueue', exclusive=False)
queue_name = queue_declare_result.method.queue queue_name = queue_declare_result.method.queue
channel.queue_bind(exchange=os.environ['PIKA_INPUT_EXCHANGE'], queue=queue_name) channel.queue_bind(exchange=os.environ['PIKA_INPUT_EXCHANGE'], routing_key='feature', queue=queue_name)
channel.basic_consume(queue=queue_name, on_message_callback=message_callback, auto_ack=True) channel.basic_qos(prefetch_count=1)
channel.basic_consume(queue=queue_name, on_message_callback=message_callback, auto_ack=False)
logging.info("Connection complete! Listening to messages...") logging.info("Connection complete! Listening to messages...")
try: try: