Add Sentry to the ingestion server (#1106)

WordPress · Apr 3, 2023 · 9290141 · 9290141
1 parent 6a3c97f
commit 9290141
Show file tree

Hide file tree

Showing 4 changed files with 51 additions and 18 deletions.
diff --git a/docker-compose.yml b/docker-compose.yml
@@ -150,6 +150,7 @@ services:
       - ./ingestion_server:/ingestion_server
     env_file:
       - ingestion_server/env.docker
+      - ingestion_server/.env
     stdin_open: true
     tty: true
 

diff --git a/ingestion_server/Pipfile b/ingestion_server/Pipfile
@@ -23,6 +23,7 @@ psycopg2 = "~=2.9"
 python-decouple = "~=3.8"
 PyYAML = "~=6.0"
 tldextract = "~=3.3"
+sentry-sdk = {extras = ["falcon"], version = "*"}
 
 [requires]
 python_version = "3.10"
diff --git a/ingestion_server/Pipfile.lock b/ingestion_server/Pipfile.lock
diff --git a/ingestion_server/ingestion_server/api.py b/ingestion_server/ingestion_server/api.py
@@ -1,14 +1,17 @@
 """A small RPC API server for scheduling data refresh and indexing tasks."""
 
 import logging
+import os
 import sys
 import time
 import uuid
 from multiprocessing import Process, Value
 from urllib.parse import urlparse
 
 import falcon
+import sentry_sdk
 from falcon.media.validators import jsonschema
+from sentry_sdk.integrations.falcon import FalconIntegration
 
 from ingestion_server import slack
 from ingestion_server.constants.media_types import MEDIA_TYPES, MediaType
@@ -23,6 +26,15 @@
 CALLBACK_URL = "callback_url"
 SINCE_DATE = "since_date"
 
+sentry_sdk.init(
+    dsn=os.environ.get("SENTRY_DSN"),
+    integrations=[
+        FalconIntegration(),
+    ],
+    traces_sample_rate=os.environ.get("SENTRY_SAMPLE_RATE", 1.0),
+    environment=os.environ.get("ENVIRONMENT"),
+)
+
 
 class HealthResource:
     @staticmethod