Convert 2 examples from Activity::wait_any to ActivitySet

[simgrid.git] / examples / cpp / activityset-testany / s4u-activityset-testany.cpp

diff --git a/examples/cpp/activityset-testany/s4u-activityset-testany.cpp b/examples/cpp/activityset-testany/s4u-activityset-testany.cpp
new file mode 100644 (file)
index 0000000..8128c24
--- /dev/null
+++ b/examples/cpp/activityset-testany/s4u-activityset-testany.cpp
@@ -0,0 +1,71 @@
+/* Copyright (c) 2010-2023. The SimGrid Team. All rights reserved.          */
+
+/* This program is free software; you can redistribute it and/or modify it
+ * under the terms of the license (GNU LGPL) which comes with this package. */
+
+#include "simgrid/s4u.hpp"
+#include <cstdlib>
+#include <iostream>
+#include <string>
+namespace sg4 = simgrid::s4u;
+
+XBT_LOG_NEW_DEFAULT_CATEGORY(s4u_activity_testany, "Messages specific for this s4u example");
+
+static void bob()
+{
+  sg4::Mailbox* mbox    = sg4::Mailbox::by_name("mbox");
+  const sg4::Disk* disk = sg4::Host::current()->get_disks().front();
+  std::string* payload;
+
+  XBT_INFO("Create my asynchronous activities");
+  auto exec = sg4::this_actor::exec_async(5e9);
+  auto comm = mbox->get_async(&payload);
+  auto io   = disk->read_async(3e8);
+
+  sg4::ActivitySet pending_activities;
+  pending_activities.push(exec);
+  pending_activities.push(comm);
+  pending_activities.push(io);
+
+  XBT_INFO("Sleep_for a while");
+  sg4::this_actor::sleep_for(1);
+
+  XBT_INFO("Test for completed activities");
+  while (not pending_activities.empty()) {
+    auto completed_one = pending_activities.test_any();
+    if (completed_one != nullptr) {
+      if (boost::dynamic_pointer_cast<sg4::Comm>(completed_one))
+        XBT_INFO("Completed a Comm");
+      if (boost::dynamic_pointer_cast<sg4::Exec>(completed_one))
+        XBT_INFO("Completed an Exec");
+      if (boost::dynamic_pointer_cast<sg4::Io>(completed_one))
+        XBT_INFO("Completed an I/O");
+    } else {
+      XBT_INFO("Nothing matches, test again in 0.5s");
+      sg4::this_actor::sleep_for(.5);
+    }
+  }
+  XBT_INFO("Last activity is complete");
+  delete payload;
+}
+
+static void alice()
+{
+  auto* payload = new std::string("Message");
+  XBT_INFO("Send '%s'", payload->c_str());
+  sg4::Mailbox::by_name("mbox")->put(payload, 6e8);
+}
+
+int main(int argc, char* argv[])
+{
+  sg4::Engine e(&argc, argv);
+
+  e.load_platform(argv[1]);
+
+  sg4::Actor::create("bob", e.host_by_name("bob"), bob);
+  sg4::Actor::create("alice", e.host_by_name("alice"), alice);
+
+  e.run();
+
+  return 0;
+}