File tree Expand file tree Collapse file tree 1 file changed +1
-37
lines changed
Expand file tree Collapse file tree 1 file changed +1
-37
lines changed Original file line number Diff line number Diff line change @@ -26,7 +26,7 @@ fn main() {
2626async fn main_inner ( ) {
2727 env_logger:: init ( ) ;
2828
29- let cancel_token = shutdown_token ( ) ;
29+ let cancel_token = CancellationToken :: new ( ) ;
3030 let env_vars = Arc :: new ( EnvVars :: from_env ( ) . unwrap ( ) ) ;
3131 let opt = opt:: Opt :: parse ( ) ;
3232
@@ -65,39 +65,3 @@ async fn main_inner() {
6565 )
6666 . await ;
6767}
68-
69- fn shutdown_token ( ) -> CancellationToken {
70- use tokio:: signal;
71-
72- let cancel_token = CancellationToken :: new ( ) ;
73- let cancel_token_clone = cancel_token. clone ( ) ;
74-
75- async fn shutdown_signal_handler ( ) {
76- let ctrl_c = async {
77- signal:: ctrl_c ( ) . await . unwrap ( ) ;
78- } ;
79-
80- #[ cfg( unix) ]
81- let terminate = async {
82- signal:: unix:: signal ( signal:: unix:: SignalKind :: terminate ( ) )
83- . unwrap ( )
84- . recv ( )
85- . await ;
86- } ;
87-
88- #[ cfg( not( unix) ) ]
89- let terminate = std:: future:: pending :: < ( ) > ( ) ;
90-
91- tokio:: select! {
92- _ = ctrl_c => { } ,
93- _ = terminate => { } ,
94- } ;
95- }
96-
97- tokio:: spawn ( async move {
98- shutdown_signal_handler ( ) . await ;
99- cancel_token_clone. cancel ( ) ;
100- } ) ;
101-
102- cancel_token
103- }
You can’t perform that action at this time.
0 commit comments