apt_dht_Khashmir/knode.py

   1 ## Copyright 2002-2004 Andrew Loewenstern, All Rights Reserved
   2 # see LICENSE.txt for license information
   3
   4 from twisted.python import log
   5
   6 from node import Node, NULL_ID
   7
   8 class KNodeBase(Node):
   9     def checkSender(self, dict):
  10         try:
  11             senderid = dict['rsp']['id']
  12         except KeyError:
  13             log.msg("No peer id in response")
  14             raise Exception, "No peer id in response."
  15         else:
  16             if self.id != NULL_ID and senderid != self.id:
  17                 log.msg("Got response from different node than expected.")
  18                 self.table.invalidateNode(self)
  19
  20         return dict
  21
  22     def errBack(self, err):
  23         log.err(err)
  24         return err
  25
  26     def ping(self, id):
  27         df = self.conn.sendRequest('ping', {"id":id})
  28         df.addErrback(self.errBack)
  29         df.addCallback(self.checkSender)
  30         return df
  31
  32     def join(self, id):
  33         df = self.conn.sendRequest('join', {"id":id})
  34         df.addErrback(self.errBack)
  35         df.addCallback(self.checkSender)
  36         return df
  37
  38     def findNode(self, target, id):
  39         df = self.conn.sendRequest('find_node', {"target" : target, "id": id})
  40         df.addErrback(self.errBack)
  41         df.addCallback(self.checkSender)
  42         return df
  43
  44 class KNodeRead(KNodeBase):
  45     def findValue(self, key, id):
  46         df =  self.conn.sendRequest('find_value', {"key" : key, "id" : id})
  47         df.addErrback(self.errBack)
  48         df.addCallback(self.checkSender)
  49         return df
  50
  51 class KNodeWrite(KNodeRead):
  52     def storeValue(self, key, value, token, id):
  53         df = self.conn.sendRequest('store_value', {"key" : key, "value" : value, "token" : token, "id": id})
  54         df.addErrback(self.errBack)
  55         df.addCallback(self.checkSender)
  56         return df