examples/c/activityset-waitany/activityset-waitany.c

   1 /* Copyright (c) 2010-2023. The SimGrid Team. All rights reserved.          */
   2
   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. */
   5
   6 #include "simgrid/activity_set.h"
   7 #include "simgrid/actor.h"
   8 #include "simgrid/comm.h"
   9 #include "simgrid/engine.h"
  10 #include "simgrid/exec.h"
  11 #include "simgrid/host.h"
  12 #include "simgrid/mailbox.h"
  13
  14 #include "xbt/log.h"
  15 #include "xbt/sysdep.h"
  16
  17 XBT_LOG_NEW_DEFAULT_CATEGORY(s4u_activity_waittany, "Messages specific for this s4u example");
  18
  19 static void bob()
  20 {
  21   XBT_INFO("Create my asynchronous activities");
  22   sg_exec_t exec = sg_actor_exec_init(5e9);
  23   sg_exec_start(exec);
  24
  25   sg_mailbox_t mbox = sg_mailbox_by_name("mbox");
  26   void* payload     = NULL;
  27   sg_comm_t comm    = sg_mailbox_get_async(mbox, &payload);
  28
  29   sg_activity_set_t pending_activities = sg_activity_set_init();
  30   sg_activity_set_push(pending_activities, (sg_activity_t)exec);
  31   sg_activity_set_push(pending_activities, (sg_activity_t)comm);
  32
  33   XBT_INFO("Wait for asynchronous activities to complete");
  34   while (!sg_activity_set_empty(pending_activities)) {
  35
  36     sg_activity_t completed_one = sg_activity_set_wait_any(pending_activities);
  37     if (sg_comm_isinstance(completed_one))
  38       XBT_INFO("Completed a Comm");
  39     else if (sg_exec_isinstance(completed_one))
  40       XBT_INFO("Completed an Exec");
  41     else
  42       xbt_die("This activity set is supposed to only contain Comm or Exec");
  43     sg_activity_unref(completed_one);
  44   }
  45   XBT_INFO("Last activity is complete");
  46   free(payload);
  47 }
  48
  49 static void alice()
  50 {
  51   char* payload = xbt_strdup("Message");
  52   XBT_INFO("Send '%s'", payload);
  53   sg_mailbox_put(sg_mailbox_by_name("mbox"), payload, 6e8);
  54 }
  55
  56 int main(int argc, char* argv[])
  57 {
  58   simgrid_init(&argc, argv);
  59   xbt_assert(argc > 1,
  60              "Usage: %s platform_file\n"
  61              "\tExample: %s hosts_with_disks.xml\n",
  62              argv[0], argv[0]);
  63
  64   simgrid_load_platform(argv[1]);
  65
  66   sg_actor_create("alice", sg_host_by_name("alice"), alice, 0, NULL);
  67   sg_actor_create("bob", sg_host_by_name("bob"), bob, 0, NULL);
  68
  69   simgrid_run();
  70
  71   return 0;
  72 }