discovery.rs

use std::collections::BTreeMap;

use product_config::writer::to_hadoop_xml;
use snafu::{ResultExt, Snafu};
use stackable_operator::{
    builder::{configmap::ConfigMapBuilder, meta::ObjectMetaBuilder},
    commons::product_image_selection::ResolvedProductImage,
    k8s_openapi::api::core::v1::ConfigMap,
    kube::runtime::reflector::ObjectRef,
    utils::cluster_info::KubernetesClusterInfo,
};

use crate::{
    crd::{HBASE_SITE_XML, HbaseRole, v1alpha1},
    hbase_controller::build_recommended_labels,
    kerberos::{self, kerberos_discovery_config_properties},
    zookeeper::ZookeeperConnectionInformation,
};

type Result<T, E = Error> = std::result::Result<T, E>;

#[derive(Snafu, Debug)]
pub enum Error {
    #[snafu(display("object {hbase} is missing metadata to build owner reference"))]
    ObjectMissingMetadataForOwnerRef {
        source: stackable_operator::builder::meta::Error,
        hbase: ObjectRef<v1alpha1::HbaseCluster>,
    },

    #[snafu(display("failed to build ConfigMap"))]
    BuildConfigMap {
        source: stackable_operator::builder::configmap::Error,
    },

    #[snafu(display("failed to build object meta data"))]
    ObjectMeta {
        source: stackable_operator::builder::meta::Error,
    },

    #[snafu(display("failed to add Kerberos discovery"))]
    AddKerberosDiscovery { source: kerberos::Error },
}

/// Creates a discovery config map containing the `hbase-site.xml` for clients.
pub fn build_discovery_configmap(
    hbase: &v1alpha1::HbaseCluster,
    cluster_info: &KubernetesClusterInfo,
    zookeeper_connection_information: &ZookeeperConnectionInformation,
    resolved_product_image: &ResolvedProductImage,
) -> Result<ConfigMap> {
    let mut hbase_site = zookeeper_connection_information.as_hbase_settings();
    hbase_site.extend(
        kerberos_discovery_config_properties(hbase, cluster_info)
            .context(AddKerberosDiscoverySnafu)?,
    );

    ConfigMapBuilder::new()
        .metadata(
            ObjectMetaBuilder::new()
                .name_and_namespace(hbase)
                .ownerreference_from_resource(hbase, None, Some(true))
                .with_context(|_| ObjectMissingMetadataForOwnerRefSnafu {
                    hbase: ObjectRef::from_obj(hbase),
                })?
                .with_recommended_labels(build_recommended_labels(
                    hbase,
                    &resolved_product_image.app_version_label_value,
                    &HbaseRole::RegionServer.to_string(),
                    "discovery",
                ))
                .context(ObjectMetaSnafu)?
                .build(),
        )
        .add_data(
            HBASE_SITE_XML,
            to_hadoop_xml(
                hbase_site
                    .into_iter()
                    .map(|(k, v)| (k, Some(v)))
                    .collect::<BTreeMap<_, _>>()
                    .iter(),
            ),
        )
        .build()
        .context(BuildConfigMapSnafu)
}