From 8f4a8ef4d53ab35998bd801e29b82d65d3a3a2de Mon Sep 17 00:00:00 2001
From: Bryan Stitt <bryan@llamanodes.com>
Date: Wed, 1 Nov 2023 22:30:11 -0700
Subject: [PATCH] set max to head block of no to_block

---
 web3_proxy/src/jsonrpc/request_builder.rs | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/web3_proxy/src/jsonrpc/request_builder.rs b/web3_proxy/src/jsonrpc/request_builder.rs
index e4b6bb34..ea2b189a 100644
--- a/web3_proxy/src/jsonrpc/request_builder.rs
+++ b/web3_proxy/src/jsonrpc/request_builder.rs
@@ -348,6 +348,7 @@ impl ValidatedRequest {
         }
 
         // now that kafka has logged the user's original params, we can calculate the cache key
+        // calculating the cache key might alter the params
 
         // TODO: modify CacheMode::new to wait for a future block if one is requested! be sure to update head_block too!
         let cache_mode = match &mut request {
@@ -498,7 +499,13 @@ impl ValidatedRequest {
 
     #[inline]
     pub fn max_block_needed(&self) -> Option<U64> {
-        self.cache_mode.to_block().map(|x| x.num())
+        if let Some(to_block) = self.cache_mode.to_block() {
+            Some(to_block.num())
+        } else {
+            self.head_block
+                .as_ref()
+                .map(|head_block| head_block.number())
+        }
     }
 
     #[inline]