53 lines
1.4 KiB
Python
53 lines
1.4 KiB
Python
import asyncore
|
|
import socket
|
|
import time
|
|
|
|
requestCount = 0
|
|
parallel = 50
|
|
duration = 60
|
|
|
|
|
|
class HTTPClient(asyncore.dispatcher):
|
|
"""An asyncore dispatcher"""
|
|
port = 12345
|
|
|
|
def __init__(self, host, path, connect=True):
|
|
if not hasattr(self, '_map'):
|
|
asyncore.dispatcher.__init__(self)
|
|
if connect:
|
|
self.create_socket(socket.AF_INET, socket.SOCK_STREAM)
|
|
self.connect((host, HTTPClient.port))
|
|
self.buffer = 'GET %s HTTP/1.0\r\n\r\n' % path
|
|
|
|
def handle_close(self):
|
|
# pylint: disable=global-statement
|
|
global requestCount
|
|
requestCount += 1
|
|
self.close()
|
|
|
|
def handle_read(self):
|
|
# print self.recv(8192)
|
|
self.recv(8192)
|
|
|
|
def writable(self):
|
|
return len(self.buffer) > 0
|
|
|
|
def handle_write(self):
|
|
sent = self.send(self.buffer)
|
|
self.buffer = self.buffer[sent:]
|
|
|
|
|
|
if __name__ == "__main__":
|
|
# initial fill
|
|
for i in range(parallel):
|
|
HTTPClient('127.0.0.1', '/')
|
|
start = time.time()
|
|
while time.time() - start < duration:
|
|
if len(asyncore.socket_map) < parallel:
|
|
for i in range(parallel - len(asyncore.socket_map)):
|
|
HTTPClient('127.0.0.1', '/')
|
|
print "Active connections: %i" % (len(asyncore.socket_map))
|
|
asyncore.loop(count=len(asyncore.socket_map) / 2)
|
|
if requestCount % 100 == 0:
|
|
print "Processed %i total messages" % (requestCount)
|