Golang 检查文件扩展名

在 Golang 中，我们可以通过判断文件名的后缀来实现判断文件类型的功能，而后缀即为文件扩展名。以下是一段用于检查文件扩展名的示例代码。

package main

import (
    "fmt"
    "strings"
)

func main() {
    fileName := "example.jpg"
    if strings.HasSuffix(fileName, ".jpg") {
        fmt.Println("This is a jpg file")
    } else if strings.HasSuffix(fileName, ".png") {
        fmt.Println("This is a png file")
    } else {
        fmt.Println("Unknown file type")
    }
}

上面的代码通过 strings.HasSuffix 函数来判断文件名的后缀是否为 .jpg 或 .png，从而判断文件类型。若不是这两种类型，则输出 Unknown file type。

除了上述示例代码之外，我们还可以通过 path/filepath 包提供的函数来获取和操作文件路径及其组成部分的信息。

package main

import (
    "fmt"
    "path/filepath"
)

func getFileExtension(fileName string) string {
    extension := filepath.Ext(fileName)
    return extension
}

func main() {
    fileName := "example.jpg"
    extension := getFileExtension(fileName)
    fmt.Println("File extension:", extension)
}

上面的代码通过 filepath.Ext 函数来获取文件名的后缀，从而实现了获取文件扩展名的功能。

总的来说，Golang 提供了多种方式来实现文件扩展名的判断和获取，开发者可以根据实际需求选择适合自己的方法。