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

Finished stuff

Browse Source
This commit is contained in:
Pünkösd Marcell
2020-07-28 18:10:19 +02:00
parent a45abf5870
commit fe0552fb09
3 changed files with 80 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
cnn_classification_service/main.py
View File

@@ -8,12 +8,19 @@ import json
from sentry_sdk.integrations.logging import LoggingIntegration
import sentry_sdk
from cnn_classifier import Classifier
from magic_doer import run_everything
def message_callback(ch, method, properties, body):
msg = json.loads(body.decode('utf-8'))
# TODO
results = run_everything(msg)
# TODO: Ez azért elég gettó, de legalább csatlakozik
connection = pika.BlockingConnection(pika.connection.URLParameters(os.environ['PIKA_URL']))
channel = connection.channel()
channel.exchange_declare(exchange=os.environ['PIKA_OUTPUT_EXCHANGE'], exchange_type='fanout')
channel.basic_publish(exchange=os.environ['PIKA_OUTPUT_EXCHANGE'], routing_key='classification-result',
body=json.dumps(results).encode("utf-8"))
def main():
@@ -37,12 +44,12 @@ def main():
logging.info("Connecting to MQ service...")
connection = pika.BlockingConnection(pika.connection.URLParameters(os.environ['PIKA_URL']))
channel = connection.channel()
channel.exchange_declare(exchange=os.environ['PIKA_EXCHANGE_NAME'], exchange_type='fanout')
channel.exchange_declare(exchange=os.environ['PIKA_INPUT_EXCHANGE'], exchange_type='fanout')
queue_declare_result = channel.queue_declare(queue='', exclusive=True)
queue_name = queue_declare_result.method.queue
channel.queue_bind(exchange=os.environ['PIKA_EXCHANGE_NAME'], queue=queue_name)
channel.queue_bind(exchange=os.environ['PIKA_INPUT_EXCHANGE'], queue=queue_name)
channel.basic_consume(queue=queue_name, on_message_callback=message_callback, auto_ack=True)
logging.info("Connection complete! Listening to messages...")
@@ -54,4 +61,4 @@ def main():
if __name__ == '__main__':
main()
main()