Add headless service, move listener logic to own file

Author: Maleware

rust/operator-binary/src/crd/mod.rs

@@ -19,7 +19,7 @@ use stackable_operator::{
         fragment::{self, Fragment, ValidationError},
         merge::Merge,
     },
-    k8s_openapi::apimachinery::pkg::api::resource::Quantity,
+    k8s_openapi::{api::core::v1::ServicePort, apimachinery::pkg::api::resource::Quantity},
     kube::{CustomResource, ResourceExt, runtime::reflector::ObjectRef},
     memory::{BinaryMultiple, MemoryQuantity},
     product_config_utils::{self, Configuration},
@@ -32,7 +32,10 @@ use stackable_operator::{
 };
 use strum::{Display, EnumIter, EnumString, IntoEnumIterator};
 
-use crate::crd::v1alpha1::{SupersetConfigFragment, SupersetRoleConfig};
+use crate::{
+    crd::v1alpha1::{SupersetConfigFragment, SupersetRoleConfig},
+    listener::default_listener_class,
+};
 
 pub mod affinity;
 pub mod authentication;
@@ -49,9 +52,6 @@ pub const MAX_LOG_FILES_SIZE: MemoryQuantity = MemoryQuantity {
     unit: BinaryMultiple::Mebi,
 };
 
-pub const LISTENER_VOLUME_NAME: &str = "listener";
-pub const LISTENER_VOLUME_DIR: &str = "/stackable/listener";
-
 pub const APP_PORT_NAME: &str = "http";
 pub const APP_PORT: u16 = 8088;
 pub const METRICS_PORT_NAME: &str = "metrics";
@@ -311,10 +311,6 @@ impl Default for v1alpha1::SupersetRoleConfig {
     }
 }
 
-fn default_listener_class() -> String {
-    "cluster-internal".to_string()
-}
-
 #[derive(Clone, Debug, Deserialize, Eq, PartialEq, Serialize)]
 #[serde(rename_all = "camelCase")]
 pub struct SupersetCredentials {
@@ -528,6 +524,43 @@ impl v1alpha1::SupersetCluster {
         }
     }
 
+    /// Set of functions to define service names on rolegroup level.
+    /// Headless service for cluster internal purposes only.
+    // TODO: Move to operator-rs
+    pub fn rolegroup_headless_service_name(
+        &self,
+        rolegroup: &RoleGroupRef<v1alpha1::SupersetCluster>,
+    ) -> String {
+        format!("{name}-headless", name = rolegroup.object_name())
+    }
+
+    /// Headless metrics service exposes Prometheus endpoint only
+    // TODO: Move to operator-rs
+    pub fn rolegroup_headless_metrics_service_name(
+        &self,
+        rolegroup: &RoleGroupRef<v1alpha1::SupersetCluster>,
+    ) -> String {
+        format!("{name}-metrics", name = rolegroup.object_name())
+    }
+
+    pub fn metrics_ports(&self) -> Vec<ServicePort> {
+        vec![ServicePort {
+            name: Some(METRICS_PORT_NAME.to_string()),
+            port: METRICS_PORT.into(),
+            protocol: Some("TCP".to_string()),
+            ..ServicePort::default()
+        }]
+    }
+
+    pub fn service_ports(&self) -> Vec<ServicePort> {
+        vec![ServicePort {
+            name: Some(APP_PORT_NAME.to_string()),
+            port: APP_PORT.into(),
+            protocol: Some("TCP".to_string()),
+            ..ServicePort::default()
+        }]
+    }
+
     pub fn generic_role_config(&self, role: &SupersetRole) -> Option<GenericRoleConfig> {
         self.get_role_config(role).map(|r| r.common.to_owned())
     }

rust/operator-binary/src/listener.rs

@@ -0,0 +1,61 @@
+use snafu::{ResultExt, Snafu};
+use stackable_operator::{builder::meta::ObjectMetaBuilder, crd::listener, kvp::ObjectLabels};
+
+use crate::crd::{APP_PORT, APP_PORT_NAME, v1alpha1};
+
+pub const LISTENER_VOLUME_NAME: &str = "listener";
+pub const LISTENER_VOLUME_DIR: &str = "/stackable/listener";
+
+#[derive(Snafu, Debug)]
+pub enum Error {
+    #[snafu(display("object is missing metadata to build owner reference"))]
+    ObjectMissingMetadataForOwnerRef {
+        source: stackable_operator::builder::meta::Error,
+    },
+    #[snafu(display("failed to build Metadata"))]
+    MetadataBuild {
+        source: stackable_operator::builder::meta::Error,
+    },
+}
+
+pub fn build_group_listener(
+    superset: &v1alpha1::SupersetCluster,
+    object_labels: ObjectLabels<v1alpha1::SupersetCluster>,
+    listener_class: String,
+    listener_group_name: String,
+) -> Result<listener::v1alpha1::Listener, Error> {
+    let metadata = ObjectMetaBuilder::new()
+        .name_and_namespace(superset)
+        .name(listener_group_name)
+        .ownerreference_from_resource(superset, None, Some(true))
+        .context(ObjectMissingMetadataForOwnerRefSnafu)?
+        .with_recommended_labels(object_labels)
+        .context(MetadataBuildSnafu)?
+        .build();
+
+    let spec = listener::v1alpha1::ListenerSpec {
+        class_name: Some(listener_class),
+        ports: Some(listener_ports()),
+        ..Default::default()
+    };
+
+    let listener = listener::v1alpha1::Listener {
+        metadata,
+        spec,
+        status: None,
+    };
+
+    Ok(listener)
+}
+
+pub fn listener_ports() -> Vec<listener::v1alpha1::ListenerPort> {
+    vec![listener::v1alpha1::ListenerPort {
+        name: APP_PORT_NAME.to_owned(),
+        port: APP_PORT.into(),
+        protocol: Some("TCP".to_owned()),
+    }]
+}
+
+pub fn default_listener_class() -> String {
+    "cluster-internal".to_string()
+}

rust/operator-binary/src/main.rs

@@ -42,6 +42,7 @@ mod config;
 mod controller_commons;
 mod crd;
 mod druid_connection_controller;
+mod listener;
 mod operations;
 mod product_logging;
 mod rbac;