Add convenient method for subscribing to stream requests

autonomys · May 10, 2024 · e127680 · e127680
1 parent aa5fa80
commit e127680
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/crates/subspace-farmer/src/cluster/nats_client.rs b/crates/subspace-farmer/src/cluster/nats_client.rs
@@ -765,6 +765,20 @@ impl NatsClient {
             .await
     }
 
+    /// Simple subscription that will produce decoded stream requests, while skipping messages that
+    /// fail to decode
+    pub async fn subscribe_to_stream_requests<Request>(
+        &self,
+        instance: Option<&str>,
+        queue_group: Option<String>,
+    ) -> Result<SubscriberWrapper<StreamRequest<Request>>, SubscribeError>
+    where
+        Request: GenericStreamRequest,
+    {
+        self.simple_subscribe(Request::SUBJECT, instance, queue_group)
+            .await
+    }
+
     /// Simple subscription that will produce decoded notifications, while skipping messages that
     /// fail to decode
     pub async fn subscribe_to_notifications<Notification>(