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
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106

//! Kubernetes network proxy component.
//!
//! Runs `kube-proxy` which maintains network rules on nodes, enabling
//! Kubernetes Service abstraction by forwarding traffic to backend pods.

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

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

impl Component for ProxyComponent {
    fn name(&self) -> &str {
        "Proxy"
    }

    fn phase(&self) -> Phase {
        // Proxy only needs the API server to sync caches, not the
        // kubelets, so it starts in the Controller phase.
        Phase::Controller
    }

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

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

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

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

        let mut process = Process::start(
            &dir,
            "Proxy",
            "kube-proxy",
            &[
                &format!("--config={}", cfg.display()),
                &format!(
                    "--hostname-override={}",
                    if config.multi_node() {
                        Node::name(config, network, 0)
                    } else {
                        network.hostname().into()
                    }
                ),
            ],
        )?;

        process.wait_ready("Caches are synced")?;
        Ok(Box::new(Proxy { process }))
    }
}

impl Stoppable for Proxy {
    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/proxy.yml"),
            "/tmp/proxy.kubeconfig", "10.10.0.0/18",
        );
        assert!(yml.contains("kind: KubeProxyConfiguration"));
        assert!(yml.contains("kubeconfig: \"/tmp/proxy.kubeconfig\""));
        assert!(yml.contains("clusterCIDR: \"10.10.0.0/18\""));
        assert!(yml.contains("mode: \"iptables\""));
    }

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