Skip to content

Commit 797ca28

Browse files
author
rusty
committed
Use stop_token in listeners
1 parent ac0de5c commit 797ca28

File tree

2 files changed

+4
-2
lines changed

2 files changed

+4
-2
lines changed

src/listener/tcp_listener.rs

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,6 @@
11
use super::{is_transient_error, ListenInfo};
22

3+
use crate::cancellation::StopStreamExt;
34
use crate::listener::Listener;
45
use crate::{log, Server};
56

@@ -98,7 +99,7 @@ where
9899
.take()
99100
.expect("`Listener::bind` must be called before `Listener::accept`");
100101

101-
let mut incoming = listener.incoming();
102+
let mut incoming = listener.incoming().stop_on(server.stop_token.clone());
102103

103104
while let Some(stream) = incoming.next().await {
104105
match stream {

src/listener/unix_listener.rs

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,6 @@
11
use super::{is_transient_error, ListenInfo};
22

3+
use crate::cancellation::StopStreamExt;
34
use crate::listener::Listener;
45
use crate::{log, Server};
56

@@ -96,7 +97,7 @@ where
9697
.take()
9798
.expect("`Listener::bind` must be called before `Listener::accept`");
9899

99-
let mut incoming = listener.incoming();
100+
let mut incoming = listener.incoming().stop_on(server.stop_token.clone());
100101

101102
while let Some(stream) = incoming.next().await {
102103
match stream {

0 commit comments

Comments
 (0)