31 lines
697 B
Python

from dataclasses import dataclass
from framework.framework import Framework
MESSAGE_SIZE = 1
@dataclass
class Message:
_id: int
sent_time: float
def id(self) -> int:
return self._id
def __len__(self) -> int:
# Return any number here, since we don't use Sphinx encoding for queuesim and byte serialization.
# This must be matched with NomssipConfig.msg_size.
return MESSAGE_SIZE
class MessageBuilder:
def __init__(self, framework: Framework):
self.framework = framework
self.next_id = 0
def next(self) -> Message:
msg = Message(self.next_id, self.framework.now())
self.next_id += 1
return msg