Close #105

[simgrid.git] / teshsuite / s4u / cloud-interrupt-migration / cloud-interrupt-migration.cpp

diff --git a/teshsuite/s4u/cloud-interrupt-migration/cloud-interrupt-migration.cpp b/teshsuite/s4u/cloud-interrupt-migration/cloud-interrupt-migration.cpp
new file mode 100644 (file)
index 0000000..afb79fb
--- /dev/null
+++ b/teshsuite/s4u/cloud-interrupt-migration/cloud-interrupt-migration.cpp
@@ -0,0 +1,72 @@
+/* Copyright (c) 2017. 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/live_migration.h"
+#include "simgrid/s4u.hpp"
+#include "simgrid/s4u/VirtualMachine.hpp"
+
+XBT_LOG_NEW_DEFAULT_CATEGORY(s4u_cloud_interrupt_migration, "Messages specific for this example");
+
+static void vm_migrate(simgrid::s4u::VirtualMachine* vm, simgrid::s4u::Host* dst_pm)
+{
+  simgrid::s4u::Host* src_pm = vm->getPm();
+  double mig_sta             = simgrid::s4u::Engine::getClock();
+  sg_vm_migrate(vm, dst_pm);
+  double mig_end = simgrid::s4u::Engine::getClock();
+
+  XBT_INFO("%s migrated: %s->%s in %g s", vm->getCname(), src_pm->getCname(), dst_pm->getCname(), mig_end - mig_sta);
+}
+
+static simgrid::s4u::ActorPtr vm_migrate_async(simgrid::s4u::VirtualMachine* vm, simgrid::s4u::Host* dst_pm)
+{
+  return simgrid::s4u::Actor::createActor("mig_wrk", simgrid::s4u::Host::current(), vm_migrate, vm, dst_pm);
+}
+
+static void master_main()
+{
+  simgrid::s4u::Host* pm0 = simgrid::s4u::Host::by_name("Fafard");
+  simgrid::s4u::Host* pm1 = simgrid::s4u::Host::by_name("Tremblay");
+
+  simgrid::s4u::VirtualMachine* vm0 = new simgrid::s4u::VirtualMachine("VM0", pm0, 1);
+  vm0->setRamsize(1e9); // 1Gbytes
+  vm0->start();
+
+  XBT_INFO("Start the migration of %s from %s to %s", vm0->getCname(), pm0->getCname(), pm1->getCname());
+  simgrid::s4u::ActorPtr migration = vm_migrate_async(vm0, pm1);
+
+  simgrid::s4u::this_actor::sleep_for(2);
+  XBT_INFO("Wait! change my mind, shutdown %s. This ends the migration", vm0->getCname());
+  vm0->shutdown();
+
+  simgrid::s4u::this_actor::sleep_for(8);
+
+  XBT_INFO("Start again the migration of %s from %s to %s", vm0->getCname(), pm0->getCname(), pm1->getCname());
+
+  vm0->start();
+  vm_migrate_async(vm0, pm1);
+
+  XBT_INFO("Wait for the completion of the migration this time");
+  simgrid::s4u::this_actor::sleep_for(200);
+  vm0->destroy();
+}
+
+int main(int argc, char* argv[])
+{
+  /* Get the arguments */
+  simgrid::s4u::Engine e(&argc, argv);
+  sg_vm_live_migration_plugin_init();
+
+  /* load the platform file */
+  e.loadPlatform(argv[1]);
+
+  simgrid::s4u::Actor::createActor("master_", simgrid::s4u::Host::by_name("Fafard"), master_main);
+
+  e.run();
+
+  XBT_INFO("Bye (simulation time %g)", simgrid::s4u::Engine::getClock());
+
+  return 0;
+}