dhash_store.h

来自「基于DHT的对等协议」· C头文件 代码 · 共 85 行

#ifndef __DHASH_STORE__H_
#define __DHASH_STORE__H_

#include <dhash_prot.h>
#include <dhash_common.h>
#include <dhash.h>
#include <route.h>
#include <chord.h>

typedef callback<void, dhash_stat, chordID, bool>::ref cbclistore_t;

//store a block/fragment by sending RPCs smaller than the MTU
class dhash_store  {

public:
  
  static void execute (ptr<vnode> clntnode, ptr<location> dest, 
		       blockID bid,
		       str data, 
		       u_int32_t exp,
		       cbclistore_t cb, 
		       store_status store_type = DHASH_STORE,
		       int nonce = 0)
  {
    vNew dhash_store (clntnode, dest, bid, data, exp, store_type, nonce, cb);
  }


protected:
  uint npending;
  bool error;
  dhash_stat status;

  ptr<location> dest;
  str data;
  u_int32_t expiration;
  blockID bid;
  cbclistore_t cb;
  dhash_ctype ctype;
  int procno;
  u_int nonce;
  store_status store_type;
  ptr<vnode> clntnode;

  int num_retries;
  timecb_t *dcb;
  bool present;

  ptr<bool> deleted;

  dhash_store (ptr<vnode> clntnode, ptr<location> dest, blockID bid,
               str _block, u_int32_t exp,
	       store_status store_type, int nonce,
	       cbclistore_t cb)
    : npending (0),
      error (false),
      status (DHASH_OK),
      dest (dest), data (_block), expiration (exp), bid (bid), cb (cb),
      ctype (bid.ctype), 
      procno ((nonce > 0) ? DHASHPROC_FETCHCOMPLETE : DHASHPROC_STORE), 
      nonce (nonce),
      store_type (store_type),
      clntnode (clntnode), num_retries (0),
      dcb (NULL),
      present (false),
      deleted (New refcounted<bool> (false))
  {
    start (deleted);
  }
  

  ~dhash_store ()
  {
    if (dcb) timecb_remove (dcb);
  }

  void done (bool present);
  void timed_out (ptr<bool> deleted);
  void start (ptr<bool> deleted);
  void finish (ptr<bool> deleted, ptr<dhash_storeres> res, int num, clnt_stat err);
  void store (char *buf, size_t len, size_t off, int num);  
};

#endif