本文将介绍如何在 Golang 中获取通过 `multipart.File` 上传的文件信息。我们将重点讲解如何获取文件的大小和 MIME 类型，并提供示例代码帮助你理解和应用这些技术。通过本文，你将能够轻松地从上传的文件中提取所需的信息，从而进行后续处理，例如文件类型验证、大小限制等。
在 Golang 中处理文件上传时，multipart.File 类型提供了对上传文件的访问能力。然而，直接从 multipart.File 对象中获取文件大小和 MIME 类型可能并不直接。本文将介绍如何利用 Golang 的标准库来获取这些信息。
获取文件大小
multipart.File 接口本身并没有直接提供获取文件大小的方法。但是，它通常会实现 io.Seeker 接口，允许我们移动文件指针。更重要的是，它会实现一个非导出的 Size() 方法。因此，我们可以通过类型断言来获取文件的大小。
以下代码片段展示了如何获取文件大小：
立即学习“go语言免费学习笔记（深入）”；
type Sizer interface {
Size() int64
}
file, multipartFileHeader, err := r.FormFile("file")
if err != nil {
// Handle error
return err
}
// 获取文件大小
fileSize := file.(Sizer).Size()
log.Printf("Size: %d\n", fileSize)注意事项：
- 确保你的 multipart.File 实现了 Sizer 接口。
- 如果类型断言失败，程序会 panic。在实际应用中，应该进行错误处理。
获取 MIME 类型
要获取文件的 MIME 类型，我们需要读取文件的前几个字节，然后使用 http.DetectContentType 函数进行检测。
以下代码片段展示了如何获取 MIME 类型：
file, multipartFileHeader, err := r.FormFile("file")
if err != nil {
// Handle error
return err
}
// 创建一个缓冲区来存储文件头
fileHeader := make([]byte, 512)
// 将文件头复制到缓冲区中
if _, err := file.Read(fileHeader); err != nil {
return err
}
// 将文件指针重置到文件开头
if _, err := file.Seek(0, 0); err != nil {
return err
}
// 检测 MIME 类型
mimeType := http.DetectContentType(fileHeader)
log.Printf("MIME: %s\n", mimeType)注意事项：
- http.DetectContentType 函数通过检查文件的前 512 个字节来推断 MIME 类型。
- 如果文件小于 512 字节，该函数仍然可以工作，但结果可能不准确。
- 读取文件头后，务必将文件指针重置到文件开头，以便后续操作。
完整示例
以下是一个完整的示例，展示了如何在 HTTP 处理程序中获取上传文件的大小和 MIME 类型：
package main
import (
"fmt"
"log"
"net/http"
)
// Size constants
const (
MB = 1 << 20
)
type Sizer interface {
Size() int64
}
func uploadHandler(w http.ResponseWriter, r *http.Request) {
if err := r.ParseMultipartForm(5 * MB); err != nil {
http.Error(w, err.Error(), http.StatusBadRequest)
return
}
// Limit upload size
r.Body = http.MaxBytesReader(w, r.Body, 5*MB) // 5 Mb
file, multipartFileHeader, err := r.FormFile("file")
if err != nil {
http.Error(w, err.Error(), http.StatusBadRequest)
return
}
defer file.Close()
// Create a buffer to store the header of the file in
fileHeader := make([]byte, 512)
// Copy the headers into the FileHeader buffer
if _, err := file.Read(fileHeader); err != nil {
http.Error(w, err.Error(), http.StatusInternalServerError)
return
}
// set position back to start.
if _, err := file.Seek(0, 0); err != nil {
http.Error(w, err.Error(), http.StatusInternalServerError)
return
}
fileSize := file.(Sizer).Size()
mimeType := http.DetectContentType(fileHeader)
log.Printf("Name: %#v\n", multipartFileHeader.Filename)
log.Printf("Size: %d\n", fileSize)
log.Printf("MIME: %s\n", mimeType)
fmt.Fprintf(w, "File uploaded successfully!\n")
fmt.Fprintf(w, "Name: %s\n", multipartFileHeader.Filename)
fmt.Fprintf(w, "Size: %d bytes\n", fileSize)
fmt.Fprintf(w, "MIME Type: %s\n", mimeType)
}
func main() {
http.HandleFunc("/upload", uploadHandler)
log.Println("Server listening on port 8080")
log.Fatal(http.ListenAndServe(":8080", nil))
}使用方法：
- 将以上代码保存为 main.go。
- 运行 go run main.go。
- 使用浏览器或 curl 等工具向 http://localhost:8080/upload 发送包含文件上传的 POST 请求。
示例 curl 命令：
curl -F "file=@your_file.txt" http://localhost:8080/upload
将 your_file.txt 替换为你想要上传的文件。
总结
本文介绍了如何在 Golang 中获取通过 multipart.File 上传的文件的大小和 MIME 类型。通过类型断言和 http.DetectContentType 函数，我们可以轻松地提取这些信息。请记住，在实际应用中，应该进行适当的错误处理，以确保程序的健壮性。这些技术可以帮助你更好地处理文件上传，并根据文件类型和大小执行相应的操作。
