-
Notifications
You must be signed in to change notification settings - Fork 3
/
_protocol.py
48 lines (36 loc) · 1.16 KB
/
_protocol.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
"""
Base class for protocol benchmarks.
"""
from benchlib import Client
class Client(Client):
# override in subclasses, Protocol class:
_protocol = None
# override in subclasses, string to be passed to protocol:
_parseString = None
_bytes = 0
def _request(self):
s = self._parseString
for chunkSize in (16, 64, 256, 1024):
# deliver string to protocol with given chunk size:
p = self._protocol()
i = 0
while i < len(s):
p.dataReceived(s[i : i + chunkSize])
i += chunkSize
self._bytes += chunkSize
self._reactor.callLater(0.0, self._continue, None)
def _stop(self, reason):
if self._finished is not None:
finished = self._finished
self._finished = None
finished.callback(self._bytes)
def makeMain(protocol, string):
class ProtocolClient(Client):
_protocol = protocol
_parseString = string
def main(reactor, duration):
concurrency = 1
client = ProtocolClient(reactor)
d = client.run(concurrency, duration)
return d
return main