From 27996b3bc50a5c5f7e137315293057ad738ed975 Mon Sep 17 00:00:00 2001
From: Mark Veidemanis <m@zm.is>
Date: Tue, 2 Aug 2022 22:22:22 +0100
Subject: [PATCH] Add Pyroscope profiler

---
 app/settings.py                   | 10 ++++++++++
 docker-compose.yml                |  9 +++++++++
 docker/prod/requirements.prod.txt |  1 +
 docker/requirements.dev.txt       |  1 +
 4 files changed, 21 insertions(+)

diff --git a/app/settings.py b/app/settings.py
index 0ee988e..b28bbb3 100644
--- a/app/settings.py
+++ b/app/settings.py
@@ -135,3 +135,13 @@ REST_FRAMEWORK = {
 }
 
 from app.local_settings import *  # noqa
+if PROFILER:
+    import pyroscope
+    pyroscope.configure(
+        application_name = "neptune",
+        server_address   = "http://pyroscope:4040",
+        auth_token       = os.getenv("PYROSCOPE_AUTH_TOKEN", ""),
+        # tags           = {
+        #     "region":   f'{os.getenv("REGION")}',
+        # }
+    )
\ No newline at end of file
diff --git a/docker-compose.yml b/docker-compose.yml
index 45fd8c9..ecb6bf7 100644
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -13,6 +13,15 @@ services:
     env_file:
       - .env
 
+  pyroscope:
+    image: pyroscope/pyroscope
+    environment:
+      - PYROSCOPE_LOG_LEVEL=debug
+    ports:
+      - '4040:4040'
+    command:
+      - 'server'
+
 networks:
    default:
       external:
diff --git a/docker/prod/requirements.prod.txt b/docker/prod/requirements.prod.txt
index e48344f..3d57592 100644
--- a/docker/prod/requirements.prod.txt
+++ b/docker/prod/requirements.prod.txt
@@ -7,3 +7,4 @@ stripe
 django-rest-framework
 numpy
 uwsgi
+pyroscope-io
diff --git a/docker/requirements.dev.txt b/docker/requirements.dev.txt
index 9ea9856..d2d54cd 100644
--- a/docker/requirements.dev.txt
+++ b/docker/requirements.dev.txt
@@ -6,3 +6,4 @@ opensearch-py
 stripe
 django-rest-framework
 numpy
+pyroscope-io