本文将介绍如何在 Golang 中处理通过 `multipart/form-data` 方式上传的文件，并获取其相关信息，例如文件大小和 MIME 类型。通过示例代码，你将学习如何从 `multipart.File` 中读取文件头，进而检测文件类型，并获取文件的大小信息。
当用户通过 HTML 表单上传文件时，通常会使用 multipart/form-data 编码方式。在 Golang 中，我们可以使用 r.FormFile("file") 方法来获取上传的文件，它返回一个 multipart.File 接口，一个 multipart.FileHeader 结构体和一个 error。本文将重点介绍如何从 multipart.File 和 multipart.FileHeader 中提取有用的信息，例如文件大小和 MIME 类型。
获取文件大小和 MIME 类型
以下代码展示了如何获取上传文件的名称、大小和 MIME 类型：
package main
import (
"log"
"net/http"
)
// Size constants
const (
MB = 1 << 20
)
// Sizer interface to get file size
type Sizer interface {
Size() int64
}
func uploadHandler(w http.ResponseWriter, r *http.Request) error {
// 设置最大内存，防止Dos攻击
if err := r.ParseMultipartForm(5 * MB); err != nil {
return err
}
// 限制上传大小
r.Body = http.MaxBytesReader(w, r.Body, 5*MB) // 5 Mb
// 获取文件
file, multipartFileHeader, err := r.FormFile("file")
if err != nil {
return err
}
defer file.Close()
// 创建一个 buffer 来存储文件头
fileHeader := make([]byte, 512)
// 将文件头复制到 buffer 中
if _, err := file.Read(fileHeader); err != nil {
return err
}
// 将文件指针重置到文件开始处
if _, err := file.Seek(0, 0); err != nil {
return err
}
log.Printf("Name: %#v\n", multipartFileHeader.Filename)
log.Printf("Size: %#v\n", file.(Sizer).Size())
log.Printf("MIME: %#v\n", http.DetectContentType(fileHeader))
return nil
}
func main() {
http.HandleFunc("/upload", func(w http.ResponseWriter, r *http.Request) {
err := uploadHandler(w, r)
if err != nil {
http.Error(w, err.Error(), http.StatusInternalServerError)
return
}
w.WriteHeader(http.StatusOK)
w.Write([]byte("File uploaded successfully!"))
})
log.Fatal(http.ListenAndServe(":8080", nil))
}代码解释：
立即学习“go语言免费学习笔记（深入）”；
- 设置最大内存: r.ParseMultipartForm(5 * MB) 用于解析 multipart 表单，并设置最大内存为 5MB。这有助于防止恶意用户上传过大的文件导致服务器资源耗尽。
- 限制上传大小: http.MaxBytesReader(w, r.Body, 5*MB) 进一步限制了请求体的大小，确保上传的文件不会超过 5MB。
- 获取文件: r.FormFile("file") 用于获取名为 "file" 的上传文件。它返回 multipart.File 接口, multipart.FileHeader 结构体和错误信息。
- 读取文件头: 为了获取 MIME 类型，需要读取文件的前 512 字节。file.Read(fileHeader) 将文件头读取到 fileHeader 缓冲区中。
- 重置文件指针: 读取文件头后，需要将文件指针重置到文件开始处，以便后续处理。file.Seek(0, 0) 将文件指针移动到文件的开头。
-
获取文件信息:
- multipartFileHeader.Filename 包含上传文件的原始名称。
- file.(Sizer).Size() 获取文件的大小。这里使用类型断言将 multipart.File 接口转换为 Sizer 接口，该接口定义了 Size() 方法。
- http.DetectContentType(fileHeader) 使用读取的文件头来检测文件的 MIME 类型。
示例输出：
假设上传的文件名为 logo_35x30_black.png，大小为 18674 字节，那么输出可能如下所示：
2023/10/27 10:00:00 Name: "logo_35x30_black.png" 2023/10/27 10:00:00 Size: 18674 2023/10/27 10:00:00 MIME: "image/png"
注意事项
- 错误处理: 代码中包含了错误处理，例如检查 r.ParseMultipartForm 和 file.Read 的返回值。在实际应用中，应该根据具体情况处理这些错误。
- 文件大小限制: 应该根据实际需求设置合理的文件大小限制，以防止恶意上传或服务器资源耗尽。
- MIME 类型检测: http.DetectContentType 函数基于文件内容进行 MIME 类型检测，但它并非总是准确的。如果需要更精确的 MIME 类型检测，可以考虑使用第三方库。
- 安全: 在处理上传文件时，务必注意安全问题，例如防止恶意文件上传、文件覆盖等。
总结
本文介绍了如何在 Golang 中获取通过 multipart/form-data 方式上传的文件的相关信息，包括文件大小和 MIME 类型。通过示例代码，你可以快速上手，并将其应用到你的项目中。记住，在实际应用中，要根据具体需求进行适当的修改和扩展，并注意安全问题。
