@@ -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