@@ -6,7 +6,6 @@ use std::sync::Arc;
66use std:: time:: Duration ;
77
88use anyhow:: { Context , Result } ;
9- use tokio:: time:: sleep;
109use cert_client:: CertRequestClient ;
1110use dstack_guest_agent_rpc:: {
1211 dstack_guest_server:: { DstackGuestRpc , DstackGuestServer } ,
@@ -30,6 +29,7 @@ use ring::rand::{SecureRandom, SystemRandom};
3029use serde_json:: json;
3130use sha3:: { Digest , Keccak256 } ;
3231use tdx_attest:: eventlog:: read_event_logs;
32+ use tokio:: time:: sleep;
3333
3434use crate :: config:: Config ;
3535
@@ -59,7 +59,7 @@ impl AppState {
5959 . await
6060 . context ( "Failed to create cert signer" ) ?;
6161 let key = KeyPair :: generate ( ) . context ( "Failed to generate demo key" ) ?;
62-
62+
6363 // Retry certificate request with exponential backoff
6464 let mut retries = 0 ;
6565 const MAX_RETRIES : u32 = 5 ;
@@ -199,7 +199,11 @@ impl DstackGuestRpc for InternalRpcHandler {
199199 . state
200200 . inner
201201 . cert_client
202- . request_cert ( & derived_key, config. clone ( ) , self . state . config ( ) . simulator . enabled )
202+ . request_cert (
203+ & derived_key,
204+ config. clone ( ) ,
205+ self . state . config ( ) . simulator . enabled ,
206+ )
203207 . await
204208 {
205209 Ok ( cert) => break cert,
@@ -359,7 +363,11 @@ impl TappdRpc for InternalRpcHandlerV0 {
359363 . state
360364 . inner
361365 . cert_client
362- . request_cert ( & derived_key, config. clone ( ) , self . state . config ( ) . simulator . enabled )
366+ . request_cert (
367+ & derived_key,
368+ config. clone ( ) ,
369+ self . state . config ( ) . simulator . enabled ,
370+ )
363371 . await
364372 {
365373 Ok ( cert) => break cert,
0 commit comments