mirror of https://github.com/MagicStack/uvloop.git
96 lines
2.9 KiB
Python
96 lines
2.9 KiB
Python
# Copied with minimal modifications from curio
|
|
# https://github.com/dabeaz/curio
|
|
|
|
|
|
import argparse
|
|
import concurrent.futures
|
|
import socket
|
|
import ssl
|
|
import time
|
|
|
|
|
|
if __name__ == '__main__':
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument('--msize', default=1000, type=int,
|
|
help='message size in bytes')
|
|
parser.add_argument('--mpr', default=1, type=int,
|
|
help='messages per request')
|
|
parser.add_argument('--num', default=200000, type=int,
|
|
help='number of messages')
|
|
parser.add_argument('--times', default=1, type=int,
|
|
help='number of times to run the test')
|
|
parser.add_argument('--workers', default=3, type=int,
|
|
help='number of workers')
|
|
parser.add_argument('--addr', default='127.0.0.1:25000', type=str,
|
|
help='address:port of echoserver')
|
|
parser.add_argument('--ssl', default=False, action='store_true')
|
|
args = parser.parse_args()
|
|
|
|
client_context = None
|
|
if args.ssl:
|
|
print('with SSL')
|
|
client_context = ssl.SSLContext(ssl.PROTOCOL_SSLv23)
|
|
if hasattr(client_context, 'check_hostname'):
|
|
client_context.check_hostname = False
|
|
client_context.verify_mode = ssl.CERT_NONE
|
|
|
|
unix = False
|
|
if args.addr.startswith('file:'):
|
|
unix = True
|
|
addr = args.addr[5:]
|
|
else:
|
|
addr = args.addr.split(':')
|
|
addr[1] = int(addr[1])
|
|
addr = tuple(addr)
|
|
print('will connect to: {}'.format(addr))
|
|
|
|
MSGSIZE = args.msize
|
|
REQSIZE = MSGSIZE * args.mpr
|
|
|
|
msg = b'x'*(MSGSIZE - 1) + b'\n'
|
|
if args.mpr:
|
|
msg *= args.mpr
|
|
|
|
def run_test(n):
|
|
print('Sending', NMESSAGES, 'messages')
|
|
if args.mpr:
|
|
n //= args.mpr
|
|
|
|
if unix:
|
|
sock = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
|
|
else:
|
|
sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
|
|
|
|
try:
|
|
sock.setsockopt(socket.IPPROTO_TCP, socket.TCP_NODELAY, 1)
|
|
except (OSError, NameError):
|
|
pass
|
|
|
|
if client_context:
|
|
sock = client_context.wrap_socket(sock)
|
|
|
|
sock.connect(addr)
|
|
|
|
while n > 0:
|
|
sock.sendall(msg)
|
|
nrecv = 0
|
|
while nrecv < REQSIZE:
|
|
resp = sock.recv(REQSIZE)
|
|
if not resp:
|
|
raise SystemExit()
|
|
nrecv += len(resp)
|
|
n -= 1
|
|
|
|
TIMES = args.times
|
|
N = args.workers
|
|
NMESSAGES = args.num
|
|
start = time.time()
|
|
for _ in range(TIMES):
|
|
with concurrent.futures.ProcessPoolExecutor(max_workers=N) as e:
|
|
for _ in range(N):
|
|
e.submit(run_test, NMESSAGES)
|
|
end = time.time()
|
|
duration = end-start
|
|
print(NMESSAGES * N * TIMES, 'in', duration)
|
|
print(NMESSAGES * N * TIMES / duration, 'requests/sec')
|