NetworkProximityService.h

 
#ifndef WRENCH_NETWORKPROXIMITYSERVICE_H
#define WRENCH_NETWORKPROXIMITYSERVICE_H
 
#include <complex>
#include <random>
#include <cfloat>
#include "wrench/services/Service.h"
#include "wrench/services/network_proximity/NetworkProximityServiceProperty.h"
#include "wrench/services/network_proximity/NetworkProximityDaemon.h"
 
namespace wrench{
 
    class NetworkProximityService: public Service {
 
    private:
        std::map<std::string, std::string> default_property_values = {
                {NetworkProximityServiceProperty::LOOKUP_OVERHEAD,                            "0.0"},
                {NetworkProximityServiceProperty::NETWORK_PROXIMITY_SERVICE_TYPE,             "ALLTOALL"},
                {NetworkProximityServiceProperty::NETWORK_PROXIMITY_MESSAGE_SIZE,             "1024"},
                {NetworkProximityServiceProperty::NETWORK_PROXIMITY_MEASUREMENT_PERIOD,       "60"},
                {NetworkProximityServiceProperty::NETWORK_PROXIMITY_MEASUREMENT_PERIOD_MAX_NOISE, "20"},
                {NetworkProximityServiceProperty::NETWORK_DAEMON_COMMUNICATION_COVERAGE,    "1.0"},
                {NetworkProximityServiceProperty::NETWORK_PROXIMITY_PEER_LOOKUP_SEED, "1"}
        };
 
        std::map<std::string, double> default_messagepayload_values = {
                {NetworkProximityServiceMessagePayload::STOP_DAEMON_MESSAGE_PAYLOAD,                1024},
                {NetworkProximityServiceMessagePayload::DAEMON_STOPPED_MESSAGE_PAYLOAD,             1024},
                {NetworkProximityServiceMessagePayload::NETWORK_DAEMON_MEASUREMENT_REPORTING_PAYLOAD,  1024},
                {NetworkProximityServiceMessagePayload::NETWORK_DB_LOOKUP_REQUEST_MESSAGE_PAYLOAD,  1024},
                {NetworkProximityServiceMessagePayload::NETWORK_DB_LOOKUP_ANSWER_MESSAGE_PAYLOAD,   1024},
                {NetworkProximityServiceMessagePayload::NETWORK_DAEMON_CONTACT_REQUEST_PAYLOAD,      1024},
                {NetworkProximityServiceMessagePayload::NETWORK_DAEMON_CONTACT_ANSWER_PAYLOAD,      1024},
        };
 
    public:
 
 
 
        /***********************/
        /***********************/
 
        static constexpr double NOT_AVAILABLE = DBL_MAX;
 
        ~NetworkProximityService() override;
 
        /***********************/
        /***********************/
 
        NetworkProximityService(std::string db_hostname,
                                std::vector<std::string> hosts_in_network,
                                std::map<std::string, std::string> property_list = {},
                                std::map<std::string, double> messagepayload_list = {}
        );
 
        /***********************/
        /***********************/
 
 
        std::vector<std::string> getHostnameList();
 
        std::pair<double, double> getHostPairDistance(std::pair<std::string, std::string> hosts);
        std::pair<std::pair<double, double>, double>  getHostCoordinate(std::string);
 
        std::string getNetworkProximityServiceType();
 
        /***********************/
        /***********************/
 
    private:
 
        friend class Simulation;
 
        std::vector<std::shared_ptr<NetworkProximityDaemon>> network_daemons;
        std::vector<std::string> hosts_in_network;
 
        std::default_random_engine master_rng;
 
        int main() override;
 
        bool processNextMessage();
 
        void addEntryToDatabase(std::pair<std::string,std::string> pair_hosts,double proximity_value);
 
        std::map<std::pair<std::string,std::string>,std::pair<double,double>> entries;
 
        std::map<std::string, std::pair<std::complex<double>, double>> coordinate_lookup_table;
 
        std::shared_ptr<NetworkProximityDaemon> getCommunicationPeer(const std::shared_ptr<NetworkProximityDaemon>  sender_daemon);
 
        void vivaldiUpdate(double proximityValue, std::string sender_hostname, std::string peer_hostname);
 
        void validateProperties();
 
    };
}
 
 
 
 
#endif //WRENCH_NETWORKPROXIMITYSERVICE_H