From a64b14b06f4c1cb102ae2786c803a7b82abf38e5 Mon Sep 17 00:00:00 2001
From: marcsello <punkosdmarcell@rocketmail.com>
Date: Tue, 20 Oct 2020 01:00:04 +0200
Subject: [PATCH] Added sentry performance tracking

---
 cnn_classification_service/main.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/cnn_classification_service/main.py b/cnn_classification_service/main.py
index 7285351..0789212 100644
--- a/cnn_classification_service/main.py
+++ b/cnn_classification_service/main.py
@@ -6,6 +6,7 @@ import pika
 import json
 
 from sentry_sdk.integrations.logging import LoggingIntegration
+from sentry_sdk import start_transaction
 import sentry_sdk
 
 from magic_doer import MagicDoer
@@ -18,7 +19,8 @@ def message_callback(channel, method, properties, body):
         logging.warning(f"Invalid message recieved: {e}")
         return
 
-    results = MagicDoer.run_everything(msg)  # <- This is where the magic happens
+    with start_transaction(op="cnn-classification-service", name="classify-soundfile"):
+        results = MagicDoer.run_everything(msg)  # <- This is where the magic happens
 
     channel.basic_publish(
         exchange=os.environ['PIKA_OUTPUT_EXCHANGE'],