Golang 中的 Bufio 包詳解之 Bufio.Scanner
作者：路多辛
Bufio.Scanner 在讀取緩沖區(qū)時，會將讀取的數(shù)據(jù)保存在內(nèi)部緩沖區(qū)中。因此，在每次調(diào)用 Scanner.Scan() 方法時，都會從輸入中讀取一個新的數(shù)據(jù)塊，并保存在內(nèi)部緩沖區(qū)中。如果需要讀取輸入緩沖區(qū)中的所有數(shù)據(jù)，需要不斷調(diào)用 Scanner.Scan() 方法，直到返回 False 為止。
bufio.Scanner
bufio.Scanner 是 Golang 中一個用于逐個讀取輸入緩沖區(qū)的掃描器，通常與 bufio.Reader 一起使用，bufio.Reader 用于從輸入中讀取數(shù)據(jù)，而 bufio.Scanner 則用于逐個讀取輸入緩沖區(qū)的內(nèi)容。
bufio.Scanner 可以將輸入數(shù)據(jù)分解為邏輯上的行并返回。Scanner 通過定義一個 Split 函數(shù)來將輸入分解為行。結(jié)構(gòu)體定義和對應(yīng)的方法如下：
type Scanner struct {
r io.Reader // The reader provided by the client.
split SplitFunc // The function to split the tokens.
maxTokenSize int // Maximum size of a token; modified by tests.
token []byte // Last token returned by split.
buf []byte // Buffer used as argument to split.
start int // First non-processed byte in buf.
end int // End of data in buf.
err error // Sticky error.
empties int // Count of successive empty tokens.
scanCalled bool // Scan has been called; buffer is in use.
done bool // Scan has finished.
}下面是 bufio.Writer 提供的一些主要方法:
- func (s *Scanner) Scan() bool，用于讀取輸入緩沖區(qū)中的下一個數(shù)據(jù)塊，并將其保存在內(nèi)部的緩沖區(qū)中。如果讀取成功，則返回 true；如果已經(jīng)讀取了所有數(shù)據(jù)或者發(fā)生了錯誤，則返回 false。
- func (s *Scanner) Text() string，用于獲取內(nèi)部緩沖區(qū)中的文本內(nèi)容，通常與 Scan() 方法一起使用，用于獲取讀取的數(shù)據(jù)。
- func (s *Scanner) Bytes() []byte，用于獲取內(nèi)部緩沖區(qū)中的字節(jié)內(nèi)容，通常與 Scan() 方法一起使用，用于獲取讀取的數(shù)據(jù)。
- func (s *Scanner) Err() error，用于獲取在讀取輸入時發(fā)生的錯誤信息，如果讀取過程中沒有發(fā)生錯誤，則返回 nil；否則，返回一個非 nil 的錯誤對象。
- func (s *Scanner) Buffer(buf []byte, max int)， 用于自定義輸入緩沖區(qū)大小，接受一個 []byte 類型的參數(shù)，用于指定緩沖區(qū)的大小。
- func (s *Scanner) Split(split SplitFunc)，用于指定一個分割函數(shù)，將輸入分割成多個數(shù)據(jù)塊，接受一個 func([]byte) bool 類型的參數(shù)，該函數(shù)在每次讀取輸入時被調(diào)用，用于判斷是否需要將當(dāng)前數(shù)據(jù)塊分割成多個小塊。通常用于處理非常大的數(shù)據(jù)塊，以避免內(nèi)存溢出等問題。
使用示例
簡單使用示例如下：
package main
import (
"bufio"
"fmt"
"strings"
)
func main() {
input := "路多辛的所思所想\n很值得一看哦！\n"
scanner := bufio.NewScanner(strings.NewReader(input))
// 逐行遍歷
for scanner.Scan() {
fmt.Println(scanner.Text())
}
// 錯誤處理
if err := scanner.Err(); err != nil {
fmt.Println("Error:", err)
}
// 自定義分隔符
scanner = bufio.NewScanner(strings.NewReader("路多辛,的,所思所想"))
scanner.Split(func(data []byte, atEOF bool) (advance int, token []byte, err error) {
// 分隔符為逗號
for i, d := range data {
if d == ',' {
return i + 1, data[:i], nil
}
}
if atEOF && len(data) > 0 {
return len(data), data, nil
}
return 0, nil, nil
})
for scanner.Scan() {
fmt.Println(scanner.Text())
}
}運(yùn)行看下效果：
$ go run main.go
路多辛的所思所想
很值得一看哦
路多辛
的
所思所想第一個示例中，使用了默認(rèn)的分割方式，即按行讀取輸入。第二個示例中，使用自定義分隔符，將輸入的字符串按照逗號進(jìn)行分隔。
小結(jié)
bufio.Scanner 在讀取緩沖區(qū)時，會將讀取的數(shù)據(jù)保存在內(nèi)部緩沖區(qū)中。因此，在每次調(diào)用 scanner.Scan() 方法時，都會從輸入中讀取一個新的數(shù)據(jù)塊，并保存在內(nèi)部緩沖區(qū)中。如果需要讀取輸入緩沖區(qū)中的所有數(shù)據(jù)，需要不斷調(diào)用 scanner.Scan() 方法，直到返回 false 為止。
責(zé)任編輯：姜華
來源：
今日頭條
