kubernix 0.3.1

Kubernetes development cluster bootstrapping with Nix packages
Documentation 
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
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90

//! Kubernetes scheduler component.
//!
//! Runs `kube-scheduler` which assigns pods to nodes based on resource
//! availability, constraints, and scheduling policies.

use crate::{
    component::{ClusterContext, Component, Phase},
    config::Config,
    kubeconfig::KubeConfig,
    process::{Process, ProcessState, Stoppable},
    write_if_changed,
};
use anyhow::Result;
use std::fs::create_dir_all;

/// Component wrapper for registry-based startup.
pub struct SchedulerComponent;

impl Component for SchedulerComponent {
    fn name(&self) -> &str {
        "Scheduler"
    }

    fn phase(&self) -> Phase {
        Phase::Controller
    }

    fn start(&self, ctx: &ClusterContext<'_>) -> ProcessState {
        Scheduler::start(ctx.config, ctx.kubeconfig)
    }
}

/// Manages the `kube-scheduler` process lifecycle.
pub struct Scheduler {
    process: Process,
}

impl Scheduler {
    /// Start the scheduler with the given cluster configuration.
    pub fn start(config: &Config, kubeconfig: &KubeConfig) -> ProcessState {
        let dir = config.root().join("scheduler");
        create_dir_all(&dir)?;

        let yml = format!(
            include_str!("assets/scheduler.yml"),
            kubeconfig.scheduler().display()
        );
        let cfg = &dir.join("config.yml");
        write_if_changed(cfg, &yml)?;

        let mut process = Process::start(
            &dir,
            "Scheduler",
            "kube-scheduler",
            &[&format!("--config={}", cfg.display()), "--v=2"],
        )?;

        process.wait_ready("Serving securely")?;
        Ok(Box::new(Self { process }))
    }
}

impl Stoppable for Scheduler {
    fn stop(&mut self) -> Result<()> {
        self.process.stop()
    }
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn config_template_renders() {
        let yml = format!(
            include_str!("assets/scheduler.yml"),
            "/tmp/scheduler.kubeconfig"
        );
        assert!(yml.contains("apiVersion: kubescheduler.config.k8s.io/v1"));
        assert!(yml.contains("kubeconfig: \"/tmp/scheduler.kubeconfig\""));
        assert!(yml.contains("leaderElect: false"));
    }

    #[test]
    fn component_metadata() {
        let c = SchedulerComponent;
        assert_eq!(c.name(), "Scheduler");
        assert_eq!(c.phase(), Phase::Controller);
    }
}