简单网络通信

wasmedge_wasi_socket 让 Rust 开发者可以创建简单的网络通信应用程序，然后将它编译为在 WasmEdge Runtime 中运行的 WebAssembly。WasmEdge 的一项关键特性便是对非阻塞套接字的支持，可以让一个单线程的 WASM 应用程序处理并发的网络请求。比如，当程序在等待一个连接传输的数据时，它可以开始或处理另一个连接。

在本章节中，我们将从一个简单的 HTTP 客户端以及服务器示例开始。然后在下个章节中，我们会介绍更为复杂的非阻塞示例。

HTTP 客户端示例

HTTP 客户端的源代码 如下。

1. use wasmedge_http_req::request;

3. fn main() {
4. let mut writer = Vec::new(); //container for body of a response
5. let res = request::get("http://127.0.0.1:1234/get", &mut writer).unwrap();

7. println!("GET");
8. println!("Status: {} {}", res.status_code(), res.reason());
9. println!("Headers {}", res.headers());
10. println!("{}", String::from_utf8_lossy(&writer));

12. let mut writer = Vec::new(); //container for body of a response
13. const BODY: &[u8; 27] = b"field1=value1&field2=value2";
14. // let res = request::post("https://httpbin.org/post", BODY, &mut writer).unwrap();
15. // no https , no dns
16. let res = request::post("http://127.0.0.1:1234/post", BODY, &mut writer).unwrap();

18. println!("POST");
19. println!("Status: {} {}", res.status_code(), res.reason());
20. println!("Headers {}", res.headers());
21. println!("{}", String::from_utf8_lossy(&writer));
22. }

使用如下命令来编译 Rust 程序。

cargo build --target wasm32-wasi --release

使用如下命令在 WasmEdge 中运行程序。

wasmedge target/wasm32-wasi/release/http_client.wasm

HTTP 服务器示例

HTTP 服务器的源代码 如下。

1. use bytecodec::DecodeExt;
2. use httpcodec::{HttpVersion, ReasonPhrase, Request, RequestDecoder, Response, StatusCode};
3. use std::io::{Read, Write};
4. #[cfg(feature = "std")]
5. use std::net::{Shutdown, TcpListener, TcpStream};
6. #[cfg(not(feature = "std"))]
7. use wasmedge_wasi_socket::{Shutdown, TcpListener, TcpStream};

9. fn handle_http(req: Request<String>) -> bytecodec::Result<Response<String>> {
10. Ok(Response::new(
11. HttpVersion::V1_0,
12. StatusCode::new(200)?,
13. ReasonPhrase::new("")?,
14. format!("echo: {}", req.body()),
15. ))
16. }

18. fn handle_client(mut stream: TcpStream) -> std::io::Result<()> {
19. let mut buff = [0u8; 1024];
20. let mut data = Vec::new();

22. loop {
23. let n = stream.read(&mut buff)?;
24. data.extend_from_slice(&buff[0..n]);
25. if n < 1024 {
26. break;
27. }
28. }

30. let mut decoder =
31. RequestDecoder::<httpcodec::BodyDecoder<bytecodec::bytes::Utf8Decoder>>::default();

33. let req = match decoder.decode_from_bytes(data.as_slice()) {
34. Ok(req) => handle_http(req),
35. Err(e) => Err(e),
36. };

38. let r = match req {
39. Ok(r) => r,
40. Err(e) => {
41. let err = format!("{:?}", e);
42. Response::new(
43. HttpVersion::V1_0,
44. StatusCode::new(500).unwrap(),
45. ReasonPhrase::new(err.as_str()).unwrap(),
46. err.clone(),
47. )
48. }
49. };

51. let write_buf = r.to_string();
52. stream.write(write_buf.as_bytes())?;
53. stream.shutdown(Shutdown::Both)?;
54. Ok(())
55. }

57. fn main() -> std::io::Result<()> {
58. let port = std::env::var("PORT").unwrap_or(1234.to_string());
59. println!("new connection at {}", port);
60. let listener = TcpListener::bind(format!("0.0.0.0:{}", port))?;
61. loop {
62. let _ = handle_client(listener.accept()?.0);
63. }
64. }

使用如下命令来编译 Rust 程序。

cargo build --target wasm32-wasi --release

使用如下命令在 WasmEdge 中运行程序。

$ wasmedge target/wasm32-wasi/release/http_server.wasm new connection at 1234

你可以用 curl 发送一个 HTTP 请求，来测试 HTTP 服务器。

$ curl -d "name=WasmEdge" -X POST http://127.0.0.1:1234 echo: name=WasmEdge