inital commit

2026-03-05 00:04:26 -06:00
commit f8515a2883
269 changed files with 47081 additions and 0 deletions
@@ -0,0 +1 @@
+k6 run stress/ws-flood-10k.js --env HOST=127.0.0.1 --env PORT=5173 --env SCHEME=wss --env AGENT_TOKEN=dev-token-insecure-local-only --env TLS_INSECURE=true --env TARGET_VUS=10000 --env RAMP_SEC=300 --env HOLD_SEC=600 --env ROLE=client --env HELLO=1 --env RECONNECT=true --env RECONNECT_DELAY_MS=2000 --env SLICE_SEC=60
@@ -0,0 +1,26 @@
+k6 run stress/ws-soak.js \
+  --env HOST=127.0.0.1 \
+  --env PORT=5173 \
+  --env SCHEME=wss \
+  --env AGENT_TOKEN=dev-token-insecure-local-only \
+  --env TLS_INSECURE=true \
+  --env VUS=2000 \
+  --env STEP=500 \
+  --env STAGE_SEC=60 \
+  --env CLIENT_PREFIX=soak \
+  --env ROLE=viewer \
+  --env HEARTBEAT_MS=15000 \
+  --env HELLO=1
+
+# Example 10k flood
+# k6 run stress/ws-flood-10k.js \
+#   --env HOST=127.0.0.1 \
+#   --env PORT=5173 \
+#   --env SCHEME=wss \
+#   --env AGENT_TOKEN=dev-token-insecure-local-only \
+#   --env TLS_INSECURE=true \
+#   --env TARGET_VUS=10000 \
+#   --env RAMP_SEC=300 \
+#   --env HOLD_SEC=600 \
+#   --env ROLE=client \
+#   --env HELLO=1
@@ -0,0 +1,125 @@
+import ws from "k6/ws";
+import { sleep, check } from "k6";
+
+const HOST = __ENV.HOST || "127.0.0.1";
+const PORT = __ENV.PORT || "5173";
+const SCHEME = __ENV.SCHEME || "wss";
+const ROLE = __ENV.ROLE || "client";
+const CLIENT_PREFIX = __ENV.CLIENT_PREFIX || "flood";
+const AGENT_TOKEN = __ENV.AGENT_TOKEN || "";
+const TLS_INSECURE = __ENV.TLS_INSECURE === "1" || __ENV.TLS_INSECURE === "true";
+const HEARTBEAT_MS = Number(__ENV.HEARTBEAT_MS || 15000);
+const INCLUDE_HELLO = __ENV.HELLO !== "0";
+const RECONNECT = __ENV.RECONNECT === "1" || __ENV.RECONNECT === "true";
+const RECONNECT_DELAY_MS = Number(__ENV.RECONNECT_DELAY_MS || 2000);
+const SESSION_SEC = Number(__ENV.SESSION_SEC || 0);
+const SLICE_SEC = Number(__ENV.SLICE_SEC || 60);
+
+const TARGET_VUS = Number(__ENV.TARGET_VUS || 10000);
+const RAMP_SEC = Number(__ENV.RAMP_SEC || 300);
+const HOLD_SEC = Number(__ENV.HOLD_SEC || 600);
+const RAMPDOWN_SEC = Number(__ENV.RAMPDOWN_SEC || 120);
+
+export const options = {
+  scenarios: {
+    ws_flood: {
+      executor: "ramping-vus",
+      startVUs: 0,
+      stages: [
+        { duration: `${RAMP_SEC}s`, target: TARGET_VUS },
+        { duration: `${HOLD_SEC}s`, target: TARGET_VUS },
+        { duration: `${RAMPDOWN_SEC}s`, target: 0 },
+      ],
+      gracefulRampDown: "30s",
+    },
+  },
+  thresholds: {
+    ws_connecting: ["p(95)<1500"],
+    ws_sessions: ["count>0"],
+    checks: ["rate>0.99"],
+  },
+  insecureSkipTLSVerify: TLS_INSECURE,
+};
+
+function buildUrl(clientId) {
+  return `${SCHEME}://${HOST}:${PORT}/api/clients/${clientId}/stream/ws?role=${ROLE}`;
+}
+
+function connectAndHold(url, params, clientId, holdMs) {
+  let closed = false;
+
+  const res = ws.connect(url, params, (socket) => {
+    socket.on("open", () => {
+      if (INCLUDE_HELLO) {
+        socket.send(
+          JSON.stringify({
+            type: "hello",
+            id: clientId,
+            host: "flood",
+            os: "test",
+            arch: "x64",
+            version: "k6",
+            user: "k6",
+            monitors: 1,
+            country: "ZZ",
+          }),
+        );
+      }
+      socket.send(JSON.stringify({ type: "ping", ts: Date.now() }));
+      socket.setInterval(() => {
+        socket.send(JSON.stringify({ type: "ping", ts: Date.now() }));
+      }, HEARTBEAT_MS);
+    });
+
+    socket.on("message", () => {
+      // consume messages to avoid backpressure
+    });
+
+    socket.on("error", (e) => {
+      console.error(`ws error: ${e.error()}`);
+      closed = true;
+    });
+
+    socket.on("close", () => {
+      closed = true;
+    });
+
+    const start = Date.now();
+    while (!closed && Date.now() - start < holdMs) {
+      sleep(1);
+    }
+  });
+
+  check(res, {
+    "status 101": (r) => r && r.status === 101,
+  });
+
+  return res;
+}
+
+export default function () {
+  const clientId = `${CLIENT_PREFIX}-${__VU}-${__ITER}`;
+  const url = buildUrl(clientId);
+
+  const params = {};
+  if (AGENT_TOKEN) {
+    params.headers = { "X-Agent-Token": AGENT_TOKEN };
+  }
+
+  const totalSec = SESSION_SEC > 0 ? SESSION_SEC : HOLD_SEC + 60;
+
+  if (!RECONNECT) {
+    connectAndHold(url, params, clientId, totalSec * 1000);
+    return;
+  }
+
+  let remaining = totalSec;
+  while (remaining > 0) {
+    const slice = Math.min(remaining, SLICE_SEC);
+    connectAndHold(url, params, clientId, slice * 1000);
+    remaining -= slice;
+    if (remaining > 0) {
+      sleep(Math.max(RECONNECT_DELAY_MS, 0) / 1000);
+    }
+  }
+}
@@ -0,0 +1,142 @@
+import ws from "k6/ws";
+import { sleep, check } from "k6";
+
+// 50k-cap ramp: env-tunable, defaults aim for 50k concurrent sockets.
+const HOST = __ENV.HOST || "127.0.0.1";
+const PORT = __ENV.PORT || "5173";
+const SCHEME = __ENV.SCHEME || "wss";
+const AGENT_TOKEN = __ENV.AGENT_TOKEN || "";
+const TLS_INSECURE = __ENV.TLS_INSECURE === "1" || __ENV.TLS_INSECURE === "true";
+const ROLE = __ENV.ROLE || "viewer";
+const CLIENT_PREFIX = __ENV.CLIENT_PREFIX || "soak50k";
+const HEARTBEAT_MS = Number(__ENV.HEARTBEAT_MS || 15000);
+const INCLUDE_HELLO = __ENV.HELLO !== "0";
+const METRIC_URL_TAG = __ENV.METRIC_URL_TAG || "ws-stream";
+
+// Two-phase ramp then long hold and very slow ramp-down to avoid disconnect storms.
+const MAX_VUS = Number(__ENV.VUS || 50000);
+const FAST_TARGET = Number(__ENV.FAST_TARGET || 30000);
+const FAST_STEP = Number(__ENV.FAST_STEP || 5000);
+const FAST_STAGE_SEC = Number(__ENV.FAST_STAGE_SEC || 60);
+const SLOW_STEP = Number(__ENV.SLOW_STEP || 1000);
+const SLOW_STAGE_SEC = Number(__ENV.SLOW_STAGE_SEC || 150);
+const HOLD_SEC = Number(__ENV.HOLD_SEC || 900); // 15m hold by default
+const RAMPDOWN_STEP = Number(__ENV.RAMPDOWN_STEP || 2000);
+const RAMPDOWN_STAGE_SEC = Number(__ENV.RAMPDOWN_STAGE_SEC || 180);
+
+const stages = [];
+// Rapid phase
+for (
+  let target = FAST_STEP;
+  target <= Math.min(FAST_TARGET, MAX_VUS);
+  target += FAST_STEP
+) {
+  stages.push({ duration: `${FAST_STAGE_SEC}s`, target });
+}
+// Slow creep beyond fast target
+for (
+  let target = FAST_TARGET + SLOW_STEP;
+  target <= MAX_VUS;
+  target += SLOW_STEP
+) {
+  stages.push({ duration: `${SLOW_STAGE_SEC}s`, target });
+}
+// Hold peak for a long time to minimize churn
+stages.push({ duration: `${HOLD_SEC}s`, target: MAX_VUS });
+// Slow ramp-down to avoid close storms
+for (
+  let target = MAX_VUS - RAMPDOWN_STEP;
+  target > 0;
+  target -= RAMPDOWN_STEP
+) {
+  stages.push({
+    duration: `${RAMPDOWN_STAGE_SEC}s`,
+    target: Math.max(target, 0),
+  });
+}
+stages.push({ duration: `${RAMPDOWN_STAGE_SEC}s`, target: 0 });
+
+export const options = {
+  scenarios: {
+    ws_capacity: {
+      executor: "ramping-vus",
+      startVUs: 0,
+      stages,
+      gracefulRampDown: "60s",
+    },
+  },
+  thresholds: {
+    ws_connecting: ["p(95)<1500"],
+    ws_sessions: ["count>0"],
+    checks: ["rate>0.99"],
+  },
+  insecureSkipTLSVerify: TLS_INSECURE,
+};
+
+function buildUrl(clientId) {
+  return `${SCHEME}://${HOST}:${PORT}/api/clients/${clientId}/stream/ws?role=${ROLE}`;
+}
+
+export default function () {
+  const clientId = `${CLIENT_PREFIX}-${__VU}-${__ITER}`;
+  const url = buildUrl(clientId);
+
+  const params = { tags: { url: METRIC_URL_TAG, name: METRIC_URL_TAG } };
+  if (AGENT_TOKEN) {
+    params.headers = { "X-Agent-Token": AGENT_TOKEN };
+  }
+
+  const res = ws.connect(
+    url,
+    params,
+    (socket) => {
+      socket.on("open", () => {
+        if (INCLUDE_HELLO) {
+          socket.send(
+            JSON.stringify({
+              type: "hello",
+              id: clientId,
+              host: "soak",
+              os: "test",
+              arch: "x64",
+              version: "k6",
+              user: "k6",
+              monitors: 1,
+              country: "ZZ",
+            }),
+          );
+        }
+
+        socket.send(JSON.stringify({ type: "ping", ts: Date.now() }));
+        socket.setInterval(() => {
+          socket.send(JSON.stringify({ type: "ping", ts: Date.now() }));
+        }, HEARTBEAT_MS);
+        // Additional small tick to ensure ws_msgs_sent is >0 even if pings are filtered
+        socket.setInterval(
+          () => {
+            socket.send("tick");
+          },
+          Math.max(HEARTBEAT_MS / 2, 1000),
+        );
+      });
+
+      socket.on("message", () => {
+        // consume messages to avoid backpressure
+      });
+
+      socket.on("error", (e) => {
+        console.error(`ws error: ${e.error()}`);
+      });
+
+      socket.on("close", () => {
+        // closed
+      });
+
+      sleep(7200); // hold connections for the full run
+    },
+  );
+
+  check(res, {
+    "status 101": (r) => r && r.status === 101,
+  });
+}
@@ -0,0 +1,99 @@
+import ws from "k6/ws";
+import { sleep, check } from "k6";
+
+// Env-tunable ramp and endpoints so you can reuse this locally and on a VPS.
+const HOST = __ENV.HOST || "127.0.0.1";
+const PORT = __ENV.PORT || "5173";
+const SCHEME = __ENV.SCHEME || "wss";
+const AGENT_TOKEN = __ENV.AGENT_TOKEN || "";
+const TLS_INSECURE = __ENV.TLS_INSECURE === "1" || __ENV.TLS_INSECURE === "true";
+const ROLE = __ENV.ROLE || "viewer";
+const CLIENT_PREFIX = __ENV.CLIENT_PREFIX || "soak";
+const HEARTBEAT_MS = Number(__ENV.HEARTBEAT_MS || 15000);
+const INCLUDE_HELLO = __ENV.HELLO !== "0"; // send hello frame once to populate DB/state
+
+// Ramp shape: default 500 VU every 60s until VUS max. Override via env.
+const MAX_VUS = Number(__ENV.VUS || 2000);
+const STEP = Number(__ENV.STEP || 500);
+const STAGE_SEC = Number(__ENV.STAGE_SEC || 60);
+const stages = [];
+for (let target = STEP; target <= MAX_VUS; target += STEP) {
+  stages.push({ duration: `${STAGE_SEC}s`, target });
+}
+stages.push({ duration: `${STAGE_SEC}s`, target: MAX_VUS });
+
+export const options = {
+  scenarios: {
+    ws_capacity: {
+      executor: "ramping-vus",
+      startVUs: 0,
+      stages,
+      gracefulRampDown: "30s",
+    },
+  },
+  thresholds: {
+    ws_connecting: ["p(95)<1000"], // connect latency
+    ws_session_duration: ["p(95)>300000"], // keep sessions alive
+    ws_sessions: ["count>0"],
+    ws_msgs_sent: ["count>0"],
+    checks: ["rate>0.99"],
+  },
+  insecureSkipTLSVerify: TLS_INSECURE,
+};
+
+function buildUrl(clientId) {
+  return `${SCHEME}://${HOST}:${PORT}/api/clients/${clientId}/stream/ws?role=${ROLE}`;
+}
+
+export default function () {
+  const clientId = `${CLIENT_PREFIX}-${__VU}-${__ITER}`;
+  const url = buildUrl(clientId);
+
+  const params = {};
+  if (AGENT_TOKEN) {
+    params.headers = { "X-Agent-Token": AGENT_TOKEN };
+  }
+
+  const res = ws.connect(url, params, (socket) => {
+    socket.on("open", () => {
+      if (INCLUDE_HELLO) {
+        socket.send(
+          JSON.stringify({
+            type: "hello",
+            id: clientId,
+            host: "soak",
+            os: "test",
+            arch: "x64",
+            version: "k6",
+            user: "k6",
+            monitors: 1,
+            country: "ZZ",
+          }),
+        );
+      }
+
+      socket.send(JSON.stringify({ type: "ping", ts: Date.now() }));
+      socket.setInterval(() => {
+        socket.send(JSON.stringify({ type: "ping", ts: Date.now() }));
+      }, HEARTBEAT_MS);
+    });
+
+    socket.on("message", () => {
+      // consume messages to avoid backpressure
+    });
+
+    socket.on("error", (e) => {
+      console.error(`ws error: ${e.error()}`);
+    });
+
+    socket.on("close", () => {
+      // closed
+    });
+
+    sleep(3600); // hold connection for the scenario duration
+  });
+
+  check(res, {
+    "status 101": (r) => r && r.status === 101,
+  });
+}
				`@@ -0,0 +1 @@`
				`k6 run stress/ws-flood-10k.js --env HOST=127.0.0.1 --env PORT=5173 --env SCHEME=wss --env AGENT_TOKEN=dev-token-insecure-local-only --env TLS_INSECURE=true --env TARGET_VUS=10000 --env RAMP_SEC=300 --env HOLD_SEC=600 --env ROLE=client --env HELLO=1 --env RECONNECT=true --env RECONNECT_DELAY_MS=2000 --env SLICE_SEC=60`