use super::connection::Web3Connection; use super::provider::Web3Provider; use std::fmt; use std::sync::atomic; use std::sync::Arc; use tokio::time::{sleep, Duration, Instant}; use tracing::warn; use tracing::{instrument, trace}; #[derive(Debug)] pub enum OpenRequestResult { Handle(OpenRequestHandle), /// Unable to start a request. Retry at the given time. RetryAt(Instant), /// Unable to start a request. Retrying will not succeed. RetryNever, } /// Make RPC requests through this handle and drop it when you are done. #[derive(Debug)] pub struct OpenRequestHandle(Arc); impl OpenRequestHandle { pub fn new(connection: Arc) -> Self { // TODO: attach a unique id to this? customer requests have one, but not internal queries // TODO: what ordering?! connection .active_requests .fetch_add(1, atomic::Ordering::AcqRel); // TODO: handle overflows? // TODO: what ordering? connection .total_requests .fetch_add(1, atomic::Ordering::Relaxed); Self(connection) } pub fn clone_connection(&self) -> Arc { self.0.clone() } /// Send a web3 request /// By having the request method here, we ensure that the rate limiter was called and connection counts were properly incremented /// By taking self here, we ensure that this is dropped after the request is complete #[instrument(skip_all)] pub async fn request( &self, method: &str, params: T, ) -> Result where T: fmt::Debug + serde::Serialize + Send + Sync, R: serde::Serialize + serde::de::DeserializeOwned + fmt::Debug, { // TODO: use tracing spans properly // TODO: requests from customers have request ids, but we should add // TODO: including params in this is way too verbose trace!("Sending {} to {}", method, self.0); let mut provider = None; while provider.is_none() { match self.0.provider.read().await.as_ref() { None => { warn!(rpc=%self.0, "no provider!"); // TODO: how should this work? a reconnect should be in progress. but maybe force one now? // TODO: sleep how long? subscribe to something instead? sleep(Duration::from_millis(100)).await } Some(found_provider) => provider = Some(found_provider.clone()), } } let response = match &*provider.unwrap() { Web3Provider::Http(provider) => provider.request(method, params).await, Web3Provider::Ws(provider) => provider.request(method, params).await, }; // TODO: i think ethers already has trace logging (and does it much more fancy) // TODO: at least instrument this with more useful information // trace!("Reply from {}: {:?}", self.0, response); trace!("Reply from {}", self.0); response } } impl Drop for OpenRequestHandle { fn drop(&mut self) { self.0 .active_requests .fetch_sub(1, atomic::Ordering::AcqRel); } }