add health endpoint (#752)

kkrt-labs · Feb 2, 2024 · b499179 · b499179
1 parent 0807ed0
commit b499179
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/docker/rpc/Dockerfile b/docker/rpc/Dockerfile
@@ -1,5 +1,6 @@
 FROM python:3.9.13 as compiler
 # install poetry
+ENV POETRY_VERSION=1.7.1
 RUN curl -sSL https://install.python-poetry.org | python3 -
 ENV PATH="$PATH:/root/.local/bin:/root/.foundry/bin"
 RUN poetry config virtualenvs.create false

diff --git a/src/eth_rpc/mod.rs b/src/eth_rpc/mod.rs
@@ -9,6 +9,7 @@ pub mod rpc;
 pub mod servers;
 
 use eyre::Result;
+use jsonrpsee::server::middleware::http::{InvalidPath, ProxyGetRequestLayer};
 use jsonrpsee::server::{ServerBuilder, ServerHandle};
 use jsonrpsee::RpcModule;
 use thiserror::Error;
@@ -20,6 +21,8 @@ pub enum RpcError {
     IoError(#[from] std::io::Error),
     #[error(transparent)]
     ParseError(#[from] AddrParseError),
+    #[error(transparent)]
+    JsonRpcError(#[from] InvalidPath),
 }
 
 /// # Errors
@@ -33,7 +36,8 @@ pub async fn run_server(
 
     let cors = CorsLayer::new().allow_methods(Any).allow_origin(Any).allow_headers(Any);
 
-    let http_middleware = tower::ServiceBuilder::new().layer(cors);
+    let http_middleware =
+        tower::ServiceBuilder::new().layer(ProxyGetRequestLayer::new("/health", "net_health")?).layer(cors);
 
     let server = ServerBuilder::default()
         .max_connections(std::env::var("RPC_MAX_CONNECTIONS").unwrap_or("100".to_string()).parse().unwrap())