Add WebSocket support

Cargo.toml

@@ -12,23 +12,33 @@ keywords = ["bitcoin", "mempool"]
 
 [features]
 # Default features
-default = ["rustls"]
+default = ["rustls", "ws"]
 # Enable rust TLS
 rustls = ["reqwest/rustls-tls"]
 # Enable natile TLS (openssl)
 nativetls = ["reqwest/native-tls"]
 # Enable socks proxy
 socks = ["reqwest/socks"]
+# Enable WebSocket client
+ws = ["dep:futures-util", "dep:serde_json", "dep:tokio", "dep:tokio-tungstenite", "dep:tracing"]
 
 [dependencies]
 bitcoin = { version = "0.32", default-features = false, features = ["std", "serde"] }
 reqwest = { version = "0.12", default-features = false, features = ["json"] }
 serde = { version = "1.0", features = ["derive"] }
 url = "2.5"
 
+# WebSocket
+futures-util = { version = "0.3", optional = true }
+serde_json = { version = "1.0", optional = true }
+tokio = { version = "1", features = ["sync", "time"], optional = true }
+tokio-tungstenite = { version = "0.27", features = ["rustls-tls-webpki-roots"], optional = true }
+tracing = { version = "0.1", optional = true }
+
 [dev-dependencies]
 serde_json = "1.0"
 tokio = { version = "1.46", features = ["macros", "rt-multi-thread"] }
 
 [[example]]
 name = "mempool"
+required-features = ["ws"]

examples/mempool.rs

@@ -1,6 +1,7 @@
 use std::str::FromStr;
 
 use mempoolspace::prelude::*;
+use tokio::sync::mpsc::UnboundedReceiver;
 
 #[tokio::main]
 async fn main() {
@@ -28,4 +29,26 @@ async fn main() {
     // Get recommended fees
     let fees = client.get_recommended_fees().await.unwrap();
     println!("{:?}", fees);
+
+    // Subscribe
+    let req = MempoolSubscriptionRequest::LiveData {
+        action: LiveDataAction::Want,
+        data: vec![LiveDataType::Stats],
+    };
+    let sub = client.subscribe(req).await.unwrap();
+
+    tokio::select! {
+        _ = sub.worker => {
+            println!("Worker exited");
+        },
+        _ = handle_messages(sub.receiver) => {
+            println!("Receiver exited");
+        }
+    }
+}
+
+async fn handle_messages(mut rx: UnboundedReceiver<MempoolSubscriptionResponse>) {
+    while let Some(message) = rx.recv().await {
+        println!("{message:?}");
+    }
 }