Skip to content

Commit

Permalink
fixes #160 (#161)
Browse files Browse the repository at this point in the history
  • Loading branch information
wolf4ood committed Feb 15, 2023
1 parent 8a28909 commit 634adb6
Showing 1 changed file with 14 additions and 14 deletions.
28 changes: 14 additions & 14 deletions src/consumer.rs
Original file line number Diff line number Diff line change
Expand Up @@ -84,6 +84,20 @@ impl ConsumerBuilder {
}

let subscription_id = 1;
let (tx, rx) = channel(10000);
let consumer = Arc::new(ConsumerInternal {
subscription_id,
stream: stream.to_string(),
client: client.clone(),
sender: tx,
closed: Arc::new(AtomicBool::new(false)),
waker: AtomicWaker::new(),
metrics_collector: collector,
});

let msg_handler = ConsumerMessageHandler(consumer.clone());
client.set_handler(msg_handler).await;

let response = client
.subscribe(
subscription_id,
Expand All @@ -95,20 +109,6 @@ impl ConsumerBuilder {
.await?;

if response.is_ok() {
let (tx, rx) = channel(10000);
let consumer = Arc::new(ConsumerInternal {
subscription_id,
stream: stream.to_string(),
client: client.clone(),
sender: tx,
closed: Arc::new(AtomicBool::new(false)),
waker: AtomicWaker::new(),
metrics_collector: collector,
});

let msg_handler = ConsumerMessageHandler(consumer.clone());
client.set_handler(msg_handler).await;

Ok(Consumer {
receiver: rx,
internal: consumer,
Expand Down

0 comments on commit 634adb6

Please sign in to comment.