Danh sách thư mục trong Go

Tôi đã cố gắng tìm ra cách đơn giản là liệt kê các tệp và thư mục trong một thư mục trong Go.

Tôi đã tìm thấy filepath.Walk, nhưng nó tự động đi vào các thư mục con mà tôi không muốn. Tất cả các tìm kiếm khác của tôi đã không làm mọi thứ tốt hơn.

Tôi chắc chắn rằng chức năng này tồn tại, nhưng nó thực sự rất khó tìm. Hãy cho tôi biết nếu có ai biết tôi nên tìm ở đâu. Cảm ơn.

Bạn có thể thử sử dụng chức năng ReadDir trong io/ioutilgói. Theo các tài liệu:

ReadDir đọc thư mục được đặt tên bởi dirname và trả về một danh sách các mục nhập thư mục được sắp xếp.

Các lát kết quả chứa os.FileInfocác loại, cung cấp các phương thức được liệt kê ở đây . Dưới đây là một ví dụ cơ bản liệt kê tên của mọi thứ trong thư mục hiện tại (các thư mục được bao gồm nhưng không được đánh dấu đặc biệt - bạn có thể kiểm tra xem một mục có phải là thư mục hay không bằng cách sử dụng IsDir()phương thức):

package main

import (
    "fmt"
    "io/ioutil"
     "log"
)

func main() {
    files, err := ioutil.ReadDir("./")
    if err != nil {
        log.Fatal(err)
    }

    for _, f := range files {
            fmt.Println(f.Name())
    }
}

Thậm chí đơn giản hơn, sử dụng path/filepath:

package main    

import (
    "fmt"
    "log"
    "path/filepath"
)

func main() {
    files, err := filepath.Glob("*")
    if err != nil {
        log.Fatal(err)
    }
    fmt.Println(files) // contains a list of all files in the current directory
}

Chúng ta có thể lấy danh sách các tệp trong một thư mục trên hệ thống tệp bằng các hàm thư viện chuẩn golang khác nhau.

1. filepath.Walk
2. ioutil.ReadDir
3. os.File.Readdir

package main

import (
    "fmt"
    "io/ioutil"
    "log"
    "os"
    "path/filepath"
)

func main() {
    var (
        root  string
        files []string
        err   error
    )

    root := "/home/manigandan/golang/samples"
    // filepath.Walk
    files, err = FilePathWalkDir(root)
    if err != nil {
        panic(err)
    }
    // ioutil.ReadDir
    files, err = IOReadDir(root)
    if err != nil {
        panic(err)
    }
    //os.File.Readdir
    files, err = OSReadDir(root)
    if err != nil {
        panic(err)
    }

    for _, file := range files {
        fmt.Println(file)
    }
}

1. Sử dụng filepath.Walk

Các path/filepathgói cung cấp một cách tiện dụng để quét tất cả các tập tin trong một thư mục, nó sẽ tự động quét từng thư mục con trong thư mục.

func FilePathWalkDir(root string) ([]string, error) {
    var files []string
    err := filepath.Walk(root, func(path string, info os.FileInfo, err error) error {
        if !info.IsDir() {
            files = append(files, path)
        }
        return nil
    })
    return files, err
}

2. Sử dụng ioutil.ReadDir

ioutil.ReadDir đọc thư mục được đặt tên bởi dirname và trả về một danh sách các mục nhập thư mục được sắp xếp theo tên tệp.

func IOReadDir(root string) ([]string, error) {
    var files []string
    fileInfo, err := ioutil.ReadDir(root)
    if err != nil {
        return files, err
    }

    for _, file := range fileInfo {
        files = append(files, file.Name())
    }
    return files, nil
}

3. Sử dụng os.File.Readdir

Readdir đọc nội dung của thư mục được liên kết với tệp và trả về một lát lên đến n giá trị FileInfo, như được trả về bởi Lstat, theo thứ tự thư mục. Các cuộc gọi tiếp theo trên cùng một tệp sẽ mang lại FileInfos hơn nữa.

func OSReadDir(root string) ([]string, error) {
    var files []string
    f, err := os.Open(root)
    if err != nil {
        return files, err
    }
    fileInfo, err := f.Readdir(-1)
    f.Close()
    if err != nil {
        return files, err
    }

    for _, file := range fileInfo {
        files = append(files, file.Name())
    }
    return files, nil
}

Kết quả điểm chuẩn.

Nhận thêm thông tin chi tiết về bài đăng trên blog này

ioutil.ReadDirlà một tìm kiếm tốt, nhưng nếu bạn nhấp và nhìn vào nguồn bạn thấy rằng nó gọi phương thức Readdir của os.File . Nếu bạn ổn với thứ tự thư mục và không cần danh sách được sắp xếp, thì phương thức Readdir này là tất cả những gì bạn cần.

Từ mô tả của bạn, những gì bạn có thể muốn là os.Readdirnames .

func (f *File) Readdirnames(n int) (names []string, err error)

Readdirnames đọc nội dung của thư mục được liên kết với tệp và trả về một lát lên đến n tên của các tệp trong thư mục, theo thứ tự thư mục. Các cuộc gọi tiếp theo trên cùng một tệp sẽ mang lại thêm tên.

...

Nếu n <= 0, Readdirnames trả về tất cả các tên từ thư mục trong một lát.

Đoạn trích:

file, err := os.Open(path)
if err != nil {
    return err
}
names, err := file.Readdirnames(0)
if err != nil {
    return err
}
fmt.Println(names)

Tín dụng cho bình luận của SquattedSlavInTracksuit ; Tôi đã đề nghị thúc đẩy bình luận của họ để trả lời nếu tôi có thể.