Skip to content

Commit 8983b45

Browse files
committed
Box a big future
1 parent 61bf4b6 commit 8983b45

1 file changed

Lines changed: 20 additions & 18 deletions

File tree

rust/rust-sysdb/src/server.rs

Lines changed: 20 additions & 18 deletions
Original file line numberDiff line numberDiff line change
@@ -82,25 +82,27 @@ impl SysdbService {
8282
.await;
8383

8484
let backends = self.backends.clone();
85-
Server::builder()
86-
.layer(chroma_tracing::GrpcServerTraceLayer)
87-
.add_service(health_service)
88-
.add_service(SysDbServer::new(self))
89-
.serve_with_shutdown(addr, async move {
90-
// TODO(Sanket): Drain existing requests before shutting down.
91-
select! {
92-
_ = sigterm.recv() => {
93-
backends.close().await;
94-
tracing::info!("Received SIGTERM, shutting down server");
85+
Box::pin(
86+
Server::builder()
87+
.layer(chroma_tracing::GrpcServerTraceLayer)
88+
.add_service(health_service)
89+
.add_service(SysDbServer::new(self))
90+
.serve_with_shutdown(addr, async move {
91+
// TODO(Sanket): Drain existing requests before shutting down.
92+
select! {
93+
_ = sigterm.recv() => {
94+
backends.close().await;
95+
tracing::info!("Received SIGTERM, shutting down server");
96+
}
97+
_ = sigint.recv() => {
98+
backends.close().await;
99+
tracing::info!("Received SIGINT, shutting down server");
100+
}
95101
}
96-
_ = sigint.recv() => {
97-
backends.close().await;
98-
tracing::info!("Received SIGINT, shutting down server");
99-
}
100-
}
101-
})
102-
.await
103-
.expect("Server failed");
102+
}),
103+
)
104+
.await
105+
.expect("Server failed");
104106
}
105107
}
106108

0 commit comments

Comments
 (0)