xref: /wasmtime-44.0.1/crates/test-programs/src/bin/p2_cli_serve_keyvalue.rs (revision b315a0a8)

use test_programs::proxy;
use test_programs::wasi::{
    http::types::{Fields, IncomingRequest, OutgoingBody, OutgoingResponse, ResponseOutparam},
    keyvalue,
};

struct T;

proxy::export!(T);

impl proxy::exports::wasi::http::incoming_handler::Guest for T {
    fn handle(_: IncomingRequest, outparam: ResponseOutparam) {
        let fields = Fields::new();
        let resp = OutgoingResponse::new(fields);
        let body = resp.body().expect("outgoing response");

        ResponseOutparam::set(outparam, Ok(resp));

        let out = body.write().expect("outgoing stream");
        let bucket = keyvalue::store::open("").unwrap();
        let data = bucket.get("hello").unwrap().unwrap();
        out.blocking_write_and_flush(&data)
            .expect("writing response");

        drop(out);
        OutgoingBody::finish(body, None).expect("outgoing-body.finish");
    }
}

fn main() {}