#![allow(unused_imports)]
use super::builders::*;
use super::client::AgentServiceClient;
use unitycatalog_common::models::agents::v0alpha1::*;
#[derive(Clone)]
pub struct AgentClient {
pub(crate) catalog_name: String,
pub(crate) schema_name: String,
pub(crate) agent_name: String,
pub(crate) client: AgentServiceClient,
}
impl AgentClient {
pub fn new(
catalog_name: impl Into<String>,
schema_name: impl Into<String>,
agent_name: impl Into<String>,
client: AgentServiceClient,
) -> Self {
Self {
catalog_name: catalog_name.into(),
schema_name: schema_name.into(),
agent_name: agent_name.into(),
client,
}
}
pub fn from_full_name(full_name: impl Into<String>, client: AgentServiceClient) -> Self {
let full_name = full_name.into();
let mut parts = full_name.splitn(3usize, '.');
let catalog_name = parts.next().unwrap_or_default();
let schema_name = parts.next().unwrap_or_default();
let agent_name = parts.next().unwrap_or_default();
Self::new(catalog_name, schema_name, agent_name, client)
}
pub fn catalog_name(&self) -> &str {
&self.catalog_name
}
pub fn schema_name(&self) -> &str {
&self.schema_name
}
pub fn name(&self) -> &str {
&self.agent_name
}
pub fn full_name(&self) -> String {
format!(
"{}.{}.{}",
self.catalog_name, self.schema_name, self.agent_name
)
}
pub fn get(&self) -> GetAgentBuilder {
GetAgentBuilder::new(
self.client.clone(),
format!(
"{}.{}.{}",
self.catalog_name, self.schema_name, self.agent_name
),
)
}
pub fn update(&self) -> UpdateAgentBuilder {
UpdateAgentBuilder::new(
self.client.clone(),
format!(
"{}.{}.{}",
self.catalog_name, self.schema_name, self.agent_name
),
)
}
pub fn delete(&self) -> DeleteAgentBuilder {
DeleteAgentBuilder::new(
self.client.clone(),
format!(
"{}.{}.{}",
self.catalog_name, self.schema_name, self.agent_name
),
)
}
}