67 lines
1.9 KiB
Rust
67 lines
1.9 KiB
Rust
use std::collections::HashMap;
|
|
use serde::Deserialize;
|
|
|
|
#[derive(Debug, Deserialize)]
|
|
pub struct ServiceNode {
|
|
pub ServiceAddress: String,
|
|
pub ServicePort: u16,
|
|
pub ServiceTags: Vec<String>,
|
|
pub ServiceMeta: HashMap<String, String>,
|
|
}
|
|
|
|
pub async fn get_service_address(consul_url: &str, service_name: &str) -> Result<Vec<ServiceNode>, Box<dyn std::error::Error>> {
|
|
let client = reqwest::Client::new();
|
|
let consul_service_url = format!("{}/v1/catalog/service/{}", consul_url, service_name);
|
|
|
|
let response = client
|
|
.get(&consul_service_url)
|
|
.send()
|
|
.await?;
|
|
|
|
if !response.status().is_success() {
|
|
return Err(format!(
|
|
"Failed to fetch service nodes for '{}': {}",
|
|
service_name, response.status()
|
|
)
|
|
.into());
|
|
}
|
|
|
|
// Deserialize the response into a Vec<ServiceNode>
|
|
let nodes: Vec<ServiceNode> = response.json().await?;
|
|
|
|
if nodes.is_empty() {
|
|
Err(format!("No nodes found for service '{}'", service_name).into())
|
|
} else {
|
|
Ok(nodes)
|
|
}
|
|
}
|
|
async fn get_services_with_tag(
|
|
service_name: &str,
|
|
tag: &str,
|
|
consul_url: &str,
|
|
) -> Result<Vec<ServiceNode>, Box<dyn std::error::Error>> {
|
|
let url = format!("{}/v1/catalog/service/{}", consul_url, service_name);
|
|
|
|
let client = reqwest::Client::new();
|
|
let response = client.get(&url).send().await?;
|
|
|
|
if !response.status().is_success() {
|
|
return Err(format!(
|
|
"Failed to fetch service nodes for '{}': {}",
|
|
service_name, response.status()
|
|
)
|
|
.into());
|
|
}
|
|
|
|
// Deserialize the response into a Vec<ServiceNode>
|
|
let nodes: Vec<ServiceNode> = response.json().await?;
|
|
|
|
// Filter nodes that include the specified tag
|
|
let filtered_nodes = nodes
|
|
.into_iter()
|
|
.filter(|node| node.ServiceTags.contains(&tag.to_string()))
|
|
.collect();
|
|
|
|
Ok(filtered_nodes)
|
|
}
|