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