104 lines
2.5 KiB
Go
104 lines
2.5 KiB
Go
|
package lib
|
||
|
|
|
||
|
|
import (
|
||
|
|
"bytes"
|
||
|
|
"fmt"
|
||
|
|
"net/http"
|
||
|
|
"strings"
|
||
|
|
"sync"
|
||
|
|
"sync/atomic"
|
||
|
|
"time"
|
||
|
|
)
|
||
|
|
|
||
|
|
// Global HTTP client used for making requests.
|
||
|
|
var client = &http.Client{}
|
||
|
|
|
||
|
|
type RequestError struct {
|
||
|
|
Verb string
|
||
|
|
URL string
|
||
|
|
Err error
|
||
|
|
}
|
||
|
|
|
||
|
|
func (e *RequestError) Error() string {
|
||
|
|
return fmt.Sprintf("error making %s request to %s: %v", e.Verb, e.URL, e.Err)
|
||
|
|
}
|
||
|
|
|
||
|
|
// makeRequest sends an HTTP request and updates performance metrics.
|
||
|
|
func makeRequest(verb, url, token string, jsonData []byte, second int) error {
|
||
|
|
startTime := time.Now()
|
||
|
|
|
||
|
|
// Create a new request with the provided verb, URL, and JSON data if provided.
|
||
|
|
var req *http.Request
|
||
|
|
var err error
|
||
|
|
if jsonData != nil {
|
||
|
|
req, err = http.NewRequest(verb, url, bytes.NewBuffer(jsonData))
|
||
|
|
req.Header.Set("Content-Type", "application/json")
|
||
|
|
} else {
|
||
|
|
req, err = http.NewRequest(verb, url, nil)
|
||
|
|
}
|
||
|
|
if err != nil {
|
||
|
|
return &RequestError{Verb: verb, URL: url, Err: err}
|
||
|
|
}
|
||
|
|
|
||
|
|
// Add the bearer token to the request's Authorization header if provided.
|
||
|
|
if token != "" {
|
||
|
|
req.Header.Set("Authorization", "Bearer "+token)
|
||
|
|
}
|
||
|
|
|
||
|
|
// Send the request.
|
||
|
|
resp, err := client.Do(req)
|
||
|
|
if err != nil {
|
||
|
|
return &RequestError{Verb: verb, URL: url, Err: err}
|
||
|
|
}
|
||
|
|
defer resp.Body.Close()
|
||
|
|
|
||
|
|
// Calculate the duration of the request.
|
||
|
|
duration := time.Since(startTime)
|
||
|
|
|
||
|
|
// Update the performance metrics in a separate goroutine.
|
||
|
|
go UpdateMetrics(duration, resp, second)
|
||
|
|
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
// sendRequests sends requests at the specified rate.
|
||
|
|
func SendRequests(url, bearerToken, requestType string, jsonData []byte, maxRequests int, requestsPerSecond float64) {
|
||
|
|
// Calculate the rate limit based on the requests per second.
|
||
|
|
rateLimit := time.Second / time.Duration(requestsPerSecond)
|
||
|
|
ticker := time.NewTicker(rateLimit)
|
||
|
|
defer ticker.Stop()
|
||
|
|
|
||
|
|
// Initialize the request count.
|
||
|
|
var requestCount int32 = 0
|
||
|
|
|
||
|
|
// Wait for all goroutines to finish.
|
||
|
|
var wg sync.WaitGroup
|
||
|
|
|
||
|
|
// Log beginning of requests
|
||
|
|
fmt.Println("Starting Loadr Requests...")
|
||
|
|
|
||
|
|
// Start sending requests at the specified rate.
|
||
|
|
startTime := time.Now()
|
||
|
|
for range ticker.C {
|
||
|
|
second := int(time.Since(startTime).Seconds())
|
||
|
|
if int(requestCount) >= maxRequests {
|
||
|
|
break
|
||
|
|
}
|
||
|
|
wg.Add(1)
|
||
|
|
go func(u, t, verb string, data []byte, sec int) {
|
||
|
|
defer wg.Done()
|
||
|
|
err := makeRequest(verb, u, t, data, sec)
|
||
|
|
if err != nil {
|
||
|
|
fmt.Println(err)
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
atomic.AddInt32(&requestCount, 1)
|
||
|
|
}(url, bearerToken, strings.ToUpper(requestType), jsonData, second)
|
||
|
|
}
|
||
|
|
|
||
|
|
wg.Wait() // Wait for all requests to finish.
|
||
|
|
|
||
|
|
CalculateAndPrintMetrics(startTime, requestsPerSecond, url, requestType)
|
||
|
|
}
|