up

scripts/graph/load-test.sh

@@ -0,0 +1,47 @@
+#!/usr/bin/env bash
+set -euo pipefail
+
+# DEVOPS-GRAPH-24-001: load test graph index/adjacency APIs
+
+TARGET=${TARGET:-"http://localhost:5000"}
+OUT="out/graph-load"
+mkdir -p "$OUT"
+
+USERS=${USERS:-8}
+DURATION=${DURATION:-60}
+RATE=${RATE:-200}
+
+cat > "${OUT}/k6-graph.js" <<'EOF'
+import http from 'k6/http';
+import { sleep } from 'k6';
+
+export const options = {
+  vus: __USERS__,
+  duration: '__DURATION__s',
+  thresholds: {
+    http_req_duration: ['p(95)<500'],
+    http_req_failed: ['rate<0.01'],
+  },
+};
+
+const targets = [
+  '/graph/api/index',
+  '/graph/api/adjacency?limit=100',
+  '/graph/api/search?q=log4j',
+];
+
+export default function () {
+  const host = __TARGET__;
+  targets.forEach(path => http.get(`${host}${path}`));
+  sleep(1);
+}
+EOF
+
+sed -i "s/__USERS__/${USERS}/g" "${OUT}/k6-graph.js"
+sed -i "s/__DURATION__/${DURATION}/g" "${OUT}/k6-graph.js"
+sed -i "s@__TARGET__@\"${TARGET}\"@g" "${OUT}/k6-graph.js"
+
+echo "[graph-load] running k6..."
+k6 run "${OUT}/k6-graph.js" --summary-export "${OUT}/summary.json" --http-debug="off"
+
+echo "[graph-load] summary written to ${OUT}/summary.json"