[python_uvloop] Initial commit

Author: KowalskiThomas

scenarios/python_uvloop_3.11/Dockerfile

@@ -0,0 +1,25 @@
+ARG BASE_IMAGE="prof-python-3.11"
+FROM $BASE_IMAGE
+
+# Copy the Python target into the container
+COPY ./scenarios/python_uvloop_3.11/main.py \
+    ./scenarios/python_uvloop_3.11/requirements.txt \
+    /app/
+RUN chmod 644 /app/*
+
+
+# Set the working directory to the location of the program
+WORKDIR /app
+
+RUN pip install -r requirements.txt
+
+ENV EXECUTION_TIME_SEC="5"
+ENV DD_PROFILING_ENABLED=true
+ENV DD_TRACE_ENABLED=false
+ENV DD_TRACE_DEBUG=true
+ENV DD_PROFILING_OUTPUT_PPROF="/app/data/profiles"
+ENV EXECUTION_TIME_SEC="5"
+
+# Run the program when the container starts
+CMD python main.py
+

scenarios/python_uvloop_3.11/expected_profile.json

@@ -0,0 +1,64 @@
+{
+    "test_name": "python_uvloop_3.11",
+    "scale_by_duration": false,
+    "pprof-regex": "",
+    "stacks": [
+        {
+            "profile-type": "wall-time",
+            "pprof-regex": "",
+            "stack-content": [
+                {
+                    "regular_expression": ".*cpu_bound_work.*",
+                    "percent": 75,
+                    "error_margin": 15,
+                    "labels": [
+                        {
+                            "key": "thread name",
+                            "values": [
+                                "MainThread"
+                            ]
+                        }
+                    ]
+                }
+            ]
+        },
+        {
+            "profile-type": "wall-time",
+            "pprof-regex": "",
+            "stack-content": [
+                {
+                    "regular_expression": ".*mixed_workload.*",
+                    "percent": 15,
+                    "error_margin": 10,
+                    "labels": [
+                        {
+                            "key": "thread name",
+                            "values": [
+                                "MainThread"
+                            ]
+                        }
+                    ]
+                }
+            ]
+        },
+        {
+            "profile-type": "wall-time",
+            "pprof-regex": "",
+            "stack-content": [
+                {
+                    "regular_expression": ".*io_simulation.*",
+                    "percent": 0,
+                    "error_margin": 5,
+                    "labels": [
+                        {
+                            "key": "thread name",
+                            "values": [
+                                "MainThread"
+                            ]
+                        }
+                    ]
+                }
+            ]
+        }
+    ]
+}

scenarios/python_uvloop_3.11/main.py

@@ -0,0 +1,59 @@
+"""Test scenario for uvloop profiling."""
+
+import asyncio
+import os
+import time
+
+import uvloop
+from ddtrace.profiling import Profiler
+
+
+async def cpu_bound_work(duration: float) -> None:
+    """Perform CPU-bound work for a specified duration."""
+    end_time = time.monotonic() + duration
+    count = 0
+    while time.monotonic() < end_time:
+        count += 1
+
+
+async def io_simulation(duration: float) -> None:
+    """Simulate I/O-bound work using asyncio.sleep."""
+    await asyncio.sleep(duration)
+
+
+async def mixed_workload(cpu_duration: float, io_duration: float) -> None:
+    """Perform both CPU-bound and I/O-bound work."""
+    await cpu_bound_work(cpu_duration)
+    await io_simulation(io_duration)
+
+
+async def main() -> None:
+    """Main async function that runs the test workload."""
+    execution_time_sec = int(os.environ.get("EXECUTION_TIME_SEC", "5"))
+
+    # Run multiple concurrent tasks to exercise uvloop
+    tasks = [
+        asyncio.create_task(cpu_bound_work(execution_time_sec * 0.3)),
+        asyncio.create_task(mixed_workload(execution_time_sec * 0.2, execution_time_sec * 0.1)),
+        asyncio.create_task(io_simulation(execution_time_sec * 0.4)),
+    ]
+
+    # Also do some work in the main task
+    await cpu_bound_work(execution_time_sec * 0.3)
+
+    # Wait for all tasks to complete
+    await asyncio.gather(*tasks)
+
+    print(f"Completed execution for {execution_time_sec} seconds")
+
+
+if __name__ == "__main__":
+    # Start the profiler
+    prof = Profiler()
+    prof.start()
+
+    # Install uvloop as the event loop
+    uvloop.install()
+
+    # Run the async main function
+    asyncio.run(main())

scenarios/python_uvloop_3.11/requirements.txt

@@ -0,0 +1,3 @@
+ddtrace
+uvloop
+