+## Copyright 2002-2004 Andrew Loewenstern, All Rights Reserved
+# see LICENSE.txt for license information
+
+from const import NULL_ID
from node import Node
-from twisted.internet.defer import Deferred
-from xmlrpcclient import XMLRPCClientFactory as factory
-from const import reactor
-class KNode(Node):
- def ping(self, sender):
- df = Deferred()
- f = factory('ping', (sender,), df.callback, df.errback)
- reactor.connectTCP(self.host, self.port, f)
- return df
- def findNode(self, target, sender):
- df = Deferred()
- f = factory('find_node', (target, sender), df.callback, df.errback)
- reactor.connectTCP(self.host, self.port, f)
- return df
- def storeValue(self, key, value, sender):
- df = Deferred()
- f = factory('store_value', (key, value, sender), df.callback, df.errback)
- reactor.connectTCP(self.host, self.port, f)
- return df
- def findValue(self, key, sender):
- f = factory('find_value', (key, sender), df.callback, df.errback)
- reactor.connectTCP(self.host, self.port, f)
- df = Deferred()
- return df
+class KNodeBase(Node):
+ def checkSender(self, dict):
+ try:
+ 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."
+ self.table.invalidateNode(self)
+
+ return dict
+
+ def errBack(self, err):
+ print ">>> ", err
+ return err
+
+ 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, id):
+ df = self.conn.sendRequest('find_node', {"target" : target, "id": id})
+ df.addErrback(self.errBack)
+ df.addCallback(self.checkSender)
+ return df
+
+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 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