我有一个test.csv
,我会逐行读取,并确定每行有多少字节。
这应该是37字节以下,因为我在Windows上,前两行有\r\n
增加4个总字节。
foo,bar,baz
100,200,300
400,500,600
我想使用csv.NewReader()
简单地确定每行中有多少字节。但是,我知道csv.Reader在我下面的代码中计算字节数时并没有计算每行中的逗号和\n
。
我应该为每行中的逗号数+行数添加一些数学运算吗?为\r\n
添加2个字节,为最后一行添加一个字节,因为它没有\r\n?这感觉有点古怪,所以我宁愿看看是否有更好的解决方案来解决我的字节计数问题。
我的代码:
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)
}
}
1条答案
按热度按时间wfypjpf41#
使用Reader.InputOffset获取文件中的当前位置:
https://go.dev/play/p/cgqq9woK02w
要获得每行的字节数,请从上一个偏移量中减去。
https://go.dev/play/p/cXCfogUBcdy