lenn
/
dioxus-mirror
镜像来自 https://github.com/DioxusLabs/dioxus.git


			
				
					
						
						
							1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071
							use dioxus::prelude::*;

use dioxus_liveview::LiveViewPool;
use salvo::extra::affix;
use salvo::extra::ws::WsHandler;
use salvo::prelude::*;
use std::net::SocketAddr;
use std::sync::Arc;

fn app(cx: Scope) -> Element {
    let mut num = use_state(cx, || 0);

    cx.render(rsx! {
        div {
            "hello salvo! {num}"
            button { onclick: move |_| num += 1, "Increment" }
        }
    })
}

#[tokio::main]
async fn main() {
    pretty_env_logger::init();

    let addr: SocketAddr = ([127, 0, 0, 1], 3030).into();

    let view = LiveViewPool::new();

    let router = Router::new()
        .hoop(affix::inject(Arc::new(view)))
        .get(index)
        .push(Router::with_path("ws").get(connect));

    println!("Listening on http://{}", addr);

    Server::new(TcpListener::bind(addr)).serve(router).await;
}

#[handler]
fn index(_depot: &mut Depot, res: &mut Response) {
    let addr: SocketAddr = ([127, 0, 0, 1], 3030).into();
    res.render(Text::Html(format!(
        r#"
            <!DOCTYPE html>
            <html>
                <head> <title>Dioxus LiveView with Warp</title>  </head>
                <body> <div id="main"></div> </body>
                {glue}
            </html>
            "#,
        glue = dioxus_liveview::interpreter_glue(&format!("ws://{addr}/ws"))
    )));
}

#[handler]
async fn connect(
    req: &mut Request,
    depot: &mut Depot,
    res: &mut Response,
) -> Result<(), StatusError> {
    let view = depot.obtain::<Arc<LiveViewPool>>().unwrap().clone();
    let fut = WsHandler::new().handle(req, res)?;

    tokio::spawn(async move {
        if let Some(ws) = fut.await {
            _ = view.launch(dioxus_liveview::salvo_socket(ws), app).await;
        }
    });

    Ok(())
}