GroupManagement.hpp

Go to the documentation of this file.

#ifndef GROUPMANAGEMENT_HPP_
#define GROUPMANAGEMENT_HPP_

#include "CBroker.hpp"
#include "IDGIModule.hpp"
#include "CPeerNode.hpp"
#include "PeerSets.hpp"

#include "messages/ModuleMessage.pb.h"

#include <boost/interprocess/sync/interprocess_mutex.hpp>
#include <boost/shared_ptr.hpp>
#include <boost/thread/mutex.hpp>

namespace freedm {

namespace broker {

namespace gm {

class GMAgent
  : public IDGIModule
{
  public:
    enum { NORMAL,DOWN,RECOVERY,REORGANIZATION,ELECTION };
    GMAgent();
    ~GMAgent();
    int Run();
    static PeerSet ProcessPeerList(const PeerListMessage& msg);

  private:
    void Recovery();
    bool IsCoordinator() const { return (Coordinator() == GetUUID()); };

    // Handlers
    void HandleIncomingMessage(boost::shared_ptr<const ModuleMessage> msg, CPeerNode peer);
    void HandlePeerList(const PeerListMessage& msg,CPeerNode peer);
    void HandleAccept(const AcceptMessage& msg,CPeerNode peer);
    void HandleAreYouCoordinator(const AreYouCoordinatorMessage& msg,CPeerNode peer);
    void HandleAreYouThere(const AreYouThereMessage& msg,CPeerNode peer);
    void HandleInvite(const InviteMessage& msg,CPeerNode peer);
    void HandleResponseAYC(const AreYouCoordinatorResponseMessage& msg,CPeerNode peer);
    void HandleResponseAYT(const AreYouThereResponseMessage& msg,CPeerNode peere);
    void HandlePeerListQuery(const PeerListQueryMessage& msg, CPeerNode peer);

    //Routines
    void Check( const boost::system::error_code& err );
    void Timeout( const boost::system::error_code& err );
    void Recovery( const boost::system::error_code& err );
    void Premerge( const boost::system::error_code &err );
    void Merge( const boost::system::error_code& err );
    void PushPeerList();

    // Messages
    ModuleMessage AreYouCoordinator();
    ModuleMessage Invitation();
    ModuleMessage AreYouCoordinatorResponse(std::string payload, int seq);
    ModuleMessage AreYouThereResponse(std::string payload,int seq);
    ModuleMessage Accept();
    ModuleMessage AreYouThere();
    ModuleMessage PeerList(std::string requester="all");
    static ModuleMessage PeerListQuery(std::string requester);

    //Peer Set Manipulation
    CPeerNode AddPeer(std::string uuid);
    CPeerNode AddPeer(CPeerNode peer);
    CPeerNode GetPeer(const std::string& uuid);

    int GetStatus() const;
    void SetStatus(int status);
    void InviteGroupNodes( const boost::system::error_code& err, PeerSet p_tempSet );
    void Reorganize( const boost::system::error_code& err );
    void SystemState();
    void StartMonitor(const boost::system::error_code& err);
    std::string Coordinator() const { return m_GroupLeader; }


    static ModuleMessage PrepareForSending(
        const GroupManagementMessage& message, std::string recipient = "gm");

    PeerSet m_UpNodes;
    PeerSet m_Coordinators;
    TimedPeerSet m_AYCResponse;
    TimedPeerSet m_AYTResponse;

    google::protobuf::uint32 m_GroupID;
    std::string  m_GroupLeader;
    unsigned int m_GrpCounter;

    /* IO and Timers */
    boost::asio::io_service m_localservice;
    CBroker::TimerHandle m_timer;
    CBroker::TimerHandle m_fidtimer;

    // Timeouts
    boost::posix_time::time_duration CHECK_TIMEOUT;
    boost::posix_time::time_duration TIMEOUT_TIMEOUT;
    boost::posix_time::time_duration FID_TIMEOUT;
    boost::posix_time::time_duration AYC_RESPONSE_TIMEOUT;
    boost::posix_time::time_duration AYT_RESPONSE_TIMEOUT;
    boost::posix_time::time_duration INVITE_RESPONSE_TIMEOUT;

    int m_groupsformed ;
    int m_groupsbroken ;
    int m_groupselection;
    int m_groupsjoined;
    int m_membership;
    int m_membershipchecks;
    int m_status;
    std::map< std::string , bool > m_fidstate;
};

} // namespace gm

} // namespace broker

} // namespace freedm

#endif