PeerDiscovery.cpp

Go to the documentation of this file.

#include "PeerDiscovery.h"
#include "madara/utility/Utility.h"

namespace utility = madara::utility;
typedef madara::knowledge::KnowledgeRecord::Integer   Integer;

madara::filters::PeerDiscovery::PeerDiscovery (
  const std::string & prefix,
  knowledge::KnowledgeRecord::Integer heart_beat)
  : initialized_ (false), prefix_ (prefix),
    heart_beat_ (heart_beat), last_clear_ (0)
{
}

madara::filters::PeerDiscovery::~PeerDiscovery ()
{
}


void
madara::filters::PeerDiscovery::filter (
  knowledge::KnowledgeMap & /*records*/,
  const transport::TransportContext & transport_context,
  knowledge::Variables & vars)
{
  if (!initialized_)
  {
    peers_.set_name (prefix_, vars);
    initialized_ = true;
  }

  std::string originator (transport_context.get_originator ());
  Integer cur_time = (Integer)transport_context.get_current_time ();

  peers_.set (originator, cur_time);

  if (heart_beat_ > 0 && last_clear_ != cur_time)
  {
    std::vector <std::string> keys;
    peers_.sync_keys ();

    peers_.keys (keys);

    for (size_t i = 0; i < keys.size (); ++i)
    {
      // if the current originator is old, erase it
      if (cur_time - peers_[keys[i]].to_integer () > heart_beat_)
      {
        peers_.erase (keys[i]);
      }
    }
  }
}