Setter for node_version property (solana-labs#29594)

Add: set_node_version method Co-authored-by: Charles Raimbault <[email protected]>
takeuchi5512 · Jan 10, 2023 · 0a6ff82 · 0a6ff82
1 parent db277d3
commit 0a6ff82
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git 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<semver::Version> {
         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
@@ -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<semver::Version, RpcError> {
         let r_node_version = self.node_version.read().await;
         if let Some(version) = &*r_node_version {