Go 标准库的神秘功能:如何轻松识别任何文件类型-技术圈

阅读本文大概需要 2 分钟。

文件类型识别是在很多应用场景中都需要用到的功能，比如在 Web 开发中，我们需要根据上传文件的类型来进行不同的处理，或者在文件管理系统中，我们需要根据文件类型来显示不同的图标和操作。文件类型识别的常用方法有两种：一种是根据文件的扩展名来判断，比如 .jpg 表示 JPEG 图像，.mp3 表示 MP3 音频等；另一种是根据文件内容的特征来判断，比如 JPEG 图像的前几个字节是 FF D8 FF，MP3 音频的前几个字节是 ID3 等。这两种方法各有优缺点，扩展名方法简单快速，但是容易被伪造或者缺失；内容特征方法准确可靠，但是需要读取文件内容并进行分析。

Go 语言的标准库 net/http 包中提供了一个简单的判断文件类型的方法 DetectContentType() 。它读取文件内容的前 512 个字节内容，返回一个 MIME 类型字符串，例如 image/jpeg。它使用了 mimesniff 算法²⁴，根据一组预定义的规则来匹配文件内容的特征和对应的 MIME 类型。这个方法既不依赖于文件扩展名，也不需要完整地读取文件内容，因此既快速又准确。

下面我们来看一个使用 Go 语言识别 MP3 文件格式的示例：

      package main

import (
 "fmt"
 "net/http"
 "os"
)

func main() {
 // 打开一个 MP3 文件
 file, err := os.Open("example.mp3")
 if err != nil {
  fmt.Println(err)
  return
 }
 defer file.Close()

 // 读取文件的前 512 个字节
 buffer := make([]byte, 512)
 n, err := file.Read(buffer)
 if err != nil {
  fmt.Println(err)
  return
 }

 // 调用 http.DetectContentType 方法判断文件类型
  // 实际上，如果字节数超过 512，该函数也只会使用前 512 个字节
 contentType := http.DetectContentType(buffer[:n])
 fmt.Println(contentType) // 输出 audio/mpeg
}

这个示例中，我们首先打开了一个 MP3 文件，并读取了它的前 512 个字节到一个缓冲区中。然后我们调用了 http.DetectContentType 方法，并传入了缓冲区中的数据。这个方法会返回一个字符串表示文件的 MIME 类型，对于 MP3 文件来说，就是 audio/mpeg。

此外，返回的 contentType 可以通过以下方式返回具体的扩展名：

      // 上面 mp3 文件返回的会是 .mp3，注意包含 .
ext := mime.ExtensionsByType(contentType)

在本文中，我们介绍了文件类型识别的应用场景和常用方法，并且以 Go 语言为例，展示了如何使用 http.DetectContentType 方法来识别 MP3 文件格式。这个方法是 Go 标准库提供的一个简单而有效的判断文件类型的方法，可以在很多场合中使用。我们希望这篇文章能够对你有所帮助。

如果你遇到了无法识别的文件或识别不准，怎么办？请期待后续文章。

往期推荐

XML 处理，Go 标准库太简单了怎么办？

我是 polarisxu，北大硕士毕业，曾在 360 等知名互联网公司工作，10多年技术研发与架构经验！2012 年接触 Go 语言并创建了 Go 语言中文网！著有《 Go语言编程之旅》、开源图书《 Go语言标准库》等。

坚持输出技术（包括 Go、Rust 等技术）、职场心得和创业感悟！欢迎关注「polarisxu」一起成长！也欢迎加我微信好友交流： gopherstudio