Golang 中的 bufio 包是帶緩沖 I/O 操作的標(biāo)準(zhǔn)庫(kù)，之前的幾篇文章詳細(xì)講解了 bufio.Reader、bufio.Writer 和 bufio.Scanner 這個(gè)幾個(gè)結(jié)構(gòu)體的使用方法、特性和使用場(chǎng)景，本文介紹一下 bufio 包中的函數(shù)。

介紹常用函數(shù)之前，先簡(jiǎn)單介紹下另一個(gè)結(jié)構(gòu)體類型和一個(gè)函數(shù)類型：bufio.ReadWriter 和 bufio.SplitFunc，對(duì)應(yīng)的定義如下：

type ReadWriter struct {
	*Reader
	*Writer
}

type SplitFunc func(data []byte, atEOF bool) (advance int, token []byte, err error)

ReadWriter 結(jié)構(gòu)體實(shí)現(xiàn)了 io.ReadWriter 接口，SplitFunc 是用來(lái)指定自定義分割規(guī)則的函數(shù)類型，通常結(jié)合 bufio.Scanner 使用。

創(chuàng)建對(duì)象函數(shù)

• func NewReader(rd io.Reader) *Reader，創(chuàng)建一個(gè)帶緩沖的 Reader 對(duì)象。
• func NewWriter(w io.Writer) *Writer，創(chuàng)建一個(gè)帶緩沖的 Writer 對(duì)象。
• func NewReadWriter(r *Reader, w *Writer) *ReadWriter，創(chuàng)建一個(gè)帶緩沖的 ReadWriter 對(duì)象。
• func NewReaderSize(rd io.Reader, size int) *Reader，創(chuàng)建一個(gè)帶緩沖的 Reader 對(duì)象，使用 size 指定緩沖區(qū)的最小值。
• func NewWriterSize(w io.Writer, size int) *Writer，創(chuàng)建一個(gè)帶緩沖的 Writer 對(duì)象，使用 size 指定緩沖區(qū)的最小值。
• func NewScanner(r io.Reader) *Scanner，創(chuàng)建一個(gè)帶緩沖的 Scanner 對(duì)象。

SplitFunc 類型函數(shù)

以下幾個(gè)函數(shù)都是 SplitFunc 類型，通常與 bufio.NewScanner 一起使用。

• func ScanLines(data []byte, atEOF bool) (advance int, token []byte, err error)，用于逐行讀取輸入流中的數(shù)據(jù)。
• func ScanBytes(data []byte, atEOF bool) (advance int, token []byte, err error)，用于逐個(gè)字節(jié)讀取輸入流中的數(shù)據(jù)。
• func ScanRunes(data []byte, atEOF bool) (advance int, token []byte, err error)，用于逐個(gè) unicode 編碼讀取輸入流中的數(shù)據(jù)。
• func ScanWords(data []byte, atEOF bool) (advance int, token []byte, err error)，用于逐個(gè)單詞（空格分割）讀取輸入流中的數(shù)據(jù)。

使用示例

package main

import (
	"bufio"
	"fmt"
	"os"
	"strings"
)

func main() {
	// 確定讀取文件的路徑
	path := "file.txt"

	// 打開(kāi)文件并處理錯(cuò)誤
	file, err := os.Open(path)
	if err != nil {
		panic(err)
	}
	defer file.Close()

	// 創(chuàng)建Scanner類型對(duì)象
	scanner := bufio.NewScanner(file)

	// 設(shè)置分割函數(shù)
	scanner.Split(bufio.ScanWords)

	// 創(chuàng)建一個(gè)map對(duì)象，用于存儲(chǔ)單詞出現(xiàn)的次數(shù)
	counts := make(map[string]int)

	// 讀取文件內(nèi)容并統(tǒng)計(jì)單詞出現(xiàn)次數(shù)
	for scanner.Scan() {
		word := strings.ToLower(scanner.Text())
		counts[word]++
	}

	// 檢查Scanner是否出錯(cuò)
	if err := scanner.Err(); err != nil {
		panic(err)
	}

	// 輸出結(jié)果到控制臺(tái)
	for word, count := range counts {
		fmt.Printf("%s: %d\n", word, count)
	}

	// 讓用戶根據(jù)輸入的關(guān)鍵字進(jìn)行篩選
	fmt.Print("Enter keyword to filter results: ")
	reader := bufio.NewReader(os.Stdin)
	input, _ := reader.ReadString('\n')
	keyword := strings.TrimSpace(input)

	if keyword != "" {
		for word, count := range counts {
			if strings.Contains(word, keyword) {
				fmt.Printf("%s: %d\n", word, count)
			}
		}
	}
}

首先創(chuàng)建了一個(gè) Scanner 類型的對(duì)象，指定了 ScanWords 作為分隔函數(shù)，然后遍歷文件內(nèi)容，統(tǒng)計(jì)單詞出現(xiàn)的次數(shù)并將結(jié)果存儲(chǔ)到一個(gè)map中，最后讓用戶輸入關(guān)鍵字進(jìn)行篩選，根據(jù)關(guān)鍵字過(guò)濾并輸出結(jié)果。

小結(jié)

bufio 是一個(gè)提供了帶緩沖讀寫(xiě)操作的包，通過(guò)使用 bufio 包提供的類型和方法，可以高效地讀寫(xiě)數(shù)據(jù)，特別是當(dāng)涉及到大量數(shù)據(jù)的讀寫(xiě)時(shí)，可以大大提高程序的性能。