首页 > 后端开发 > Golang > 正文

Golang:获取 Multipart 文件信息,如大小和 MIME 类型

霞舞
发布: 2025-10-15 11:38:37
原创
538人浏览过

golang:获取 multipart 文件信息,如大小和 mime 类型

本文将介绍如何在 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 类型:

文心大模型
文心大模型

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

文心大模型 56
查看详情 文心大模型
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))
}
登录后复制

使用方法:

  1. 将以上代码保存为 main.go
  2. 运行 go run main.go。
  3. 使用浏览器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 类型的详细内容,更多请关注php中文网其它相关文章!

最佳 Windows 性能的顶级免费优化软件
最佳 Windows 性能的顶级免费优化软件

每个人都需要一台速度更快、更稳定的 PC。随着时间的推移,垃圾文件、旧注册表数据和不必要的后台进程会占用资源并降低性能。幸运的是,许多工具可以让 Windows 保持平稳运行。

下载
来源:php中文网
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn
最新问题
开源免费商场系统广告
热门教程
更多>
最新下载
更多>
网站特效
网站源码
网站素材
前端模板
关于我们 免责申明 意见反馈 讲师合作 广告合作 最新更新 English
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号 技术交流群
PHP中文网订阅号
每天精选资源文章推送
PHP中文网APP
随时随地碎片化学习

Copyright 2014-2025 https://www.php.cn/ All Rights Reserved | php.cn | 湘ICP备2023035733号