def _get_validator_state(state, validator_id=None):
if validator_id is None:
address = _get_address('validator_map')
validator_state = ValidatorMap()
else:
validator_state = ValidatorInfo()
address = _get_address(validator_id)
try:
entries_list = state.get([address], timeout=STATE_TIMEOUT_SEC)
except FutureTimeoutError:
LOGGER.warning('Timeout occured on state.get([%s])', address)
raise InternalError('Unable to get {}'.format(address))
if len(entries_list) != 0:
validator_state.ParseFromString(entries_list[0].data)
return validator_state
def create_set_request_validator_map(self):
address = self._key_to_address("validator_map")
validator_map = ValidatorMap()
validator_map.entries.add(key=self.pubkey_hash, value=self.public_key)
data = validator_map.SerializeToString()
return self._factory.create_set_request({address: data})
def create_get_empty_resposne_validator_map(self):
address = self._key_to_address("validator_map")
data = ValidatorMap().SerializeToString()
return self._factory.create_get_response({address: data})