Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
replay-storage becomes replay-io ... with disks
[simgrid.git] / examples / s4u / replay-io / s4u-replay-io.cpp
diff --git a/examples/s4u/replay-io/s4u-replay-io.cpp b/examples/s4u/replay-io/s4u-replay-io.cpp
new file mode 100644 (file)
index 0000000..93af1ff
--- /dev/null
@@ -0,0 +1,130 @@
+/* Copyright (c) 2017-2019. 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/plugins/file_system.h>
+#include <simgrid/s4u.hpp>
+#include <xbt/replay.hpp>
+#include <xbt/str.h>
+
+#include <boost/algorithm/string/join.hpp>
+
+XBT_LOG_NEW_DEFAULT_CATEGORY(replay_io, "Messages specific for this example");
+
+static std::unordered_map<std::string, simgrid::s4u::File*> opened_files;
+
+#define ACT_DEBUG(...)                                                                                                 \
+  if (XBT_LOG_ISENABLED(replay_io, xbt_log_priority_verbose)) {                                                        \
+    std::string NAME = boost::algorithm::join(action, " ");                                                            \
+    XBT_DEBUG(__VA_ARGS__);                                                                                            \
+  } else                                                                                                               \
+    ((void)0)
+
+static void log_action(simgrid::xbt::ReplayAction& action, double date)
+{
+  if (XBT_LOG_ISENABLED(replay_io, xbt_log_priority_verbose)) {
+    std::string s = boost::algorithm::join(action, " ");
+    XBT_VERB("%s %f", s.c_str(), date);
+  }
+}
+
+static simgrid::s4u::File* get_file_descriptor(const std::string& file_name)
+{
+  std::string full_name = simgrid::s4u::this_actor::get_name() + ":" + file_name;
+
+  return opened_files.at(full_name);
+}
+
+class Replayer {
+public:
+  explicit Replayer(std::vector<std::string> args)
+  {
+    const char* actor_name = args[0].c_str();
+    simgrid::xbt::replay_runner(actor_name, nullptr);
+  }
+
+  void operator()()
+  {
+    // Nothing to do here
+  }
+
+  /* My actions */
+  static void open(simgrid::xbt::ReplayAction& action)
+  {
+    std::string file_name = action[2];
+    double clock          = simgrid::s4u::Engine::get_clock();
+    std::string full_name = simgrid::s4u::this_actor::get_name() + ":" + file_name;
+
+    ACT_DEBUG("Entering Open: %s (filename: %s)", NAME.c_str(), file_name.c_str());
+    simgrid::s4u::File* file = new simgrid::s4u::File(file_name, NULL);
+
+    opened_files.insert({full_name, file});
+
+    log_action(action, simgrid::s4u::Engine::get_clock() - clock);
+  }
+
+  static void read(simgrid::xbt::ReplayAction& action)
+  {
+    std::string file_name = action[2];
+    sg_size_t size        = std::stoul(action[3]);
+    double clock          = simgrid::s4u::Engine::get_clock();
+
+    simgrid::s4u::File* file = get_file_descriptor(file_name);
+
+    ACT_DEBUG("Entering Read: %s (size: %llu)", NAME.c_str(), size);
+    file->read(size);
+
+    log_action(action, simgrid::s4u::Engine::get_clock() - clock);
+  }
+
+  static void close(simgrid::xbt::ReplayAction& action)
+  {
+    std::string file_name = action[2];
+    double clock          = simgrid::s4u::Engine::get_clock();
+
+    simgrid::s4u::File* file = get_file_descriptor(file_name);
+
+    ACT_DEBUG("Entering Close: %s (filename: %s)", NAME.c_str(), file_name.c_str());
+    delete file;
+
+    log_action(action, simgrid::s4u::Engine::get_clock() - clock);
+  }
+};
+
+int main(int argc, char* argv[])
+{
+  simgrid::s4u::Engine e(&argc, argv);
+  sg_storage_file_system_init();
+
+  xbt_assert(argc > 3,
+             "Usage: %s platform_file deployment_file [action_files]\n"
+             "\texample: %s platform.xml deployment.xml actions # if all actions are in the same file\n"
+             "\t# if actions are in separate files, specified in deployment\n"
+             "\texample: %s platform.xml deployment.xml",
+             argv[0], argv[0], argv[0]);
+
+  e.load_platform(argv[1]);
+  e.register_actor<Replayer>("p0");
+  e.load_deployment(argv[2]);
+
+  /*   Action registration */
+  xbt_replay_action_register("open", Replayer::open);
+  xbt_replay_action_register("read", Replayer::read);
+  xbt_replay_action_register("close", Replayer::close);
+
+  if (argv[3]) {
+    simgrid::xbt::action_fs = new std::ifstream(argv[3], std::ifstream::in);
+  }
+
+  e.run();
+
+  if (argv[3]) {
+    delete simgrid::xbt::action_fs;
+    simgrid::xbt::action_fs = nullptr;
+  }
+
+  XBT_INFO("Simulation time %g", e.get_clock());
+
+  return 0;
+}