@@ -15,9 +15,8 @@ use tracing::{error, info};
1515use tracing_subscriber:: FmtSubscriber ;
1616use tracing_subscriber:: filter:: { EnvFilter , LevelFilter } ;
1717
18- use crate :: core:: types:: { JsonRpcError , JsonRpcErrorResponse , ServerState } ;
1918use crate :: core:: light_client:: handle_request;
20-
19+ use crate :: core :: types :: { JsonRpcError , JsonRpcErrorResponse , ServerState } ;
2120
2221pub fn create_error_response (
2322 code : i32 ,
@@ -89,7 +88,7 @@ pub fn start_server(
8988
9089 runtime. spawn ( async move {
9190 info ! ( target: "helios::server" , "Starting HTTP server on {}" , address) ;
92-
91+
9392 let make_svc = make_service_fn ( move |_conn| {
9493 let state = state_clone. clone ( ) ;
9594 async move {
@@ -98,19 +97,18 @@ pub fn start_server(
9897 } ) )
9998 }
10099 } ) ;
101-
100+
102101 let server = Server :: bind ( & address) . serve ( make_svc) ;
103-
104102 let shutdown_signal = {
105103 let mut state = state. lock ( ) . unwrap ( ) ;
106104 state. shutdown_signal . take ( ) . unwrap ( )
107105 } ;
108-
106+
109107 let server = server. with_graceful_shutdown ( async {
110108 shutdown_signal. await . ok ( ) ;
111109 info ! ( target: "helios::server" , "Server shutting down gracefully" ) ;
112110 } ) ;
113-
111+
114112 info ! ( target: "helios::server" , "HTTP server running at {}" , address) ;
115113 if let Err ( e) = server. await {
116114 error ! ( target: "helios::server" , "Server error: {}" , e) ;
0 commit comments