load-testing/run_load_test.py at main · OpenSocket/load-testing · 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
import concurrent.futures
import requests
import httpx
import time

NUM_REQUESTS = 1000
NUM_THREADS = 50
# URL = "http://rest.ably.io/time"
# URL = "http://localhost:3000/"
URL = "http://timeapi.io/api/Time/current/zone?timeZone=Asia/Kolkata"

print("URL:          ", URL)
print("NUM_REQUESTS: ", NUM_REQUESTS)
print("NUM_THREADS:  ", NUM_THREADS)
print()

# requests
client = requests.Session()
durations = []
success, failure = 0, 0


def send():
    for _ in range(NUM_REQUESTS):
        start = time.monotonic()
        client.get(URL)
        end = time.monotonic()
        duration = end - start
        durations.append(duration)


all_start = time.monotonic()
with concurrent.futures.ThreadPoolExecutor(max_workers=NUM_THREADS) as executor:
    for _ in range(NUM_THREADS):
        executor.submit(send)
all_end = time.monotonic()

durations = sorted(durations)
totalTimeTaken = sum(durations)
print("requests")
print("--------")
print("Total requests: ", len(durations))
print("Average: %.3f" % (sum(durations) / len(durations)))
print("Median:  %.3f" % (durations[int(len(durations) * 0.5)]))
print("95th:    %.3f" % (durations[int(len(durations) * 0.95)]))
print("99th:    %.3f" % (durations[int(len(durations) * 0.99)]))
print("total time taken %.3f" % totalTimeTaken)
print("total time taken concurrent %.3f" % (all_end - all_start))
print()

# httpx
timeout = httpx.Timeout(10, connect=60)
client = httpx.Client(timeout=timeout)
durations = []


def send():
    for _ in range(NUM_REQUESTS):
        start = time.monotonic()
        client.get(URL)
        end = time.monotonic()
        duration = end - start
        durations.append(duration)


all_start = time.monotonic()
with concurrent.futures.ThreadPoolExecutor(max_workers=NUM_THREADS) as executor:
    for _ in range(NUM_THREADS):
        executor.submit(send)
all_end = time.monotonic()

durations = sorted(durations)
totalTimeTaken = sum(durations)

print("httpx")
print("-----")
print("Total requests: ", len(durations))
print("Average: %.3f" % (sum(durations) / len(durations)))
print("Median:  %.3f" % (durations[int(len(durations) * 0.5)]))
print("95th:    %.3f" % (durations[int(len(durations) * 0.95)]))
print("99th:    %.3f" % (durations[int(len(durations) * 0.99)]))
print("total time taken %.3f" % totalTimeTaken)
print("total time taken concurrent %.3f" % (all_end - all_start))
print()