def __init__(self, env, *args, **kwargs):
self.work_queue = []
self.message_size_gen = Uniform(MIN_MSG_SIZE, MAX_MSG_SIZE, 'int')
self.message_value_gen = Uniform(MIN_MSG_VALUE, MAX_MSG_VALUE, 'int')
self.msg_received = env.event()
super(PingProgram, self).__init__(env, *args, **kwargs)
def script(self):
rack_options = {
'size': RACK_SIZE
}
node_options = {
'cpus': NODE_CPUS,
'memory': NODE_MEMORY
}
gen = Cluster.create(
self.env,
size=CLUSTER_SIZE,
rack_options=rack_options,
node_options=node_options,
)
cluster = gen.next()
self.cluster = cluster
# create program generator
pgen = PingProgram.create(self.env, cpus=1, memory=4, ports=[3333, 4444])
# create nodes using cluster's node generator
nodes = [cluster.add() for i in range(NODE_COUNT)]
# assign new programs to each node
for n in nodes:
n.assign(pgen.next())
# set some programs to start working instead of waiting
starters = random.sample(nodes, START_COUNT)
work_maker = Uniform(10, 50, 'int')
for n in starters:
p = n.programs[n.programs.keys()[0]]
p.work_queue.append(work_maker.next())
class CarGenerator(Process):
"""
Generates new cars that arrive at the gas station.
"""
def __init__(self, env, station):
self.station = station
self.uniform = Uniform(*T_INTER)
super(CarGenerator, self).__init__(env)
def run(self):
"""
Create a new car at the station at each random interval.
"""
for idx in Sequence():
yield self.env.timeout(self.uniform.get())
Car(self.env, self.station)
class CarGenerator(Process):
"""
Generates new cars that arrive at the gas station.
"""
def __init__(self, env, station):
self.station = station
self.uniform = Uniform(*T_INTER)
super(CarGenerator, self).__init__(env)
def run(self):
"""
Create a new car at the station at each random interval.
"""
for idx in Sequence():
yield self.env.timeout(self.uniform.get())
Car(self.env, self.station)
def __init__(self, env, station):
self.station = station
self.uniform = Uniform(*T_INTER)
super(CarGenerator, self).__init__(env)
def __init__(self, env, station):
self.station = station
self.fuel_level = Uniform(*FUELTANK_LEVEL).get()
super(Car, self).__init__(env)
class PingProgram(Program):
def __init__(self, env, *args, **kwargs):
self.work_queue = []
self.message_size_gen = Uniform(MIN_MSG_SIZE, MAX_MSG_SIZE, 'int')
self.message_value_gen = Uniform(MIN_MSG_VALUE, MAX_MSG_VALUE, 'int')
self.msg_received = env.event()
super(PingProgram, self).__init__(env, *args, **kwargs)
def recv(self, message):
# print "Program {}: received a message with value {} at {}\n".format(self.id, value, self.env.now)
self.work_queue.append(message.value)
self.msg_received.succeed()
self.msg_received = self.env.event()
def wait(self):
"""
Wait until we get a new message
"""
# print "Program {}: waiting for recv at {}\n".format(self.id, self.env.now)
yield self.msg_received
# print "Program {}: received message! {}\n".format(self.id, self.env.now)
def work(self):
"""
Simulate work by going to sleep according to the oldest value in the
work queue.
"""
# print "Program {}: starting work at {}\n".format(self.id, self.env.now)
yield self.env.timeout(self.work_queue.pop(0))
# print "Program {}: done working at {}\n".format(self.id, self.env.now)
def send(self):
"""
Send a message to another node with varying size and value. The value
of the message will be used by the recipient computer as the amount of
time to "work".
"""
# print "Program {}: sending at {}\n".format(self.id, self.env.now)
yield self.env.timeout(1)
# find other node and send the message
recip = self.node.rack.cluster.random(
lambda n: n.id != self.node.id and n.programs
)
self.node.send(
dst=recip.address._replace(port=3333),
size=self.message_size_gen.next(),
value=self.message_value_gen.next()
)
# print "Program {}: done sending at {}\n".format(self.id, self.env.now)
def run(self):
"""
Kicks off execution of a simulated program. This method contains a loop
to cycle through specific behaviors such as:
- wait for recv
- sleep for random time
- does a send to one or more other Programs/Nodes
- repeat
"""
while True:
if not self.work_queue:
waiting = self.env.process(self.wait())
yield waiting
working = self.env.process(self.work())
yield working
sending = self.env.process(self.send())
yield sending
def __init__(self, env, station):
self.station = station
self.uniform = Uniform(*T_INTER)
super(CarGenerator, self).__init__(env)