#![allow(unused_imports)]
use super::builders::*;
use super::client::RegisteredModelServiceClient;
use unitycatalog_common::models::registered_models::v1::*;
#[derive(Clone)]
pub struct RegisteredModelClient {
pub(crate) catalog_name: String,
pub(crate) schema_name: String,
pub(crate) registered_model_name: String,
pub(crate) client: RegisteredModelServiceClient,
}
impl RegisteredModelClient {
pub fn new(
catalog_name: impl Into<String>,
schema_name: impl Into<String>,
registered_model_name: impl Into<String>,
client: RegisteredModelServiceClient,
) -> Self {
Self {
catalog_name: catalog_name.into(),
schema_name: schema_name.into(),
registered_model_name: registered_model_name.into(),
client,
}
}
pub fn from_full_name(
full_name: impl Into<String>,
client: RegisteredModelServiceClient,
) -> 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 registered_model_name = parts.next().unwrap_or_default();
Self::new(catalog_name, schema_name, registered_model_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.registered_model_name
}
pub fn full_name(&self) -> String {
format!(
"{}.{}.{}",
self.catalog_name, self.schema_name, self.registered_model_name
)
}
pub fn get(&self) -> GetRegisteredModelBuilder {
GetRegisteredModelBuilder::new(
self.client.clone(),
format!(
"{}.{}.{}",
self.catalog_name, self.schema_name, self.registered_model_name
),
)
}
pub fn update(&self) -> UpdateRegisteredModelBuilder {
UpdateRegisteredModelBuilder::new(
self.client.clone(),
format!(
"{}.{}.{}",
self.catalog_name, self.schema_name, self.registered_model_name
),
)
}
pub fn delete(&self) -> DeleteRegisteredModelBuilder {
DeleteRegisteredModelBuilder::new(
self.client.clone(),
format!(
"{}.{}.{}",
self.catalog_name, self.schema_name, self.registered_model_name
),
)
}
}