spwf 0.0.2

Simple personal website framework
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43

use std::sync::Arc;
use aoko::{standard::{functions::ext::StdAnyExt, asynchronies::tokio::AsyncExt}, no_std::functions::ext::Utf8Ext};
use spwf::{SharedData, server::handlers::{Index, Handler, StaticFile, VisitCount, NotFound, Echo}};
use tokio::{net::{TcpListener, TcpStream}, io::{AsyncReadExt}, sync::Mutex};

#[tokio::main]
async fn main() {
    let listener = TcpListener::bind("127.0.0.1:5000").await.unwrap();

    let data = SharedData { visit_count: 0 }.into_tokio_arc_mutex();

    loop {
        let (mut stream, _) = listener.accept().await.unwrap();

        let data = data.clone();

        tokio::spawn(async move {
            let buf = &mut [0; 1024];

            stream.read(buf).await.unwrap();
            // buf.to_str_lossy().echo();  // 打印请求信息

            route(&mut stream, buf, data).await;
        });
    }
}

async fn route(stream: &mut TcpStream, buf: &[u8], shared_data: Arc<Mutex<SharedData>>) {
    if buf.starts_with(b"GET / HTTP/1.1") {
        // Index page
        Index.handle(stream, shared_data).await;
    } else if buf.starts_with(b"GET /static") {
        // Static file
        StaticFile { path_buf: buf }.handle(stream, shared_data).await;
    } else if buf.starts_with(b"GET /count") {
        // Visit count
        VisitCount.handle(stream, shared_data).await
    } else if buf.starts_with(b"GET /echo") {
        Echo { path_buf: buf }.handle(stream, shared_data).await;
    } else {
        NotFound.handle(stream, shared_data).await;
    }
}