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::info("Master Sending "+ task.name + " to " + mailbox + " with Comput Size " +
  32            task.compSize.to_s)
  33       task.send(mailbox)
  34       MSG::info("Master Done Sending " + task.name + " to " + mailbox)
  35     end
  36
  37    # Sending Finalize MSG::Tasks
  38    MSG::info("Master: All tasks have been dispatched. Let's tell everybody the computation is over.")
  39    for i in 0..slaveCount-1
  40      mailbox = "slave " + i.to_s
  41      MSG::info("Master Sending Finalize to " + mailbox)
  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        info("Ready to Receive Task")
  62        task = MSG::Task.receive(mailbox)
  63        MSG::info("Task Received : " + task.name)
  64        if (task.name == "finalize")
  65                MSG::info("Slave " + mailbox + " got finalize msg")
  66                break
  67        end
  68        MSG::info("Slave " + mailbox + " ...Processing" + task.name)
  69        task.execute
  70        MSG::info("task "+ task.name + " Executed !!")
  71     end
  72     MSG::info("Slave " + mailbox +  "I'm Done , See You !!")
  73   end
  74 end
  75
  76 #################################################
  77 # main chunck
  78 #################################################
  79
  80 if (ARGV.length == 2)
  81         MSG.createEnvironment(ARGV[0])
  82         MSG.deployApplication(ARGV[1])
  83 else
  84         MSG.createEnvironment("platform.xml")
  85         MSG.deployApplication("deploy.xml")
  86
  87 end
  88
  89 MSG.run
  90 puts "Simulation time : " + MSG.getClock .to_s
  91 MSG.exit