golang-101-hacks(25)——Buffered read

bufio包提供缓冲读取的函数。如下面例子所示

(1) 首先创建test.txt

# cat test.txt
abcd
efg
hijk
lmn

test.txt 文件包含4行内容

(2) 接着看下面程序:

package main

import (
        "bufio"
        "fmt"
        "io"
        "log"
        "os"
)

func main() {
        f, err := os.Open("test.txt")
        if err != nil {
                log.Fatal(err)
        }

        r := bufio.NewReader(f)
        for {
                if s, err := r.ReadSlice('\n'); err == nil || err == io.EOF {
                        fmt.Printf("%s", s)
                        if err == io.EOF {
                                break
                        }
                } else {
                        log.Fatal(err)
                }

        }
}

(a)打开test.txt文件

f, err := os.Open("test.txt")

(b)

r := bufio.NewReader(f)

bufio.NewReader(f)生成一个 bufio.Reader结构体,其实现了缓冲读取的方法 (c)

for {
    if s, err := r.ReadSlice('\n'); err == nil || err == io.EOF {
        fmt.Printf("%s", s)
        if err == io.EOF {
            break
        }
    } else {
        log.Fatal(err)
    }

}

读取并打印每一行 运行结果如下

abcd
efg
hijk
lmn

可以使用bufio.Scanner来实现"打印每行"功能:

package main

import (
        "bufio"
        "fmt"
        "log"
        "os"
)

func main() {
        f, err := os.Open("test.txt")
        if err != nil {
                log.Fatal(err)
        }

        s := bufio.NewScanner(f)

        for s.Scan() {
                fmt.Println(s.Text())
        }
}

(a)

s := bufio.NewScanner(f)

bufio.NewScanner(f)创建一个bufio.Scanner结构体,默认情况下按行分割内容。

bufio.NewScanner(f) creates a new bufio.Scanner struct which splits the content by line by default.

(b)

for s.Scan() {
    fmt.Println(s.Text())
}

s.Scan()前进bufio.Scanner到下一个标记(在这种情况下,它是一个可选的回车后跟一个强制换行),我们可以使用s.Text()函数来获取内容。

我们还可以自定义SplitFunc函数,它不会按行分隔。查看下面代码:

package main

import (
        "bufio"
        "fmt"
        "log"
        "os"
)

func main() {
        f, err := os.Open("test.txt")
        if err != nil {
                log.Fatal(err)
        }

        s := bufio.NewScanner(f)
        split := func(data []byte, atEOF bool) (advance int, token []byte, err error) {
                for i := 0; i < len(data); i++ {
                        if data[i] == 'h' {
                                return i + 1, data[:i], nil
                        }
                }

                return 0, data, bufio.ErrFinalToken
        }
        s.Split(split)
        for s.Scan() {
                fmt.Println(s.Text())
        }
}

split函数用“ h” 分隔内容,运行结果为

abcd
efg

ijk
lmn

本文参与腾讯云自媒体分享计划,欢迎正在阅读的你也加入,一起分享。

发表于

我来说两句

0 条评论
登录 后参与评论

扫码关注云+社区

领取腾讯云代金券