fix: missing token
This commit is contained in:
parent
9f4fb6c274
commit
4e3b561f61
2 changed files with 12 additions and 2 deletions
|
|
@ -1,4 +1,4 @@
|
||||||
use std::{fs::File};
|
use std::{env, fs::File};
|
||||||
|
|
||||||
use env_logger::Builder;
|
use env_logger::Builder;
|
||||||
use libtbr::recipe_functions::common;
|
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 name = cfg_clone.get("SERVICE_NAME").expect("missing service name");
|
||||||
let url = cfg_clone.get("SERVICE_URL").expect("missing service url");
|
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?;
|
let mut client = RegistryClient::connect(register_server_address.clone()).await?;
|
||||||
reg::register_service(&mut client, name, url).await;
|
reg::register_service(&mut client, name, url).await;
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -1,5 +1,6 @@
|
||||||
use std::time::Duration;
|
use std::time::Duration;
|
||||||
|
|
||||||
|
use log::{error, info};
|
||||||
use tokio::time::sleep;
|
use tokio::time::sleep;
|
||||||
use tonic::transport::Channel;
|
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(),
|
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) {
|
pub async fn heartbeat_loop(name: String, url: String) {
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue