examples/lua/SimSplay/sim_splay.lua

   1 require "simgrid"
   2
   3 -- Splay global modules
   4 rpc = {}
   5 log = {}
   6 job = {}
   7 events = {}
   8 os = {}
   9 start = {}
  10 misc = {}
  11
  12 -- Splay global variables
  13 job.me = {}
  14 job.nodes = {}
  15 job.list_type = "random"
  16
  17 -- Init nodes tables
  18 function init_nodes()
  19   for i = 1, simgrid.host.number() do
  20     job.nodes[i] = { ip = simgrid.host.get_prop_value(simgrid.host.at(i), "ip"),
  21                      port = simgrid.host.get_prop_value(simgrid.host.at(i), "port") }
  22   end
  23 end
  24
  25 function init_jobs()
  26   init_nodes()
  27 end
  28
  29 -- Job methods
  30 function job.me.ip()
  31   return simgrid.host.get_prop_value(simgrid.host.self(), "ip")
  32 end
  33
  34 function job.me.port()
  35   return simgrid.host.get_prop_value(simgrid.host.self(), "port")
  36 end
  37
  38
  39 function job.position()
  40   return simgrid.host.get_prop_value(simgrid.host.self(), "position")
  41 end
  42
  43 -- log Methods
  44 function log:print(msg)
  45   simgrid.info(msg);
  46 end
  47
  48 -- rpc Methods
  49 function rpc.call(node, call)
  50   --init_nodes();
  51   func = "empty"
  52   arg = "empty"
  53   mailbox = node
  54
  55   if type(node) == "table" then
  56     mailbox = node.ip..":"..node.port
  57   end
  58
  59   if type(call) == "table" then
  60     func = call[1]
  61     arg = call[2]
  62   end
  63   task_call = simgrid.task.new("splay_task", 10000, 10000)
  64   task_call['func_call_name'] = func
  65   task_call['func_call_arg'] = arg
  66   log:print("Sending Task to mailbox "..mailbox.." to call '"..func.."' with arg '"..arg.."'")
  67   simgrid.task.send(task_call, mailbox)
  68
  69 end
  70
  71 function rpc.server(port)
  72   -- nothing really to do : no need to open Socket since it's a Simulation
  73 end
  74
  75 -- event Methods
  76 function events.sleep(time)
  77   my_mailbox = job.me.ip()..":"..job.me.port()
  78   task = simgrid.task.recv(my_mailbox, time)
  79
  80   if task ~= nil then
  81     -- an RPC call just woke me up
  82     call_function(task['func_call_name'], task['func_call_arg'])
  83   end
  84 end
  85
  86 -- main function for each process, this is equivalent to the deployment file
  87 function events.thread(main_func)
  88   dofile("platform_script.lua")
  89   init_jobs()
  90 end
  91
  92 -- OS methods
  93 function os.exit()
  94   simgrid.host.destroy(simgrid.host.self())
  95 end
  96
  97 -- Start Methods
  98 function start.loop()
  99   simgrid.run()
 100   --simgrid.clean()
 101 end
 102
 103 -- Misc Methods
 104 function misc.between(a, b)
 105   return a
 106 end
 107
 108 -- useful functions
 109 function call_function(fct, arg)
 110   _G[fct](arg)
 111 end
 112
 113 function SPLAYschool(arg)
 114   simgrid.info("Calling me..."..arg)
 115 end
 116