
本文将介绍如何在 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)注意事项:
要获取文件的 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 处理程序中获取上传文件的大小和 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))
}使用方法:
示例 curl 命令:
curl -F "file=@your_file.txt" http://localhost:8080/upload
将 your_file.txt 替换为你想要上传的文件。
本文介绍了如何在 Golang 中获取通过 multipart.File 上传的文件的大小和 MIME 类型。通过类型断言和 http.DetectContentType 函数,我们可以轻松地提取这些信息。请记住,在实际应用中,应该进行适当的错误处理,以确保程序的健壮性。这些技术可以帮助你更好地处理文件上传,并根据文件类型和大小执行相应的操作。
以上就是Golang:获取 Multipart 文件信息,如大小和 MIME 类型的详细内容,更多请关注php中文网其它相关文章!
每个人都需要一台速度更快、更稳定的 PC。随着时间的推移,垃圾文件、旧注册表数据和不必要的后台进程会占用资源并降低性能。幸运的是,许多工具可以让 Windows 保持平稳运行。
Copyright 2014-2025 https://www.php.cn/ All Rights Reserved | php.cn | 湘ICP备2023035733号