diff --git a/src/main.rs b/src/main.rs index 2d59208..4ed980a 100644 --- a/src/main.rs +++ b/src/main.rs @@ -1,4 +1,4 @@ -use std::{fs::File}; +use std::{env, fs::File}; use env_logger::Builder; use libtbr::recipe_functions::common; @@ -27,6 +27,11 @@ async fn setup_registry(config: gcm::Configure) -> gcm::StandardResult { let name = cfg_clone.get("SERVICE_NAME").expect("missing service name"); let url = cfg_clone.get("SERVICE_URL").expect("missing service url"); + let token = cfg_clone.get("REGISTRY_TOKEN").expect("token not provided"); + unsafe { + env::set_var("REGISTRY_TOKEN", token); + } + let mut client = RegistryClient::connect(register_server_address.clone()).await?; reg::register_service(&mut client, name, url).await; diff --git a/src/reg.rs b/src/reg.rs index acac43b..099ead8 100644 --- a/src/reg.rs +++ b/src/reg.rs @@ -1,5 +1,6 @@ use std::time::Duration; +use log::{error, info}; use tokio::time::sleep; use tonic::transport::Channel; @@ -17,7 +18,11 @@ pub async fn register_service(client: &mut RegistryClient, name: &str, token: std::env::var("REGISTRY_TOKEN").unwrap_or_default(), }); - let _ = client.register(req).await; + let resp = client.register(req).await; + match resp { + Ok(r) => info!("{:?}", r), + Err(e) => error!("error register: {:?}", e) + } } pub async fn heartbeat_loop(name: String, url: String) {