如何處理Go語言中的并發文件上傳問題

169次閱讀

共計 936 個字符，預計需要花費 3 分鐘才能閱讀完成。

在 Go 語言中處理并發文件上傳問題時，可以使用 goroutine 來實現并發上傳。下面是一個簡單的示例代碼：

package main
import (
"fmt"
"io"
"net/http"
"os"
"path/filepath"
)
func main() {http.HandleFunc("/upload", uploadHandler)
http.ListenAndServe(":8080", nil)
}
func uploadHandler(w http.ResponseWriter, r *http.Request) {
if r.Method == "POST" {r.ParseMultipartForm(32 << 20) // 設置最大文件大小
file, handler, err := r.FormFile("file")
if err != nil {fmt.Println(err)
return
}
defer file.Close()
filename := handler.Filename
savePath := filepath.Join("./uploads", filename)
out, err := os.Create(savePath)
if err != nil {fmt.Println(err)
return
}
defer out.Close()
_, err = io.Copy(out, file)
if err != nil {fmt.Println(err)
return
}
fmt.Fprintf(w, " 文件上傳成功！")
}
}

在這個示例中，我們通過 http.HandleFunc 函數將 uploadHandler 函數綁定到 /upload 路徑上。在 uploadHandler 函數中，我們使用 r.ParseMultipartForm 函數解析表單數據，并通過 r.FormFile 函數獲取上傳的文件。
然后，我們通過 os.Create 函數創建一個文件，之后使用 io.Copy 函數將上傳的文件內容拷貝到新創建的文件中。最后，使用 fmt.Fprintf 函數向客戶端返回上傳成功的消息。
這個示例中的代碼是串行執行的，如果需要并發處理多個文件上傳，可以將文件上傳的處理放在一個 goroutine 中。具體實現方式可以根據實際需求進行調整。