StateCollection.hpp

Go to the documentation of this file.

#ifndef CSTATECOLLECTION_HPP_
#define CSTATECOLLECTION_HPP_

#include "PeerSets.hpp"
#include "IDGIModule.hpp"
#include "CPeerNode.hpp"
#include "messages/ModuleMessage.pb.h"

#include <map>
#include <memory>
#include <vector>

#include <boost/shared_ptr.hpp>
#include <boost/property_tree/ptree.hpp>

namespace freedm
{

namespace broker
{

namespace sc
{

using boost::property_tree::ptree;


class SCAgent
      : public IDGIModule
{
    public:
        SCAgent();

    private:
        //Marker structure
        typedef std::pair< std::string, int >  StateVersion;

        //Handler
        void HandleAccept(CPeerNode peer);
        void HandlePeerList(const gm::PeerListMessage& msg, CPeerNode peer);
        void HandleRequest(const RequestMessage& msg, CPeerNode peer);
        void HandleMarker(const MarkerMessage& msg, CPeerNode peer);
        void HandleState(const StateMessage& msg, CPeerNode peer);
        void HandleIncomingMessage(boost::shared_ptr<const ModuleMessage> msg, CPeerNode peer);

        //Internal
        void    Initiate();
        void    TakeSnapshot(const std::vector<std::string>& devicelist);
        void    SendStateBack();
        void    StateResponse();
        void    SaveForward(StateVersion latest, const MarkerMessage& msg);

        //Peer set operations
        CPeerNode AddPeer(CPeerNode peer);
        CPeerNode GetPeer(std::string uuid);

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

        std::multimap<StateVersion, StateMessage> collectstate;
        std::multimap<StateVersion, StateMessage>::iterator it;

        unsigned int m_countstate;
        unsigned int m_countmarker;
        unsigned int m_countdone;

        std::string m_scleader;

        bool m_NotifyToSave;

        std::string m_module;

        //For multidevices state collection the following variables have to be changed
        std::vector<std::string> m_device;

        StateVersion        m_curversion;
        StateMessage m_curstate;

        PeerSet m_AllPeers;
};

} // namespace sc

} // namespace broker

} // namespace freedm

#endif