feat: add close()

Author: tejasbadadare

lazer/sdk/rust/client/examples/subscribe_price_feeds.rs

@@ -41,7 +41,7 @@ async fn main() -> anyhow::Result<()> {
     while let Some(msg) = stream.next().await {
         println!("Received update: {:?}", msg?);
         count += 1;
-        if count >= 50 {
+        if count >= 10 {
             break;
         }
     }
@@ -51,5 +51,6 @@ async fn main() -> anyhow::Result<()> {
     println!("Unsubscribed from {:?}", subscription_id);
 
     tokio::time::sleep(tokio::time::Duration::from_millis(500)).await;
+    client.close().await?;
     Ok(())
 }

lazer/sdk/rust/client/src/lib.rs

@@ -179,4 +179,15 @@ impl LazerClient {
             anyhow::bail!("WebSocket connection not started")
         }
     }
+
+    /// Closes the WebSocket connection
+    pub async fn close(&mut self) -> Result<()> {
+        if let Some(sender) = &mut self.ws_sender {
+            sender.send(Message::Close(None)).await?;
+            self.ws_sender = None;
+            Ok(())
+        } else {
+            anyhow::bail!("WebSocket connection not started")
+        }
+    }
 }