From 4e3efb729595bda3fd74b4939d07369ebfeb893f Mon Sep 17 00:00:00 2001
From: marcsello <punkosdmarcell@rocketmail.com>
Date: Tue, 10 Aug 2021 14:16:32 +0200
Subject: [PATCH] Added basic tracing

---
 requirements.txt |  7 ++++++-
 src/app.py       | 15 ++++++++++++++-
 2 files changed, 20 insertions(+), 2 deletions(-)

diff --git a/requirements.txt b/requirements.txt
index 7edd6a5..f9cf75d 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -16,4 +16,9 @@ py-healthcheck
 Flask-InfluxDB
 tzdata
 tzlocal
-apscheduler~=3.7.0
\ No newline at end of file
+apscheduler~=3.7.0
+
+opentracing~=2.4.0
+jaeger-client
+requests-opentracing
+Flask-Opentracing
\ No newline at end of file
diff --git a/src/app.py b/src/app.py
index 1e65402..eeaef59 100644
--- a/src/app.py
+++ b/src/app.py
@@ -19,6 +19,10 @@ from apscheduler.schedulers.background import BackgroundScheduler
 from magic_amqp import magic_amqp
 from error_handlers import register_all_error_handlers
 
+import jaeger_client
+import opentracing
+from flask_opentracing import FlaskTracing
+
 """
 Main Flask RESTful API
 """
@@ -32,7 +36,7 @@ if Config.SENTRY_DSN:
     sentry_sdk.init(
         dsn=Config.SENTRY_DSN,
         integrations=[FlaskIntegration(), SqlalchemyIntegration()],
-        traces_sample_rate=1.0,
+        traces_sample_rate=0.0,
         send_default_pii=True,
         release=Config.RELEASE_ID,
         environment=Config.RELEASEMODE,
@@ -66,6 +70,15 @@ def init_db():
         influx_db.database.create(Config.INFLUXDB_DATABASE)
     db.create_all()
 
+# Setup tracing
+def initialize_tracer():
+    app.logger.info("Initializing jaeger...")
+    jaeger_cfg = jaeger_client.Config(config={}, service_name='input-service', validate=True)
+    tracer = jaeger_cfg.initialize_tracer()
+    return tracer
+
+
+tracing = FlaskTracing(initialize_tracer, True, app)
 
 api.add_resource(SampleResource, "/sample")
 api.add_resource(SampleParameterResource, '/sample/<tag>')