main.py

#!/usr/bin/env python3
# https://community.grafana.com/t/configuring-grafana-and-pyroscope-phlare-with-docker-compose/102023
# https://github.com/grafana/pyroscope/blob/main/examples/python/simple/main.py
# 
import logging
import os
import pyroscope

l = logging.getLogger()
l.setLevel(logging.DEBUG)

addr = os.getenv("PYROSCOPE_SERVER_ADDRESS") or "http://pyroscope:4040"
print(addr)

pyroscope.configure(
	application_name = "simple.python.app",
	server_address = addr,
	enable_logging = True,
)

def work(n):
	i = 0
	while i < n:
		i += 1

def fast_function():
	with pyroscope.tag_wrapper({ "function": "fast" }):
		work(20000)

def slow_function():
	with pyroscope.tag_wrapper({ "function": "slow" }):
	    work(80000)

if __name__ == "__main__":
	while True:
		fast_function()
		slow_function()