Added documentation + detailed logging
This commit is contained in:
parent
5dad2d5275
commit
078791ee64
3 changed files with 241 additions and 124 deletions
94
README.md
94
README.md
|
@ -1,6 +1,15 @@
|
||||||
# Loadr
|
# Loadr
|
||||||
|
|
||||||
A lightweight REST load testing tool with robust support for different request patterns, token auth, and performance reports.
|
A lightweight REST load testing tool with flexible request patterns, token auth, and detailed performance metrics output.
|
||||||
|
|
||||||
|
## Stack
|
||||||
|
- Core: Pure Golang
|
||||||
|
- Output: Console + CSV/TXT reporting
|
||||||
|
- Pattern support: Sequential and probabilistic request patterns
|
||||||
|
- Transport: Standard Go HTTP client
|
||||||
|
|
||||||
|
## Requirements
|
||||||
|
- Golang 1.22.0+
|
||||||
|
|
||||||
## Installation
|
## Installation
|
||||||
|
|
||||||
|
@ -11,58 +20,85 @@ go build
|
||||||
## Quick Start
|
## Quick Start
|
||||||
|
|
||||||
```bash
|
```bash
|
||||||
# Simple pattern: 5 POST requests
|
# Simple pattern: 5 POST requests at 20 req/sec
|
||||||
./loadr -rate=20 -max=100 -url=http://api.example.com -pattern=5p
|
./loadr -rate=20 -max=100 -url=http://api.example.com -pattern=5p
|
||||||
|
|
||||||
# Mixed pattern: 1 POST followed by 5 GETs, repeating
|
# Mixed pattern: 1 POST followed by 5 GETs, repeating
|
||||||
./loadr -rate=20 -max=100 -url=http://api.example.com -pattern=1p5g
|
./loadr -rate=20 -max=100 -url=http://api.example.com -pattern=1p5g
|
||||||
|
|
||||||
# With authentication and request body
|
# Probabilistic pattern: 20% POSTs, 80% GETs
|
||||||
|
./loadr -rate=20 -max=100 -url=http://api.example.com -pattern=20%p80%g
|
||||||
|
|
||||||
|
# With auth and request body
|
||||||
./loadr -rate=20 -max=100 -url=http://api.example.com -pattern=2p3g -json=./data.json -token=YourBearerToken
|
./loadr -rate=20 -max=100 -url=http://api.example.com -pattern=2p3g -json=./data.json -token=YourBearerToken
|
||||||
```
|
```
|
||||||
|
|
||||||
## Request Patterns
|
## Request Patterns
|
||||||
|
|
||||||
The `-pattern` flag supports flexible request patterns:
|
The `-pattern` flag supports three types of request patterns:
|
||||||
|
|
||||||
### Simple Patterns
|
### Simple Patterns
|
||||||
- `5p` : 5 POST requests
|
- `5p` : 5 POST requests
|
||||||
- `3g` : 3 GET requests
|
- `3g` : 3 GET requests
|
||||||
|
- Default is `1g` if no pattern specified
|
||||||
|
|
||||||
### Sequential Patterns
|
### Sequential Patterns
|
||||||
- `1p5g` : 1 POST followed by 5 GETs
|
- `1p5g` : 1 POST followed by 5 GETs, repeating
|
||||||
- `2p3g` : 2 POSTs followed by 3 GETs
|
- `2p3g` : 2 POSTs followed by 3 GETs
|
||||||
- `3g2p` : 3 GETs followed by 2 POSTs
|
- `3g2p` : 3 GETs followed by 2 POSTs
|
||||||
|
|
||||||
### Probabalistic Patterns
|
### Probabilistic Patterns
|
||||||
- `20%p80%g` : 20% POST and by 80% GETs
|
- `20%p80%g` : Random selection with 20% POST and 80% GET probability
|
||||||
|
- Percentages must sum to 100
|
||||||
### Pattern Rules
|
|
||||||
- Numbers specify how many requests of each type
|
|
||||||
- 'p' or 'P' specifies POST requests
|
|
||||||
- 'g' or 'G' specifies GET requests
|
|
||||||
- '%' indicates probabilistic requests
|
|
||||||
- If no number is specified, 1 is assumed (e.g., "pg" = "1p1g")
|
|
||||||
- Pattern repeats until max requests is reached
|
|
||||||
|
|
||||||
## Command Line Flags
|
## Command Line Flags
|
||||||
|
|
||||||
- `-rate`: Number of requests per second (default: 10)
|
```bash
|
||||||
- `-max`: Maximum number of requests to send (default: 50)
|
-rate Number of requests per second (default: 10)
|
||||||
- `-url`: Target URL (default: "https://example.com")
|
-max Maximum number of requests to send (default: 50)
|
||||||
- `-pattern`: Request pattern (e.g., "5p", "1p5g", "3g2p", "10%p90%g")
|
-url Target URL (default: "https://example.com")
|
||||||
- `-json`: Path to JSON file for request body
|
-pattern Request pattern (e.g., "5p", "1p5g", "20%p80%g")
|
||||||
- `-token`: Bearer token for authorization
|
-json Path to JSON file for request body
|
||||||
- `-v`, `-version`: Print version information
|
-token Bearer token for authorization
|
||||||
|
-v Print version information
|
||||||
|
```
|
||||||
|
|
||||||
## Reports
|
## Performance Metrics
|
||||||
|
|
||||||
Test results are automatically:
|
Loadr generates two types of reports:
|
||||||
1. Displayed in the console
|
|
||||||
2. Saved to `.reports/[timestamp].txt`
|
|
||||||
|
|
||||||
Reports include:
|
### Summary Report (.reports/summary_[timestamp].txt)
|
||||||
- Total requests sent and received
|
- Total requests sent/received
|
||||||
- Average, maximum, and minimum latency
|
- Average, maximum, and minimum latency
|
||||||
- Requests per second (sent and received)
|
- Requests/sec (target and actual)
|
||||||
|
- Latency percentiles (p50, p95, p99)
|
||||||
- Pattern information
|
- Pattern information
|
||||||
|
|
||||||
|
### Detailed CSV (.reports/detailed_metrics_[timestamp].csv)
|
||||||
|
- Per-request timestamps
|
||||||
|
- Individual request latencies
|
||||||
|
- Status codes
|
||||||
|
- Request types
|
||||||
|
|
||||||
|
## Technical Details
|
||||||
|
|
||||||
|
### Rate Limiting
|
||||||
|
Loadr uses a time-based approach to approximate the target requests per second:
|
||||||
|
|
||||||
|
1. Calculates ideal interval between requests: `interval = 1second / requestsPerSecond`
|
||||||
|
2. Tracks next scheduled request time
|
||||||
|
3. Uses sleep to maintain timing between requests
|
||||||
|
4. Launches requests asynchronously to maintain timing accuracy
|
||||||
|
|
||||||
|
### Request Execution
|
||||||
|
- Uses a global HTTP client for connection reuse
|
||||||
|
- Requests are executed concurrently using goroutines
|
||||||
|
- Metrics are updated synchronously using mutex protection
|
||||||
|
|
||||||
|
### Metrics Collection
|
||||||
|
- Real-time tracking of request latencies
|
||||||
|
- Thread-safe counters for requests and responses
|
||||||
|
- Calculates percentiles from stored request durations
|
||||||
|
- Support for detailed CSV export for external analysis
|
||||||
|
|
||||||
|
Note: The actual request rate may vary slightly from the target rate due to system load and network conditions. The detailed CSV output can be used to analyze the actual timing distribution.
|
||||||
|
|
146
lib/metrics.go
146
lib/metrics.go
|
@ -1,24 +1,26 @@
|
||||||
package lib
|
package lib
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"encoding/csv"
|
||||||
"fmt"
|
"fmt"
|
||||||
"math"
|
"math"
|
||||||
"net/http"
|
"net/http"
|
||||||
"os"
|
"os"
|
||||||
"path/filepath"
|
"path/filepath"
|
||||||
|
"sort"
|
||||||
"strings"
|
"strings"
|
||||||
"time"
|
"time"
|
||||||
)
|
)
|
||||||
|
|
||||||
// Global metrics instance to track performance during load testing
|
// Global metrics instance to track performance during load testing
|
||||||
var metrics = PerformanceMetrics{
|
var metrics = PerformanceMetrics{
|
||||||
// Initialize MinLatency to the maximum possible duration
|
|
||||||
MinLatency: time.Duration(math.MaxInt64),
|
MinLatency: time.Duration(math.MaxInt64),
|
||||||
// Initialize response counters map
|
|
||||||
ResponseCounters: make(map[int]int32),
|
ResponseCounters: make(map[int]int32),
|
||||||
|
RequestLatencies: make([]RequestMetric, 0),
|
||||||
}
|
}
|
||||||
|
|
||||||
// UpdateMetrics synchronously updates performance metrics for each request
|
// UpdateMetrics records metrics for each individual request
|
||||||
|
// Parameters:
|
||||||
// - duration: time taken to complete the request
|
// - duration: time taken to complete the request
|
||||||
// - resp: HTTP response from the request
|
// - resp: HTTP response from the request
|
||||||
// - second: elapsed seconds since the start of the test
|
// - second: elapsed seconds since the start of the test
|
||||||
|
@ -26,16 +28,23 @@ func UpdateMetrics(duration time.Duration, resp *http.Response, second int) {
|
||||||
metrics.Mu.Lock()
|
metrics.Mu.Lock()
|
||||||
defer metrics.Mu.Unlock()
|
defer metrics.Mu.Unlock()
|
||||||
|
|
||||||
// Increment total requests
|
// Create and store individual request metric
|
||||||
|
metric := RequestMetric{
|
||||||
|
Timestamp: time.Now(),
|
||||||
|
Duration: duration,
|
||||||
|
StatusCode: resp.StatusCode,
|
||||||
|
Verb: resp.Request.Method,
|
||||||
|
}
|
||||||
|
metrics.RequestLatencies = append(metrics.RequestLatencies, metric)
|
||||||
|
|
||||||
|
// Update aggregate metrics
|
||||||
metrics.TotalRequests++
|
metrics.TotalRequests++
|
||||||
// Add current request's latency to total
|
|
||||||
metrics.TotalLatency += duration
|
metrics.TotalLatency += duration
|
||||||
|
|
||||||
// Update maximum latency if current duration is higher
|
// Update max/min latencies
|
||||||
if duration > metrics.MaxLatency {
|
if duration > metrics.MaxLatency {
|
||||||
metrics.MaxLatency = duration
|
metrics.MaxLatency = duration
|
||||||
}
|
}
|
||||||
// Update minimum latency if current duration is lower
|
|
||||||
if duration < metrics.MinLatency {
|
if duration < metrics.MinLatency {
|
||||||
metrics.MinLatency = duration
|
metrics.MinLatency = duration
|
||||||
}
|
}
|
||||||
|
@ -45,13 +54,26 @@ func UpdateMetrics(duration time.Duration, resp *http.Response, second int) {
|
||||||
metrics.TotalResponses++
|
metrics.TotalResponses++
|
||||||
metrics.ResponseCounters[second]++
|
metrics.ResponseCounters[second]++
|
||||||
}
|
}
|
||||||
|
|
||||||
// Debug log of current metrics
|
|
||||||
fmt.Printf("Current metrics - Total Requests: %d, Total Responses: %d\n",
|
|
||||||
metrics.TotalRequests, metrics.TotalResponses)
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// CalculateAndPrintMetrics generates a comprehensive report of load test performance
|
// calculatePercentile calculates the nth percentile of latencies
|
||||||
|
// Parameters:
|
||||||
|
// - latencies: sorted slice of request durations
|
||||||
|
// - percentile: desired percentile (e.g., 50 for p50, 95 for p95)
|
||||||
|
// Returns: the duration at the specified percentile
|
||||||
|
func calculatePercentile(latencies []time.Duration, percentile float64) time.Duration {
|
||||||
|
if len(latencies) == 0 {
|
||||||
|
return 0
|
||||||
|
}
|
||||||
|
|
||||||
|
index := int(math.Ceil((percentile/100.0)*float64(len(latencies)))) - 1
|
||||||
|
if index < 0 {
|
||||||
|
index = 0
|
||||||
|
}
|
||||||
|
return latencies[index]
|
||||||
|
}
|
||||||
|
|
||||||
|
// CalculateAndPrintMetrics generates and saves comprehensive test results
|
||||||
// Parameters:
|
// Parameters:
|
||||||
// - startTime: when the load test began
|
// - startTime: when the load test began
|
||||||
// - requestsPerSecond: target request rate
|
// - requestsPerSecond: target request rate
|
||||||
|
@ -64,51 +86,60 @@ func CalculateAndPrintMetrics(startTime time.Time, requestsPerSecond float64, en
|
||||||
metrics.Mu.Lock()
|
metrics.Mu.Lock()
|
||||||
defer metrics.Mu.Unlock()
|
defer metrics.Mu.Unlock()
|
||||||
|
|
||||||
|
// Save detailed per-request metrics to CSV
|
||||||
|
saveDetailedMetrics()
|
||||||
|
|
||||||
|
// Prepare latencies for percentile calculations
|
||||||
|
latencies := make([]time.Duration, len(metrics.RequestLatencies))
|
||||||
|
for i, metric := range metrics.RequestLatencies {
|
||||||
|
latencies[i] = metric.Duration
|
||||||
|
}
|
||||||
|
sort.Slice(latencies, func(i, j int) bool {
|
||||||
|
return latencies[i] < latencies[j]
|
||||||
|
})
|
||||||
|
|
||||||
|
// Calculate various percentiles
|
||||||
|
p50 := calculatePercentile(latencies, 50)
|
||||||
|
p95 := calculatePercentile(latencies, 95)
|
||||||
|
p99 := calculatePercentile(latencies, 99)
|
||||||
|
|
||||||
// Calculate average latency
|
// Calculate average latency
|
||||||
averageLatency := time.Duration(0)
|
averageLatency := time.Duration(0)
|
||||||
if metrics.TotalRequests > 0 {
|
if metrics.TotalRequests > 0 {
|
||||||
averageLatency = metrics.TotalLatency / time.Duration(metrics.TotalRequests)
|
averageLatency = metrics.TotalLatency / time.Duration(metrics.TotalRequests)
|
||||||
}
|
}
|
||||||
|
|
||||||
// Calculate total test duration and total responses
|
// Calculate total duration and responses
|
||||||
totalDuration := time.Since(startTime).Seconds()
|
totalDuration := time.Since(startTime).Seconds()
|
||||||
totalResponses := int32(0)
|
totalResponses := int32(0)
|
||||||
for _, count := range metrics.ResponseCounters {
|
for _, count := range metrics.ResponseCounters {
|
||||||
totalResponses += count
|
totalResponses += count
|
||||||
}
|
}
|
||||||
|
|
||||||
// Ensure MinLatency is not left at its initial max value
|
// Reset min latency if unused
|
||||||
if metrics.MinLatency == time.Duration(math.MaxInt64) {
|
if metrics.MinLatency == time.Duration(math.MaxInt64) {
|
||||||
metrics.MinLatency = 0
|
metrics.MinLatency = 0
|
||||||
}
|
}
|
||||||
|
|
||||||
// Build detailed results string
|
// Build detailed results string
|
||||||
results := fmt.Sprintf("Load Test Report\n")
|
results := fmt.Sprintf("Load Test Report\n=============\n\n")
|
||||||
results += fmt.Sprintf("=============\n\n")
|
|
||||||
|
|
||||||
// Report endpoint and request pattern
|
|
||||||
results += fmt.Sprintf("Endpoint: %s\n", endpoint)
|
results += fmt.Sprintf("Endpoint: %s\n", endpoint)
|
||||||
results += fmt.Sprintf("Pattern: ")
|
results += fmt.Sprintf("Pattern: ")
|
||||||
|
|
||||||
|
// Format request patterns
|
||||||
for i, p := range patterns {
|
for i, p := range patterns {
|
||||||
if i > 0 {
|
if i > 0 {
|
||||||
results += " → "
|
results += " → "
|
||||||
}
|
}
|
||||||
|
|
||||||
var patternDesc string
|
|
||||||
if p.Percentage > 0 && p.Percentage < 100 {
|
if p.Percentage > 0 && p.Percentage < 100 {
|
||||||
// Probabilistic pattern (e.g., "20%p80%g")
|
results += fmt.Sprintf("%.0f%%%s", p.Percentage, strings.ToLower(p.Verb[:1]))
|
||||||
patternDesc = fmt.Sprintf("%.0f%%%s", p.Percentage, strings.ToLower(p.Verb[:1]))
|
|
||||||
} else {
|
} else {
|
||||||
// Simple or sequential pattern (e.g., "5p", "3g", "1p5g")
|
results += fmt.Sprintf("%d%s", p.Sequence, strings.ToLower(p.Verb[:1]))
|
||||||
patternDesc = fmt.Sprintf("%d%s", p.Sequence, strings.ToLower(p.Verb[:1]))
|
|
||||||
}
|
}
|
||||||
|
|
||||||
results += patternDesc
|
|
||||||
}
|
}
|
||||||
results += "\n\n"
|
results += "\n\n"
|
||||||
|
|
||||||
// Detailed performance metrics
|
// Add performance metrics
|
||||||
results += fmt.Sprintf("Performance Metrics\n")
|
results += fmt.Sprintf("Performance Metrics\n")
|
||||||
results += fmt.Sprintf("-----------------\n")
|
results += fmt.Sprintf("-----------------\n")
|
||||||
results += fmt.Sprintf("Total Requests Sent: %d\n", metrics.TotalRequests)
|
results += fmt.Sprintf("Total Requests Sent: %d\n", metrics.TotalRequests)
|
||||||
|
@ -120,27 +151,68 @@ func CalculateAndPrintMetrics(startTime time.Time, requestsPerSecond float64, en
|
||||||
results += fmt.Sprintf("Requests/sec (Actual): %.2f\n", float64(metrics.TotalRequests)/totalDuration)
|
results += fmt.Sprintf("Requests/sec (Actual): %.2f\n", float64(metrics.TotalRequests)/totalDuration)
|
||||||
results += fmt.Sprintf("Responses/sec: %.2f\n", float64(totalResponses)/totalDuration)
|
results += fmt.Sprintf("Responses/sec: %.2f\n", float64(totalResponses)/totalDuration)
|
||||||
|
|
||||||
// Print and save the report
|
// Add percentile information
|
||||||
|
results += fmt.Sprintf("\nLatency Percentiles\n")
|
||||||
|
results += fmt.Sprintf("-----------------\n")
|
||||||
|
results += fmt.Sprintf("50th percentile (p50): %s\n", p50)
|
||||||
|
results += fmt.Sprintf("95th percentile (p95): %s\n", p95)
|
||||||
|
results += fmt.Sprintf("99th percentile (p99): %s\n", p99)
|
||||||
|
|
||||||
fmt.Println(results)
|
fmt.Println(results)
|
||||||
saveReport(results)
|
saveReport(results)
|
||||||
}
|
}
|
||||||
|
|
||||||
// saveReport writes the load test results to a timestamped file in the .reports directory
|
// saveDetailedMetrics writes per-request metrics to a CSV file
|
||||||
// Parameters:
|
// The CSV includes timestamp, duration, status code, and request type for each request
|
||||||
// - results: formatted results string to be saved
|
func saveDetailedMetrics() {
|
||||||
func saveReport(results string) {
|
|
||||||
// Ensure .reports directory exists
|
|
||||||
resultsDir := ".reports"
|
resultsDir := ".reports"
|
||||||
os.MkdirAll(resultsDir, os.ModePerm)
|
os.MkdirAll(resultsDir, os.ModePerm)
|
||||||
|
|
||||||
// Create a unique filename based on current timestamp
|
csvFile := filepath.Join(resultsDir, fmt.Sprintf("detailed_metrics_%d.csv", time.Now().Unix()))
|
||||||
resultsFile := filepath.Join(resultsDir, fmt.Sprintf("%d.txt", time.Now().Unix()))
|
file, err := os.Create(csvFile)
|
||||||
|
if err != nil {
|
||||||
|
fmt.Println("Error creating CSV file:", err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
defer file.Close()
|
||||||
|
|
||||||
|
writer := csv.NewWriter(file)
|
||||||
|
defer writer.Flush()
|
||||||
|
|
||||||
|
// Write CSV header
|
||||||
|
writer.Write([]string{
|
||||||
|
"Timestamp",
|
||||||
|
"Duration (ms)",
|
||||||
|
"Status Code",
|
||||||
|
"Request Type",
|
||||||
|
})
|
||||||
|
|
||||||
|
// Write individual request metrics
|
||||||
|
for _, metric := range metrics.RequestLatencies {
|
||||||
|
writer.Write([]string{
|
||||||
|
metric.Timestamp.Format(time.RFC3339),
|
||||||
|
fmt.Sprintf("%.2f", float64(metric.Duration.Milliseconds())),
|
||||||
|
fmt.Sprintf("%d", metric.StatusCode),
|
||||||
|
metric.Verb,
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
fmt.Println("Detailed metrics saved:", csvFile)
|
||||||
|
}
|
||||||
|
|
||||||
|
// saveReport writes the summary report to a text file
|
||||||
|
// Parameters:
|
||||||
|
// - results: formatted string containing the complete test results
|
||||||
|
func saveReport(results string) {
|
||||||
|
resultsDir := ".reports"
|
||||||
|
os.MkdirAll(resultsDir, os.ModePerm)
|
||||||
|
|
||||||
|
resultsFile := filepath.Join(resultsDir, fmt.Sprintf("summary_%d.txt", time.Now().Unix()))
|
||||||
|
|
||||||
// Write results to file
|
|
||||||
if err := os.WriteFile(resultsFile, []byte(results), 0644); err != nil {
|
if err := os.WriteFile(resultsFile, []byte(results), 0644); err != nil {
|
||||||
fmt.Println("Error saving report:", err)
|
fmt.Println("Error saving report:", err)
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
fmt.Println("Report saved:", resultsFile)
|
fmt.Println("Summary report saved:", resultsFile)
|
||||||
}
|
}
|
||||||
|
|
|
@ -15,6 +15,15 @@ type PerformanceMetrics struct {
|
||||||
MaxLatency time.Duration
|
MaxLatency time.Duration
|
||||||
MinLatency time.Duration
|
MinLatency time.Duration
|
||||||
ResponseCounters map[int]int32
|
ResponseCounters map[int]int32
|
||||||
|
RequestLatencies []RequestMetric
|
||||||
|
}
|
||||||
|
|
||||||
|
// RequestMetric represents a single request's performance metrics
|
||||||
|
type RequestMetric struct {
|
||||||
|
Timestamp time.Time
|
||||||
|
Duration time.Duration
|
||||||
|
StatusCode int
|
||||||
|
Verb string
|
||||||
}
|
}
|
||||||
|
|
||||||
// RequestError represents a detailed error that occurs during an HTTP request
|
// RequestError represents a detailed error that occurs during an HTTP request
|
||||||
|
|
Loading…
Add table
Reference in a new issue