From 66d9921702f9af7013a2f34f169904299749b1fa Mon Sep 17 00:00:00 2001
From: Jose Celano <josecelano@gmail.com>
Date: Tue, 21 Nov 2023 14:59:58 +0000
Subject: [PATCH] feat: [#508] add new binary HTTP health check

It makes a request to an HTTP endpoint to check that the service is
healthy.
---
 Cargo.toml                   |  1 +
 src/bin/http_health_check.rs | 37 ++++++++++++++++++++++++++++++++++++
 2 files changed, 38 insertions(+)
 create mode 100644 src/bin/http_health_check.rs

diff --git a/Cargo.toml b/Cargo.toml
index 51a1ac00f..2316a1edf 100644
--- a/Cargo.toml
+++ b/Cargo.toml
@@ -1,4 +1,5 @@
 [package]
+default-run = "torrust-tracker"
 name = "torrust-tracker"
 readme = "README.md"
 
diff --git a/src/bin/http_health_check.rs b/src/bin/http_health_check.rs
new file mode 100644
index 000000000..2c39f2a79
--- /dev/null
+++ b/src/bin/http_health_check.rs
@@ -0,0 +1,37 @@
+//! Minimal `curl` or `wget` to be used for container health checks.
+//!
+//! It's convenient to avoid using third-party libraries because:
+//!
+//! - They are harder to maintain.
+//! - They introduce new attack vectors.
+use std::{env, process};
+
+#[tokio::main]
+async fn main() {
+    let args: Vec<String> = env::args().collect();
+    if args.len() != 2 {
+        eprintln!("Usage:   cargo run --bin health_check <HEALTH_URL>");
+        eprintln!("Example: cargo run --bin http_health_check http://localhost:1212/api/v1/stats?token=MyAccessToken");
+        std::process::exit(1);
+    }
+
+    println!("Health check ...");
+
+    let url = &args[1].clone();
+
+    match reqwest::get(url).await {
+        Ok(response) => {
+            if response.status().is_success() {
+                println!("STATUS: {}", response.status());
+                process::exit(0);
+            } else {
+                println!("Non-success status received.");
+                process::exit(1);
+            }
+        }
+        Err(err) => {
+            println!("ERROR: {err}");
+            process::exit(1);
+        }
+    }
+}
\ No newline at end of file