Allow specifying extra extensions to enable in Druid

CHANGELOG.md

@@ -11,6 +11,7 @@ All notable changes to this project will be documented in this file.
 - Deploy default and support custom affinities ([#406]).
 - Log aggregation added ([#407]).
 - Added the ability to mount extra volumes for files that may be needed for ingestion tasks to work ([#415])
+- Add support for specifying additional extensions to enable ([#416])
 
 ### Changed
 
@@ -30,6 +31,7 @@ All notable changes to this project will be documented in this file.
 [#407]: https://github.com/stackabletech/druid-operator/pull/407
 [#408]: https://github.com/stackabletech/druid-operator/pull/408
 [#415]: https://github.com/stackabletech/druid-operator/pull/415
+[#416]: https://github.com/stackabletech/druid-operator/pull/416
 
 ## [23.1.0] - 2023-01-23
 

deploy/helm/druid-operator/crds/crds.yaml

@@ -1298,6 +1298,11 @@ spec:
                 clusterConfig:
                   description: Common cluster wide configuration that can not differ or be overridden on a role or role group level
                   properties:
+                    additionalExtensions:
+                      description: Additional extensions to enable in Druid The operator will automatically enable all extensions needed based on the cluster configuration, but for extra functionality which the operator cannot anticipate, it can sometimes be necessary to load additional extensions Anything specified here will be added to the list generated by the operator, not fully replace it.
+                      items:
+                        type: string
+                      type: array
                     authentication:
                       default: []
                       description: List of Authentication classes using like TLS or LDAP to authenticate users

rust/crd/src/lib.rs

@@ -236,6 +236,14 @@ pub struct DruidClusterConfig {
     /// These volumes will be mounted into all pods below `/stackable/userdata/{volumename}`
     #[serde(default, skip_serializing_if = "Vec::is_empty")]
     pub extra_volumes: Vec<Volume>,
+    /// Additional extensions to enable in Druid
+    /// The operator will automatically enable all extensions needed based on the cluster
+    /// configuration, but for extra functionality which the operator cannot anticipate, it can
+    /// sometimes be necessary to load additional extensions
+    /// Anything specified here will be added to the list generated by the operator, not fully
+    /// replace it.
+    #[serde(default, skip_serializing_if = "Vec::is_empty")]
+    pub additional_extensions: Vec<String>,
 }
 
 /// Common configuration for all role groups

rust/operator-binary/src/extensions.rs

@@ -38,5 +38,18 @@ pub fn get_extension_list(
         extensions.push(EXT_S3.to_string());
     }
 
+    // Add user specified extensions to the list of loaded extensions if any are present
+    for additional_extension in &druid.spec.cluster_config.additional_extensions {
+        if extensions.contains(additional_extension) {
+            tracing::warn!("Skipping user specified extension [{additional_extension}] as it was already added to the extensions.");
+        } else {
+            tracing::info!(
+                "Adding user specified extension [{additional_extension}] to list of enabled extensions."
+            );
+            extensions.push(additional_extension.to_string());
+        }
+    }
+
+    tracing::debug!(?extensions, "Finished generating extension list",);
     extensions
 }