Golang 获取 Multipart 文件信息：大小、MIME 类型等

本文将介绍如何在 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语言免费学习笔记（深入）”；

文心大模型

百度飞桨-文心大模型 ERNIE 3.0 文本理解与创作

56

查看详情

1. 设置最大内存: r.ParseMultipartForm(5 * MB) 用于解析 multipart 表单，并设置最大内存为 5MB。这有助于防止恶意用户上传过大的文件导致服务器资源耗尽。
2. 限制上传大小: http.MaxBytesReader(w, r.Body, 5*MB) 进一步限制了请求体的大小，确保上传的文件不会超过 5MB。
3. 获取文件: r.FormFile("file") 用于获取名为 "file" 的上传文件。它返回 multipart.File 接口, multipart.FileHeader 结构体和错误信息。
4. 读取文件头: 为了获取 MIME 类型，需要读取文件的前 512 字节。file.Read(fileHeader) 将文件头读取到 fileHeader 缓冲区中。
5. 重置文件指针: 读取文件头后，需要将文件指针重置到文件开始处，以便后续处理。file.Seek(0, 0) 将文件指针移动到文件的开头。
6. 获取文件信息:
   • 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 类型。通过示例代码，你可以快速上手，并将其应用到你的项目中。记住，在实际应用中，要根据具体需求进行适当的修改和扩展，并注意安全问题。

以上就是Golang 获取 Multipart 文件信息：大小、MIME 类型等的详细内容，更多请关注php中文网其它相关文章！