github/darkfi
1
1
Fork 0
mirror of https://github.com/darkrenaissance/darkfi.git synced 2026-04-28 03:00:18 -04:00
Code Issues Packages Projects Releases Wiki Activity

rpc/server/test: Stop all active connections when the server is stopped.

Browse Source
This commit is contained in:
parazyd
2023-08-31 10:18:07 +02:00
parent d030a19be4
commit 882ef29351
1 changed files with 9 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
src/rpc/server.rs
View File

@@ -238,13 +238,21 @@ mod tests {
drop(listener);
let rpc_server = Arc::new(RpcServer { rpc_connections: Mutex::new(HashSet::new()) });
let rpc_server_ = rpc_server.clone();
let server_task = StoppableTask::new();
server_task.clone().start(
listen_and_serve(endpoint.clone(), rpc_server.clone(), None, executor.clone()),
|res| async move {
match res {
Ok(()) | Err(Error::RpcServerStopped) => {}
Ok(()) | Err(Error::RpcServerStopped) => {
eprintln!("Stopping connections");
for (i, task) in rpc_server_.get_connections().await.iter().enumerate()
{
eprintln!("Stopping connection #{}", i);
task.stop().await;
}
}
Err(e) => panic!("{}", e),
}
},