Go mime.TypeByExtension 怎样根据扩展名判断 MIME 类型

Go 中根据文件扩展名判断 MIME 类型，可以直接调用 mime.TypeByExtension。关键是传入带点的扩展名，例如 .jpg，而不是把完整文件名直接传进去；如果标准库或系统映射中没有对应类型，函数会返回空字符串。

官方资料：https://pkg.go.dev/mime#TypeByExtension

一、先明确 TypeByExtension 的输入和返回值

mime.TypeByExtension 的签名很短：传入扩展名字符串，返回 MIME 类型字符串。官方约定扩展名应以点开头，例如 .html；没有关联类型时返回空字符串。

package main

import (
	"fmt"
	"mime"
)

func main() {
	// 扩展名带点时，标准库可以按内置或系统映射查询。
	for _, ext := range []string{".html", ".json", ".unknown"} {
		typ := mime.TypeByExtension(ext)
		// 空字符串表示没有找到映射，不能误当成 application/octet-stream。
		fmt.Printf("%s => %q\n", ext, typ)
	}
}

这里的 .html、.json 只是常见输入示例。函数返回的是 MIME 描述，不是对文件内容的真实性证明。

二、从文件名提取扩展名再查询

业务代码通常拿到的是上传文件名或本地路径，先用 filepath.Ext 提取最后一个扩展名，再交给 TypeByExtension。不要手写字符串截取，否则容易遗漏没有扩展名、隐藏文件和多段后缀等情况。

package main

import (
	"fmt"
	"mime"
	"path/filepath"
	"strings"
)

func mimeFromName(name string) (string, bool) {
	// Ext 返回最后一个点开始的后缀，例如 report.HTML 得到 .HTML。
	ext := filepath.Ext(name)
	if ext == "" {
		return "", false
	}

	// 统一大小写让业务日志和白名单比较更容易保持一致。
	typ := mime.TypeByExtension(strings.ToLower(ext))
	if typ == "" {
		return "", false
	}
	return typ, true
}

func main() {
	// 文件名只用于演示扩展名到 MIME 的映射过程。
	typ, ok := mimeFromName("report.HTML")
	fmt.Println(typ, ok)
}

标准库本身已经支持大小写不敏感查找，但在业务层统一扩展名仍有价值：日志、规则表和后续比较可以用同一种格式。若文件名来自用户输入，还应把它当作标签而不是信任依据。

三、处理大小写、文本 charset 与未知类型

当前 Go 文档说明，扩展名先按大小写敏感方式查找，再按不区分大小写方式查找。文本类型还可能带上默认的 charset=utf-8，因此不要把返回值简单截断成只有斜杠前后的主类型。

官方实现还会读取平台可用的 MIME 数据：Unix 可能使用系统 MIME-info 数据库或 mime.types 文件，Windows 则从注册表提取。因此同一扩展名在不同部署环境中可能有差异，跨环境需要稳定结果时，应在业务层维护允许的映射。

四、需要自定义映射时注册扩展名

如果业务有明确的自定义后缀，可以在初始化阶段调用 mime.AddExtensionType。扩展名同样需要以点开头，注册失败要直接记录或返回，避免应用以为映射已经生效。

package main

import (
	"fmt"
	"mime"
)

func init() {
	// 自定义扩展名必须带前导点，类型值保持完整的 media type 形式。
	if err := mime.AddExtensionType(".reportx", "application/x-report"); err != nil {
		panic(err)
	}
}

func main() {
	// 注册完成后，业务代码可以按普通扩展名查询。
	fmt.Println(mime.TypeByExtension(".reportx"))
}

这类注册是进程级的全局映射，适合在应用启动时一次完成，不建议在每个请求中动态修改。更复杂的租户或版本规则，可以把 TypeByExtension 当作默认值，再由业务白名单覆盖。

五、把扩展名判断放在正确的安全边界

文件名扩展名很容易被修改，所以它只能用于初步分类、选择响应头或决定展示方式。上传接口若需要判断实际内容，可以读取文件头交给 http.DetectContentType，并同时限制大小、允许类型和存储路径。

package main

import (
	"fmt"
	"net/http"
)

func sniff(data []byte) string {
	// DetectContentType 依据前 512 字节内的内容做类型探测。
	// 它仍然不是完整的业务安全策略，最终要结合允许列表。
	return http.DetectContentType(data)
}

func main() {
	// 示例数据只用于说明内容探测入口，不代表任何真实上传文件。
	fmt.Println(sniff([]byte("plain text")))
}

1. 不要把空字符串直接写成成功类型。未知扩展名应该进入显式兜底分支。
2. 不要只看最后的字符串。多段后缀、无后缀和伪造文件名都需要单独处理。
3. 不要把扩展名当作内容校验。需要安全边界时，加入内容探测、大小限制和白名单。
4. 不要忽略平台差异。依赖系统 MIME 数据时，要在部署环境中固定或记录映射来源。

总结：TypeByExtension 适合做哪一层

相关问题

为什么传入完整文件名会返回空字符串？

函数接收的是扩展名，不是路径或完整文件名。先用 filepath.Ext 取出类似 .json 的值，再进行查询。

未知类型应该返回 application/octet-stream 吗？

标准库只负责返回映射结果，未知时返回空字符串。是否使用 application/octet-stream 应由业务的下载、上传或响应策略决定。

TypeByExtension 能判断文件是否真的是图片吗？

不能。它依据扩展名和 MIME 映射工作；需要判断实际内容时，还应读取文件头并配合允许列表、大小限制和安全存储策略。