ipp.cpp 7.6 KB
Newer Older
Millian Poquet's avatar
Millian Poquet committed
1 2
/**
 * @file ipp.cpp
Millian Poquet's avatar
Millian Poquet committed
3
 * @brief Inter-Process Protocol (within Batsim, not with the Decision real process)
Millian Poquet's avatar
Millian Poquet committed
4 5
 */

6 7 8 9 10 11
#include "ipp.hpp"

#include <simgrid/msg.h>

using namespace std;

Millian Poquet's avatar
Millian Poquet committed
12
XBT_LOG_NEW_DEFAULT_CATEGORY(ipp, "ipp"); //!< Logging
13

Millian Poquet's avatar
Millian Poquet committed
14 15 16 17
void generic_send_message(const std::string & destination_mailbox,
                          IPMessageType type,
                          void * data,
                          bool detached)
18 19 20 21 22 23 24
{
    IPMessage * message = new IPMessage;
    message->type = type;
    message->data = data;

    msg_task_t task_to_send = MSG_task_create(NULL, 0, 1e-6, message);

25 26
    XBT_DEBUG("message from '%s' to '%s' of type '%s' with data %p",
              MSG_process_get_name(MSG_process_self()), destination_mailbox.c_str(),
Millian Poquet's avatar
Millian Poquet committed
27
              ip_message_type_to_string(type).c_str(), data);
28

29
    if (detached)
30
    {
31
        MSG_task_dsend(task_to_send, destination_mailbox.c_str(), NULL);
32
    }
33
    else
34 35 36 37 38 39 40
    {
        msg_error_t err = MSG_task_send(task_to_send, destination_mailbox.c_str());
        xbt_assert(err == MSG_OK,
                   "Sending message from '%s' to '%s' of type '%s' with data %p FAILED!",
                   MSG_process_get_name(MSG_process_self()), destination_mailbox.c_str(),
                   ip_message_type_to_string(type).c_str(), data);
    }
41

42 43
    XBT_DEBUG("message from '%s' to '%s' of type '%s' with data %p done",
              MSG_process_get_name(MSG_process_self()), destination_mailbox.c_str(),
Millian Poquet's avatar
Millian Poquet committed
44
              ip_message_type_to_string(type).c_str(), data);
45 46 47 48
}

void send_message(const std::string & destination_mailbox, IPMessageType type, void * data)
{
49
    generic_send_message(destination_mailbox, type, data, false);
50 51 52 53
}

void dsend_message(const std::string & destination_mailbox, IPMessageType type, void * data)
{
54
    generic_send_message(destination_mailbox, type, data, true);
55 56
}

Millian Poquet's avatar
Millian Poquet committed
57
std::string ip_message_type_to_string(IPMessageType type)
58
{
59 60 61
    string s;

    switch(type)
62
    {
63 64 65
        case IPMessageType::JOB_SUBMITTED:
            s = "JOB_SUBMITTED";
            break;
66 67 68
        case IPMessageType::JOB_SUBMITTED_BY_DP:
            s = "JOB_SUBMITTED_BY_DP";
            break;
69 70 71 72 73 74
        case IPMessageType::JOB_COMPLETED:
            s = "JOB_COMPLETED";
            break;
        case IPMessageType::PSTATE_MODIFICATION:
            s = "PSTATE_MODIFICATION";
            break;
Millian Poquet's avatar
Millian Poquet committed
75 76
        case IPMessageType::SCHED_EXECUTE_JOB:
            s = "SCHED_EXECUTE_JOB";
77
            break;
Millian Poquet's avatar
Millian Poquet committed
78 79
        case IPMessageType::SCHED_REJECT_JOB:
            s = "SCHED_REJECT_JOB";
80
            break;
81 82 83
        case IPMessageType::SCHED_KILL_JOB:
            s = "SCHED_KILL_JOB";
            break;
Millian Poquet's avatar
Millian Poquet committed
84 85
        case IPMessageType::SCHED_CALL_ME_LATER:
            s = "SCHED_CALL_ME_LATER";
86
            break;
87 88 89
        case IPMessageType::SCHED_TELL_ME_ENERGY:
            s = "SCHED_TELL_ME_ENERGY";
            break;
Pierre-François's avatar
Pierre-François committed
90 91 92 93 94 95
        case IPMessageType::SCHED_WAIT_ANSWER:
            s = "SCHED_WAIT_ANSWER";
            break;
        case IPMessageType::WAIT_QUERY:
            s = "WAIT_QUERY";
            break;
96 97 98 99 100 101 102 103 104
        case IPMessageType::SCHED_READY:
            s = "SCHED_READY";
            break;
        case IPMessageType::WAITING_DONE:
            s = "WAITING_DONE";
            break;
        case IPMessageType::SUBMITTER_HELLO:
            s = "SUBMITTER_HELLO";
            break;
105 106 107
        case IPMessageType::SUBMITTER_CALLBACK:
            s = "SUBMITTER_CALLBACK";
            break;
108 109 110 111 112 113 114 115 116
        case IPMessageType::SUBMITTER_BYE:
            s = "SUBMITTER_BYE";
            break;
        case IPMessageType::SWITCHED_ON:
            s = "SWITCHED_ON";
            break;
        case IPMessageType::SWITCHED_OFF:
            s = "SWITCHED_OFF";
            break;
117
        case IPMessageType::KILLING_DONE:
Millian Poquet's avatar
Millian Poquet committed
118
            s = "KILLING_DONE";
119
            break;
120 121
        case IPMessageType::END_DYNAMIC_SUBMIT:
            s = "END_DYNAMIC_SUBMIT";
122
    }
123

124
    return s;
125
}
126 127 128 129 130 131

void send_message(const char *destination_mailbox, IPMessageType type, void *data)
{
    const string str = destination_mailbox;
    send_message(str, type, data);
}
132

133 134 135 136 137 138
void dsend_message(const char *destination_mailbox, IPMessageType type, void *data)
{
    const string str = destination_mailbox;
    dsend_message(str, type, data);
}

139 140 141 142 143 144 145 146 147
IPMessage::~IPMessage()
{
    switch (type)
    {
        case IPMessageType::JOB_SUBMITTED:
        {
            JobSubmittedMessage * msg = (JobSubmittedMessage *) data;
            delete msg;
        } break;
148 149 150 151 152
        case IPMessageType::JOB_SUBMITTED_BY_DP:
        {
            JobSubmittedByDPMessage * msg = (JobSubmittedByDPMessage *) data;
            delete msg;
        } break;
153 154 155 156 157 158 159 160 161 162
        case IPMessageType::JOB_COMPLETED:
        {
            JobCompletedMessage * msg = (JobCompletedMessage *) data;
            delete msg;
        } break;
        case IPMessageType::PSTATE_MODIFICATION:
        {
            PStateModificationMessage * msg = (PStateModificationMessage *) data;
            delete msg;
        } break;
Millian Poquet's avatar
Millian Poquet committed
163
        case IPMessageType::SCHED_EXECUTE_JOB:
164
        {
165
            ExecuteJobMessage * msg = (ExecuteJobMessage *) data;
166
            // The Allocations themselves are not memory-deallocated there but at the end of the job execution.
167 168
            delete msg;
        } break;
Millian Poquet's avatar
Millian Poquet committed
169
        case IPMessageType::SCHED_REJECT_JOB:
170 171 172 173
        {
            JobRejectedMessage * msg = (JobRejectedMessage *) data;
            delete msg;
        } break;
174 175 176 177 178
        case IPMessageType::SCHED_KILL_JOB:
        {
            KillJobMessage * msg = (KillJobMessage *) data;
            delete msg;
        } break;
Millian Poquet's avatar
Millian Poquet committed
179
        case IPMessageType::SCHED_CALL_ME_LATER:
180
        {
Millian Poquet's avatar
Millian Poquet committed
181
            CallMeLaterMessage * msg = (CallMeLaterMessage*) data;
182 183
            delete msg;
        } break;
184 185 186
        case IPMessageType::SCHED_TELL_ME_ENERGY:
        {
        } break;
Pierre-François's avatar
Pierre-François committed
187 188 189 190 191 192 193 194
        case IPMessageType::WAIT_QUERY:
        {
            WaitQueryMessage * msg = (WaitQueryMessage *) data;
            delete msg;
        } break;
        case IPMessageType::SCHED_WAIT_ANSWER:
        {
            SchedWaitAnswerMessage * msg = (SchedWaitAnswerMessage *) data;
195
            delete msg;
Pierre-François's avatar
Pierre-François committed
196
        } break;
197 198 199 200 201
        case IPMessageType::SCHED_READY:
        {
        } break;
        case IPMessageType::SUBMITTER_HELLO:
        {
202 203 204 205 206 207 208
            SubmitterHelloMessage * msg = (SubmitterHelloMessage *) data;
            delete msg;
        } break;
        case IPMessageType::SUBMITTER_CALLBACK:
        {
            SubmitterJobCompletionCallbackMessage * msg = (SubmitterJobCompletionCallbackMessage *) data;
            delete msg;
209 210 211
        } break;
        case IPMessageType::SUBMITTER_BYE:
        {
212 213
            SubmitterByeMessage * msg = (SubmitterByeMessage *) data;
            delete msg;
214 215 216
        } break;
        case IPMessageType::SWITCHED_ON:
        {
217
            SwitchMessage * msg = (SwitchMessage *) data;
218 219 220 221
            delete msg;
        } break;
        case IPMessageType::SWITCHED_OFF:
        {
222
            SwitchMessage * msg = (SwitchMessage *) data;
223 224
            delete msg;
        } break;
225 226 227
        case IPMessageType::WAITING_DONE:
        {
        } break;
228 229 230 231 232
        case IPMessageType::KILLING_DONE:
        {
            KillingDoneMessage * msg = (KillingDoneMessage *) data;
            delete msg;
        } break;
233 234 235
        case IPMessageType::END_DYNAMIC_SUBMIT:
        {
        } break;
236 237 238 239
    }

    data = nullptr;
}
240

241 242 243 244 245 246 247
JobIdentifier::JobIdentifier(const string &workload_name, int job_number) :
    workload_name(workload_name),
    job_number(job_number)
{

}

248 249 250 251
string JobIdentifier::to_string() const
{
    return workload_name + '!' + std::to_string(job_number);
}
252 253 254 255 256

bool operator<(const JobIdentifier &ji1, const JobIdentifier &ji2)
{
    return ji1.to_string() < ji2.to_string();
}