diff --git a/pubsub-client/src/nonblocking/pubsub_client.rs b/pubsub-client/src/nonblocking/pubsub_client.rs index 1f746bfa5a..e62b1e8a68 100644 --- a/pubsub-client/src/nonblocking/pubsub_client.rs +++ b/pubsub-client/src/nonblocking/pubsub_client.rs @@ -295,6 +295,12 @@ impl PubsubClient { self.ws.await.unwrap() // WS future should not be cancelled or panicked } + pub async fn set_node_version(&self, version: semver::Version) -> Result<(), ()> { + let mut w_node_version = self.node_version.write().await; + *w_node_version = Some(version); + Ok(()) + } + async fn get_node_version(&self) -> PubsubClientResult { let r_node_version = self.node_version.read().await; if let Some(version) = &*r_node_version { diff --git a/rpc-client/src/nonblocking/rpc_client.rs b/rpc-client/src/nonblocking/rpc_client.rs index 1eeb3ab4a0..c6a6de39d4 100644 --- a/rpc-client/src/nonblocking/rpc_client.rs +++ b/rpc-client/src/nonblocking/rpc_client.rs @@ -514,6 +514,12 @@ impl RpcClient { self.sender.url() } + pub async fn set_node_version(&self, version: semver::Version) -> Result<(), ()> { + let mut w_node_version = self.node_version.write().await; + *w_node_version = Some(version); + Ok(()) + } + async fn get_node_version(&self) -> Result { let r_node_version = self.node_version.read().await; if let Some(version) = &*r_node_version {