config.cc

   1 #include "config.h"
   2 #include "handle.h"
   3
   4 // The config module maintains views. As a node joins or leaves a
   5 // view, the next view will be the same as previous view, except with
   6 // the new node added or removed. The first view contains only node
   7 // 1. If node 2 joins after the first node (it will download the views
   8 // from node 1), it will learn about view 1 with the first node as the
   9 // only member.  It will then invoke Paxos to create the next view.
  10 // It will tell Paxos to ask the nodes in view 1 to agree on the value
  11 // {1, 2}.  If Paxos returns success, then it moves to view 2 with
  12 // {1,2} as the members. When node 3 joins, the config module runs
  13 // Paxos with the nodes in view 2 and the proposed value to be
  14 // {1,2,3}. And so on.  When a node discovers that some node of the
  15 // current view is not responding, it kicks off Paxos to propose a new
  16 // value (the current view minus the node that isn't responding). The
  17 // config module uses Paxos to create a total order of views, and it
  18 // is ensured that the majority of the previous view agrees to the
  19 // next view.  The Paxos log contains all the values (i.e., views)
  20 // agreed on.
  21 //
  22 // The RSM module informs config to add nodes. The config module
  23 // runs a heartbeater thread that checks in with nodes.  If a node
  24 // doesn't respond, the config module will invoke Paxos's proposer to
  25 // remove the node.  Higher layers will learn about this change when a
  26 // Paxos acceptor accepts the new proposed value through
  27 // paxos_commit().
  28 //
  29 // To be able to bring other nodes up to date to the latest formed
  30 // view, each node will have a complete history of all view numbers
  31 // and their values that it knows about. At any time a node can reboot
  32 // and when it re-joins, it may be many views behind; by remembering
  33 // all views, the other nodes can bring this re-joined node up to
  34 // date.
  35
  36 config::config(const string &_first, const string &_me, config_view_change *_vc)
  37     : my_view_id(0), first(_first), me(_me), vc(_vc),
  38       paxos(this, me == _first, me, me)
  39 {
  40     get_rpcs()->reg(paxos_protocol::heartbeat, &config::heartbeat, this);
  41     lock cfg_mutex_lock(cfg_mutex);
  42     reconstruct(cfg_mutex_lock);
  43     thread(&config::heartbeater, this).detach();
  44 }
  45
  46 void config::restore(const string &s) {
  47     lock cfg_mutex_lock(cfg_mutex);
  48     paxos.restore(s);
  49     reconstruct(cfg_mutex_lock);
  50 }
  51
  52 void config::get_view(unsigned instance, vector<string> &m) {
  53     lock cfg_mutex_lock(cfg_mutex);
  54     get_view(instance, m, cfg_mutex_lock);
  55 }
  56
  57 void config::get_view(unsigned instance, vector<string> &m, lock &) {
  58     string value = paxos.value(instance);
  59     LOG("get_view(" << instance << "): returns " << value);
  60     m = members(value);
  61 }
  62
  63 vector<string> config::members(const string &value) const {
  64     return explode(value);
  65 }
  66
  67 string config::value(const vector<string> &members) const {
  68     return implode(members);
  69 }
  70
  71 void config::reconstruct(lock &cfg_mutex_lock) {
  72     VERIFY(cfg_mutex_lock);
  73     my_view_id = paxos.instance();
  74     if (my_view_id > 0) {
  75         get_view(my_view_id, mems, cfg_mutex_lock);
  76         LOG("view " << my_view_id << " " << mems);
  77     }
  78 }
  79
  80 // Called by Paxos's acceptor.
  81 void config::paxos_commit(unsigned instance, const string &value) {
  82     lock cfg_mutex_lock(cfg_mutex);
  83
  84     vector<string> newmem = members(value);
  85     LOG("instance " << instance << ": " << newmem);
  86
  87     for (auto mem : mems) {
  88         LOG("is " << mem << " still a member?");
  89         if (!isamember(mem, newmem) && me != mem) {
  90             LOG("delete " << mem);
  91             invalidate_handle(mem);
  92         }
  93     }
  94
  95     mems = newmem;
  96     my_view_id = instance;
  97     if (vc) {
  98         cfg_mutex_lock.unlock();
  99         vc->commit_change(instance);
 100         cfg_mutex_lock.lock();
 101     }
 102 }
 103
 104 bool config::ismember(const string &m, unsigned vid) {
 105     lock cfg_mutex_lock(cfg_mutex);
 106     vector<string> v;
 107     get_view(vid, v, cfg_mutex_lock);
 108     return isamember(m, v);
 109 }
 110
 111 bool config::add(const string &new_m, unsigned vid) {
 112     lock cfg_mutex_lock(cfg_mutex);
 113     LOG("adding " << new_m << " to " << vid);
 114     if (vid != my_view_id) {
 115         LOG("that's not my view id, " << my_view_id << "!");
 116         return false;
 117     }
 118     LOG("calling down to paxos layer");
 119     vector<string> m(mems), cmems(mems);
 120     m.push_back(new_m);
 121     LOG("old mems " << cmems << " " << value(cmems));
 122     LOG("new mems " << m << " " << value(m));
 123     unsigned nextvid = my_view_id + 1;
 124     cfg_mutex_lock.unlock();
 125     bool r = paxos.run(nextvid, cmems, value(m));
 126     cfg_mutex_lock.lock();
 127     LOG("paxos proposer returned " << (r ? "success" : "failure"));
 128     return r;
 129 }
 130
 131 // caller should hold cfg_mutex
 132 bool config::remove(const string &m, lock &cfg_mutex_lock) {
 133     LOG("my_view_id " << my_view_id << " remove? " << m);
 134     vector<string> n;
 135     for (auto mem : mems) {
 136         if (mem != m)
 137             n.push_back(mem);
 138     }
 139     vector<string> cmems = mems;
 140     unsigned nextvid = my_view_id + 1;
 141     cfg_mutex_lock.unlock();
 142     bool r = paxos.run(nextvid, cmems, value(n));
 143     cfg_mutex_lock.lock();
 144     LOG("proposer returned " << (r ? "success" : "failure"));
 145     return r;
 146 }
 147
 148 void config::heartbeater() [[noreturn]] {
 149     lock cfg_mutex_lock(cfg_mutex);
 150
 151     while (1) {
 152         auto next_timeout = steady_clock::now() + seconds(3);
 153         LOG("go to sleep");
 154         config_cond.wait_until(cfg_mutex_lock, next_timeout);
 155
 156         unsigned vid = my_view_id;
 157         vector<string> cmems;
 158         get_view(vid, cmems, cfg_mutex_lock);
 159         LOG("current membership " << cmems);
 160
 161         if (!isamember(me, cmems)) {
 162             LOG("not member yet; skip hearbeat");
 163             continue;
 164         }
 165
 166         // who has the smallest ID?
 167         string m = min(me, *min_element(cmems.begin(), cmems.end()));
 168
 169         if (m == me) {
 170             // ping the other nodes
 171             for (string mem : cmems) {
 172                 if (mem == me || doheartbeat(mem, cfg_mutex_lock) == OK)
 173                     continue;
 174                 if (vid == my_view_id)
 175                     remove(mem, cfg_mutex_lock);
 176                 break;
 177             }
 178         } else {
 179             // ping the node with the smallest ID
 180             if (doheartbeat(m, cfg_mutex_lock) != OK && vid == my_view_id)
 181                 remove(m, cfg_mutex_lock);
 182         }
 183     }
 184 }
 185
 186 paxos_protocol::status config::heartbeat(int &r, string m, unsigned vid) {
 187     lock cfg_mutex_lock(cfg_mutex);
 188     r = (int) my_view_id;
 189     LOG("heartbeat from " << m << "(" << vid << ") my_view_id " << my_view_id);
 190     if (vid == my_view_id)
 191         return paxos_protocol::OK;
 192     else if (paxos.isrunning()) {
 193         VERIFY (vid == my_view_id + 1 || vid + 1 == my_view_id);
 194         return paxos_protocol::OK;
 195     }
 196     return paxos_protocol::ERR;
 197 }
 198
 199 config::heartbeat_t config::doheartbeat(const string &m, lock &cfg_mutex_lock) {
 200     unsigned vid = my_view_id;
 201     LOG("heartbeat to " << m << " (" << vid << ")");
 202     handle h(m);
 203
 204     cfg_mutex_lock.unlock();
 205     int r = 0, ret = rpc_const::bind_failure;
 206     if (rpcc *cl = h.safebind())
 207         ret = cl->call_timeout(paxos_protocol::heartbeat, rpcc::to(1000), r, me, vid);
 208     cfg_mutex_lock.lock();
 209
 210     heartbeat_t res = OK;
 211     switch (ret) {
 212         case paxos_protocol::OK:
 213             break;
 214         case rpc_const::atmostonce_failure:
 215         case rpc_const::oldsrv_failure:
 216             invalidate_handle(m);
 217             break;
 218         default:
 219             LOG("problem with " << m << " (" << ret << ") my vid " << vid << " his vid " << r);
 220             res = (ret < 0) ? FAILURE : VIEWERR;
 221     }
 222     LOG("done " << res);
 223     return res;
 224 }