fix: formatting, example

Author: tejasbadadare

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

@@ -10,8 +10,8 @@ use pyth_lazer_sdk::LazerClient;
 async fn main() -> anyhow::Result<()> {
     // Create and start the client
     let mut client = LazerClient::new(
-        "wss://hermes.pyth.network",
-        "YOUR_ACCESS_TOKEN".to_string(),
+        "wss://pyth-lazer.dourolabs.app/v1/stream",
+        "YOUR_ACCESS_TOKEN",
     )?;
     let mut stream = client.start().await?;
 
@@ -36,12 +36,12 @@ async fn main() -> anyhow::Result<()> {
 
     println!("Subscribed to BTC/USD price feed. Waiting for updates...");
 
-    // Process the first 5 updates
+    // Process the first 100 updates
     let mut count = 0;
     while let Some(msg) = stream.next().await {
         println!("Received update: {:?}", msg?);
         count += 1;
-        if count >= 5 {
+        if count >= 100 {
             break;
         }
     }

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

@@ -57,12 +57,14 @@ pub enum BinaryResponse {
 pub struct LazerClient {
     endpoint: Url,
     access_token: String,
-    ws_sender: Option<futures_util::stream::SplitSink<
-        tokio_tungstenite::WebSocketStream<
-            tokio_tungstenite::MaybeTlsStream<tokio::net::TcpStream>,
+    ws_sender: Option<
+        futures_util::stream::SplitSink<
+            tokio_tungstenite::WebSocketStream<
+                tokio_tungstenite::MaybeTlsStream<tokio::net::TcpStream>,
+            >,
+            Message,
         >,
-        Message,
-    >>,
+    >,
 }
 
 impl LazerClient {
@@ -74,8 +76,9 @@ impl LazerClient {
     ///
     /// # Returns
     /// Returns a new client instance (not yet connected)
-    pub fn new(endpoint: &str, access_token: String) -> Result<Self> {
+    pub fn new(endpoint: &str, access_token: &str) -> Result<Self> {
         let endpoint = Url::parse(endpoint)?;
+        let access_token = access_token.to_string();
         Ok(Self {
             endpoint,
             access_token,
@@ -89,8 +92,9 @@ impl LazerClient {
     /// Returns a stream of responses from the server
     pub async fn start(&mut self) -> Result<impl futures_util::Stream<Item = Result<Response>>> {
         let url = self.endpoint.clone();
-        let mut request = tokio_tungstenite::tungstenite::client::IntoClientRequest::into_client_request(url)?;
-        
+        let mut request =
+            tokio_tungstenite::tungstenite::client::IntoClientRequest::into_client_request(url)?;
+
         request.headers_mut().insert(
             "Authorization",
             format!("Bearer {}", self.access_token).parse().unwrap(),
@@ -166,7 +170,7 @@ impl LazerClient {
             }
         });
 
-        Ok((client, response_stream))
+        Ok(response_stream)
     }
 
     /// Subscribes to price feed updates