Fix notify progress with TLS
This commit is contained in:
parent
1d9c288bc7
commit
ad2c0909b4
1 changed files with 5 additions and 1 deletions
|
|
@ -201,6 +201,7 @@ pub async fn play(
|
|||
let limiter = limiter.clone();
|
||||
let running = running.clone();
|
||||
let dummy_bytes = dummy_bytes.clone();
|
||||
let notify_socket = notify_socket.clone();
|
||||
handles.push(tokio::spawn(async move {
|
||||
let mut running_guard = running.lock().await;
|
||||
running_guard.insert(*conn_id);
|
||||
|
|
@ -275,6 +276,9 @@ pub async fn play(
|
|||
println!("Client: {} / {}", cnt + 1, total);
|
||||
}
|
||||
drop(limiter);
|
||||
if let Some(notify_socket) = ¬ify_socket {
|
||||
notify_socket.send(&cnt.to_be_bytes()).unwrap();
|
||||
}
|
||||
let mut running_guard = running.lock().await;
|
||||
running_guard.remove(conn_id);
|
||||
drop(running_guard);
|
||||
|
|
@ -361,10 +365,10 @@ pub async fn play(
|
|||
if debug {
|
||||
println!("Client: {} / {}", cnt, total);
|
||||
}
|
||||
drop(limiter);
|
||||
if let Some(notify_socket) = ¬ify_socket {
|
||||
notify_socket.send(&cnt.to_be_bytes()).unwrap();
|
||||
}
|
||||
drop(limiter);
|
||||
let mut running_guard = running.lock().await;
|
||||
running_guard.remove(conn_id);
|
||||
drop(running_guard);
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue