src/bindings/ruby/MasterSlave.rb

   1 # Debug it with this command:
   2 # make -C ../.. && valgrind ruby MasterSlave.rb --log=ruby.thres:debug 2>&1 | less
   3
   4 require 'simgrid'
   5
   6 include MSG
   7
   8 #################################################
   9 # Class Master
  10 #################################################
  11
  12 class Master < MSG::Process
  13   # main : that function that will be executed when running simulation
  14
  15   def main(args) # args is an array containing arguments for function master
  16    size = args.size
  17    for i in 0..size-1
  18      MSG::info("args["+String(i)+"]="+args[i])
  19    end
  20
  21    raise "Master needs 3 arguments" if size < 3
  22    numberOfTask = Integer(args[0])
  23    taskComputeSize = Float(args[1])
  24    taskCommunicationSize = Float(args[2])
  25    slaveCount = Integer(args[3])
  26
  27    # Creates and sends the tasks
  28     for i in 0..numberOfTask-1
  29      task = MSG::Task.new("Task_"+ i.to_s, taskComputeSize , taskCommunicationSize);
  30       mailbox = "slave " + (i%slaveCount).to_s
  31       MSG::debug("Master Sending "+ task.name + " to " + mailbox + " with Comput Size " +
  32            task.compSize.to_s)
  33       task.send(mailbox)
  34       MSG::debug("Master Done Sending " + task.name + " to " + mailbox)
  35     end
  36
  37    # Sending Finalize MSG::Tasks
  38    #MSG::info("Master: All "+numberOfTask+" tasks have been dispatched. Let's tell everybody the computation is over.")
  39    MSG::info("Master: All tasks have been dispatched. Let's tell everybody the computation is over.")
  40    for i in 0..slaveCount-1
  41      mailbox = "slave " + i.to_s
  42      finalize_task = Task.new("finalize",0,0)
  43      finalize_task.send(mailbox)
  44    end
  45    MSG::info("Master : Everything's Done")
  46   end
  47 end
  48
  49 #################################################
  50 # Class Slave
  51 #################################################
  52 class Slave < MSG::Process
  53
  54   def main(args)
  55     mailbox = "slave " + args[0]
  56     for i in 0..args.size-1
  57       MSG::debug("args["+String(i)+"]="+args[i])
  58     end
  59
  60     while true
  61        task = MSG::Task.receive(mailbox)
  62        if (task.name == "finalize")
  63                break
  64        end
  65        task.execute
  66        MSG::debug("Slave '" + mailbox + "' done executing task "+ task.name + ".")
  67     end
  68     MSG::info("I'm done, see you")
  69   end
  70 end
  71
  72 #################################################
  73 # main chunck
  74 #################################################
  75
  76 if (ARGV.length == 2)
  77         MSG.createEnvironment(ARGV[0])
  78         MSG.deployApplication(ARGV[1])
  79 else
  80         MSG.createEnvironment("platform.xml")
  81         MSG.deployApplication("deploy.xml")
  82 end
  83
  84 MSG.run
  85 puts "Simulation time : " + MSG.getClock .to_s
  86 MSG.exit