Rust编程实战：Rust实现简单的Web服务，单线程性能问题
知识点
- tcp 服务
- 多线程处理
实现功能
启动web服务，访问链接获取页面内容。
单线程web服务
TcpListener
使用 TcpListener 开启服务端口
let listener = TcpListener::bind("127.0.0.1:7878").unwrap();
处理客户端连接：
for stream in listener.incoming() {println!("Connection established!");}
当浏览器访问 http://127.0.0.1:7878/ 就打印 “Connection established!”, 表示服务启动成功。
处理接口请求
设计2个接口：
- 访问 http://127.0.0.1:7878 时，显示index.html 的内容
- 访问 http://127.0.0.1:7878/sleep 时，做一个延迟，模拟单线程服务的问题
准备html文件
- index.html
<!DOCTYPE html>
<html lang="en"><head><meta charset="utf-8"><title>Hello!</title></head><body><h1>Hello!</h1><p>Hi from Rust</p></body>
</html>
- 404.html
<!DOCTYPE html>
<html lang="en"><head><meta charset="utf-8"><title>你好!</title></head><body><h1>很抱歉!</h1><p>由于运维删库跑路，我们的数据全部丢失，总监也已经准备跑路，88</p></body>
</html>
实现
- 实现请求路由判断
- 读取 html 文件内容，并且返回
请求路由判断
通过 BufReader 获取请求数据
// 解析buffer. 获取请求的数据let buf_reader = BufReader::new(&mut stream);// 读取请求行let http_request: Vec<_> = buf_reader.lines().map(|result| result.unwrap()).take_while(|line| !line.is_empty()).collect();http_requset 得到数据如下：
["GET / HTTP/1.1","Host: 127.0.0.1:7878","Connection: keep-alive",...]
由此数据可得，只需要判断数组第一个数据就知道请求到哪个路由了。
let request_line = if http_request.len( ) > 0 { &http_request[0] } else {""};let (status_line, filename) = match &request_line[..] {"GET / HTTP/1.1" => ("HTTP/1.1 200 OK", "index.html"),"GET /sleep HTTP/1.1" => {std::thread::sleep(Duration::from_secs(5)); // 添加 5 s 延迟("HTTP/1.1 200 OK", "index.html")}_ => ("HTTP/1.1 404 NOT FOUND", "404.html"),};
处理html文件
通过 文件 处理模块 fs 获取文件的内容。
let contens = fs::read_to_string(filename).unwrap();
设置返回数据
获取文件的内容之后，把数据返回给客户端, 通过 write_all 方法。
let content_length = contens.len();// 组合返回数据let response = format!("{}\r\nContent-Length: {}\r\n\r\n{}",status_line,content_length,contens);stream.write_all(response.as_bytes()).unwrap();
效果
当客户端访问 http://127.0.0.1:7878 就会看到 index.html 的内容。如果客户端先访问 http://127.0.0.1:7878/sleep 新起标签页面访问 http://127.0.0.1:7878，如下图。你会发现 http://127.0.0.1:7878 的请求并没有立即显示，而是等待 sleep 请求结束才会显示。相当于接口会存在排队的处理，这对于web服务是个不好的体验。因此需要多线程去处理这些问题。
处理多线程请求
将每个请求都用 thread 去处理
thread::spawn(|| {handle_connection(stream);});
效果如下，并没有发生排队的问题。这只是简单处理，实际上现在成熟的框架处理单线程的问题。
源码