[python_greenlet] Initial commit

Author: KowalskiThomas

scenarios/python_greenlet_3.11/Dockerfile

@@ -0,0 +1,22 @@
+ARG BASE_IMAGE="prof-python-3.11"
+FROM $BASE_IMAGE
+
+# Copy the Python target into the container
+COPY ./scenarios/python_greenlet_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
+
+COPY ./scenarios/python_greenlet_3.11/main.py /app/
+
+ENV DD_PROFILING_ENABLED=true
+ENV DD_TRACE_ENABLED=false
+ENV DD_TRACE_DEBUG=false
+ENV DD_PROFILING_OUTPUT_PPROF="/app/data/profiles"
+ENV EXECUTION_TIME_SEC="4"
+
+# Run the program when the container starts
+CMD ddtrace-run python main.py

scenarios/python_greenlet_3.11/expected_profile.json

@@ -0,0 +1,102 @@
+{
+  "test_name": "python_greenlet_3.11",
+  "scale_by_duration": false,
+  "pprof-regex": "",
+  "stacks": [
+    {
+      "profile-type": "wall-time",
+      "pprof-regex": "",
+      "stack-content": [
+        {
+          "regular_expression": ".*work_a.*",
+          "percent": 50,
+          "error_margin": 10,
+          "labels": [
+            {
+              "key": "thread name",
+              "values": [
+                "MainThread"
+              ]
+            }
+          ]
+        }
+      ]
+    },
+    {
+      "profile-type": "wall-time",
+      "pprof-regex": "",
+      "stack-content": [
+        {
+          "regular_expression": ".*work_b.*",
+          "percent": 50,
+          "error_margin": 10,
+          "labels": [
+            {
+              "key": "thread name",
+              "values": [
+                "MainThread"
+              ]
+            }
+          ]
+        }
+      ]
+    },
+    {
+      "profile-type": "wall-time",
+      "pprof-regex": "",
+      "stack-content": [
+        {
+          "regular_expression": ".*work_dependency.*",
+          "percent": 25,
+          "error_margin": 10,
+          "labels": [
+            {
+              "key": "thread name",
+              "values": [
+                "MainThread"
+              ]
+            }
+          ]
+        }
+      ]
+    },
+    {
+      "profile-type": "wall-time",
+      "pprof-regex": "",
+      "stack-content": [
+        {
+          "regular_expression": ".*work_a;.*work_dependency.*",
+          "percent": 13,
+          "error_margin": 10,
+          "labels": [
+            {
+              "key": "thread name",
+              "values": [
+                "MainThread"
+              ]
+            }
+          ]
+        }
+      ]
+    },
+    {
+      "profile-type": "wall-time",
+      "pprof-regex": "",
+      "stack-content": [
+        {
+          "regular_expression": ".*work_b;.*work_dependency.*",
+          "percent": 13,
+          "error_margin": 10,
+          "labels": [
+            {
+              "key": "thread name",
+              "values": [
+                "MainThread"
+              ]
+            }
+          ]
+        }
+      ]
+    }
+  ]
+}

scenarios/python_greenlet_3.11/main.py

@@ -0,0 +1,68 @@
+"""Test scenario for greenlet profiling."""
+
+import os
+import time
+
+import greenlet
+
+
+class GreenletBurner:
+    def __init__(self) -> None:
+        self.greenlet_a = greenlet.greenlet(self.work_a)
+        self.greenlet_b = greenlet.greenlet(self.work_b)
+
+        self.total_count_a = 0
+        self.total_count_b = 0
+
+    def work_dependency(self, count:int)->tuple[bool, int]:
+        count += 1
+        if count % 10000 == 0:
+            return True, count
+
+        return False, count
+
+    def work_a(self) -> None:
+        """Perform work in greenlet A."""
+        end_time = time.monotonic() + 0.1
+        count = 0
+        while time.monotonic() < end_time:
+            has_dependency, count = self.work_dependency(count)
+            if has_dependency:
+                self.greenlet_b.switch()
+
+            self.total_count_a += 1
+
+    def work_b(self) -> None:
+        """Perform work in greenlet B."""
+        end_time = time.monotonic() + 0.1
+        count = 0
+        while time.monotonic() < end_time:
+            has_dependency, count = self.work_dependency(count)
+            if has_dependency:
+                self.greenlet_a.switch()
+
+            self.total_count_b += 1
+
+    def work(self) -> None:
+        self.greenlet_a.switch()
+        self.greenlet_b.switch()
+
+        self.greenlet_a = greenlet.greenlet(self.work_a)
+        self.greenlet_b = greenlet.greenlet(self.work_b)
+
+def main()->None:
+    execution_time_sec = int(os.environ.get("EXECUTION_TIME_SEC", "10"))
+
+    greenlet_burner = GreenletBurner()
+
+    # Run for specified duration
+    end_time = time.monotonic() + execution_time_sec
+
+    while time.monotonic() < end_time:
+        # Start switching between greenlets
+        greenlet_burner.work()
+
+    print(f"Completed execution for {execution_time_sec} seconds")
+
+if __name__ == "__main__":
+    main()

scenarios/python_greenlet_3.11/requirements.txt

@@ -0,0 +1,2 @@
+ddtrace
+greenlet