OmNetMediator.h

Go to the documentation of this file.

/*
 Copyright (2016) Michael Baer, Daniel Bonrath, All rights reserved.

 Licensed under the Apache License, Version 2.0 (the "License");
 you may not use this file except in compliance with the License.
 You may obtain a copy of the License at

 http://www.apache.org/licenses/LICENSE-2.0

 Unless required by applicable law or agreed to in writing, software
 distributed under the License is distributed on an "AS IS" BASIS,
 WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 See the License for the specific language governing permissions and
 limitations under the License.
 */

#ifndef __M2ETIS_SIM_OMNETMEDIATOR_H__
#define __M2ETIS_SIM_OMNETMEDIATOR_H__

#include <string>

#include "m2etis/config/GeneratedEventTypes.h"

#include "m2etis/message/MessageType.h"
#include "m2etis/message/NetworkMessage.h"

#include "m2etis/net/NetworkInterface.h"
#include "m2etis/net/NetworkCallbackInterface.h"
#include "m2etis/net/NetworkType.h"
#include "m2etis/net/NodeHandle.h"

#include "m2etis/sim/OmNetCallbackInterface.h"

namespace m2etis {
namespace sim {

        class PubSubWrapper;

        class OmNetMediator : public net::NetworkInterface<net::NetworkType<net::OMNET> > {
        public:
                OmNetMediator(const std::string & ip, const int port, const std::string & known_hostname, const int known_hostport) : _nodeHandle(new net::NodeHandle<net::NetworkType<net::OMNET>>), _callback(NULL), _ocallback(NULL), _name(ip), _hostName(known_hostname), _listenPort(port), _hostPort(known_hostport), _root(_hostName + std::string(":") + std::to_string(_hostPort)), _pubsub() {
                }

                ~OmNetMediator() {}

                void registerOmNetCallback(OmNetCallbackInterface * o);

                void setCallback(net::NetworkCallbackInterface<net::NetworkType<net::OMNET>> * cb);

                void send(const message::ActionType type, const std::string & to, const message::M2Message<SimulationEventType>::Ptr payload, net::NodeHandle<net::NetworkType<net::OMNET>>::Ptr_const hint, unsigned int size);

                void send(const net::NetworkType<net::OMNET>::Message::Ptr msg, net::NodeHandle<net::NetworkType<net::OMNET>>::Ptr_const hint);

                void send(const net::NetworkType<net::OMNET>::Message::Ptr msg);

                net::NodeHandle<net::NetworkType<net::OMNET> >::Ptr getSelfNodeHandle() const;

                void registerMessageType(const message::ActionType type, const bool ack) const;

                void registerMessageType(const net::NetworkType<net::OMNET>::MessageType type, const bool ack) const;

                bool forward(std::string & key, message::M2Message<SimulationEventType>::Ptr msg, const net::NodeHandle<net::NetworkType<net::OMNET>>::Ptr hint);

                void deliver(message::NetworkMessage<net::NetworkType<net::OMNET>>::Ptr msg);

                void update(const std::string & key, const net::NodeHandle<net::NetworkType<net::OMNET>>::Ptr_const handle, bool joined);

                std::string getSHA1(int channel_name);

                bool isResponsibleFor(const net::NetworkType<net::OMNET>::Key & key, const net::NetworkType<net::OMNET>::Key & node) const;

                net::NetworkType<net::OMNET>::Key getRoot() const { return _root; }

                void registerWrapper(PubSubWrapper * p) {
                        _pubsub = p;
                }

        private:
                net::NodeHandle<net::NetworkType<net::OMNET>>::Ptr _nodeHandle;
                net::NetworkCallbackInterface<net::NetworkType<net::OMNET>> * _callback;
                OmNetCallbackInterface * _ocallback;
                std::string _name;
                std::string _hostName;
                int _listenPort;
                int _hostPort;
                net::NetworkType<net::OMNET>::Key _root;
                PubSubWrapper * _pubsub;

                OmNetMediator(const OmNetMediator &);
                OmNetMediator & operator=(const OmNetMediator &);
        };

} /* namespace sim */
} /* namespace m2etis */

#endif /* __M2ETIS_SIM_OMNETMEDIATOR_H__ */