def test_josn_support(self):
'''It should convert to json and back'''
kvp_info_list = KVPExchangeInfoList(
[KVPExchangeInfo(KeyBytes(), KeyBytes(), 123, 456)])
self.assertEqual(kvp_info_list,
KVPExchangeInfoList.from_json_loadable(
kvp_info_list.to_json_dumpable()))
def _received_find_value_rpc(self, data_packet):
'''Find value rpc callback'''
_logger.debug('Find value %s←%s', self.address,
data_packet.address)
key = KeyBytes.new_silent(data_packet.dict_obj.get(JSONKeys.KEY,
'fake'))
index = KeyBytes.new_silent(data_packet.dict_obj.get(JSONKeys.INDEX))
if not key:
_logger.debug('Find value %s←%s bad key', self.address,
data_packet.address)
return
_logger.debug('Find value %s←%s k=%s i=%s', self.address,
data_packet.address, key, index)
kvpid = KVPID(key, index)
if index and kvpid in self._kvp_table:
kvp_record = self._kvp_table.record(kvpid)
d = self._template_dict()
d[JSONKeys.VALUES] = KVPExchangeInfoList([
KVPExchangeInfo.from_kvp_record(kvp_record)
]).to_json_dumpable()
self._network.send_answer_reply(data_packet, d)
elif self._kvp_table.indices(key):
kvp_record_list = self._kvp_table.records_by_key(key)
d = self._template_dict()
d[JSONKeys.VALUES] = KVPExchangeInfoList.from_kvp_record_list(
kvp_record_list).to_json_dumpable()
self._network.send_answer_reply(data_packet, d)
else:
self._reply_find_node(data_packet, key)
def run(self, controller, node, key, index):
d = controller._template_dict()
d[JSONKeys.RPC] = JSONKeys.RPCs.FIND_VALUE
d[JSONKeys.KEY] = key.base64
if index:
d[JSONKeys.INDEX] = index.base64
future = controller._network.send(node.address, d, timeout=True)
data_packet = future.result()
if not data_packet:
_logger.debug('Find value timeout %s←%s', controller.node, node)
return
controller._update_routing_table_from_data_packet(data_packet)
dict_obj = data_packet.dict_obj
if JSONKeys.VALUES in dict_obj:
kvp_info_list = KVPExchangeInfoList.from_json_loadable(
dict_obj[JSONKeys.VALUES])
_logger.debug('Find value %s←%s dictlen=%d',
controller.node, node, len(kvp_info_list))
return FindValueFromNodeResult(kvp_info_list, None)
elif JSONKeys.NODES in dict_obj:
try:
nodes = NodeList.from_json_loadable(dict_obj[JSONKeys.NODES])
except ValueError as e:
_logger.debug('Find value node invalid %s←%s err=%s',
controller.node, node, e)
else:
_logger.debug('Find value nodes %s←%s len=%d',
controller.node, node, len(nodes))
return FindValueFromNodeResult(None, nodes)
