add steps to Latency

This commit is contained in:
Youngjoon Lee 2024-12-20 20:30:40 +09:00
parent a5e5daad20
commit e597303e2f
No known key found for this signature in database
GPG Key ID: 303963A54A81DD4D

View File

@ -21,21 +21,17 @@ class Event:
class Latency:
start_event: Event
end_event: Optional[Event] = None
steps: Optional[int] = None
def finish(self, event: Event):
assert self.end_event is None
assert event.step_id >= self.start_event.step_id
self.end_event = event
self.steps = self.end_event.step_id - self.start_event.step_id
def finished(self) -> bool:
return self.end_event is not None
@property
def value(self) -> Optional[int]:
if self.end_event is None:
return None
return self.end_event.step_id - self.start_event.step_id
@dataclass
class Message:
@ -77,7 +73,7 @@ class Message:
@property
def latency(self) -> Optional[int]:
return self.total_latency.value
return self.total_latency.steps
def __eq__(self, other):
if not isinstance(other, Message):