use super::*;
pub trait StatefulSetExt {
fn new(name: impl ToString) -> Self;
fn spec(self, spec: appsv1::StatefulSetSpec) -> Self;
fn replicas(self, replicas: i32) -> Self;
fn selector(self, selector: metav1::LabelSelector) -> Self;
fn match_labels(
self,
match_labels: impl IntoIterator<Item = (impl ToString, impl ToString)>,
) -> Self;
fn service_name(self, service_name: impl ToString) -> Self;
fn template(self, template: corev1::PodTemplateSpec) -> Self;
fn ordinals(self, ordinals: i32) -> Self;
}
impl StatefulSetExt for appsv1::StatefulSet {
fn new(name: impl ToString) -> Self {
let metadata = metadata(name);
Self {
metadata,
..default()
}
}
fn spec(self, spec: appsv1::StatefulSetSpec) -> Self {
Self {
spec: Some(spec),
..self
}
}
fn replicas(mut self, replicas: i32) -> Self {
self.spec_mut().replicas.replace(replicas);
self
}
fn selector(mut self, selector: metav1::LabelSelector) -> Self {
self.spec_mut().selector = selector;
self
}
fn match_labels(
mut self,
match_labels: impl IntoIterator<Item = (impl ToString, impl ToString)>,
) -> Self {
self.spec_mut().selector = metav1::LabelSelector::match_labels(match_labels);
self
}
fn service_name(mut self, service_name: impl ToString) -> Self {
self.spec_mut()
.service_name
.replace(service_name.to_string());
self
}
fn template(mut self, template: corev1::PodTemplateSpec) -> Self {
self.spec_mut().template = template;
self
}
fn ordinals(mut self, start: i32) -> Self {
self.spec_mut().ordinals = Some(appsv1::StatefulSetOrdinals { start: Some(start) });
self
}
}
impl HasSpec for appsv1::StatefulSet {
type Spec = appsv1::StatefulSetSpec;
fn spec_mut(&mut self) -> &mut Self::Spec {
self.spec.get_or_insert_default()
}
}