本文将介绍如何在 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 mainimport ( "log" "net/http")// Size constantsconst ( MB = 1 << 20)// Sizer interface to get file sizetype 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: %#vn", multipartFileHeader.Filename) log.Printf("Size: %#vn", file.(Sizer).Size()) log.Printf("MIME: %#vn", 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语言免费学习笔记（深入）”；
Otter.ai
一个自动的会议记录和笔记工具，会议内容生成和实时转录
91 查看详情 设置最大内存: 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: 186742023/10/27 10:00:00 MIME: "image/png"
注意事项
错误处理: 代码中包含了错误处理，例如检查 r.ParseMultipartForm 和 file.Read 的返回值。在实际应用中，应该根据具体情况处理这些错误。文件大小限制: 应该根据实际需求设置合理的文件大小限制，以防止恶意上传或服务器资源耗尽。MIME 类型检测: http.DetectContentType 函数基于文件内容进行 MIME 类型检测，但它并非总是准确的。如果需要更精确的 MIME 类型检测，可以考虑使用第三方库。安全: 在处理上传文件时，务必注意安全问题，例如防止恶意文件上传、文件覆盖等。
总结
本文介绍了如何在 Golang 中获取通过 multipart/form-data 方式上传的文件的相关信息，包括文件大小和 MIME 类型。通过示例代码，你可以快速上手，并将其应用到你的项目中。记住，在实际应用中，要根据具体需求进行适当的修改和扩展，并注意安全问题。
以上就是Golang 获取 Multipart 文件信息：大小、MIME 类型等的详细内容，更多请关注创想鸟其它相关文章！
版权声明：本文内容由互联网用户自发贡献，该文观点仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。
如发现本站有涉嫌抄袭侵权/违法违规的内容， 请发送邮件至 chuangxiangniao@163.com 举报，一经查实，本站将立刻删除。
发布者：程序猿，转转请注明出处：https://www.chuangxiangniao.com/p/1113068.html
微信扫一扫 
支付宝扫一扫 
