Golang：如何有效地确定文件中的行数？

本篇文章主要是结合我之前面试的各种经历和实战开发中遇到的问题解决经验整理的，希望这篇《Golang：如何有效地确定文件中的行数？》对你有很大帮助！欢迎收藏，分享给更多的需要的朋友学习~

问题内容

在 Golang 中，我正在寻找一种有效的方法来确定文件的行数。

当然，我总是可以遍历整个文件，但似乎效率不高。

file, _ := os.Open("/path/to/filename")
fileScanner := bufio.NewScanner(file)
lineCount := 0
for fileScanner.Scan() {
    lineCount++
}
fmt.Println("number of lines:", lineCount)

有没有更好（更快、更便宜）的方法来找出一个文件有多少行？

正确答案

这是一个更快的行计数器，bytes.Count用于查找换行符。

它更快，因为它消除了返回整行所需的所有额外逻辑和缓冲，并利用 bytes 包提供的一些汇编优化函数来搜索字节片中的字符。

更大的缓冲区在这里也有帮助，尤其是对于更大的文件。在我的系统上，使用我用于测试的文件，32k 缓冲区是最快的。

func lineCounter(r io.Reader) (int, error) {
    buf := make([]byte, 32*1024)
    count := 0
    lineSep := []byte{'\n'}

    for {
        c, err := r.Read(buf)
        count += bytes.Count(buf[:c], lineSep)

        switch {
        case err == io.EOF:
            return count, nil

        case err != nil:
            return count, err
        }
    }
}

和基准输出：

BenchmarkBuffioScan   500      6408963 ns/op     4208 B/op    2 allocs/op
BenchmarkBytesCount   500      4323397 ns/op     8200 B/op    1 allocs/op
BenchmarkBytes32k     500      3650818 ns/op     65545 B/op   1 allocs/op

今天带大家了解了golang的相关知识，希望对你有所帮助；关于Golang的技术知识我们会一点点深入介绍，欢迎大家关注golang学习网公众号，一起学习编程~