diff --git a/shenyu-client-rust/src/core.rs b/shenyu-client-rust/src/core.rs index 2da6279..eaa0248 100644 --- a/shenyu-client-rust/src/core.rs +++ b/shenyu-client-rust/src/core.rs @@ -62,11 +62,15 @@ impl ShenyuClient { /// Register to shenyu admin. pub fn register(&self) -> Result<(), Error> { if let Ok(token) = self.get_register_token() { + info!( + "[SUCCESS], get register token success, register token: {:#?}", + &token + ); _ = self .headers .insert("X-Access-Token".to_string(), token.to_string()); } else { - return Err(Error::new(ErrorKind::Other, "Can't get register token")); + error!("Can't get register token"); } self.register_all_metadata(true); self.register_uri(); @@ -316,7 +320,7 @@ impl ShenyuClient { if let Ok(true) = self.request(url, &json_data) { info!( "[SUCCESS], register metadata success, register data: {:#?}", - json_data + &json_data ); continue; } @@ -355,7 +359,7 @@ impl ShenyuClient { if let Ok(true) = self.request(url, &json_data) { info!( "[SUCCESS], register discover config success, register data: {:#?}", - json_data + &json_data ); continue; } @@ -389,7 +393,7 @@ impl ShenyuClient { if let Ok(true) = self.request(url, &json_data) { info!( "[SUCCESS], offline success, register data: {:#?}", - json_data + &json_data ); continue; } diff --git a/shenyu-client-rust/src/lib.rs b/shenyu-client-rust/src/lib.rs index 37be84f..b143479 100644 --- a/shenyu-client-rust/src/lib.rs +++ b/shenyu-client-rust/src/lib.rs @@ -427,25 +427,8 @@ mod tests_axum { let res = ShenyuClient::new(config, app.app_name(), app.uri_infos(), 9527); assert!(&res.is_ok()); let client = &mut res.unwrap(); - println!( - "shenyu-client-rust.token: {:?}", - client - .headers - .get("X-Access-Token") - .map(|r| r.clone()) - .unwrap_or("None".to_string()) - ); - - if let Ok(token) = client.get_register_token() { - _ = client - .headers - .insert("X-Access-Token".to_string(), token.to_string()); - } else { - panic!("Can't get register token"); - } - client.register_all_metadata(true); - client.register_uri(); - client.register_discovery_config(); + + client.register().unwrap(); client.offline_register(); } @@ -478,25 +461,8 @@ mod tests_actix_web { let res = ShenyuClient::new(config, app.app_name(), app.uri_infos(), 9527); assert!(&res.is_ok()); let client = &mut res.unwrap(); - println!( - "shenyu-client-rust.token: {:?}", - client - .headers - .get("X-Access-Token") - .map(|r| r.clone()) - .unwrap_or("None".to_string()) - ); - - if let Ok(token) = client.get_register_token() { - _ = client - .headers - .insert("X-Access-Token".to_string(), token.to_string()); - } else { - panic!("Can't get register token"); - } - client.register_all_metadata(true); - client.register_uri(); - client.register_discovery_config(); + + client.register().unwrap(); client.offline_register(); } }