转载转载转载!!!!!
https://liqiang.io/post/get-the-content-type-of-file-a31e7f02
Golang 获取文件的 Content-Type
@SOLUTION· 2020-03-13 23:45 · 4 min read
概述
在写 Web 程序的时候,有的时候框架并不能满足一些需求,所以有时需要使用平凡的 HTTP 响应,这个时候就需要知道返回内容的 Content-Type 了,本文就介绍一下如何获取文件的 Content-Type。
- 获取文件 Content-Type
在 Go 的标准库:net/http 中就存在可以查找一个文件的 ContentType 和 MimeType 的方法。但是这个方法的输入是文件的内容,它是通过判断文件里面的某些特征字,从而判断文件的类型。
判断文件 ContentType 的方法为:DetectContentType(),它会读取你文件的前 512 位进行判断,然后返回一个 Mime 类型,例如 “application/json” 或者 “image/jpg”。
- 示例
[root@liqiang.io]# 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
}