首页  >  文章  >  后端开发  >  确定 Golang 中 CSV 行有多少字节

确定 Golang 中 CSV 行有多少字节

王林
王林转载
2024-02-05 23:33:081016浏览

确定 Golang 中 CSV 行有多少字节

问题内容

我有一个 test.csv ,我将逐行读取它并确定每行中有多少字节。

这应该低于 37 个字节,因为我在 Windows 上,前两行有 rn 总共添加了 4 个字节。

foo,bar,baz
100,200,300
400,500,600

我想使用 csv.NewReader() 简单地确定每行中有多少字节。但是,我知道在下面的代码中进行字节计数时,csv.Reader 不会计算每行中的逗号和 n

我是否应该为每行中的逗号数+行数添加一些数学运算,为 rn -1 为最后一行添加 2 个字节,因为它没有 rn?这感觉有点老套,所以我宁愿看看是否有更好的解决方案来解决我的字节计数问题。

我的代码:

package main

import (
    "encoding/csv"
    "fmt"
    "io"
    "log"
    "os"
)

func main() {
    file, err := os.Open("test.csv")
    if err != nil {
        log.Fatal(err)
    }
    defer file.Close()

    fileInfo, err := file.Stat()
    if err != nil {
        log.Fatal(err)
    }
    fmt.Printf("file total bytes is %d\n", fileInfo.Size())

    // init reader
    reader := csv.NewReader(file)

    // extract the header
    headers, err := reader.Read()
    if err != nil {
        log.Fatal(err)
    }
    fmt.Printf("headers are: %+v\n", headers)

    byteCounter := 0
    for {
        // if we reached end of file, stop
        if err == io.EOF {
            break
        }

        // read a record
        record, err := reader.Read()
        if err != nil {
            log.Fatal(err)
        }
        // loop through each record and count how many bytes
        for _, item := range record {
            byteCounter += len(item)
            fmt.Printf("record is %d bytes\n", len(item))
        }
        fmt.Println("total bytes so far is: ", byteCounter)

    }
}

正确答案


使用 Reader.InputOffset 获取文件中的当前位置: BD44F3686574B2610B9F6F11A1F9CFEF

https://www.php.cn/link/e2c2ff9c7fb57db574e461fbe467d84e

要获取每行的字节数,请减去之前的偏移量。

prevOffset := reader.InputOffset()

for {
    ...
    // read a record
    _, err := reader.Read()
    ...
    fmt.Println("line length is: ", reader.InputOffset()-prevOffset)
    prevOffset = reader.InputOffset()

}

https://www.php.cn/link/501e3f8a108d7ab9335ceecd363d113d

以上是确定 Golang 中 CSV 行有多少字节的详细内容。更多信息请关注PHP中文网其他相关文章!

声明:
本文转载于:stackoverflow.com。如有侵权,请联系admin@php.cn删除