Define second performance test scenario (#2403)

Author: pPrecel

tests/performance/scenarios/mandelbrot/handler.js

@@ -0,0 +1,60 @@
+module.exports = {
+  main: async function (event, context) {
+    const data = event.extensions.request.headers;
+    width = data['width'];
+    height = data['height'];
+    iterations = data['iterations'];
+    zoom = data['zoom'];
+
+    console.log(`Received request with width=${width}, height=${height}, iterations=${iterations}, zoom=${zoom}`);
+
+    return generateMandelbrotJson(width, height, iterations, zoom);
+  }
+}
+
+function generateMandelbrotJson(width, height, iterations, zoom) {
+  const safeWidth = Number.parseInt(width);
+  const safeHeight = Number.parseInt(height);
+  const maxIterations = Number.parseInt(iterations);
+  const safeZoom = Number.parseFloat(zoom);
+  const values = [];
+
+  const xOffset = -0.7;
+  const yOffset = 0;
+
+  for (let y = 0; y < safeHeight; y++) {
+    const row = [];
+    const ci = (y / safeHeight - 0.5) / safeZoom + yOffset;
+
+    for (let x = 0; x < safeWidth; x++) {
+      const cr = (x / safeWidth - 0.5) / safeZoom + xOffset;
+      row.push(mandelbrotEscapeIterations(cr, ci, maxIterations));
+    }
+
+    values.push(row);
+  }
+
+  return {
+    type: "mandelbrot",
+    width: safeWidth,
+    height: safeHeight,
+    maxIterations: maxIterations,
+    zoom: safeZoom,
+    values: values
+  };
+}
+
+function mandelbrotEscapeIterations(cr, ci, maxIterations) {
+  let zr = 0;
+  let zi = 0;
+  let iter = 0;
+
+  while (iter < maxIterations && zr * zr + zi * zi < 4) {
+    const nextZr = zr * zr - zi * zi + cr;
+    zi = 2 * zr * zi + ci;
+    zr = nextZr;
+    iter++;
+  }
+
+  return iter;
+}

tests/performance/scenarios/mandelbrot/handler.py

@@ -0,0 +1,56 @@
+import json
+
+def main(event, context):
+    data = event['extensions']['request'].headers
+    width = data.get("width")
+    height = data.get("height")
+    iterations = data.get("iterations")
+    zoom = data.get("zoom")
+
+    print(f"Received request with width={width}, height={height}, iterations={iterations}, zoom={zoom}")
+
+    return generate_mandelbrot_json(width, height, iterations, zoom)
+
+
+def generate_mandelbrot_json(width, height, iterations, zoom):
+    safe_width = int(str(width).strip())
+    safe_height = int(str(height).strip())
+    max_iterations = int(str(iterations).strip())
+    safe_zoom = float(str(zoom).strip())
+    values = []
+
+    x_offset = -0.7
+    y_offset = 0.0
+
+    for y in range(safe_height):
+        row = []
+        ci = (y / safe_height - 0.5) / safe_zoom + y_offset
+
+        for x in range(safe_width):
+            cr = (x / safe_width - 0.5) / safe_zoom + x_offset
+            row.append(mandelbrot_escape_iterations(cr, ci, max_iterations))
+
+        values.append(row)
+
+    return {
+        "type": "mandelbrot",
+        "width": safe_width,
+        "height": safe_height,
+        "maxIterations": max_iterations,
+        "zoom": safe_zoom,
+        "values": values,
+    }
+
+
+def mandelbrot_escape_iterations(cr, ci, max_iterations):
+    zr = 0.0
+    zi = 0.0
+    it = 0
+
+    while it < max_iterations and (zr * zr + zi * zi) < 4:
+        next_zr = zr * zr - zi * zi + cr
+        zi = 2 * zr * zi + ci
+        zr = next_zr
+        it += 1
+
+    return it

tests/performance/scenarios/mandelbrot/k6.js

@@ -0,0 +1,68 @@
+import http from "k6/http";
+import { sleep } from 'k6';
+import exec from 'k6/execution';
+
+export const options = {
+    scenarios: {
+    warmup: {
+        executor: 'constant-vus',
+        vus: 5,
+        duration: '1m',
+        gracefulStop: '0s',
+        exec: 'call_function_then_sleep',
+    },
+    constant_load: {
+        executor: 'constant-vus',
+        vus: 50,
+        startTime: '1m',
+        duration: "2m",
+        gracefulStop: '0s',
+        exec: 'call_function_then_sleep',
+    },
+    max_load: {
+        executor: 'constant-vus',
+        vus: 100,
+        startTime: '3m',
+        duration: "2m",
+        gracefulStop: '0s',
+        exec: 'call_function',
+    },
+    ramping_max_load: {
+        executor: 'ramping-vus',
+        startVUs: 0,
+        stages: [
+        { duration: '30s', target: 175 },
+        { duration: '30s', target: 350 },
+        { duration: '30s', target: 525 },
+        { duration: '30s', target: 700 },
+        { duration: '30s', target: 875 },
+        { duration: '30s', target: 1050 },
+        ],
+        gracefulRampDown: '0s',
+        startTime: '5m',
+        gracefulStop: '0s',
+        exec: 'call_function',
+    },
+    },
+};
+
+export function call_function() {
+    get();
+};
+
+export function call_function_then_sleep() {
+    get();
+    sleep(1);
+};
+
+function get() {
+    http.get(`http://${__ENV.FUNC_HOSTNAME}:80/`, {
+        headers: { 
+            'Content-Type': 'application/json',
+            'width': 800,
+            'height': 640,
+            'iterations': exec.instance.vusActive,
+            'zoom': 1,
+        },
+    });
+};

tests/performance/scenarios/mandelbrot/package.json

@@ -0,0 +1,3 @@
+{
+  "dependencies": {}
+}

tests/performance/scenarios/mandelbrot/requirements.txt

@@ -0,0 +1 @@
+