Golang 获取文件名称与后缀

package main

import (
    "fmt"
    "path/filepath"
)

func main() {
    // 测试路径1：普通绝对路径（单后缀）
    path1 := "/data/images/flower.jpg"
    // 测试路径2：相对路径+多后缀（如压缩包）
    path2 := "./download/package.tar.gz"
    // 测试路径3：隐藏文件（无后缀）
    path3 := "/home/user/.bashrc"
    // 测试路径4：Windows 路径（含空格）
    path4 := "C:\\Users\\Admin\\Documents\\简历.pdf"

    fmt.Println("测试路径1：", path1)
    fmt.Println("测试路径2：", path2)
    fmt.Println("测试路径3：", path3)
    fmt.Println("测试路径4：", path4)
}

// 方法一：Base() + Ext() 基础组合
func getFilenameAndExtBasic(path string) (filename string, ext string) {
    // 1. 提取完整文件名（含后缀）：如 "flower.jpg" "package.tar.gz" ".bashrc"
    fullName := filepath.Base(path)
    // 2. 提取后缀：如 ".jpg" ".gz" ""（隐藏文件无后缀时返回空）
    ext = filepath.Ext(fullName)
    // 3. 截取纯名称（去掉后缀）：用字符串长度差计算截取范围
    filename = fullName[:len(fullName)-len(ext)]
    return
}

// 在 main 函数中调用
func main() {
    // 省略测试路径定义...
    name1, ext1 := getFilenameAndExtBasic(path1)
    fmt.Printf("路径1结果：名称=%s, 后缀=%s\n", name1, ext1) // 名称=flower, 后缀=.jpg

    name2, ext2 := getFilenameAndExtBasic(path2)
    fmt.Printf("路径2结果：名称=%s, 后缀=%s\n", name2, ext2) // 名称=package.tar, 后缀=.gz

    name3, ext3 := getFilenameAndExtBasic(path3)
    fmt.Printf("路径3结果：名称=%s, 后缀=%s\n", name3, ext3) // 名称=.bashrc, 后缀=
}

// 方法二：自定义拆分（支持多后缀，可指定取前N个后缀）
// isFullExt: true=取完整多后缀（如 .tar.gz）, false=取第一个后缀（如 .tar）
func getFilenameAndExtMulti(path string, isFullExt bool) (filename string, ext string) {
    fullName := filepath.Base(path)
    // 找到第一个点的索引
    firstDotIdx := -1
    for i, c := range fullName {
        if c == '.' {
            firstDotIdx = i
            break
        }
    }
    // 无点时（无后缀）
    if firstDotIdx == -1 {
        filename = fullName
        ext = ""
        return
    }
    // 处理隐藏文件（以点开头，且无其他点）
    if firstDotIdx == 0 && strings.Count(fullName, ".") == 1 {
        filename = fullName
        ext = ""
        return
    }
    // 取完整多后缀（从第一个点到结尾）或第一个后缀（从第一个点到第二个点）
    if isFullExt {
        ext = fullName[firstDotIdx:]
        filename = fullName[:firstDotIdx]
    } else {
        secondDotIdx := strings.Index(fullName[firstDotIdx+1:], ".")
        if secondDotIdx == -1 {
            // 只有一个后缀，同基础方法
            ext = fullName[firstDotIdx:]
        } else {
            // 取第一个后缀（如 .tar）
            ext = fullName[firstDotIdx : firstDotIdx+1+secondDotIdx+1]
        }
        filename = fullName[:firstDotIdx]
    }
    return
}

// 需导入 strings 包
// import "strings"

// 在 main 函数中调用
func main() {
    // 省略测试路径定义...
    // 多后缀场景：取完整后缀 .tar.gz
    name2Full, ext2Full := getFilenameAndExtMulti(path2, true)
    fmt.Printf("路径2完整后缀结果：名称=%s, 后缀=%s\n", name2Full, ext2Full) // 名称=package, 后缀=.tar.gz

    // 多后缀场景：取第一个后缀 .tar
    name2First, ext2First := getFilenameAndExtMulti(path2, false)
    fmt.Printf("路径2第一个后缀结果：名称=%s, 后缀=%s\n", name2First, ext2First) // 名称=package, 后缀=.tar
}

// 方法三：strings.Split 拆分（Golang 1.18+）
func getFilenameAndExtSplit(path string) (filename string, ext string, hasExt bool) {
    fullName := filepath.Base(path)
    // SplitN 按 "." 分割，最多分2份（确保只拆一次，兼容多后缀）
    parts := strings.SplitN(fullName, ".", 2)
    // 只有一份时（无后缀，如 "readme" 或隐藏文件 ".bashrc"）
    if len(parts) == 1 {
        filename = fullName
        ext = ""
        hasExt = false
        return
    }
    // 有后缀时（parts[0]是名称，parts[1]是后缀）
    // 处理隐藏文件（如 ".bashrc" 拆分后 parts[0]是空字符串）
    if parts[0] == "" {
        filename = fullName
        ext = ""
        hasExt = false
    } else {
        filename = parts[0]
        ext = "." + parts[1]
        hasExt = true
    }
    return
}

// 在 main 函数中调用
func main() {
    // 省略测试路径定义...
    name1, ext1, hasExt1 := getFilenameAndExtSplit(path1)
    fmt.Printf("路径1结果：名称=%s, 后缀=%s, 有无后缀=%t\n", name1, ext1, hasExt1) // 名称=flower, 后缀=.jpg, 有无后缀=true

    name3, ext3, hasExt3 := getFilenameAndExtSplit(path3)
    fmt.Printf("路径3结果：名称=%s, 后缀=%s, 有无后缀=%t\n", name3, ext3, hasExt3) // 名称=.bashrc, 后缀=, 有无后缀=false
}

// 方法四：结合 os.Stat() 获取（需文件真实存在）
func getFilenameAndExtWithStat(path string) (filename string, ext string, err error) {
    // 先判断文件是否存在
    _, err = os.Stat(path)
    if err != nil {
        return "", "", fmt.Errorf("文件不存在或无法访问：%v", err)
    }
    // 再提取名称和后缀（同基础方法）
    fullName := filepath.Base(path)
    ext = filepath.Ext(fullName)
    filename = fullName[:len(fullName)-len(ext)]
    return
}

// 需导入 os 包
// import "os"

// 在 main 函数中调用
func main() {
    // 假设 path1 对应的文件真实存在
    name1, ext1, err1 := getFilenameAndExtWithStat(path1)
    if err1 != nil {
        fmt.Printf("路径1处理失败：%v\n", err1)
    } else {
        fmt.Printf("路径1结果：名称=%s, 后缀=%s\n", name1, ext1)
    }

    // 不存在的路径测试
    invalidPath := "/data/nonexistent.txt"
    _, _, errInvalid := getFilenameAndExtWithStat(invalidPath)
    fmt.Printf("无效路径处理结果：%v\n", errInvalid) // 输出文件不存在错误
}

// 针对固定多后缀的补充判断
func getFullExt(ext string, fullName string) string {
    // 已知多后缀列表
    multiExts := []string{".tar.gz", ".tar.bz2", ".zip.bak"}
    for _, me := range multiExts {
        if strings.HasSuffix(fullName, me) {
            return me
        }
    }
    return ext
}