0. 概述

在写 Web 程序的时候，有的时候框架并不能满足一些需求，所以有时需要使用平凡的 HTTP 响应，这个时候就需要知道返回内容的 Content-Type 了，本文就介绍一下如何获取文件的 Content-Type。

1. 获取文件 Content-Type

在 Go 的标准库：net/http 中就存在可以查找一个文件的 ContentType 和 MimeType 的方法。但是这个方法的输入是文件的内容，它是通过判断文件里面的某些特征字，从而判断文件的类型。

判断文件 ContentType 的方法为：DetectContentType()，它会读取你文件的前 512 位进行判断，然后返回一个 Mime 类型，例如 “application/json” 或者 “image/jpg”。

2. 示例

[[email protected]]# cat get_content_type.go
package main

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

func main() {

    // Open File
    f, err := os.Open("golangcode.pdf")
    if err != nil {
        panic(err)
    }
    defer f.Close()

    // Get the content
    contentType, err := GetFileContentType(f)
    if err != nil {
        panic(err)
    }

    fmt.Println("Content Type: " + contentType)
}

func GetFileContentType(out *os.File) (string, error) {

    // Only the first 512 bytes are used to sniff the content type.
    buffer := make([]byte, 512)

    _, err := out.Read(buffer)
    if err != nil {
        return "", err
    }

    // Use the net/http package's handy DectectContentType function. Always returns a valid
    // content-type by returning "application/octet-stream" if no others seemed to match.
    contentType := http.DetectContentType(buffer)

    return contentType, nil
}

3. Ref