1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
|
# Copyright (c) 2010-2025. The SimGrid Team. All rights reserved.
#
# This program is free software; you can redistribute it and/or modify it
# under the terms of the license (GNU LGPL) which comes with this package.
""" This example demonstrates Comm.wait_for() and Comm.wait_until to set timeouts on waits.
"""
from argparse import ArgumentParser
from typing import List
import sys
from simgrid import Actor, Comm, Engine, Mailbox, this_actor
FINALIZE_MESSAGE = "finalize"
def create_parser() -> ArgumentParser:
parser = ArgumentParser()
parser.add_argument(
'--platform',
type=str,
required=True,
help='path to the platform description'
)
return parser
def sender(receiver_mailbox: Mailbox, messages_count: int, payload_size: int):
pending_comms: List[Comm] = []
# Start dispatching all messages to the receiver
for i in range(messages_count):
payload = f"Message {i}"
this_actor.info(f"Send '{payload}' to '{receiver_mailbox.name}'")
# Create a communication representing the ongoing communication
comm = receiver_mailbox.put_async(payload, payload_size)
# Add this comm to the vector of all known comms
pending_comms.append(comm)
# Start the finalize signal to the receiver
final_comm = receiver_mailbox.put_async(FINALIZE_MESSAGE, 0)
pending_comms.append(final_comm)
this_actor.info(f"Send '{FINALIZE_MESSAGE}' to '{receiver_mailbox.name}'")
this_actor.info("Done dispatching all messages")
# Now that all message exchanges were initiated, wait for their completion, in order of creation
while pending_comms:
comm = pending_comms[-1]
comm.wait_until(Engine.clock + 1) # same as: current_comm.wait_for(1.0)
pending_comms.pop() # remove it from the list
this_actor.info("Goodbye now!")
def receiver(mailbox: Mailbox):
this_actor.info("Wait for my first message")
finalized = False
while not finalized:
received: str = mailbox.get()
this_actor.info(f"I got a '{received}'.")
# If it's a finalize message, we're done.
if received == FINALIZE_MESSAGE:
finalized = True
def main():
settings = create_parser().parse_known_args()[0]
e = Engine(sys.argv)
e.load_platform(settings.platform)
receiver_mailbox: Mailbox = Mailbox.by_name("receiver")
e.host_by_name("Tremblay").add_actor("sender", sender, receiver_mailbox, 3, int(5e7))
e.host_by_name("Ruby").add_actor("receiver", receiver, receiver_mailbox)
e.run()
if __name__ == "__main__":
main()
|
