examples/msg/cloud/scale.c

   1 /* Copyright (c) 2007-2012. 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 <stdio.h>
   7 #include <sys/time.h>
   8 #include "msg/msg.h"
   9 #include "xbt/sysdep.h"         /* calloc, printf */
  10
  11 /* Create a log channel to have nice outputs. */
  12 #include "xbt/log.h"
  13 #include "xbt/asserts.h"
  14 XBT_LOG_NEW_DEFAULT_CATEGORY(msg_test,
  15                              "Messages specific for this msg example");
  16
  17 /*
  18  * Usage:
  19  * ./examples/msg/cloud/scale ../examples/platforms/g5k.xml
  20  *
  21  * 1. valgrind --tool=callgrind ./examples/msg/cloud/scale ../examples/platforms/g5k.xml
  22  * 2. kcachegrind
  23  **/
  24
  25 static double time_precise(void)
  26 {
  27         struct timeval tv;
  28         int ret = gettimeofday(&tv, NULL);
  29   if (ret < 0)
  30     xbt_die("gettimeofday");
  31
  32         double now = (double) tv.tv_sec + tv.tv_usec * 0.001 * 0.001;
  33
  34         return now;
  35 }
  36
  37 static int computation_fun(int argc, char *argv[])
  38 {
  39   for (;;) {
  40     // double clock_sta = time_precise();
  41
  42     msg_task_t task = MSG_task_create("Task", 10000000, 0, NULL);
  43     MSG_task_execute(task);
  44     MSG_task_destroy(task);
  45
  46     // double clock_end = time_precise();
  47
  48     // XBT_INFO("%f", clock_end - clock_sta);
  49   }
  50
  51
  52   return 0;
  53 }
  54
  55 static void launch_computation_worker(msg_host_t host)
  56 {
  57   MSG_process_create("compute", computation_fun, NULL, host);
  58 }
  59
  60 #if 0
  61 struct task_priv {
  62   msg_host_t tx_host;
  63   msg_process_t tx_proc;
  64   double clock_sta;
  65 };
  66
  67 static int communication_tx_fun(int argc, char *argv[])
  68 {
  69   xbt_assert(argc == 2);
  70   const char *mbox = argv[1];
  71
  72   msg_task_t task = MSG_task_create("Task", 1000000, 1000000, NULL);
  73
  74   struct task_priv *priv = xbt_new(struct task_priv, 1);
  75   priv->tx_proc = MSG_process_self();
  76   priv->tx_host = MSG_host_self();
  77   priv->clock_sta = MSG_get_clock();
  78
  79   MSG_task_set_data(task, priv);
  80
  81   MSG_task_send(task, mbox);
  82
  83   return 0;
  84 }
  85
  86 static int communication_rx_fun(int argc, char *argv[])
  87 {
  88   const char *pr_name = MSG_process_get_name(MSG_process_self());
  89   const char *host_name = MSG_host_get_name(MSG_host_self());
  90   xbt_assert(argc == 2);
  91   const char *mbox = argv[1];
  92
  93   msg_task_t task = NULL;
  94   MSG_task_recv(&task, mbox);
  95
  96   struct task_priv *priv = MSG_task_get_data(task);
  97   double clock_end = MSG_get_clock();
  98
  99   XBT_INFO("%s:%s to %s:%s => %g sec",
 100       MSG_host_get_name(priv->tx_host),
 101       MSG_process_get_name(priv->tx_proc),
 102       host_name, pr_name, clock_end - priv->clock_sta);
 103
 104   MSG_task_destroy(task);
 105
 106   return 0;
 107 }
 108
 109 static void launch_communication_worker(msg_host_t tx_host, msg_host_t rx_host)
 110 {
 111   char *mbox = bprintf("MBOX:%s-%s",
 112       MSG_host_get_name(tx_host),
 113       MSG_host_get_name(rx_host));
 114   char **argv = NULL;
 115
 116   const char *pr_name_tx =  "comm_tx";
 117   argv = xbt_new(char *, 3);
 118   argv[0] = xbt_strdup(pr_name_tx);
 119   argv[1] = xbt_strdup(mbox);
 120   argv[2] = NULL;
 121
 122   MSG_process_create_with_arguments(pr_name_tx, communication_tx_fun, NULL, tx_host, 2, argv);
 123
 124   const char *pr_name_rx =  "comm_rx";
 125   argv = xbt_new(char *, 3);
 126   argv[0] = xbt_strdup(pr_name_rx);
 127   argv[1] = xbt_strdup(mbox);
 128   argv[2] = NULL;
 129
 130   MSG_process_create_with_arguments(pr_name_rx, communication_rx_fun, NULL, rx_host, 2, argv);
 131
 132   xbt_free(mbox);
 133 }
 134 #endif
 135
 136
 137
 138
 139
 140
 141 static int master_main(int argc, char *argv[])
 142 {
 143   xbt_dynar_t hosts_dynar = MSG_hosts_as_dynar();
 144
 145   int npm = 10;
 146   int nvm = 1000;
 147
 148   msg_host_t *pm = xbt_new(msg_host_t, npm);
 149   msg_vm_t   *vm = xbt_new(msg_vm_t, nvm);
 150
 151   int i = 0;
 152   for (i = 0; i < npm; i++) {
 153           pm[i] = xbt_dynar_get_as(hosts_dynar, i, msg_host_t);
 154   }
 155
 156   for (i = 0; i < nvm; i++) {
 157           int pm_index = i % npm;
 158           char *vm_name = bprintf("vm%d", i);
 159           vm[i] = MSG_vm_create_core(pm[pm_index], vm_name);
 160           MSG_vm_start(vm[i]);
 161
 162     launch_computation_worker(vm[i]);
 163
 164           xbt_free(vm_name);
 165   }
 166
 167
 168   XBT_INFO("## Test (start)");
 169
 170   for (i = 0; i < 10; i++) {
 171           double clock_sta = time_precise();
 172           MSG_process_sleep(1);
 173           double clock_end = time_precise();
 174           XBT_INFO("duration %f", clock_end - clock_sta);
 175   }
 176
 177
 178   for (i = 0; i < nvm; i++) {
 179           MSG_vm_destroy(vm[i]);
 180   }
 181
 182   XBT_INFO("## Test (ended)");
 183
 184   return 0;
 185 }
 186
 187 static void launch_master(msg_host_t host)
 188 {
 189   const char *pr_name = "master_";
 190   char **argv = xbt_new(char *, 2);
 191   argv[0] = xbt_strdup(pr_name);
 192   argv[1] = NULL;
 193
 194   MSG_process_create_with_arguments(pr_name, master_main, NULL, host, 1, argv);
 195 }
 196
 197
 198 int main(int argc, char *argv[])
 199 {
 200   /* Get the arguments */
 201   MSG_init(&argc, argv);
 202
 203   /* load the platform file */
 204   xbt_assert(argc == 2);
 205   MSG_create_environment(argv[1]);
 206
 207   xbt_dynar_t hosts_dynar = MSG_hosts_as_dynar();
 208   msg_host_t pm0 = xbt_dynar_get_as(hosts_dynar, 0, msg_host_t);
 209   launch_master(pm0);
 210
 211   int res = MSG_main();
 212   XBT_INFO("Bye (simulation time %g)", MSG_get_clock());
 213
 214
 215   return !(res == MSG_OK);
 216 }