X-Git-Url: https://git.mxchange.org/?a=blobdiff_plain;f=knode.py;h=46ab780f5cf6cebdf37c96860ce74f1f0d968b89;hb=2852f1c4e2cc9afb3723ff04f4586e35281420fc;hp=3d9c4f1b3a5beb94638b191dd368f6c117532e1e;hpb=0b8fd85da9d9ec060effca7ce0911305d4d2b248;p=quix0rs-apt-p2p.git diff --git a/knode.py b/knode.py index 3d9c4f1..46ab780 100644 --- a/knode.py +++ b/knode.py @@ -1,4 +1,4 @@ -## Copyright 2002-2003 Andrew Loewenstern, All Rights Reserved +## Copyright 2002-2004 Andrew Loewenstern, All Rights Reserved # see LICENSE.txt for license information from node import Node @@ -10,32 +10,50 @@ class IDChecker: def __init__(id): self.id = id -class KNode(Node): +class KNodeBase(Node): def checkSender(self, dict): try: - senderid = dict['rsp']['sender']['id'] + senderid = dict['rsp']['id'] except KeyError: print ">>>> No peer id in response" raise Exception, "No peer id in response." else: if self.id != NULL_ID and senderid != self.id: print "Got response from different node than expected." - raise Exception, "Got response from different node than expected." + self.table.invalidateNode(self) + return dict + + def errBack(self, err): + print ">>> ", err + return err - def ping(self, sender): - df = self.conn.protocol.sendRequest('ping', {"sender":sender}) + def ping(self, id): + df = self.conn.sendRequest('ping', {"id":id}) + df.addErrback(self.errBack) df.addCallback(self.checkSender) return df - def findNode(self, target, sender): - df = self.conn.protocol.sendRequest('find_node', {"target" : target, "sender": sender}) + def findNode(self, target, id): + df = self.conn.sendRequest('find_node', {"target" : target, "id": id}) + df.addErrback(self.errBack) df.addCallback(self.checkSender) return df - def storeValue(self, key, value, sender): - df = self.conn.protocol.sendRequest('store_value', {"key" : key, "value" : value, "sender": sender}) + +class KNodeRead(KNodeBase): + def findValue(self, key, id): + df = self.conn.sendRequest('find_value', {"key" : key, "id" : id}) + df.addErrback(self.errBack) + df.addCallback(self.checkSender) + return df + +class KNodeWrite(KNodeRead): + def storeValue(self, key, value, id): + df = self.conn.sendRequest('store_value', {"key" : key, "value" : value, "id": id}) + df.addErrback(self.errBack) df.addCallback(self.checkSender) return df - def findValue(self, key, sender): - df = self.conn.protocol.sendRequest('find_value', {"key" : key, "sender" : sender}) + def storeValues(self, key, value, id): + df = self.conn.sendRequest('store_values', {"key" : key, "values" : value, "id": id}) + df.addErrback(self.errBack) df.addCallback(self.checkSender) return df