Hyperlane 文件分块上传服务端的解决方案

cargo add hyperlane

use hyperlane::*;
use std::fs;
use std::path::Path;
const UPLOAD_DIR: &str = "uploads";
pub async fn handle(ctx: Context) {
    if !Path::new(UPLOAD_DIR).exists() {
        let _ = fs::create_dir_all(UPLOAD_DIR);
    }
    // 处理文件块上传逻辑...
}

// 从请求头获取文件信息
let file_id = ctx.get_request_header("x-file-id").await.unwrap_or_default();
let chunk_index = ctx.get_request_header("x-chunk-index").await.unwrap_or_default().parse::<usize>().unwrap_or(0);
let total_chunks = ctx.get_request_header("x-total-chunks").await.unwrap_or_default().parse::<usize>().unwrap_or(0);
let file_name = ctx.get_request_header("x-file-name").await.unwrap_or_default();
// 保存文件块
let chunk_path = format!("{}/{}_{}", UPLOAD_DIR, file_id, chunk_index);
fs::write(&chunk_path, ctx.get_request_body().await).unwrap();
// 合并文件
if all_chunks_uploaded {
    let final_path = format!("{}/{}", UPLOAD_DIR, file_name);
    let mut output_file = fs::File::create(&final_path).unwrap();
    for i in 0..total_chunks {
        let chunk_path = format!("{}/{}_{}", UPLOAD_DIR, file_id, i);
        let data = fs::read(&chunk_path).unwrap();
        output_file.write_all(&data).unwrap();
        fs::remove_file(&chunk_path).unwrap();
    }
}