Merge pull request #12 from Overmuse/SR/health_check

SR/health check

server/Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "datastore-server"
-version = "0.4.6"
+version = "0.5.0"
 edition = "2018"
 
 # See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html

server/src/handlers/health.rs

@@ -0,0 +1,9 @@
+use super::last::get_last_price;
+use crate::redis::Redis;
+use warp::Rejection;
+
+#[tracing::instrument(skip(redis))]
+pub async fn health_check(redis: Redis) -> Result<impl warp::Reply, Rejection> {
+    get_last_price("AAPL".to_string(), redis).await?;
+    Ok(warp::reply())
+}

server/src/handlers/mod.rs

@@ -1,4 +1,5 @@
 pub mod aggregates;
 pub mod dividends;
+pub mod health;
 pub mod last;
 pub mod splits;

server/src/routes/health.rs

@@ -0,0 +1,19 @@
+use crate::handlers;
+use crate::redis::with_redis;
+use crate::Redis;
+use warp::Filter;
+
+pub fn health_check(
+    redis: Redis,
+) -> impl Filter<Extract = impl warp::Reply, Error = warp::Rejection> + Clone {
+    warp::path!("health_check")
+        .and(warp::get())
+        .and(with_redis(redis))
+        .and_then(handlers::health::health_check)
+}
+
+pub fn health_routes(
+    redis: Redis,
+) -> impl Filter<Extract = impl warp::Reply, Error = warp::Rejection> + Clone {
+    health_check(redis)
+}

server/src/routes/mod.rs

@@ -7,11 +7,13 @@ use warp::Filter;
 mod aggregates;
 mod backfill;
 mod dividends;
+mod health;
 mod last;
 mod splits;
 pub use aggregates::*;
 pub use backfill::*;
 pub use dividends::*;
+pub use health::*;
 pub use last::*;
 pub use splits::*;
 
@@ -22,6 +24,7 @@ pub fn routes(
     aggregates_routes(db.clone())
         .or(backfill_routes(db.clone()))
         .or(dividends_routes(db.clone()))
+        .or(health_routes(redis.clone()))
         .or(last_routes(redis))
         .or(splits_routes(db))
         .recover(handle_rejection)