mirror of https://github.com/MagicStack/uvloop.git
41 lines
924 B
Python
41 lines
924 B
Python
|
# Copied with minimal modifications from curio
|
||
|
|
# https://github.com/dabeaz/curio
|
||
|
|
|
||
|
|
from concurrent.futures import ProcessPoolExecutor
|
||
|
|
|
||
|
|
from socket import *
|
||
|
|
import time
|
||
|
|
import sys
|
||
|
|
|
||
|
|
if len(sys.argv) > 1:
|
||
|
|
MSGSIZE = int(sys.argv[1])
|
||
|
|
else:
|
||
|
|
MSGSIZE = 1000
|
||
|
|
|
||
|
|
msg = b'x'*MSGSIZE
|
||
|
|
|
||
|
|
def run_test(n):
|
||
|
|
print('Sending', NMESSAGES, 'messages')
|
||
|
|
sock = socket(AF_INET, SOCK_STREAM)
|
||
|
|
sock.connect(('localhost', 25000))
|
||
|
|
while n > 0:
|
||
|
|
sock.sendall(msg)
|
||
|
|
nrecv = 0
|
||
|
|
while nrecv < MSGSIZE:
|
||
|
|
resp = sock.recv(MSGSIZE)
|
||
|
|
if not resp:
|
||
|
|
raise SystemExit()
|
||
|
|
nrecv += len(resp)
|
||
|
|
n -= 1
|
||
|
|
|
||
|
|
N = 3
|
||
|
|
NMESSAGES = 200000
|
||
|
|
start = time.time()
|
||
|
|
with ProcessPoolExecutor(max_workers=N) as e:
|
||
|
|
for _ in range(N):
|
||
|
|
e.submit(run_test, NMESSAGES)
|
||
|
|
end = time.time()
|
||
|
|
duration = end-start
|
||
|
|
print(NMESSAGES*N,'in', duration)
|
||
|
|
print(NMESSAGES*N/duration, 'requests/sec')
|