Merge pull request #69 from code0-tech/68-add-health-server

feat: added health service

Author: raphael-goetz

.env-example

@@ -1,4 +1,7 @@
 ENVIRONMENT='development'
 MODE='dynamic'
-RABBITMQ_URL='amqp://localhost:5672'
+NATS_URL='amqp://localhost:5672'
 AQUILA_URL='http://localhost:8080'
+WITH_HEALTH_SERVICE=false
+GRPC_HOST='127.0.0.1'
+GRPC_PORT=50051

Cargo.lock

@@ -16,6 +16,5 @@ base64 = "0.22.1"
 env_logger = "0.11.8"
 async-nats = "0.42.0"
 prost = "0.14.1"
-
-[dev-dependencies]
-tempfile = "3.19.1"
+tonic-health = "0.14.1"
+tonic = "0.14.1"

Cargo.toml

@@ -15,11 +15,15 @@ pub struct Config {
     /// `hybrid`
     pub mode: Mode,
 
-    /// Verification Token required for internal communication
-    pub rabbitmq_url: String,
+    pub nats_url: String,
 
-    /// URL to the `Sagittarius` Server.
     pub aquila_url: String,
+
+    pub with_health_service: bool,
+
+    pub grpc_host: String,
+
+    pub grpc_port: u16,
 }
 
 /// Implementation for all relevant `Aquila` startup configurations
@@ -31,8 +35,11 @@ impl Config {
         Config {
             environment: env_with_default("ENVIRONMENT", Environment::Development),
             mode: env_with_default("MODE", Mode::STATIC),
-            rabbitmq_url: env_with_default("RABBITMQ_URL", String::from("amqp://localhost:5672")),
-            aquila_url: env_with_default("AQUILA_URL", String::from("http://localhost:8080")),
+            nats_url: env_with_default("RABBITMQ_URL", String::from("amqp://localhost:5672")),
+            aquila_url: env_with_default("AQUILA_URL", String::from("http://localhost:50051")),
+            with_health_service: env_with_default("WITH_HEALTH_SERVICE", false),
+            grpc_host: env_with_default("GRPC_HOST", "127.0.0.1".to_string()),
+            grpc_port: env_with_default("GRPC_PORT", 50051),
         }
     }
 }

src/configuration.rs

@@ -10,7 +10,9 @@ use code0_flow::flow_config::load_env_file;
 use context::{Context, ContextEntry, ContextResult};
 use error::RuntimeError;
 use futures_lite::StreamExt;
+use log::error;
 use prost::Message;
+use tonic_health::pb::health_server::HealthServer;
 use registry::FunctionStore;
 use tucana::shared::value::Kind;
 use tucana::shared::{ExecutionFlow, ListValue, NodeFunction, Value};
@@ -170,13 +172,34 @@ async fn main() {
     let mut store = FunctionStore::new();
     store.populate(collect());
 
-    let client = match async_nats::connect("nats://127.0.0.1:4222").await {
+    let client = match async_nats::connect(config.nats_url.clone()).await {
         Ok(client) => client,
         Err(err) => {
             panic!("Failed to connect to NATS server: {}", err);
         }
     };
 
+    if config.with_health_service {
+        let health_service =
+            code0_flow::flow_health::HealthService::new(config.nats_url.clone());
+        let address = match format!("{}:{}", config.grpc_host, config.grpc_port).parse() {
+            Ok(address) => address,
+            Err(err) => {
+                error!("Failed to parse grpc address: {:?}", err);
+                return;
+            }
+        };
+
+        tokio::spawn(async move {
+            let _ = tonic::transport::Server::builder()
+                .add_service(HealthServer::new(health_service))
+                .serve(address)
+                .await;
+        });
+
+        println!("Health server started at {}", address);
+    }
+
     let _ = match client
         .queue_subscribe(String::from("execution.*"), "taurus".into())
         .await
@@ -193,12 +216,9 @@ async fn main() {
                     }
                 };
 
-                let value = match handle_message(flow, &store) {
-                    Some(value) => value,
-                    None => Value {
-                        kind: Some(Kind::NullValue(0)),
-                    },
-                };
+                let value = handle_message(flow, &store).unwrap_or_else(|| Value {
+                    kind: Some(Kind::NullValue(0)),
+                });
 
                 // Send a response to the reply subject
                 if let Some(reply) = msg.reply {