Example #1
0
 def __init__(self, name, pipe_conn, batch_size, batch_rate, req_kind,
              buff_req, pool_config, send_mode):
     self._name = name
     self._stat = ClientStatistic()
     self._send_mode = send_mode
     self._buff_reqs = buff_req
     self._pipe_conn = pipe_conn
     self._pool_name = "pool_{}".format(random_string(24))
     self._loop = asyncio.new_event_loop()
     asyncio.set_event_loop(self._loop)
     self._pool_handle = None
     self._wallet_name = None
     self._wallet_handle = None
     self._test_did = None
     self._test_verk = None
     self._load_client_reqs = []
     self._loop.add_reader(self._pipe_conn, self.read_cb)
     self._closing = False
     self._batch_size = batch_size
     self._batch_rate = batch_rate
     self._gen_q = []
     self._send_q = []
     req_class, params = ReqTypeParser.create_req_generator(req_kind)
     self._req_generator = req_class(**params, client_stat=self._stat)
     assert self._req_generator is not None
     self._pool_config = json.dumps(pool_config) if isinstance(
         pool_config, dict) and pool_config else None
     if self._send_mode == LoadClient.SendResp and self._batch_size > 1:
         raise RuntimeError(
             "Batch size cannot be greater than 1 in response waiting mode")
     if self._send_mode == LoadClient.SendResp and buff_req != 0:
         raise RuntimeError(
             "Cannot pregenerate reqs in response waiting mode")
Example #2
0
 def __init__(self, name, pipe_conn, batch_size, batch_rate, req_kind,
              buff_req, pool_config, send_mode, short_stat, **kwargs):
     self._name = name
     self._stat = ClientStatistic(short_stat)
     self._send_mode = send_mode
     self._buff_reqs = buff_req
     self._pipe_conn = pipe_conn
     self._pool_name = "pool_{}".format(random_string(24))
     self._loop = asyncio.new_event_loop()
     asyncio.set_event_loop(self._loop)
     self._pool_handle = None
     self._wallet_name = None
     self._wallet_handle = None
     self._trustee_dids = []
     self._req_num_of_trustees = kwargs.get("trustees_num", 1)
     self._set_auth_rule = kwargs.get("set_auth_rules", False)
     self._test_did = None
     self._test_verk = None
     self._taa_text = None
     self._taa_version = None
     self._taa_time = None
     self._load_client_reqs = []
     self._loop.add_reader(self._pipe_conn, self.read_cb)
     self._closing = False
     self._batch_size = batch_size
     self._batch_rate = batch_rate
     self._auth_rule_metadata = {}
     self._gen_q = []
     self._send_q = []
     req_class, params = ReqTypeParser.create_req_generator(req_kind)
     self._req_generator = req_class(**params, client_stat=self._stat)
     assert self._req_generator is not None
     self._pool_config = json.dumps(pool_config) if isinstance(
         pool_config, dict) and pool_config else None
     if self._send_mode == LoadClient.SendResp and self._batch_size > 1:
         raise RuntimeError(
             "Batch size cannot be greater than 1 in response waiting mode")
     if self._send_mode == LoadClient.SendResp and buff_req != 0:
         raise RuntimeError(
             "Cannot pregenerate reqs in response waiting mode")
     self._logger = logging.getLogger(self._name)