StellarRouter/scripts/load-test-artillery.sh at main · ghzhost/StellarRouter · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
#!/usr/bin/env bash
#
# Load testing script for router-metrics-exporter using Artillery
#
# Prerequisites:
#   - Artillery installed (npm install -g artillery)
#   - router-metrics-exporter running on localhost:9090
#
# Usage:
#   ./scripts/load-test-artillery.sh [duration] [rps]
#
# Examples:
#   ./scripts/load-test-artillery.sh 30 100    # 30 seconds, 100 RPS
#   ./scripts/load-test-artillery.sh 60 500    # 60 seconds, 500 RPS

set -e

DURATION="${1:-30}"
RPS="${2:-100}"
BASE_URL="${BASE_URL:-http://localhost:9090}"
API_KEY="${ROUTER_API_KEY:-}"

echo "🚀 Starting load test for router-metrics-exporter (Artillery)"
echo "   Duration: ${DURATION}s"
echo "   Target RPS: $RPS"
echo "   Base URL: $BASE_URL"
echo ""

# Create temporary Artillery config
ARTILLERY_CONFIG=$(mktemp)
trap "rm -f $ARTILLERY_CONFIG" EXIT

cat > "$ARTILLERY_CONFIG" << EOF
config:
  target: "$BASE_URL"
  phases:
    - duration: 10
      arrivalRate: 1
      name: "Warm up"
    - duration: $DURATION
      arrivalRate: $RPS
      name: "Sustained load"
    - duration: 10
      arrivalRate: 1
      name: "Cool down"
  processor: "./scripts/artillery-processor.js"
  variables:
    api_key: "$API_KEY"
  http:
    timeout: 10

scenarios:
  - name: "Metrics Exporter Load Test"
    flow:
      - get:
          url: "/metrics"
          headers:
            X-Nonce: "{{ \$randomString(32) }}"
          expect:
            - statusCode: 200
      - think: 100
      - get:
          url: "/health"
          headers:
            X-Nonce: "{{ \$randomString(32) }}"
          expect:
            - statusCode: 200
      - think: 100
      - get:
          url: "/ready"
          headers:
            X-Nonce: "{{ \$randomString(32) }}"
          expect:
            - statusCode: [200, 503]
      - think: 100
EOF

# Run Artillery load test
artillery run "$ARTILLERY_CONFIG" --output /tmp/artillery-report.json

echo ""
echo "✅ Load test completed"
echo "📊 Report saved to /tmp/artillery-report.json"
echo ""
echo "View HTML report:"
echo "  artillery report /tmp/artillery-report.json"