golang获取文件类型

在golang中，获取文件类型是一个常见的需求。无论是在处理文件上传、文件下载、文件管理等场景下，我们常常需要知道文件的类型以便进行相应的处理。本文将介绍如何使用golang来获取文件类型。

使用扩展名

最简单和直接的方法就是通过文件的扩展名来获取文件类型。在golang中，可以通过path包中的Ext函数来获取文件的扩展名。代码如下：

import "path"
func GetFileTypeByExtension(filename string) string {
    ext := path.Ext(filename)
    switch ext {
        case ".txt":
            return "Text"
        case ".jpg", ".jpeg", ".png", ".gif":
            return "Image"
        case ".mp3", ".wav", ".flac":
            return "Audio"
        case ".mp4", ".mov", ".avi":
            return "Video"
        default:
            return "Unknown"
    }
}

上述代码通过switch语句判断文件的扩展名，并返回对应的类型。请注意，这只是一种简单的方法，可能并不完全准确，因为有些文件的扩展名可能是不准确或者不存在的。

使用mime类型

除了使用扩展名，我们还可以使用mime类型来获取文件的类型。golang中的mime包提供了获取文件的mime类型的功能。代码如下：

import "mime"
func GetFileTypeByMIME(filename string) (string, error) {
    file, err := os.Open(filename)
    if err != nil {
        return "", err
    }
    defer file.Close()
    header := make([]byte, 512)
    _, err = file.Read(header)
    if err != nil {
        return "", err
    }
    mimeType := http.DetectContentType(header)
    return mimeType, nil
}

上述代码首先打开文件，并读取文件的前512个字节。然后使用http包中的DetectContentType函数来获取文件的mimeType。

使用magic数字

除了扩展名和mime类型之外，我们还可以使用文件的magic数字来获取文件类型。magic数字是文件的一部分字节，用来标识文件的类型。golang中的filemagic包提供了读取文件magic数字的功能。代码如下：

import "github.com/rakyll/magicmime"
func GetFileTypeByMagicNumber(filename string) (string, error) {
    m, err := magicmime.New(magicmime.MAGIC_MIME_TYPE | magicmime.MAGIC_SYMLINK | magicmime.MAGIC_ERROR)
    if err != nil {
        return "", err
    }
    defer m.Close()
    mime, err := m.TypeByFile(filename)
    if err != nil {
        return "", err
    }
    return mime, nil
}

上述代码使用第三方库github.com/rakyll/magicmime来获取文件的magic数字。通过调用TypeByFile函数，我们可以获得文件的mime类型。

综上所述，我们可以通过多种方式来获取文件的类型。使用扩展名是最简单和直接的方法，但并不准确；使用mime类型可以更准确地判断文件的类型，但需要打开文件并读取一部分内容；使用magic数字可以更准确地判断文件的类型，但需要借助第三方库。根据具体的需求和场景，选择合适的方法来获取文件类型。