-
Notifications
You must be signed in to change notification settings - Fork 1
/
locust_connect_set_parallel.py
37 lines (26 loc) · 1 KB
/
locust_connect_set_parallel.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
# A "locustfile" which tries to determine the impact of session
# creations (caused by Connect) on the rate of "normal" operations
# (from Set, in this case). The 'weight' attribute is used to
# configure a ~1:10 ratio.
import logging
from zk_locust import ZKLocust
from zk_locust.task_sets import ZKConnectTaskSet, ZKSetTaskSet
from locust_extra.stats import register_extra_stats
from zk_metrics import register_zk_metrics
logging.basicConfig()
logging.getLogger('locust_extra.control').setLevel(logging.DEBUG)
logging.getLogger('zk_metrics').setLevel(logging.DEBUG)
_logger = logging.getLogger(__name__)
_logger.setLevel(logging.DEBUG)
register_zk_metrics()
register_extra_stats()
class Connect(ZKLocust):
weight = 1
task_set = ZKConnectTaskSet
def __init__(self):
# Unlike other locust instances, this one must not "autostart"
# the ZK client.
super(Connect, self).__init__(pseudo_root=None, autostart=False)
class Set(ZKLocust):
weight = 10
task_set = ZKSetTaskSet