add oltp from locust

Signed-off-by: Sylvain Hellegouarch <[email protected]>

CHANGELOG.md

@@ -2,7 +2,15 @@
 
 ## [Unreleased][]
 
-[Unreleased]: https://github.com/chaostoolkit-incubator/chaostoolkit-reliably/compare/0.60.0...HEAD
+[Unreleased]: https://github.com/chaostoolkit-incubator/chaostoolkit-reliably/compare/0.61.0...HEAD
+
+## [0.61.0][]
+
+[0.61.0]: https://github.com/chaostoolkit-incubator/chaostoolkit-reliably/compare/0.60.0...0.61.0
+
+### Added
+
+* OLTP support in the load test
 
 ## [0.60.0][]
 

chaosreliably/activities/load/actions.py

@@ -23,6 +23,7 @@ def inject_gradual_traffic_into_endpoint(
     vu_per_second_rate: int = 1,
     test_duration: int = 30,
     results_json_filepath: Optional[str] = None,
+    enable_opentracing: bool = False,
     configuration: Configuration = None,
     secrets: Secrets = None,
 ) -> Dict[str, Any]:
@@ -65,6 +66,17 @@ def inject_gradual_traffic_into_endpoint(
     if test_bearer_token:
         env["RELIABLY_LOCUST_ENDPOINT_TOKEN"] = test_bearer_token
 
+    if enable_opentracing:
+        c = configuration
+        env["RELIABLY_LOCUST_ENABLE_OLTP"] = "true"
+        env["OTEL_VENDOR"] = c.get("otel_vendor", os.getenv("OTEL_VENDOR"))
+        env["CHAOSTOOLKIT_OTEL_GCP_SA"] = c.get(
+            "otel_gcp_service_account", os.getenv("CHAOSTOOLKIT_OTEL_GCP_SA")
+        )
+        env["CHAOSTOOLKIT_OTEL_GCP_PROJECT_ID"] = c.get(
+            "otel_gcp_project_id", os.getenv("CHAOSTOOLKIT_OTEL_GCP_PROJECT_ID")
+        )
+
     results = {}
 
     with tempfile.TemporaryDirectory() as d:

chaosreliably/activities/load/scripts/step_load_test.py

@@ -4,6 +4,15 @@
 
 from locust import HttpUser, LoadTestShape, TaskSet, between, task
 
+try:
+    # these will be available when `chaostoolkit-opentracing` is also
+    # installed
+    from chaosopentracing import oltp
+
+    HAS_OLTP = True
+except ImportError:
+    HAS_OLTP = False
+
 
 class UserTasks(TaskSet):
     @task
@@ -37,3 +46,23 @@ def tick(self) -> Optional[Any]:
 
         current_step = math.floor(run_time / self.step_time) + 1
         return (current_step * self.step_load, self.spawn_rate)
+
+
+def initialize_otel_tracing() -> None:
+    enable_oltp = os.getenv("RELIABLY_LOCUST_ENABLE_OLTP")
+    if not enable_oltp:
+        return
+
+    if enable_oltp.lower() not in (
+        "1",
+        "t",
+        "true",
+    ):
+        return
+
+    oltp.configure_traces(configuration={})
+    oltp.configure_instrumentations(trace_request=True, trace_urllib3=True)
+
+
+if HAS_OLTP:
+    initialize_otel_tracing()

mypy.ini

@@ -30,3 +30,6 @@ ignore_missing_imports = True
 
 [mypy-importlib.metadata]
 ignore_missing_imports = True
+
+[mypy-chaosopentracing.*]
+ignore_missing_imports = True