rust_wheel 0.1.12

A project to define some public component.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41

use std::fmt::Display;

use tokio::sync::mpsc;

pub struct SseStream<T> where T: Display,
{
    pub receiver: Option<mpsc::UnboundedReceiver<T>>,
}

impl<T: std::fmt::Display> SseStream<T> {
    
}

impl<T: std::fmt::Display> futures::Stream for SseStream<T> {
    type Item = Result<actix_web::web::Bytes, actix_web::Error>;

    fn poll_next(
        mut self: std::pin::Pin<&mut Self>,
        ctx: &mut std::task::Context<'_>,
    ) -> std::task::Poll<Option<Self::Item>> {
        // get reciver data and send
        if let Some(receiver) = &mut self.receiver {
            match receiver.poll_recv(ctx) {
                std::task::Poll::Ready(Some(data)) => {
                    // Create the SSE event
                    let message = format!("{}", data);
                    // Return the event as a stream item
                    return std::task::Poll::Ready(Some(Ok(actix_web::web::Bytes::from(message))));
                }
                std::task::Poll::Ready(None) => {
                    return std::task::Poll::Ready(None);
                }
                std::task::Poll::Pending => {
                    return std::task::Poll::Pending;
                }
            }
        } else {
            return std::task::Poll::Ready(None);
        }
    }
}