如何解决 Golang filepath.walkfunc 问题？

Question

我正在尝试解决一项任务，我必须在具有相似名称和相同大小的其他文件中找到一个数据格式为 CSV 的文件，并在第 5 行第 3 列（索引 4 和 2）上打印一个数字所以我写了这段代码

package main

import (
    "encoding/csv"
    "fmt"
    "os"
    "path/filepath"
)

var s [][]string

func walkfunc(path string, info os.FileInfo, err error) error {
    if err != nil {
        return err
    }

    buf, err1 := os.Open(path)
    if err1 == nil {
        var err2 error
        r := csv.NewReader(buf)
        s, err2 = r.ReadAll()
        if err2 == nil {
            fmt.Printf("found: %v", s[4][2])
        }
    }
    defer buf.Close()
    return nil
}
func main() {
    const root = "./task/"
    if err := filepath.Walk(root, walkfunc); err != nil {
        fmt.Printf("error: %v", err)
    }
}

我在 output 得到了这个

GOROOT=/usr/local/go #gosetup
GOPATH=/usr/local/go/bin #gosetup
/usr/local/go/bin/go build -o /private/var/folders/j2/ybr0drz13yq31dc67zmvkb1w0000gn/T/GoLand/___go_build_qwasd3_go /Users/user/Downloads/zadacha/qwasd3.go #gosetup
/private/var/folders/j2/ybr0drz13yq31dc67zmvkb1w0000gn/T/GoLand/___go_build_qwasd3_go
panic: runtime error: index out of range [4] with length 3

goroutine 1 [running]:
main.walkfunc({0x14000018120?, 0x0?}, {0x14000098d88?, 0x10247fe40?}, {0x0?, 0x0?})
        /Users/user/Downloads/zadacha/qwasd3.go:23 +0x28c
path/filepath.walk({0x14000018120, 0xe}, {0x1024c9cf8, 0x140000685b0}, 0x1024c9338)
        /usr/local/go/src/path/filepath/path.go:433 +0xd0
path/filepath.walk({0x10248d4a8, 0x7}, {0x1024c9cf8, 0x140000684e0}, 0x1024c9338)
        /usr/local/go/src/path/filepath/path.go:457 +0x1fc
path/filepath.Walk({0x10248d4a8, 0x7}, 0x1024c9338)
        /usr/local/go/src/path/filepath/path.go:520 +0x6c
main.main()
        /Users/user/Downloads/zadacha/qwasd3.go:37 +0x30

Process finished with the exit code 2

我究竟做错了什么？

我试图在 MacBook 上运行这段代码。 所需文件包含带数字的表格，我需要在第 5 行和第 3 列打印一个数字。

Answer 1

正如其他评论所指出的那样，您需要检查每个 CSV 以确保它实际上与您预期的一样大。 您还可以通过查找“.csv”扩展名添加一个简单的检查来尝试并确保它是一个 CSV 文件，然后再打开它。

虽然，要直接解决您的错误...... CSV 读者可能能够将纯 txt 文件解释为 CSV 而不会返回错误，例如：

buf := strings.NewReader(`A regular text file with 3 lines.
Line2
Line3
`)

r := csv.NewReader(buf)
records, err := r.ReadAll()

if err != nil {
    fmt.Println("could not read all of CSV file!")
    return err
}

fmt.Println(records)

印刷：

[[A regular text file with 3 lines.] [Line2] [Line3]]

只是假设它是一个 CSV 具有正确的行数和列数：

fmt.Println("found", records[4][2])

给出您分享的恐慌信息：

panic: runtime error: index out of range [4] with length 3

您至少需要检查您的 CSV 是否有 5 行，如果有，则在尝试读取该字段之前检查第 5 行是否有 3 列：

if len(records) < 5 {
    fmt.Println(path, "does not have 5 rows")
    return nil
}

if len(records[4]) < 3 {
    fmt.Println(path, "5th row does not have 3 columns")
    return nil
}

fmt.Println("found", records[4][2])

您还可以在 walkfunc 中对文件路径本身进行基本检查，看看它是否看起来像 CSV：

if strings.ToLower(path[len(path)-4:]) != ".csv" {
    fmt.Println(path, "is not a CSV")
    return nil
}

我在这个 Playground 中展示了所有这些代码，以及一个完整工作/集成的示例。