Skip to content

Commit 420f917

Browse files
committed
use validated cluster in places where raw was used
1 parent 3225201 commit 420f917

2 files changed

Lines changed: 19 additions & 16 deletions

File tree

rust/operator-binary/src/service.rs

Lines changed: 11 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,7 @@ use crate::{
1414
ZOOKEEPER_LEADER_PORT_NAME, v1alpha1,
1515
},
1616
utils::build_recommended_labels,
17-
zk_controller::ZK_CONTROLLER_NAME,
17+
zk_controller::{ZK_CONTROLLER_NAME, validate::ValidatedCluster},
1818
};
1919

2020
#[derive(Snafu, Debug)]
@@ -39,17 +39,17 @@ pub enum Error {
3939
///
4040
/// This is mostly useful for internal communication between peers, or for clients that perform client-side load balancing.
4141
pub(crate) fn build_server_rolegroup_headless_service(
42-
zk: &v1alpha1::ZookeeperCluster,
42+
cluster: &ValidatedCluster,
4343
rolegroup: &RoleGroupRef<v1alpha1::ZookeeperCluster>,
4444
resolved_product_image: &ResolvedProductImage,
4545
) -> Result<Service, Error> {
4646
let metadata = ObjectMetaBuilder::new()
47-
.name_and_namespace(zk)
47+
.name_and_namespace(cluster)
4848
.name(rolegroup.rolegroup_headless_service_name())
49-
.ownerreference_from_resource(zk, None, Some(true))
49+
.ownerreference_from_resource(cluster, None, Some(true))
5050
.context(ObjectMissingMetadataForOwnerRefSnafu)?
5151
.with_recommended_labels(&build_recommended_labels(
52-
zk,
52+
cluster,
5353
ZK_CONTROLLER_NAME,
5454
&resolved_product_image.app_version_label_value,
5555
&rolegroup.role,
@@ -59,7 +59,7 @@ pub(crate) fn build_server_rolegroup_headless_service(
5959
.build();
6060

6161
let service_selector_labels =
62-
Labels::role_group_selector(zk, APP_NAME, &rolegroup.role, &rolegroup.role_group)
62+
Labels::role_group_selector(cluster, APP_NAME, &rolegroup.role, &rolegroup.role_group)
6363
.context(BuildLabelSnafu)?;
6464

6565
let service_spec = ServiceSpec {
@@ -94,18 +94,18 @@ pub(crate) fn build_server_rolegroup_headless_service(
9494

9595
/// The rolegroup [`Service`] for exposing metrics
9696
pub(crate) fn build_server_rolegroup_metrics_service(
97-
zk: &v1alpha1::ZookeeperCluster,
97+
cluster: &ValidatedCluster,
9898
rolegroup: &RoleGroupRef<v1alpha1::ZookeeperCluster>,
9999
resolved_product_image: &ResolvedProductImage,
100100
metrics_port: u16,
101101
) -> Result<Service, Error> {
102102
let metadata = ObjectMetaBuilder::new()
103-
.name_and_namespace(zk)
103+
.name_and_namespace(cluster)
104104
.name(rolegroup.rolegroup_metrics_service_name())
105-
.ownerreference_from_resource(zk, None, Some(true))
105+
.ownerreference_from_resource(cluster, None, Some(true))
106106
.context(ObjectMissingMetadataForOwnerRefSnafu)?
107107
.with_recommended_labels(&build_recommended_labels(
108-
zk,
108+
cluster,
109109
ZK_CONTROLLER_NAME,
110110
&resolved_product_image.app_version_label_value,
111111
&rolegroup.role,
@@ -117,7 +117,7 @@ pub(crate) fn build_server_rolegroup_metrics_service(
117117
.build();
118118

119119
let service_selector_labels =
120-
Labels::role_group_selector(zk, APP_NAME, &rolegroup.role, &rolegroup.role_group)
120+
Labels::role_group_selector(cluster, APP_NAME, &rolegroup.role, &rolegroup.role_group)
121121
.context(BuildLabelSnafu)?;
122122

123123
let service_spec = ServiceSpec {

rust/operator-binary/src/zk_controller.rs

Lines changed: 8 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -84,7 +84,7 @@ use crate::{
8484

8585
pub(crate) mod build;
8686
mod dereference;
87-
mod validate;
87+
pub(crate) mod validate;
8888

8989
pub const ZK_CONTROLLER_NAME: &str = "zookeepercluster";
9090
pub const ZK_FULL_CONTROLLER_NAME: &str = concatcp!(ZK_CONTROLLER_NAME, '.', OPERATOR_NAME);
@@ -366,11 +366,14 @@ pub async fn reconcile_zk(
366366
let metrics_port =
367367
build::properties::zoo_cfg::metrics_http_port(&validated_cluster, rolegroup_config);
368368

369-
let rg_headless_service =
370-
build_server_rolegroup_headless_service(zk, &rolegroup, resolved_product_image)
371-
.context(BuildServiceSnafu)?;
369+
let rg_headless_service = build_server_rolegroup_headless_service(
370+
&validated_cluster,
371+
&rolegroup,
372+
resolved_product_image,
373+
)
374+
.context(BuildServiceSnafu)?;
372375
let rg_metrics_service = build_server_rolegroup_metrics_service(
373-
zk,
376+
&validated_cluster,
374377
&rolegroup,
375378
resolved_product_image,
376379
metrics_port,

0 commit comments

Comments
 (0)