Go vs Rust：文件上傳性能比較

作者：Slagga 2023-11-19 21:17:58

在本文中，主要測試并比較了Go—Gin和Rust—Actix之間的多部分文件上傳性能。

一、設(shè)置

所有測試都在配備16G內(nèi)存的 MacBook Pro M1 上執(zhí)行。

軟件版本為：

Go v1.20.5
Rust v1.70.0

測試工具是一個基于 libcurl 并使用標準線程的自定義工具，能夠發(fā)送多部分請求。

資產(chǎn)目錄中有 100,000 個文件。每個文件的大小都是確切的 100K。這些文件數(shù)量在測試工作線程之間進行分配。同一個文件不會一遍又一遍地上傳。工作線程會循環(huán)處理分配給它們的文件。一旦它們處理完所有分配的文件，它們就會回到第一個文件重新開始。

每個請求攜帶兩個文件作為多部分請求體。請求的頭部和體部大致如下：

// -- Headers

{
  "content-length": "205150",
  "content-type": "multipart/form-data; boundary=------------------------3f6a15690b315b91",
}

// -- Body

--------------------------3f6a15690b315b91
Content-Disposition: form-data; name="files"; filename="45469"
Content-Type: application/octet-stream

<<File suppressed>>
--------------------------3f6a15690b315b91
Content-Disposition: form-data; name="files"; filename="42102"
Content-Type: application/octet-stream

<<file suppressed>>
--------------------------3f6a15690b315b91--

二、代碼

1.Go

package main

import (
    "github.com/gin-gonic/gin"
    "github.com/jaevor/go-nanoid"
)

func main() {
    dst := "/Users/mayankc/Work/source/perfComparisons/uploads/"
    canonicID, err := nanoid.Standard(21)
    if err != nil {
        panic(err)
    }

    router := gin.New()
    router.POST("/upload", func(c *gin.Context) {
        form, _ := c.MultipartForm()
        files := form.File["files"]

        for _, file := range files {
            c.SaveUploadedFile(file, dst+canonicID())
        }
        c.Writer.WriteHeader(201)
    })
    router.Run(":3000")
}

2.Rust

use actix_multipart::{
    form::{
        tempfile::{TempFile, TempFileConfig},
        MultipartForm,
    }
};
use actix_web::{middleware, web, App, Error, HttpResponse, HttpServer, Responder};
use nanoid::nanoid;

const BASE_DIR: &str = "/Users/mayankc/Work/source/perfComparisons/uploads/";

#[derive(Debug, MultipartForm)]
struct UploadForm {
    #[multipart(rename = "files")]
    files: Vec<TempFile>,
}

async fn save_files(
    MultipartForm(form): MultipartForm<UploadForm>,
) -> Result<impl Responder, Error> {
    for f in form.files {
        let path = format!("{}{}", BASE_DIR, nanoid!());
        f.file.persist(path).unwrap();
    }

    Ok(HttpResponse::Ok())
}

#[actix_web::main]
async fn main() -> std::io::Result<()> {
    HttpServer::new(|| {
        App::new()
            .wrap(middleware::Logger::default())
            .app_data(TempFileConfig::default().directory(BASE_DIR))
            .service(
                web::resource("/upload")
                    .route(web::post().to(save_files)),
            )
    })
    .bind(("127.0.0.1", 3000))?
    .run()
    .await
}

Rust代碼已在release mode下編譯。