卓越飞翔博客卓越飞翔博客

卓越飞翔 - 您值得收藏的技术分享站
技术文章34598本站已运行393

将变量引用传递给 WalkDirFunc

将变量引用传递给 walkdirfunc

php小编柚子为您介绍一个有关变量引用传递的重要概念——将变量引用传递给WalkDirFunc。在PHP中,变量引用传递是一种传递方式,它允许我们将一个变量的引用传递给函数或方法,从而在函数内部对该变量进行修改。通过这种方式,我们可以避免在函数内部创建新的变量副本,提高代码的执行效率。WalkDirFunc是一个示例函数,它可以遍历指定目录下的所有文件并对其进行操作。将变量引用传递给WalkDirFunc可以使我们在函数内部直接对传入的变量进行操作,而无需返回修改后的结果。这种方式在处理大量数据时尤为实用,可以减少内存开销并提高代码的可读性和可维护性。

问题内容

我正在努力递归地迭代给定目录并将找到的文件的名称及其大小存储在 map[string]uint32 中。我发现一篇文章描述了如何使用 path/filepath/walkdir 函数和 walkdirfunc 递归地遍历指定目录。

我对 go 比较陌生,并且无法找到一种方法将对我的地图的引用传递给 walkdirfunc,因为它的签名仅指定三个参数:

func main() {
    args := os.args
    if len(args) != 2 {
        msg := fmt.sprintf("usage: %v <path_to_directory>", args[0])
        log.fatal(msg)
    }

    fullpath := args[1]
    dirmap := make(map[string]uint32)

    filepath.walkdir(fullpath, walk)

}

// how to create parameter accepting &map[string]uint32
func walk(path string, entry fs.direntry, err error) error {
    if err != nil {
        return err
    }

    if !entry.isdir() {
        // add entry and it's size to dirmap

        file, err := os.stat(path)
        if err != nil {
            log.println("error determining file size:", err)
        }

        size := file.size()

        // add path and size to dirmap

    } else if entry.isdir() {
        // recurse here

    } else {
        log.println("unhandled branch:")
        log.println(entry.info())
    }

    return nil
}

编辑:由于walk是传递给filepath.walkdirwalkdirfunc,因此有必要传递从这里开始的地图。在 python 中,我会使用命名参数来完成此任务,但 go 中最好的方法是什么?

func main() {
    args := os.Args
    ...

    fullPath := args[1]
    dirMap := make(map[string]uint32)


    // dirMap should be passed here
    filepath.WalkDir(fullPath, walk)

}

解决方法

使用闭包。更改步行至:

func walk(dirmap map[string]uint32, path string, entry fs.direntry, err error) error {
    ...
}

像这样调用 walkdir:

dirMap := make(map[string]uint32)
filepath.WalkDir(fullPath, func(path string, entry fs.DirEntry, err error) error {
    return walk(dirMap, path, entry, err)
})

在此处运行示例:https://www.php.cn/link/178fde6398864e33aaefae5d2d855f20

卓越飞翔博客
上一篇: 惯用语 Go 精确读取 N 行(而不是一个字节)的文本?
下一篇: 返回列表
留言与评论(共有 0 条评论)
   
验证码:
隐藏边栏