Configurable Locust rate (#2806)

* Added param to configure locust user spawn rate * Rate parameter in helm * Added rate to kustomize base and template
GoogleCloudPlatform · Nov 27, 2024 · 40e5400 · 40e5400
1 parent 4c51003
commit 40e5400
Show file tree

Hide file tree

Showing 5 changed files with 9 additions and 1 deletion.
diff --git a/helm-chart/templates/loadgenerator.yaml b/helm-chart/templates/loadgenerator.yaml
@@ -112,6 +112,8 @@ spec:
           value: "{{ .Values.frontend.name }}:80"
         - name: USERS
           value: "10"
+        - name: RATE
+          value: "1"
         resources:
           {{- toYaml .Values.loadGenerator.resources | nindent 10 }}
 {{- if .Values.networkPolicies.create }}

diff --git a/kubernetes-manifests/loadgenerator.yaml b/kubernetes-manifests/loadgenerator.yaml
@@ -83,6 +83,8 @@ spec:
           value: "frontend:80"
         - name: USERS
           value: "10"
+        - name: RATE
+          value: "1"
         resources:
           requests:
             cpu: 300m

diff --git a/kustomize/base/loadgenerator.yaml b/kustomize/base/loadgenerator.yaml
@@ -83,6 +83,8 @@ spec:
           value: "frontend:80"
         - name: USERS
           value: "10"
+        - name: RATE
+          value: "1"
         resources:
           requests:
             cpu: 300m

diff --git a/release/kubernetes-manifests.yaml b/release/kubernetes-manifests.yaml
@@ -741,6 +741,8 @@ spec:
           value: "frontend:80"
         - name: USERS
           value: "10"
+        - name: RATE
+          value: "1"
         resources:
           requests:
             cpu: 300m

diff --git a/src/loadgenerator/Dockerfile b/src/loadgenerator/Dockerfile
@@ -32,4 +32,4 @@ COPY locustfile.py .
 # enable gevent support in debugger
 ENV GEVENT_SUPPORT=True
 
-ENTRYPOINT locust --host="http://${FRONTEND_ADDR}" --headless -u "${USERS:-10}" 2>&1
+ENTRYPOINT locust --host="http://${FRONTEND_ADDR}" --headless -u "${USERS:-10}" -r "${RATE:-1}" 2>&1