Go语言获取指定目录下的所有文件的大小|极客笔记

Go语言获取指定目录下的所有文件的大小

在Go语言中，我们经常需要获取指定目录下所有文件的大小，这对于一些需要统计文件大小或者进行文件操作的程序来说是非常有用的。本文将详细介绍如何使用Go语言获取指定目录下所有文件的大小。

使用os包获取指定目录下的所有文件

在Go语言中，可以使用os包提供的函数来获取指定目录下的所有文件。首先，我们需要导入os包：

import (
    "fmt"
    "os"
)

接下来，我们定义一个函数getFileSize，该函数用于获取指定文件的大小：

func getFileSize(path string) (int64, error) {
    file, err := os.Open(path)
    if err != nil {
        return 0, err
    }
    defer file.Close()

    fileInfo, err := file.Stat()
    if err != nil {
        return 0, err
    }

    return fileInfo.Size(), nil
}

然后，我们定义一个函数getAllFilesSize，该函数用于获取指定目录下所有文件的大小：

func getAllFilesSize(dirPath string) (int64, error) {
    var totalSize int64

    dir, err := os.Open(dirPath)
    if err != nil {
        return 0, err
    }
    defer dir.Close()

    fileInfos, err := dir.Readdir(0)
    if err != nil {
        return 0, err
    }

    for _, fileInfo := range fileInfos {
        if !fileInfo.IsDir() {
            fileSize, err := getFileSize(dirPath + "/" + fileInfo.Name())
            if err != nil {
                return 0, err
            }
            totalSize += fileSize
        }
    }

    return totalSize, nil
}

最后，我们调用getAllFilesSize函数获取指定目录下所有文件的大小：

func main() {
    dirPath := "/path/to/directory"
    totalSize, err := getAllFilesSize(dirPath)
    if err != nil {
        fmt.Println("Error:", err)
    } else {
        fmt.Printf("Total size of files in %v: %v bytes\n", dirPath, totalSize)
    }
}

在上述代码中，我们首先定义了getFileSize函数用于获取单个文件的大小，然后定义了getAllFilesSize函数用于获取指定目录下所有文件的大小。最后，在main函数中调用getAllFilesSize函数并打印结果。请注意，需要将dirPath替换为实际的目录路径。

运行结果示例

假设我们有一个目录/path/to/directory，其中包含文件file1.txt和file2.txt，这两个文件的大小分别为100字节和200字节。我们可以使用上述代码来获取这两个文件的总大小：

$ go run main.go
Total size of files in /path/to/directory: 300 bytes

上述运行结果显示，file1.txt和file2.txt的总大小为300字节。

通过上述代码示例，我们可以看到如何使用Go语言获取指定目录下所有文件的大小。这对于一些需要统计文件大小或进行文件操作的应用程序来说是非常有用的。

Go语言获取指定目录下的所有文件的大小