Skip to content

Commit ddf9bcd

Browse files
committed
refactor(sse): replace connection deadline with start time for SSE stream duration check
1 parent a9a9119 commit ddf9bcd

File tree

1 file changed

+3
-4
lines changed
  • apps/hermes/server/src/api/rest/v2

1 file changed

+3
-4
lines changed

apps/hermes/server/src/api/rest/v2/sse.rs

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -96,13 +96,12 @@ where
9696
// Convert the broadcast receiver into a Stream
9797
let stream = BroadcastStream::new(update_rx);
9898

99-
// Set connection deadline
100-
let connection_deadline = Instant::now() + MAX_CONNECTION_DURATION;
99+
// Set connection start time
100+
let start_time = Instant::now();
101101

102102
let sse_stream = stream
103103
.take_while(move |_| {
104-
let now = Instant::now();
105-
now < connection_deadline
104+
start_time.elapsed() < MAX_CONNECTION_DURATION
106105
})
107106
.then(move |message| {
108107
let state_clone = state.clone(); // Clone again to use inside the async block

0 commit comments

Comments
 (0)