Skip to content

Commit d6fb58d

Browse files
committed
small refactoring for service name functions
1 parent 2742b45 commit d6fb58d

2 files changed

Lines changed: 4 additions & 26 deletions

File tree

rust/operator-binary/src/druid_controller.rs

Lines changed: 2 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -88,10 +88,7 @@ use crate::{
8888
},
8989
operations::{graceful_shutdown::add_graceful_shutdown_config, pdb::add_pdbs},
9090
product_logging::extend_role_group_config_map,
91-
service::{
92-
build_rolegroup_headless_service, build_rolegroup_metrics_service,
93-
rolegroup_headless_service_name,
94-
},
91+
service::{build_rolegroup_headless_service, build_rolegroup_metrics_service},
9592
};
9693

9794
pub const DRUID_CONTROLLER_NAME: &str = "druidcluster";
@@ -1222,9 +1219,7 @@ fn build_rolegroup_statefulset(
12221219
),
12231220
..LabelSelector::default()
12241221
},
1225-
service_name: Some(rolegroup_headless_service_name(
1226-
&rolegroup_ref.object_name(),
1227-
)),
1222+
service_name: Some(rolegroup_ref.rolegroup_headless_service_name()),
12281223
template: pod_template,
12291224
volume_claim_templates: pvcs,
12301225
..StatefulSetSpec::default()

rust/operator-binary/src/service.rs

Lines changed: 2 additions & 19 deletions
Original file line numberDiff line numberDiff line change
@@ -12,9 +12,6 @@ use crate::crd::{
1212
DruidRole, METRICS_PORT, METRICS_PORT_NAME, security::DruidTlsSecurity, v1alpha1,
1313
};
1414

15-
const METRICS_SERVICE_SUFFIX: &str = "metrics";
16-
const HEADLESS_SERVICE_SUFFIX: &str = "headless";
17-
1815
#[derive(Snafu, Debug)]
1916
pub enum Error {
2017
#[snafu(display("object is missing metadata to build owner reference"))]
@@ -46,9 +43,7 @@ pub fn build_rolegroup_headless_service(
4643
Ok(Service {
4744
metadata: ObjectMetaBuilder::new()
4845
.name_and_namespace(druid)
49-
.name(rolegroup_headless_service_name(
50-
&role_group_ref.object_name(),
51-
))
46+
.name(role_group_ref.rolegroup_headless_service_name())
5247
.ownerreference_from_resource(druid, None, Some(true))
5348
.context(ObjectMissingMetadataForOwnerRefSnafu)?
5449
.with_recommended_labels(object_labels)
@@ -77,9 +72,7 @@ pub fn build_rolegroup_metrics_service(
7772
Ok(Service {
7873
metadata: ObjectMetaBuilder::new()
7974
.name_and_namespace(druid)
80-
.name(rolegroup_metrics_service_name(
81-
&role_group_ref.object_name(),
82-
))
75+
.name(role_group_ref.rolegroup_metrics_service_name())
8376
.ownerreference_from_resource(druid, None, Some(true))
8477
.context(ObjectMissingMetadataForOwnerRefSnafu)?
8578
.with_recommended_labels(object_labels)
@@ -109,16 +102,6 @@ fn metrics_service_ports() -> Vec<ServicePort> {
109102
}]
110103
}
111104

112-
/// Returns the metrics rolegroup service name `<cluster>-<role>-<rolegroup>-<METRICS_SERVICE_SUFFIX>`.
113-
fn rolegroup_metrics_service_name(role_group_ref_object_name: &str) -> String {
114-
format!("{role_group_ref_object_name}-{METRICS_SERVICE_SUFFIX}")
115-
}
116-
117-
/// Returns the headless rolegroup service name `<cluster>-<role>-<rolegroup>-<HEADLESS_SERVICE_SUFFIX>`.
118-
pub fn rolegroup_headless_service_name(role_group_ref_object_name: &str) -> String {
119-
format!("{role_group_ref_object_name}-{HEADLESS_SERVICE_SUFFIX}")
120-
}
121-
122105
/// Common annotations for Prometheus
123106
///
124107
/// These annotations can be used in a ServiceMonitor.

0 commit comments

Comments
 (0)