add a streaming example

This commit is contained in:
Evan Almloff committed 2024-02-16 16:04:12 -06:00
1 parent 31a7d2ac04
commit 2952578a2e
6 files changed
+92 -29

No files matched your search

@@ -5,6 +5,7 @@
use axum_desktop::*;
use dioxus::prelude::*;
use dioxus_fullstack::server_fn::axum::register_explicit;
#[tokio::main]
async fn main() {
@@ -18,7 +19,7 @@ async fn main() {
axum::serve(
listener,
axum::Router::new()
.register_server_fns("")
.register_server_fns()
.into_make_service(),
)
.await
@@ -0,0 +1,4 @@
dist
target
static
.dioxus
@@ -0,0 +1,24 @@
[package]
name = "axum-streaming"
version = "0.1.0"
edition = "2021"
publish = false
# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
[dependencies]
dioxus = { workspace = true, features = ["fullstack"] }
serde = "1.0.159"
simple_logger = "4.2.0"
tracing-wasm = "0.2.1"
tracing.workspace = true
tracing-subscriber = "0.3.17"
futures = "0.3.30"
tokio = { workspace = true, optional = true }
futures-util.workspace = true
once_cell = "1.19.0"
[features]
default = []
server = ["dioxus/axum", "tokio"]
web = ["dioxus/web"]
@@ -0,0 +1,41 @@
use dioxus::prelude::*;
use futures::StreamExt;
use server_fn::codec::{StreamingText, TextStream};
fn app() -> Element {
let mut response = use_signal(String::new);
rsx! {
button {
onclick: move |_| async move {
response.write().clear();
if let Ok(stream) = test_stream().await {
response.write().push_str("Stream started\n");
let mut stream = stream.into_inner();
while let Some(Ok(text)) = stream.next().await {
response.write().push_str(&text);
}
}
},
"Start stream"
}
"{response}"
}
}
#[server(output = StreamingText)]
pub async fn test_stream() -> Result<TextStream, ServerFnError> {
let (tx, rx) = futures::channel::mpsc::unbounded();
tokio::spawn(async move {
loop {
tokio::time::sleep(tokio::time::Duration::from_millis(500)).await;
let _ = tx.unbounded_send(Ok("Hello, world!".to_string()));
}
});
Ok(TextStream::new(rx))
}
fn main() {
launch(app)
}