Skip to content

Commit cedfb41

Browse files
committed
fix: consolidate merging
1 parent 5d74a83 commit cedfb41

2 files changed

Lines changed: 4 additions & 89 deletions

File tree

rust/operator-binary/src/controller/build/resource/statefulset.rs

Lines changed: 4 additions & 23 deletions
Original file line numberDiff line numberDiff line change
@@ -124,9 +124,6 @@ pub enum Error {
124124
source: stackable_operator::builder::pod::container::Error,
125125
},
126126

127-
#[snafu(display("failed to merge pod overrides"))]
128-
MergePodOverrides { source: crd::role::Error },
129-
130127
#[snafu(display("missing secret lifetime"))]
131128
MissingSecretLifetime,
132129

@@ -472,16 +469,8 @@ pub fn build_broker_rolegroup_statefulset(
472469
// Don't run kcat pod as PID 1, to ensure that default signal handlers apply
473470
pod_template_spec.share_process_namespace = Some(true);
474471

475-
pod_template.merge_from(
476-
kafka_role
477-
.role_pod_overrides(kafka)
478-
.context(MergePodOverridesSnafu)?,
479-
);
480-
pod_template.merge_from(
481-
kafka_role
482-
.role_group_pod_overrides(kafka, role_group_name.as_ref())
483-
.context(MergePodOverridesSnafu)?,
484-
);
472+
// Pod overrides were already merged (role <- role group) during validation.
473+
pod_template.merge_from(validated_rg.pod_overrides.clone());
485474

486475
Ok(StatefulSet {
487476
metadata: ObjectMetaBuilder::new()
@@ -778,16 +767,8 @@ pub fn build_controller_rolegroup_statefulset(
778767

779768
let mut pod_template = pod_builder.build_template();
780769

781-
pod_template.merge_from(
782-
kafka_role
783-
.role_pod_overrides(kafka)
784-
.context(MergePodOverridesSnafu)?,
785-
);
786-
pod_template.merge_from(
787-
kafka_role
788-
.role_group_pod_overrides(kafka, role_group_name.as_ref())
789-
.context(MergePodOverridesSnafu)?,
790-
);
770+
// Pod overrides were already merged (role <- role group) during validation.
771+
pod_template.merge_from(validated_rg.pod_overrides.clone());
791772

792773
Ok(StatefulSet {
793774
metadata: ObjectMetaBuilder::new()

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

Lines changed: 0 additions & 66 deletions
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,6 @@ use serde::{Deserialize, Serialize};
88
use snafu::{OptionExt, ResultExt, Snafu};
99
use stackable_operator::{
1010
commons::resources::{NoRuntimeLimits, Resources},
11-
k8s_openapi::api::core::v1::PodTemplateSpec,
1211
product_logging::spec::ContainerLogConfig,
1312
schemars::{self, JsonSchema},
1413
v2::config_overrides::KeyValueConfigOverrides,
@@ -118,71 +117,6 @@ impl KafkaRole {
118117
"kafka"
119118
}
120119

121-
pub fn role_pod_overrides(
122-
&self,
123-
kafka: &v1alpha1::KafkaCluster,
124-
) -> Result<PodTemplateSpec, Error> {
125-
let pod_overrides = match self {
126-
Self::Broker => kafka
127-
.broker_role()
128-
.with_context(|_| MissingRoleSnafu {
129-
role: self.to_string(),
130-
})?
131-
.config
132-
.pod_overrides
133-
.clone(),
134-
Self::Controller => kafka
135-
.controller_role()
136-
.with_context(|_| MissingRoleSnafu {
137-
role: self.to_string(),
138-
})?
139-
.config
140-
.pod_overrides
141-
.clone(),
142-
};
143-
144-
Ok(pod_overrides)
145-
}
146-
147-
pub fn role_group_pod_overrides(
148-
&self,
149-
kafka: &v1alpha1::KafkaCluster,
150-
rolegroup: &str,
151-
) -> Result<PodTemplateSpec, Error> {
152-
let pod_overrides = match self {
153-
Self::Broker => kafka
154-
.broker_role()
155-
.with_context(|_| MissingRoleSnafu {
156-
role: self.to_string(),
157-
})?
158-
.role_groups
159-
.get(rolegroup)
160-
.with_context(|| MissingRoleGroupSnafu {
161-
role: self.to_string(),
162-
rolegroup: rolegroup.to_string(),
163-
})?
164-
.config
165-
.pod_overrides
166-
.clone(),
167-
Self::Controller => kafka
168-
.controller_role()
169-
.with_context(|_| MissingRoleSnafu {
170-
role: self.to_string(),
171-
})?
172-
.role_groups
173-
.get(rolegroup)
174-
.with_context(|| MissingRoleGroupSnafu {
175-
role: self.to_string(),
176-
rolegroup: rolegroup.to_string(),
177-
})?
178-
.config
179-
.pod_overrides
180-
.clone(),
181-
};
182-
183-
Ok(pod_overrides)
184-
}
185-
186120
pub fn replicas(
187121
&self,
188122
kafka: &v1alpha1::KafkaCluster,

0 commit comments

Comments
 (0)