golang 一行一行读文件

最新推荐文章于 2025-12-30 16:44:16 发布

转载最新推荐文章于 2025-12-30 16:44:16 发布 · 6.3k 阅读

文章标签：

#golang

总结专栏收录该内容

47 篇文章

订阅专栏

本文介绍了使用Go语言实现的两种不同方法来读取文件并统计每一行出现的次数。第一种方法使用bufio包逐行扫描文件，第二种方法通过ioutil将整个文件读入内存再进行分割统计。适用于需要处理文本文件行重复情况的场景。

package main

import (
        "bufio"
        "fmt"
        "os"
)
func main() {
        counts := make(map[string]int)
        files := os.Args[1:]
        if len(files) == 0 {
                countLines(os.Stdin, counts)
        } else {
                for _, arg := range files {
                        f, err := os.Open(arg)
                        if err != nil {
                                fmt.Fprintf(os.Stderr, "dup2: %v\n", err)
                                continue
                        }
                        countLines(f, counts)
                        f.Close()
                }
        }
        for line, n := range counts {
                fmt.Printf("%d\t%s\n", n, line)
        }
}

func countLines(f *os.File, counts map[string]int) {
        input := bufio.NewScanner(f)
        for input.Scan() {
                counts[input.Text()]++
        }
        // NOTE: ignoring potential errors from input.Err()
}

go run main.go a.txt

另一种方法：

package main

import (
	"fmt"
	"io/ioutil"
	"os"
	"strings"
)

func main() {
	counts := make(map[string]int)
	for _, filename := range os.Args[1:] {
		data, err := ioutil.ReadFile(filename)
		if err != nil {
			fmt.Fprintf(os.Stderr, "dup3: %v\n", err)
			continue
		}
		for _, line := range strings.Split(string(data), "\n") {
			counts[line]++
		}
	}
	for line, n := range counts {
		if n > 1 {
			fmt.Printf("%d\t%s\n", n, line)
		}
	}
}