Golang对相同扩展名的文件进行分组

积累知识，胜过积蓄金银！毕竟在Golang开发的过程中，会遇到各种各样的问题，往往都是一些细节知识点还没有掌握好而导致的，因此基础知识点的积累是很重要的。下面本文《Golang对相同扩展名的文件进行分组》，就带大家讲解一下知识点，若是你对本文感兴趣，或者是想搞懂其中某个知识点，就请你继续往下看吧~

我正在用 golang 编写一个基本程序，它输出当前目录中的文件数量以及每个文件包含多少行代码：

我现在尝试将具有相同扩展名的文件分组，但遇到了问题。 所以，我所做的是将字符串映射声明为键，将字符串切片声明为值：

将扩展名作为键及其文件存储在该切片中

var m = make(map[string][]string)

我首先在“group”函数中接收程序读取的文件片段（q 参数），如下所示：

[.gitignore readme.md cmd\root.go example.go eyeball.exe go.mod go.sum main.go utils\utils.go]

然后我对该文件片段进行操作：

func group(q []string) {

    var ext string
    var checkedext []string

    for _, val := range q {
        ext = strings.split(val, ".")[1]
        for _, foo := range q {
            secext := strings.split(foo, ".")[1]
            if ext == secext {
                if !contains(checkedext, ext) {
                    addtogroup(ext, foo)
                    checkedext = append(checkedext, ext)
                }
            }
        }
    }
    fmt.println(m)

}

func contains(s []string, item string) bool {
    for _, el := range s {
        if el == item {
            return true
        }
    }
    return false
}

var m = make(map[string][]string)


func addtogroup(ext, file string) {

    var subgroup []string
    if val, ok := m[ext]; ok {
       val = append(val, file)
    } else {
        subgroup = append(subgroup, file)
        m[ext] = subgroup
    }
}

当我打印地图时它会输出：

map[exe:[eyeball.exe] gitignore:[.gitignore] go:[cmd\root.go] md:[readme.md] mod:[go.mod] sum:[go.sum]]

但是，上面的输出是错误的，因为我有 3 个 .go 文件。

我知道 range 会遍历地图的副本，但我无法通过指针引用它，因为地图本身就是一个引用，我找不到解决方法， 我想到了这样做：

var m = make(map[string]*Data)

但我不知道这是否是最好的事情

我努力解释我希望我的程序做什么以及它做什么，如果我遗漏了任何细节，请指出我。

解决方案

问题是，当您将元素添加到切片（位于地图内部）时，您必须重新分配结果切片，因为 append() 返回一个包含附加项目的新切片标头：

if val, ok := m[ext]; ok {
   val = append(val, file)
    m[ext] = val
} else {
    // ...
}

或者简单地说：

m[ext] = append(m[ext], file)

这也可以处理 ext 尚未在映射中的情况，因为 m[ext] 将是值类型的零值（即 []string），而零值是 nil 切片。您可以附加到 nil 切片。

要获取扩展名，请使用 filepath.Ext()。

您的 group() 函数可以像这样简单：

func group(names []string) map[string][]string {
    m := map[string][]string{}
    for _, name := range names {
        ext := filepath.ext(name)
        ext = strings.trimprefix(ext, ".")
        m[ext] = append(m[ext], name)
    }
    return m
}

测试它：

fmt.println(group([]string{"x\\a.txt", "b.txt", "c.go"}))

fmt.println(group([]string{"a.go", "xy\\b.go", "c.txt"}))

哪个输出（在 Go Playground 上尝试一下）：

map[go:[c.go] txt:[x\a.txt b.txt]]
map[go:[a.go xy\b.go] txt:[c.txt]]

查看相关问题：

How to update map values in Go

How to categorize list of objects based on a parameter

终于介绍完啦！小伙伴们，这篇关于《Golang对相同扩展名的文件进行分组》的介绍应该让你收获多多了吧！欢迎大家收藏或分享给更多需要学习的朋友吧~golang学习网公众号也会发布Golang相关知识，快来关注吧！