simgrid-java/examples/masterslave/Master.java

   1 /*
   2  * Master of a basic master/slave example in Java
   3  *
   4  * Copyright 2006-2012 The SimGrid Team. All rights reserved.
   5  *
   6  * This program is free software; you can redistribute it and/or modify it
   7  * under the terms of the license (GNU LGPL) which comes with this package.
   8  */
   9
  10 package masterslave;
  11 import org.simgrid.msg.Host;
  12 import org.simgrid.msg.Msg;
  13 import org.simgrid.msg.MsgException;
  14 import org.simgrid.msg.Task;
  15 import org.simgrid.msg.Process;;
  16
  17 public class Master extends Process {
  18         public Master(Host host, String name, String[]args) {
  19                 super(host,name,args);
  20         }
  21         public void main(String[] args) throws MsgException {
  22                 if (args.length < 4) {
  23                         Msg.info("Master needs 4 arguments");
  24                         System.exit(1);
  25                 }
  26
  27                 int tasksCount = Integer.valueOf(args[0]).intValue();
  28                 double taskComputeSize = Double.valueOf(args[1]).doubleValue();
  29                 double taskCommunicateSize = Double.valueOf(args[2]).doubleValue();
  30
  31                 int slavesCount = Integer.valueOf(args[3]).intValue();
  32
  33                 Msg.info("Hello! Got "+  slavesCount + " slaves and "+tasksCount+" tasks to process");
  34
  35                 for (int i = 0; i < tasksCount; i++) {
  36                         Task task = new Task("Task_" + i, taskComputeSize, taskCommunicateSize);
  37                         //Msg.info("Sending \"" + task.getName()+ "\" to \"slave_" + i % slavesCount + "\"");
  38                         task.send("slave_"+(i%slavesCount));
  39                 }
  40
  41                 Msg.info("All tasks have been dispatched. Let's tell everybody the computation is over.");
  42
  43                 for (int i = 0; i < slavesCount; i++) {
  44                         FinalizeTask task = new FinalizeTask();
  45                         task.send("slave_"+(i%slavesCount));
  46                 }
  47
  48                 Msg.info("Goodbye now!");
  49         }
  50 }