From d8b62f20a25fe808a59f10cd045110c1fd2e135e Mon Sep 17 00:00:00 2001
From: marcsello <punkosdmarcell@rocketmail.com>
Date: Wed, 4 Aug 2021 16:57:27 +0200
Subject: [PATCH] Added basic tracking

---
 model_service/app.py | 15 +++++++++++++++
 requirements.txt     |  5 ++++-
 2 files changed, 19 insertions(+), 1 deletion(-)

diff --git a/model_service/app.py b/model_service/app.py
index aaf9713..0d8ef1c 100644
--- a/model_service/app.py
+++ b/model_service/app.py
@@ -14,6 +14,11 @@ from views import SVMView, CNNView, RootView
 
 from config import Config
 
+# Tracing stuffs
+import jaeger_client
+import opentracing
+from flask_opentracing import FlaskTracing
+
 if Config.SENTRY_DSN:
     sentry_sdk.init(
         dsn=Config.SENTRY_DSN,
@@ -43,6 +48,16 @@ def create_db():
 # register error handlers
 register_all_error_handlers(app)
 
+
+# Setup tracing
+def initialize_tracer():
+    jaeger_cfg = jaeger_client.Config(config={}, service_name='cnn-classification-service', validate=True)
+    tracer = jaeger_cfg.initialize_tracer()
+    return tracer
+
+
+tracing = FlaskTracing(initialize_tracer, True, app)
+
 # register views
 for view in [SVMView, CNNView, RootView]:
     view.register(app, trailing_slash=False, route_prefix='/model')
diff --git a/requirements.txt b/requirements.txt
index 8c101e0..625d5da 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -30,4 +30,7 @@ python-dateutil==2.8.1
 scikit-learn==0.21.3
 scipy==1.4.1
 simplejson==3.17.0
-six==1.14.0
\ No newline at end of file
+six==1.14.0
+
+jaeger-client
+Flask-Opentracing
\ No newline at end of file