import statistics
import time

import requests

times = 5
direct_url = "https://c7b8c7e-12d0-8f3-4c7-8ef01f8a843-0.direct.us-east-1.prod.endpoints.ubiquity.aws.a2z.com/api/burst/51200"
ga_url = "https://c7b8c7e-12d0-8f3-4c7-8ef01f8a843-0.ubiquity.us-east-1.prod.endpoints.ubiquity.aws.a2z.com/api/burst/51200"

direct_result = []
ga_result = []

def print_statistics(text, v):
    print(text)
    print("  min: " + str(min(v)))
    print("  max: " + str(max(v)))
    print("  avg: " + str(statistics.mean(v)))

if __name__ == "__main__":

    for i in range(times):
        t1 = time.time()
        requests.get(direct_url)
        t2 = time.time()
        direct_result.append(t2 - t1)

        t1 = time.time()
        requests.get(ga_url)
        t2 = time.time()
        ga_result.append(t2 - t1)

    print("## Loop times: " + str(times))
    print_statistics("=== Direct ===", direct_result)
    print_statistics("=== GA ===", ga_result)

import socket
import statistics
import time

nlb_ip = "13.215.123.77"
ga_ip = "15.197.219.136"
times = 5
file_size = (
    12 * 1024 * 1024
)  # 12MB, please make sure this value is smaller than your actual response size
payload = b"GET / HTTP/1.1\r\nHost: "

no_sleep_nlb_t1 = []
no_sleep_nlb_t2 = []
no_sleep_ga_t1 = []
no_sleep_ga_t2 = []
sleep_nlb = []
sleep_ga = []

def recvall(sock, n):
    # Helper function to recv n bytes or return None if EOF is hit
    data = bytearray()
    while len(data) < n:
        packet = sock.recv(n - len(data))
        if not packet:
            return None
        data.extend(packet)
    return data

"""Without sleep

The normal behavior of users. Compare between "whole time" and "response time".
It illustrates the wrong calculation method reverses the result.
"""

for i in range(times):
    t1 = time.time()
    sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    server_address = (nlb_ip, 80)
    sock.connect(server_address)
    sock.sendall(payload + nlb_ip.encode() + b"\r\n\r\n")
    t2 = time.time()
		# sock.recv(10)
    recvall(sock, file_size)
    end = time.time()
    no_sleep_nlb_t1.append(end - t1)
    no_sleep_nlb_t2.append(end - t2)

for i in range(times):
    t1 = time.time()
    sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    server_address = (ga_ip, 80)
    sock.connect(server_address)
    sock.sendall(payload + ga_ip.encode() + b"\r\n\r\n")
    t2 = time.time()
		# sock.recv(10)
    recvall(sock, file_size)
    end = time.time()
    no_sleep_ga_t1.append(end - t1)
    no_sleep_ga_t2.append(end - t2)

"""With sleep

Make sure GA has forward the request to origin for fairness.
Just for comparison, this behavior doesn't exist in real world.
"""

for i in range(times):
    sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    server_address = (nlb_ip, 80)
    sock.connect(server_address)
    time.sleep(1)
    sock.sendall(payload + nlb_ip.encode() + b"\r\n\r\n")
    start = time.time()
		# sock.recv(10)
    recvall(sock, file_size)
    end = time.time()
    sleep_nlb.append(end - start)

for i in range(times):
    sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    server_address = (ga_ip, 80)
    sock.connect(server_address)
    time.sleep(1)
    sock.sendall(payload + ga_ip.encode() + b"\r\n\r\n")
    start = time.time()
		# sock.recv(10)
    recvall(sock, file_size)
    end = time.time()
    sleep_ga.append(end - start)

def print_statistics(text, v):
    print(text)
    print("  min: " + str(min(v)))
    print("  max: " + str(max(v)))
    print("  avg: " + str(statistics.mean(v)))

print("## Loop times: " + str(times))
print_statistics("=== No sleep NLB (whole time) ===", no_sleep_nlb_t1)
print_statistics("=== No sleep NLB (after request) ===", no_sleep_nlb_t2)
print_statistics("=== No sleep GA (whole time) ===", no_sleep_ga_t1)
print_statistics("=== No sleep GA (after request) ===", no_sleep_ga_t2)
print_statistics("=== Sleep NLB (after request) ===", sleep_nlb)
print_statistics("=== Sleep GA (after request) ===", sleep_ga)