[loadgenerator] added loadgeneratorFloodHomepage flagd

Author: bornav

docker-compose.yml

@@ -378,6 +378,7 @@ services:
       - OTEL_RESOURCE_ATTRIBUTES
       - OTEL_SERVICE_NAME=loadgenerator
       - PROTOCOL_BUFFERS_PYTHON_IMPLEMENTATION=python
+      - FLAGD_HOST
     depends_on:
       frontend:
         condition: service_started

src/flagd/demo.flagd.json

@@ -78,6 +78,15 @@
         "off": false
       },
       "defaultVariant": "off"
+    },
+    "loadgeneratorFloodHomepage": {
+      "description": "Flood the frontend with a large amount of requests.",
+      "state": "ENABLED",
+      "variants": {
+        "on": true,
+        "off": false
+      },
+      "defaultVariant": "off"
     }
   }
 }

src/loadgenerator/locustfile.py

@@ -33,6 +33,13 @@
 from opentelemetry.sdk._logs import LoggerProvider, LoggingHandler
 from opentelemetry.sdk._logs.export import BatchLogRecordProcessor
 from opentelemetry.sdk.resources import Resource
+
+from openfeature import api
+from openfeature.contrib.provider.flagd import FlagdProvider
+
+# TODO: once openfeature otel hook for python is released, this will work
+# from openfeature.contrib.hooks.otel import TracingHook
+
 from playwright.async_api import Route, Request
 
 logger_provider = LoggerProvider(resource=Resource.create(
@@ -64,6 +71,20 @@
 URLLib3Instrumentor().instrument()
 logging.info("Instrumentation complete")
 
+# Initialize Flagd provider
+api.set_provider(FlagdProvider(host=os.environ.get('FLAGD_HOST', 'flagd'), port=os.environ.get('FLAGD_PORT', 8013)))
+
+# Get flagd value  
+def check_flagd(flag_name: str):
+    # Initialize OpenFeature
+    client = api.get_client()
+    # TODO: once openfeature otel hook for python is released, this will work
+    # api.add_hooks(TracingHook())
+    return client.get_boolean_value("loadgeneratorFloodHomepage", False)
+
+# Amount of iterations
+flood_count = int(os.environ.get("FLOOD_COUNT", 100))
+
 categories = [
     "binoculars",
     "telescopes",
@@ -153,6 +174,12 @@ def checkout_multi(self):
         checkout_person["userId"] = user
         self.client.post("/api/checkout", json=checkout_person)
 
+    @task(5)
+    def flood_home(self):
+        if check_flagd("loadgeneratorFloodHomepage"):
+            for _ in range(0, flood_count):
+                self.client.get("/")
+
     def on_start(self):
         ctx = baggage.set_baggage("synthetic_request", "true")
         context.attach(ctx)

src/loadgenerator/requirements.txt

@@ -30,7 +30,9 @@ opentelemetry-proto==1.23.0
 opentelemetry-sdk==1.23.0
 opentelemetry-semantic-conventions==0.44b0
 opentelemetry-util-http==0.44b0
-protobuf==4.25.0
+openfeature-sdk==0.5.0
+openfeature-provider-flagd==0.1.3
+protobuf==4.25.2
 psutil==5.9.6
 pyzmq==25.1.1
 requests==2.31.0