feat: add keepalive to Torii client gRPC connections

crates/torii/client/src/client/mod.rs

@@ -1,6 +1,7 @@
 pub mod error;
 
 use std::sync::Arc;
+use std::time::Duration;
 
 use dojo_types::WorldMetadata;
 use dojo_world::contracts::WorldContractReader;
@@ -41,8 +42,10 @@
         rpc_url: String,
         relay_url: String,
         world: Felt,
+        keepalive: Option<Duration>,
     ) -> Result<Self, Error> {
-        let mut grpc_client = torii_grpc::client::WorldClient::new(torii_url, world).await?;
+        let mut grpc_client =
+            torii_grpc::client::WorldClient::new(torii_url, world, keepalive).await?;
 
         let relay_client = torii_relay::client::RelayClient::new(relay_url)?;
 

crates/torii/grpc/src/client.rs

@@ -1,5 +1,6 @@
 //! Client implementation for the gRPC service.
 use std::num::ParseIntError;
+use std::time::Duration;
 
 use futures_util::stream::MapOk;
 use futures_util::{Stream, StreamExt, TryStreamExt};
@@ -49,9 +50,14 @@
 
 impl WorldClient {
     #[cfg(not(target_arch = "wasm32"))]
-    pub async fn new(dst: String, world_address: Felt) -> Result<Self, Error> {
-        let endpoint =
-            Endpoint::from_shared(dst.clone()).map_err(|e| Error::Endpoint(e.to_string()))?;
+    pub async fn new(
+        dst: String,
+        world_address: Felt,
+        keepalive: Option<Duration>,
+    ) -> Result<Self, Error> {
+        let endpoint = Endpoint::from_shared(dst.clone())
+            .map_err(|e| Error::Endpoint(e.to_string()))?
+            .tcp_keepalive(keepalive);
         let channel = endpoint.connect().await.map_err(Error::Transport)?;
         Ok(Self {
             _world_address: world_address,