1 /* Copyright (c) 2010-2021. The SimGrid Team. All rights reserved. */
3 /* This program is free software; you can redistribute it and/or modify it
4 * under the terms of the license (GNU LGPL) which comes with this package. */
6 #include "simgrid/actor.h"
7 #include "simgrid/comm.h"
8 #include "simgrid/engine.h"
9 #include "simgrid/host.h"
10 #include "simgrid/mailbox.h"
14 #include "xbt/sysdep.h"
16 #include <stdio.h> /* snprintf */
18 XBT_LOG_NEW_DEFAULT_CATEGORY(comm_waitall, "Messages specific for this msg example");
20 static void sender(int argc, char* argv[])
22 xbt_assert(argc == 4, "This function expects 3 parameters from the XML deployment file");
23 long messages_count = xbt_str_parse_int(argv[1], "Invalid message count: %s");
24 long message_size = xbt_str_parse_int(argv[2], "Invalid message size: %s");
25 long receivers_count = xbt_str_parse_int(argv[3], "Invalid amount of receivers: %s");
26 xbt_assert(receivers_count > 0);
28 /* Array in which we store all ongoing communications */
29 sg_comm_t* pending_comms = xbt_malloc(sizeof(sg_comm_t) * (messages_count + receivers_count));
30 int pending_comms_count = 0;
32 /* Make an array of the mailboxes to use */
33 sg_mailbox_t* mboxes = xbt_malloc(sizeof(sg_mailbox_t) * receivers_count);
34 for (long i = 0; i < receivers_count; i++) {
35 char mailbox_name[80];
36 snprintf(mailbox_name, 79, "receiver-%ld", i);
37 sg_mailbox_t mbox = sg_mailbox_by_name(mailbox_name);
41 /* Start dispatching all messages to receivers, in a round robin fashion */
42 for (long i = 0; i < messages_count; i++) {
44 snprintf(msg_content, 79, "Message %ld", i);
45 sg_mailbox_t mbox = mboxes[i % receivers_count];
46 XBT_INFO("Send '%s' to '%s'", msg_content, sg_mailbox_get_name(mbox));
47 /* Create a communication representing the ongoing communication, and store it in pending_comms */
48 pending_comms[pending_comms_count++] = sg_mailbox_put_async(mbox, xbt_strdup(msg_content), message_size);
51 /* Start sending messages to let the workers know that they should stop */
52 for (long i = 0; i < receivers_count; i++) {
53 XBT_INFO("Send 'finalize' to 'receiver-%ld'", i);
54 char* end_msg = xbt_strdup("finalize");
55 sg_mailbox_t mbox = mboxes[i % receivers_count];
56 pending_comms[pending_comms_count++] = sg_mailbox_put_async(mbox, end_msg, 0);
59 XBT_INFO("Done dispatching all messages");
61 /* Now that all message exchanges were initiated, wait for their completion in one single call */
62 sg_comm_wait_all(pending_comms, pending_comms_count);
67 XBT_INFO("Goodbye now!");
70 static void receiver(int argc, char* argv[])
72 xbt_assert(argc == 2, "Expecting one parameter from the XML deployment file but got %d", argc);
73 int id = (int)xbt_str_parse_int(argv[1], "ID should be numerical, not %s");
74 char mailbox_name[80];
75 snprintf(mailbox_name, 79, "receiver-%d", id);
76 sg_mailbox_t mbox = sg_mailbox_by_name(mailbox_name);
77 XBT_INFO("Wait for my first message");
79 char* received = (char*)sg_mailbox_get(mbox);
80 XBT_INFO("I got a '%s'.", received);
81 if (!strcmp(received, "finalize")) { // If it's a finalize message, we're done
89 int main(int argc, char* argv[])
91 simgrid_init(&argc, argv);
93 "Usage: %s platform_file deployment_file\n"
94 "\tExample: %s msg_platform.xml msg_deployment.xml\n",
97 simgrid_load_platform(argv[1]);
99 simgrid_register_function("sender", sender);
100 simgrid_register_function("receiver", receiver);
101 simgrid_load_deployment(argv[2]);