50 lines
1.3 KiB
Go
50 lines
1.3 KiB
Go
package files
|
||
|
||
import (
|
||
"catface/app/global/my_errors"
|
||
"catface/app/global/variable"
|
||
"mime/multipart"
|
||
"net/http"
|
||
"os"
|
||
)
|
||
|
||
// 返回值说明：
|
||
// 7z、exe、doc 类型会返回 application/octet-stream 未知的文件类型
|
||
// jpg => image/jpeg
|
||
// png => image/png
|
||
// ico => image/x-icon
|
||
// bmp => image/bmp
|
||
// xlsx、docx 、zip => application/zip
|
||
// tar.gz => application/x-gzip
|
||
// txt、json、log等文本文件 => text/plain; charset=utf-8 备注：就算txt是gbk、ansi编码，也会识别为utf-8
|
||
|
||
// 通过文件名获取文件mime信息
|
||
func GetFilesMimeByFileName(filepath string) string {
|
||
f, err := os.Open(filepath)
|
||
if err != nil {
|
||
variable.ZapLog.Error(my_errors.ErrorsFilesUploadOpenFail + err.Error())
|
||
}
|
||
defer f.Close()
|
||
|
||
// 只需要前 32 个字节就可以了
|
||
buffer := make([]byte, 32)
|
||
if _, err := f.Read(buffer); err != nil {
|
||
variable.ZapLog.Error(my_errors.ErrorsFilesUploadReadFail + err.Error())
|
||
return ""
|
||
}
|
||
|
||
return http.DetectContentType(buffer)
|
||
}
|
||
|
||
// 通过文件指针获取文件mime信息
|
||
func GetFilesMimeByFp(fp multipart.File) string {
|
||
|
||
buffer := make([]byte, 32)
|
||
if _, err := fp.Read(buffer); err != nil {
|
||
variable.ZapLog.Error(my_errors.ErrorsFilesUploadReadFail + err.Error())
|
||
return ""
|
||
}
|
||
|
||
return http.DetectContentType(buffer)
|
||
}
|