在Golang的标准库中，mime包提供了处理媒体类型（MIME类型）的相关功能，能够帮助开发者完成文件扩展名与MIME类型的相互转换、自定义MIME类型映射等操作，在文件上传、文件解析等场景中非常实用。
mime包核心功能概述
mime包主要围绕MIME类型和文件扩展名的映射关系提供能力，核心功能可以分为三类：获取扩展名对应的MIME类型、获取MIME类型对应的扩展名、自定义MIME类型映射规则。这些功能都基于内置的类型映射表实现，同时也支持开发者扩展自定义的规则。
常用方法详解
1. 获取扩展名对应的MIME类型
使用TypeByExtension方法可以根据文件扩展名返回对应的MIME类型，该方法接收一个带点的扩展名字符串作为参数，返回对应的MIME类型字符串和是否找到的布尔值。
示例代码如下：
package main
import (
"fmt"
"mime"
)
func main() {
// 查询常见扩展名的MIME类型
mimeType, ok := mime.TypeByExtension(".jpg")
if ok {
fmt.Printf(".jpg对应的MIME类型是：%sn", mimeType)
} else {
fmt.Println("未找到.jpg对应的MIME类型")
}
// 查询不常见的扩展名
mimeType, ok = mime.TypeByExtension(".md")
if ok {
fmt.Printf(".md对应的MIME类型是：%sn", mimeType)
} else {
fmt.Println("未找到.md对应的MIME类型")
}
}
2. 获取MIME类型对应的扩展名
ExtensionsByType方法可以根据MIME类型返回对应的所有文件扩展名列表，该方法接收一个MIME类型字符串作为参数，返回扩展名字符串切片和是否找到的布尔值。
示例代码如下：
package main
import (
"fmt"
"mime"
)
func main() {
// 查询image/png对应的扩展名
exts, ok := mime.ExtensionsByType("image/png")
if ok {
fmt.Printf("image/png对应的扩展名有：%vn", exts)
} else {
fmt.Println("未找到image/png对应的扩展名")
}
// 查询text/html对应的扩展名
exts, ok = mime.ExtensionsByType("text/html")
if ok {
fmt.Printf("text/html对应的扩展名有：%vn", exts)
} else {
fmt.Println("未找到text/html对应的扩展名")
}
}
3. 自定义MIME类型映射
如果内置的映射表无法满足需求，开发者可以通过AddExtensionType方法添加自定义的扩展名和MIME类型的映射关系，该方法接收扩展名和MIME类型两个字符串参数，添加后后续的查询方法就可以使用自定义的映射规则。
示例代码如下：
package main
import (
"fmt"
"mime"
)
func main() {
// 添加自定义的MIME类型映射
err := mime.AddExtensionType(".myext", "application/x-myapp")
if err != nil {
fmt.Printf("添加映射失败：%vn", err)
return
}
// 查询自定义的扩展名
mimeType, ok := mime.TypeByExtension(".myext")
if ok {
fmt.Printf(".myext对应的MIME类型是：%sn", mimeType)
}
// 查询自定义的MIME类型对应的扩展名
exts, ok := mime.ExtensionsByType("application/x-myapp")
if ok {
fmt.Printf("application/x-myapp对应的扩展名有：%vn", exts)
}
}
实际场景：文件上传时判断文件类型
在文件上传场景中，除了前端校验文件类型，后端也需要通过文件内容或者扩展名判断文件类型，使用mime包结合文件扩展名可以快速完成初步校验。
完整示例代码如下：
package main
import (
"fmt"
"mime"
"net/http"
)
func uploadHandler(w http.ResponseWriter, r *http.Request) {
// 限制请求方法为POST
if r.Method != http.MethodPost {
http.Error(w, "仅支持POST请求", http.StatusMethodNotAllowed)
return
}
// 解析表单，限制上传文件大小为10MB
err := r.ParseMultipartForm(10 << 20)
if err != nil {
http.Error(w, "解析表单失败", http.StatusBadRequest)
return
}
// 获取上传的文件
file, header, err := r.FormFile("uploadFile")
if err != nil {
http.Error(w, "获取上传文件失败", http.StatusBadRequest)
return
}
defer file.Close()
// 获取文件扩展名
filename := header.Filename
ext := filename[len(filename)-4:] // 简单获取最后4个字符作为扩展名，实际场景需要处理更通用的情况
if ext[0] != '.' {
http.Error(w, "无法识别文件扩展名", http.StatusBadRequest)
return
}
// 通过mime包查询扩展名对应的MIME类型
mimeType, ok := mime.TypeByExtension(ext)
if !ok {
http.Error(w, "不支持的文件类型", http.StatusBadRequest)
return
}
// 允许的文件MIME类型列表
allowedTypes := map[string]bool{
"image/jpeg": true,
"image/png": true,
"application/pdf": true,
}
if !allowedTypes[mimeType] {
http.Error(w, "不支持的文件类型", http.StatusBadRequest)
return
}
fmt.Fprintf(w, "文件上传成功，文件类型为：%s", mimeType)
}
func main() {
http.HandleFunc("/upload", uploadHandler)
fmt.Println("服务启动在 :8080")
http.ListenAndServe(":8080", nil)
}
注意事项
TypeByExtension方法对扩展名的大小写不敏感，传入".JPG"和".jpg"会得到相同的结果。- 内置的MIME类型映射表是基于系统的MIME类型数据库构建的，不同系统可能存在细微差异，如果需要在跨平台场景保持一致的映射关系，建议自定义添加必要的映射规则。
- 仅通过扩展名判断文件类型并不完全可靠，恶意用户可能修改文件扩展名绕过校验，重要场景下建议结合文件内容的魔术数字进行二次校验。