examples/s4u/dht-chord/s4u-dht-chord.hpp

   1 /* Copyright (c) 2016-2017. The SimGrid Team. All rights reserved.          */
   2
   3 /* This program is free software; you can redistribute it and/or modify it
   4  * under the terms of the license (GNU LGPL) which comes with this package. */
   5
   6 #ifndef S4U_CHORD_HPP
   7 #define S4U_CHORD_HPP
   8 #include "simgrid/s4u.hpp"
   9 #include <string>
  10 #include <xbt/RngStream.h>
  11 #include <xbt/ex.hpp>
  12 #include <xbt/str.h>
  13
  14 #define MAX_SIMULATION_TIME 1000
  15 #define PERIODIC_STABILIZE_DELAY 20
  16 #define PERIODIC_FIX_FINGERS_DELAY 120
  17 #define PERIODIC_CHECK_PREDECESSOR_DELAY 120
  18 #define PERIODIC_LOOKUP_DELAY 10
  19 #define SLEEP_DELAY 4.9999
  20
  21 extern int nb_bits;
  22 extern int nb_keys;
  23 extern int timeout;
  24 extern int* powers2;
  25
  26 class HostChord {
  27   RngStream stream_;
  28   simgrid::s4u::Host* host = nullptr;
  29
  30 public:
  31   static simgrid::xbt::Extension<simgrid::s4u::Host, HostChord> EXTENSION_ID;
  32
  33   explicit HostChord(simgrid::s4u::Host* ptr) : host(ptr)
  34   {
  35     std::string descr = std::string("RngSream<") + host->getCname() + ">";
  36     stream_           = RngStream_CreateStream(descr.c_str());
  37   }
  38
  39   ~HostChord() { RngStream_DeleteStream(&stream_); };
  40
  41   RngStream getStream() { return stream_; };
  42 };
  43
  44 /* Types of tasks exchanged between nodes. */
  45 typedef enum {
  46   FIND_SUCCESSOR,
  47   FIND_SUCCESSOR_ANSWER,
  48   GET_PREDECESSOR,
  49   GET_PREDECESSOR_ANSWER,
  50   NOTIFY,
  51   SUCCESSOR_LEAVING,
  52   PREDECESSOR_LEAVING,
  53   PREDECESSOR_ALIVE,
  54   PREDECESSOR_ALIVE_ANSWER
  55 } e_message_type_t;
  56
  57 class ChordMessage {
  58 public:
  59   e_message_type_t type;              // type of message
  60   std::string issuer_host_name;       // used for logging
  61   int request_id     = -1;            // id (used by some types of messages)
  62   int request_finger = 1;             // finger parameter (used by some types of messages)
  63   int answer_id      = -1;            // answer (used by some types of messages)
  64   simgrid::s4u::MailboxPtr answer_to; // mailbox to send an answer to (if any)
  65
  66   explicit ChordMessage(e_message_type_t type)
  67       : type(type), issuer_host_name(simgrid::s4u::this_actor::getHost()->getName())
  68   {
  69   }
  70
  71   ~ChordMessage() = default;
  72
  73   static void destroy(void* message);
  74 };
  75
  76 class Node {
  77   int known_id_      = -1;
  78   double start_time_ = -1;
  79   double deadline_   = -1;
  80   bool joined        = false;
  81   int id_;                           // my id
  82   int pred_id_ = -1;                 // predecessor id
  83   simgrid::s4u::MailboxPtr mailbox_; // my mailbox
  84   int* fingers_;                     // finger table,(fingers[0] is my successor)
  85   int next_finger_to_fix;            // index of the next finger to fix in fix_fingers()
  86   RngStream stream;
  87
  88 public:
  89   explicit Node(std::vector<std::string> args);
  90   Node(const Node&) = delete;
  91   ~Node();
  92   void join(int known_id);
  93   void leave();
  94   void notifyAndQuit();
  95
  96   void randomLookup();
  97   void setFinger(int finger_index, int id);
  98   void fixFingers();
  99   void printFingerTable();
 100
 101   void setPredecessor(int predecessor_id);
 102   void checkPredecessor();
 103   int remoteGetPredecessor(int ask_to);
 104   int closestPrecedingFinger(int id);
 105   int findSuccessor(int id);
 106   int remoteFindSuccessor(int ask_to, int id);
 107
 108   void notify(int predecessor_candidate_id);
 109   void remoteNotify(int notify_id, int predecessor_candidate_id);
 110   void stabilize();
 111   void handleMessage(ChordMessage* message);
 112
 113   void operator()()
 114   {
 115     simgrid::s4u::this_actor::sleep_for(start_time_);
 116     if (known_id_ == -1) {
 117       setPredecessor(-1); // -1 means that I have no predecessor
 118       printFingerTable();
 119       joined = true;
 120     } else {
 121       join(known_id_);
 122     }
 123
 124     if (not joined)
 125       return;
 126     void* data                         = nullptr;
 127     double now                         = simgrid::s4u::Engine::getClock();
 128     double next_stabilize_date         = start_time_ + PERIODIC_STABILIZE_DELAY;
 129     double next_fix_fingers_date       = start_time_ + PERIODIC_FIX_FINGERS_DELAY;
 130     double next_check_predecessor_date = start_time_ + PERIODIC_CHECK_PREDECESSOR_DELAY;
 131     double next_lookup_date            = start_time_ + PERIODIC_LOOKUP_DELAY;
 132     simgrid::s4u::CommPtr comm_receive = nullptr;
 133     while ((now < (start_time_ + deadline_)) && now < MAX_SIMULATION_TIME) {
 134       if (comm_receive == nullptr)
 135         comm_receive = mailbox_->get_async(&data);
 136       while ((now < (start_time_ + deadline_)) && now < MAX_SIMULATION_TIME && not comm_receive->test()) {
 137         // no task was received: make some periodic calls
 138         if (now >= next_stabilize_date) {
 139           stabilize();
 140           next_stabilize_date = simgrid::s4u::Engine::getClock() + PERIODIC_STABILIZE_DELAY;
 141         } else if (now >= next_fix_fingers_date) {
 142           fixFingers();
 143           next_fix_fingers_date = simgrid::s4u::Engine::getClock() + PERIODIC_FIX_FINGERS_DELAY;
 144         } else if (now >= next_check_predecessor_date) {
 145           checkPredecessor();
 146           next_check_predecessor_date = simgrid::s4u::Engine::getClock() + PERIODIC_CHECK_PREDECESSOR_DELAY;
 147         } else if (now >= next_lookup_date) {
 148           randomLookup();
 149           next_lookup_date = simgrid::s4u::Engine::getClock() + PERIODIC_LOOKUP_DELAY;
 150         } else {
 151           // nothing to do: sleep for a while
 152           simgrid::s4u::this_actor::sleep_for(SLEEP_DELAY);
 153         }
 154         now = simgrid::s4u::Engine::getClock();
 155       }
 156
 157       if (data != nullptr) {
 158         ChordMessage* message = static_cast<ChordMessage*>(data);
 159         handleMessage(message);
 160         comm_receive = nullptr;
 161         data         = nullptr;
 162       }
 163       now = simgrid::s4u::Engine::getClock();
 164     }
 165     if (comm_receive != nullptr) {
 166       if (comm_receive->test())
 167         delete static_cast<ChordMessage*>(data);
 168       else
 169         comm_receive->cancel();
 170     }
 171     // leave the ring
 172     leave();
 173   }
 174 };
 175
 176 #endif