fix: missing token

This commit is contained in:
Pakin 2026-01-09 09:46:51 +07:00
parent 9f4fb6c274
commit 4e3b561f61
2 changed files with 12 additions and 2 deletions

View file

@ -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;

View file

@ -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<Channel>, 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) {