version 0.2.0

benchmarks/load_test/benchmark.py

@@ -0,0 +1,71 @@
+import requests
+import time
+import math
+from concurrent.futures import ThreadPoolExecutor
+from http.client import HTTPConnection
+
+HTTPConnection._http_vsn_str = 'HTTP/1.0'
+urls = {  'anthracite': 'http://localhost:8081/50MB.zip','nginx': 'http://localhost:8082/50MB.zip', 'apache': 'http://localhost:8083/50MB.zip' }
+num_requests = 1000
+num_users = 100 # number of threads
+response_times = {} 
+
+def percentile(N, percent, key=lambda x:x):
+    if not N:
+        return None
+    N.sort()
+    k = (len(N)-1) * percent
+    f = math.floor(k)
+    c = math.ceil(k)
+    if f == c:
+        return key(N[int(k)])
+    d0 = key(N[int(f)]) * (c-k)
+    d1 = key(N[int(c)]) * (k-f)
+    return d0+d1
+
+# Define a function to make an HTTP request and measure the response time
+def make_request(request_number, server_name):
+    start_time = time.time()
+    response = requests.get(urls[server_name])
+    end_time = time.time()
+
+    if response.status_code == 200:
+        response_time = end_time - start_time
+        response_times[server_name].append(response_time)
+    else:
+        print(f'Request {request_number}: Request failed with status code {response.status_code}')
+
+print('=====[ Anthracite Benchmarking Tool ]=====')
+print(f'Requests     : {num_requests}')
+print(f'Users/Threads: {num_users}')
+print(f'Test         : Load Test\n\n')
+start_all_time = time.time()
+
+futures = []
+
+for server_name in urls:
+    response_times[server_name] = []
+    with ThreadPoolExecutor(max_workers=num_users) as executor:
+        futures += [executor.submit(make_request, i + 1, server_name) for i in range(num_requests)]
+
+# Wait for all requests to complete
+for future in futures:
+    future.result()
+
+end_all_time = time.time()
+
+for server_name in urls:
+    print(f'====[ {server_name} ]=====')
+    average_response_time = sum(response_times[server_name]) / len(response_times[server_name])
+    total_response_time = sum(response_times[server_name])
+    print(f'Average Response Time: {average_response_time:.4f} seconds')
+    print(f'p995 Response Time   : {percentile(response_times[server_name], .995):.4f} seconds')
+    print(f'p99  Response Time   : {percentile(response_times[server_name], .99):.4f} seconds')
+    print(f'p90  Response Time   : {percentile(response_times[server_name], .90):.4f} seconds')
+    print(f'p75  Response Time   : {percentile(response_times[server_name], .75):.4f} seconds')
+    print(f'p50  Response Time   : {percentile(response_times[server_name], .50):.4f} seconds')
+    print(f'Total Response Time  : {total_response_time:.4f} seconds')
+
+total_test_time = end_all_time - start_all_time
+print('==========')
+print(f'Total Test Time      : {total_test_time:.4f} seconds')