stream sub structure

jakubwro · Jan 27, 2024 · dd95bcd · dd95bcd
1 parent 978c24b
commit dd95bcd
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 5 deletions.
diff --git a/src/jetstream/stream/stream.jl b/src/jetstream/stream/stream.jl
@@ -1,4 +1,10 @@
 
+struct StreamSub
+    sub::NATS.Sub
+    consumer::ConsumerInfo
+end
+
+
 include("manage.jl")
 include("info.jl")
 include("message.jl")

diff --git a/src/jetstream/stream/subscribe.jl b/src/jetstream/stream/subscribe.jl
@@ -29,5 +29,5 @@ function stream_subscribe(f, connection, subject)
             f_typed(msg)
         end
     end
-    sub, consumer
+    StreamSub(sub, consumer)
 end
diff --git a/src/jetstream/stream/unsubscribe.jl b/src/jetstream/stream/unsubscribe.jl
@@ -3,8 +3,7 @@ $(SIGNATURES)
 
 Unsubscribe stream subscription.
 """
-function stream_unsubscribe(connection, subscription::Tuple{NATS.Sub, JetStream.ConsumerInfo})
-    sub, cons = subscription
-    NATS.unsubscribe(connection, sub)
-    consumer_delete(connection, cons)
+function stream_unsubscribe(connection, stream_sub::StreamSub)
+    NATS.unsubscribe(connection, stream_sub.sub)
+    consumer_delete(connection, stream_sub.consumer)
 end