本文将介绍如何在 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: %dn", 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: %sn", mimeType)
注意事项：
http.DetectContentType 函数通过检查文件的前 512 个字节来推断 MIME 类型。如果文件小于 512 字节，该函数仍然可以工作，但结果可能不准确。读取文件头后，务必将文件指针重置到文件开头，以便后续操作。
完整示例
以下是一个完整的示例，展示了如何在 HTTP 处理程序中获取上传文件的大小和 MIME 类型：
package mainimport ( "fmt" "log" "net/http")// Size constantsconst ( 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: %#vn", multipartFileHeader.Filename) log.Printf("Size: %dn", fileSize) log.Printf("MIME: %sn", mimeType) fmt.Fprintf(w, "File uploaded successfully!n") fmt.Fprintf(w, "Name: %sn", multipartFileHeader.Filename) fmt.Fprintf(w, "Size: %d bytesn", fileSize) fmt.Fprintf(w, "MIME Type: %sn", 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 函数，我们可以轻松地提取这些信息。请记住，在实际应用中，应该进行适当的错误处理，以确保程序的健壮性。这些技术可以帮助你更好地处理文件上传，并根据文件类型和大小执行相应的操作。
以上就是Golang：获取 Multipart 文件信息，如大小和 MIME 类型的详细内容，更多请关注创想鸟其它相关文章！
版权声明：本文内容由互联网用户自发贡献，该文观点仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。
如发现本站有涉嫌抄袭侵权/违法违规的内容， 请发送邮件至 chuangxiangniao@163.com 举报，一经查实，本站将立刻删除。
发布者：程序猿，转转请注明出处：https://www.chuangxiangniao.com/p/1414086.html
微信扫一扫 
支付宝扫一扫 
