Setting an event type with poem_openapi::payload::EventStream?

[bryanlarsen]

poem::web::sse::Event gives a mechanism (event_type()) for adding an event type to an sse event.

poem_openapi::payload::EventStream does not, at least as far as I can tell. Is this possible?

[sunli829]

Here is the example to show how to convert the message to SSE event:

poem/poem-openapi/src/payload/event_stream.rs

Line 70 in 91614e4

pub fn to_event(self, f: impl FnMut(T::Item) -> Event + Send + 'static) -> Self {

[bryanlarsen]

Awesome! 2.0.11 isn't on cargo yet, so I switched to git to try it out.

I was working from your poem_openapi sse example, so I ended up with something like this:

    #[oai(path = "/foo", method = "get")]
    async fn foo(&self) -> EventStream<BoxStream<'static, serde_json::Value>> {
        EventStream::new(
            async_stream::stream! {
                for i in 0.. {
                    tokio::time::sleep(Duration::from_secs(1)).await;
                    yield serde_json::json!({"a": 3, "b": i});
                }
            }
            .boxed(),
        )
        .to_event(|ev| Event::message(ev.to_string()).event_type("foo"))
    }

Feel free to put this example into your repo.