run Tainter (#11)

Cargo.toml

@@ -12,7 +12,7 @@ tracing-subscriber = { version = "0.3.18", features = ["json"] }
 kube = { version = "0.90.0", features = ["runtime", "derive"] }
 k8s-openapi = { version = "0.21.1", features = ["latest"] }
 futures = "0.3.30"
-tokio = { version = "1.37.0", features = ["macros"] }
+tokio = { version = "1.37.0", features = ["macros", "rt-multi-thread", "rt"] }
 tower-test = "0.4.0"
 tower = "0.4.13"
 http = "1.1.0"
@@ -25,6 +25,12 @@ strum_macros = "0.26.4"
 strum = "0.26.3"
 validator = { version = "0.18.1", features = ["derive"] }
 thiserror = "1.0.64"
+clap = { version = "4.5.20", features = ["derive"] }
 
 [dev-dependencies]
 test-case = "3.3.1"
+
+[[bin]]
+edition = "2021"
+name = "tainter"
+path = "src/main.rs"

Makefile

@@ -1,6 +1,6 @@
 .PHONY: run
 run:
-	cargo run
+	cargo run -- --config-file "./settings/tainter.toml"
 
 .PHONY: test
 test:

settings/tainter.toml

@@ -0,0 +1,20 @@
+[server]
+host = "0.0.0.0"
+port = "8080"
+
+[log]
+max_level = "info"
+
+[[reconciler.matchers]]
+[reconciler.matchers.taint]
+effect = "NoExecute"
+key = "pressure"
+value = "memory"
+
+[[reconciler.matchers.conditions]]
+type = "NetworkInterfaceCard"
+status = "Kaput|Ruined"
+
+[[reconciler.matchers.conditions]]
+type = "PrivateLink"
+status = "severed"

src/main.rs

@@ -1,23 +1,45 @@
-use k8s_openapi::serde::Deserialize;
+use clap::Parser;
+use kube::Client;
+use std::error::Error;
+
+use crate::settings::Settings;
 
 mod reconciler;
 mod settings;
 mod tainter;
 
+#[derive(Parser, Debug)]
+#[command(version, about, long_about = None)]
+struct Args {
+    /// Path to TOML file from which configuration is read.
+    #[arg(short, long)]
+    config_file: String,
+}
+
+// Adding the actix_web::main attribute also implicitly adds tokio::main.
+// See https://stackoverflow.com/a/66419283.
 #[actix_web::main]
-async fn main() {
-    // TODO read settings file and initialize.
-    // let settings = Config::builder().add_source(config::File::with_name("example_config")).build().unwrap();
-    //
-    // println!("{:?}", settings);
+async fn main() -> Result<(), Box<dyn Error>> {
+    let args = Args::parse();
+    println!(
+        "Reading configuration from file at path {}",
+        args.config_file
+    );
+    let settings = Settings::new(args.config_file.as_str())?;
 
     tracing_subscriber::fmt()
         .json()
-        // TODO ability to configure log level.
-        .with_max_level(tracing::Level::INFO)
+        .with_max_level(settings.log.max_level)
         .with_current_span(false)
         .init();
 
-    // let tainter = tainter::Tainter::new("localhost".to_string(), 8000);
-    // tainter.start().await.unwrap();
+    tracing::info!("Initializing Kubernetes client");
+
+    let client = Client::try_default().await?;
+
+    let tainter = tainter::Tainter::new(settings, client);
+
+    tainter.start().await?;
+
+    Ok(())
 }

src/reconciler/mod.rs

@@ -15,13 +15,13 @@ use regex::Regex;
 
 #[derive(Debug)]
 pub struct Condition {
-    type_: Regex,
-    status: Regex,
+    pub type_: Regex,
+    pub status: Regex,
 }
 
 pub struct Configuration {
-    conditions: Vec<Condition>,
-    taint: Taint,
+    pub conditions: Vec<Condition>,
+    pub taint: Taint,
 }
 
 pub struct Reconciler {