NetworkProximityService.h

 
#ifndef WRENCH_NETWORKPROXIMITYSERVICE_H
#define WRENCH_NETWORKPROXIMITYSERVICE_H
 
#include <cfloat>
#include <complex>
#include <random>
#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