[loadgenerator] added loadgeneratorFloodHomepage flagd

Author: bornav

docker-compose.minimal.yml

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

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": 100,
+        "off": 0
+      },
+      "defaultVariant": "off"
     }
   }
 }

src/loadgenerator/locustfile.py

@@ -33,6 +33,11 @@
 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
+from openfeature.exception import OpenFeatureError
+
 from playwright.async_api import Route, Request
 
 logger_provider = LoggerProvider(resource=Resource.create(
@@ -64,6 +69,14 @@
 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)))
+
+def get_flagd_value(FlagName):
+    # Initialize OpenFeature
+    client = api.get_client()
+    return client.get_integer_value(FlagName, 0)
+
 categories = [
     "binoculars",
     "telescopes",
@@ -153,6 +166,11 @@ def checkout_multi(self):
         checkout_person["userId"] = user
         self.client.post("/api/checkout", json=checkout_person)
 
+    @task(5)
+    def flood_home(self):
+        for _ in range(0, get_flagd_value("loadgeneratorFloodHomepage")):
+            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