k8s_openapi_ext/ext/
taint.rs

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
use super::*;

pub trait TaintExt {
    fn no_schedule(key: impl ToString) -> Self;
    fn prefer_no_schedule(key: impl ToString) -> Self;
    fn no_execute(key: impl ToString) -> Self;
    fn value(self, value: impl ToString) -> Self;
}

impl TaintExt for corev1::Taint {
    fn no_schedule(key: impl ToString) -> Self {
        let effect = Effect::NoSchedule.to_string();
        let key = key.to_string();
        Self {
            effect,
            key,
            ..default()
        }
    }

    fn prefer_no_schedule(key: impl ToString) -> Self {
        let effect = Effect::PreferNoSchedule.to_string();
        let key = key.to_string();
        Self {
            effect,
            key,
            ..default()
        }
    }

    fn no_execute(key: impl ToString) -> Self {
        let effect = Effect::NoExecute.to_string();
        let key = key.to_string();
        let time_added = Some(metav1::Time::now());
        Self {
            effect,
            key,
            time_added,
            ..default()
        }
    }

    fn value(self, value: impl ToString) -> Self {
        Self {
            value: Some(value.to_string()),
            ..self
        }
    }
}