mod.rs

pub mod black_hole;
pub mod commons;
pub mod config;
pub mod delta_lake;
pub mod generic;
pub mod google_sheet;
pub mod hive;
pub mod iceberg;
pub mod postgresql;
pub mod tpcds;
pub mod tpch;

use async_trait::async_trait;
use snafu::Snafu;
use stackable_operator::{client::Client, v2::types::kubernetes::NamespaceName};

use crate::controller::dereference::TrinoCatalogName;

use self::config::CatalogConfig;

#[derive(Debug, Snafu)]
#[snafu(module)]
pub enum FromTrinoCatalogError {
    #[snafu(display("failed to configure S3 connection"))]
    ConfigureS3 {
        source: stackable_operator::crd::s3::v1alpha1::ConnectionError,
    },

    #[snafu(display("trino does not support disabling the TLS verification of S3 servers"))]
    S3TlsNoVerificationNotSupported,

    #[snafu(display("trino 469 and greater require TLS for S3"))]
    S3TlsRequired,

    #[snafu(display("trino catalog has no name set"))]
    InvalidCatalogSpec,

    #[snafu(display("failed to resolve [{catalog}] discovery config map [{cm_name}]"))]
    FailedToGetDiscoveryConfigMap {
        source: stackable_operator::client::Error,
        catalog: String,
        cm_name: String,
    },

    #[snafu(display("failed to retrieve [{catalog}] discovery config map [{cm_name}] data field"))]
    FailedToGetDiscoveryConfigMapData { catalog: String, cm_name: String },

    #[snafu(display(
        "failed to retrieve [{catalog}] discovery config map [{cm_name}] data key [{data_key}]"
    ))]
    FailedToGetDiscoveryConfigMapDataKey {
        catalog: String,
        cm_name: String,
        data_key: String,
    },

    #[snafu(display("failed to get PostgreSQL connection details"))]
    GetPostgresConnectionDetails {
        source: stackable_operator::database_connections::Error,
    },
}

#[async_trait]
pub trait ToCatalogConfig {
    async fn to_catalog_config(
        &self,
        catalog_name: &TrinoCatalogName,
        catalog_namespace: &NamespaceName,
        client: &Client,
        trino_version: u16,
    ) -> Result<CatalogConfig, FromTrinoCatalogError>;
}

#[async_trait]
pub trait ExtendCatalogConfig {
    async fn extend_catalog_config(
        &self,
        catalog_config: &mut CatalogConfig,
        catalog_name: &TrinoCatalogName,
        catalog_namespace: &NamespaceName,
        client: &Client,
        trino_version: u16,
    ) -> Result<(), FromTrinoCatalogError>;
}