Nikolai Rodionov
cba4da6d84
All checks were successful
ci/woodpecker/push/build-container Pipeline was successful
Signed-off-by: Nikolai Rodionov <iam@allanger.xyz>
104 lines
2.9 KiB
Rust
104 lines
2.9 KiB
Rust
mod conditions;
|
|
mod controllers;
|
|
mod rc;
|
|
mod cli;
|
|
mod config;
|
|
|
|
use crate::controllers::{rustfs_instance};
|
|
|
|
use actix_web::{App, HttpRequest, HttpResponse, HttpServer, Responder, get, middleware};
|
|
use clap::Parser;
|
|
use kube::{Client, CustomResourceExt};
|
|
use tracing_subscriber::EnvFilter;
|
|
|
|
use self::config::read_config_from_file;
|
|
use self::controllers::{rustfs_bucket, rustfs_user};
|
|
|
|
use api::api::v1beta1_rustfs_instance::RustFSInstance;
|
|
use api::api::v1beta_rustfs_bucket::RustFSBucket;
|
|
use api::api::v1beta_rustfs_user::RustFSUser;
|
|
|
|
/// Simple program to greet a person
|
|
#[derive(Parser, Debug)]
|
|
#[command(version, about, long_about = None)]
|
|
struct Args {
|
|
#[arg(long, default_value_t = 60000)]
|
|
/// The address the metric endpoint binds to.
|
|
metrics_port: u16,
|
|
#[arg(long, default_value_t = 8081)]
|
|
/// The address the probe endpoint binds to.
|
|
health_probe_port: u16,
|
|
#[arg(long, default_value_t = true)]
|
|
/// Enabling this will ensure there is only one active controller manager.
|
|
// DB Operator feature flags
|
|
#[arg(long, default_value_t = false)]
|
|
/// If enabled, DB Operator will run full reconciliation only
|
|
/// when changes are detected
|
|
is_change_check_nabled: bool,
|
|
#[arg(long, default_value = "/src/config/config.json")]
|
|
/// A path to a config file
|
|
config: String,
|
|
/// Set to true to generate crds
|
|
#[arg(long, default_value_t = false)]
|
|
crdgen: bool,
|
|
}
|
|
|
|
#[get("/health")]
|
|
async fn health(_: HttpRequest) -> impl Responder {
|
|
HttpResponse::Ok().json("healthy")
|
|
}
|
|
|
|
fn crdgen() {
|
|
println!(
|
|
"---\n{}",
|
|
serde_yaml::to_string(&RustFSInstance::crd()).unwrap()
|
|
);
|
|
println!(
|
|
"---\n{}",
|
|
serde_yaml::to_string(&RustFSBucket::crd()).unwrap()
|
|
);
|
|
println!(
|
|
"---\n{}",
|
|
serde_yaml::to_string(&RustFSUser::crd()).unwrap()
|
|
);
|
|
|
|
}
|
|
|
|
#[tokio::main]
|
|
async fn main() -> anyhow::Result<()> {
|
|
let args = Args::parse();
|
|
if args.crdgen {
|
|
crdgen();
|
|
return Ok(());
|
|
}
|
|
tracing_subscriber::fmt()
|
|
.json()
|
|
.with_env_filter(EnvFilter::from_default_env())
|
|
.init();
|
|
let client = Client::try_default()
|
|
.await
|
|
.expect("failed to create kube Client");
|
|
let config = read_config_from_file(args.config)?;
|
|
let rustfs_instance_ctrl = rustfs_instance::run(client.clone());
|
|
let rustfs_bucket_ctrl = rustfs_bucket::run(client.clone(), config.clone());
|
|
let rustfs_user_ctrl = rustfs_user::run(client.clone());
|
|
// Start web server
|
|
let server = HttpServer::new(move || {
|
|
App::new()
|
|
.wrap(middleware::Logger::default().exclude("/health"))
|
|
.service(health)
|
|
})
|
|
.bind("0.0.0.0:8080")?
|
|
.shutdown_timeout(5);
|
|
|
|
// Both runtimes implements graceful shutdown, so poll until both are done
|
|
tokio::join!(
|
|
rustfs_instance_ctrl,
|
|
rustfs_bucket_ctrl,
|
|
rustfs_user_ctrl,
|
|
server.run()
|
|
)
|
|
.3?;
|
|
Ok(())
|
|
}
|