ipp.cpp 7.93 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
    string s;

61 62
    // Do not remove the switch. If one adds a new IPMessageType but forgets to handle it in the
    // switch, a compilation warning should help avoiding this bug.
63
    switch(type)
64
    {
65 66 67
        case IPMessageType::JOB_SUBMITTED:
            s = "JOB_SUBMITTED";
            break;
68 69 70
        case IPMessageType::JOB_SUBMITTED_BY_DP:
            s = "JOB_SUBMITTED_BY_DP";
            break;
71 72 73 74 75 76
        case IPMessageType::JOB_COMPLETED:
            s = "JOB_COMPLETED";
            break;
        case IPMessageType::PSTATE_MODIFICATION:
            s = "PSTATE_MODIFICATION";
            break;
Millian Poquet's avatar
Millian Poquet committed
77 78
        case IPMessageType::SCHED_EXECUTE_JOB:
            s = "SCHED_EXECUTE_JOB";
79
            break;
Millian Poquet's avatar
Millian Poquet committed
80 81
        case IPMessageType::SCHED_REJECT_JOB:
            s = "SCHED_REJECT_JOB";
82
            break;
83 84 85
        case IPMessageType::SCHED_KILL_JOB:
            s = "SCHED_KILL_JOB";
            break;
Millian Poquet's avatar
Millian Poquet committed
86 87
        case IPMessageType::SCHED_CALL_ME_LATER:
            s = "SCHED_CALL_ME_LATER";
88
            break;
89 90 91
        case IPMessageType::SCHED_TELL_ME_ENERGY:
            s = "SCHED_TELL_ME_ENERGY";
            break;
Pierre-François's avatar
Pierre-François committed
92 93 94 95 96 97
        case IPMessageType::SCHED_WAIT_ANSWER:
            s = "SCHED_WAIT_ANSWER";
            break;
        case IPMessageType::WAIT_QUERY:
            s = "WAIT_QUERY";
            break;
98 99 100 101 102 103 104 105 106
        case IPMessageType::SCHED_READY:
            s = "SCHED_READY";
            break;
        case IPMessageType::WAITING_DONE:
            s = "WAITING_DONE";
            break;
        case IPMessageType::SUBMITTER_HELLO:
            s = "SUBMITTER_HELLO";
            break;
107 108 109
        case IPMessageType::SUBMITTER_CALLBACK:
            s = "SUBMITTER_CALLBACK";
            break;
110 111 112 113 114 115 116 117 118
        case IPMessageType::SUBMITTER_BYE:
            s = "SUBMITTER_BYE";
            break;
        case IPMessageType::SWITCHED_ON:
            s = "SWITCHED_ON";
            break;
        case IPMessageType::SWITCHED_OFF:
            s = "SWITCHED_OFF";
            break;
119
        case IPMessageType::KILLING_DONE:
Millian Poquet's avatar
Millian Poquet committed
120
            s = "KILLING_DONE";
121
            break;
122 123
        case IPMessageType::END_DYNAMIC_SUBMIT:
            s = "END_DYNAMIC_SUBMIT";
124
    }
125

126
    return s;
127
}
128 129 130 131 132 133

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

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

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

    data = nullptr;
}
244

245 246 247 248 249 250 251
JobIdentifier::JobIdentifier(const string &workload_name, int job_number) :
    workload_name(workload_name),
    job_number(job_number)
{

}

252 253 254 255
string JobIdentifier::to_string() const
{
    return workload_name + '!' + std::to_string(job_number);
}
256 257 258 259 260

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