feat(BA-504): Add Pyroscope profiler to common module

BUILD

@@ -10,6 +10,7 @@ python_requirements(
         "python-dateutil": ["dateutil", "dateutil.parser", "dateutil.tz"],
         "python-json-logger": ["pythonjsonlogger"],
         "pyhumps": ["humps"],
+        "pyroscope-io": ["pyroscope"],
         "pyzmq": ["zmq"],
         "PyYAML": ["yaml"],
         "typing-extensions": ["typing_extensions"],

changes/3459.feature.md

@@ -0,0 +1 @@
+Integrate Pyroscope with Backend.AI

configs/account-manager/halfstack.toml

@@ -28,6 +28,13 @@ ssl-enabled = false
 pid-file = "./account-manager.pid"             # env: BACKEND_PID_FILE
 
 
+[pyroscope]
+enabled = true
+app_name = "backendai-half-account-manager"
+server_addr = "http://localhost:4040"
+sample-rate = 100
+
+
 [logging]
 level = "INFO"
 drivers = ["console"]

configs/account-manager/sample.toml

@@ -65,6 +65,12 @@ ssl-enabled = false
 pid-file = "./account-manager.pid"             # env: BACKEND_PID_FILE
 
 
+[pyroscope]
+enabled = true
+app-name = "backendai-half-account-manager"
+server-addr = "http://localhost:4040"
+sample-rate = 100
+
 [logging]
 level = "INFO"
 drivers = ["console"]

configs/agent/ci.toml

@@ -39,6 +39,13 @@ target-service = "backendai-agent.service"
 soft-reset-available = false
 
 
+[pyroscope]
+enabled = false
+# app-name = "backendai-half-agent"
+# server-addr = "http://localhost:4040"
+# sample-rate = 100
+
+
 [logging]
 level = "INFO"
 drivers = ["console"]

configs/agent/halfstack.toml

@@ -47,6 +47,13 @@ target-service = "backendai-agent.service"
 soft-reset-available = false
 
 
+[pyroscope]
+enabled = true
+app-name = "backendai-half-agent"
+server-addr = "http://localhost:4040"
+sample-rate = 100
+
+
 [logging]
 level = "INFO"
 drivers = ["console"]

configs/agent/sample.toml

@@ -216,6 +216,13 @@ target-service = "backendai-agent.service"
 soft-reset-available = false
 
 
+[pyroscope]
+enabled = true
+app-name = "backendai-half-agent"
+server-addr = "http://localhost:4040"
+sample-rate = 100
+
+
 [logging]
 # One of: "NOTSET", "DEBUG", "INFO", "WARNING", "ERROR", "CRITICAL"
 # Set the global logging level.