Send messages to SSE through tokio::sync::mpsc::channel::<String>(5), but ReceiverStream does not consume the messages.

[dialogflowchatbot]

Summary

Thank you for providing such a great tool.

The problem I'm encountering is:

I created a Sender and Receiver using tokio::sync::mpsc::channel::<String>(5), and then I passed the Sender into a tokio::spawn task: completion::completion(&q.robot_id, &q.prompt, sender).await. The specific code is here: src/ai/crud.rs#L68.

This method passes the Sender to another function huggingface, and the code is here: src/ai/completion.rs#L40.

It is then passed to: src/ai/completion.rs#L91.

Finally, it is passed to: src/ai/llama.rs#L31.

And it is used here: src/ai/llama.rs#L139.

However, the Receiver does not consume the messages until the queue is full (5 messages), and then the program panic (intentionally) consumes all 5 messages at once, causing an error.

I originally thought the SSE configuration might be incorrect, so I wrote a simple push code: src/ai/crud.rs#L67.

async fn ttt(sender: tokio::sync::mpsc::Sender<String>) {
    for _ in 0..5 {
        let s = &sender;
        s.try_send(String::from("1"));
        tokio::time::sleep(std::time::Duration::from_secs(30)).await;
    }
}

In this code, the Receiver can consume messages normally.

It's very strange. Could it be because I passed the Sender multiple times?

axum version

axum = {version = "0.7", features = ["query", "tokio"]}

[mladedav]

Sorry, I don't understand. What panics, the receiver?

Do you have a reason to believe your issue is caused by axum?

[dialogflowchatbot]

What panics, the receiver?
The sender of tokio::sync::mpsc, when the channel was full, I made program panic intentionally.

Do you have a reason to believe your issue is caused by axum?
No. but I don't know what cause the receiver didn't consume the messages.
Either Receiver of tokio::sync::mpsc didn't concume messages at all, or SSE of Axum integrate ReceiverStream not right (maybe some code were missing)

I also tried passing pointer (&Sender) to function but with no luck.