mirror of
https://github.com/Estom/notes.git
synced 2026-04-14 02:10:47 +08:00
84 lines
3.0 KiB
Go
84 lines
3.0 KiB
Go
// 缓冲包 bufio 常用函数
|
||
// bufio 中与 io.Reader 和 io.Writer 打交道
|
||
// 可以从 Reader 中缓冲读数据
|
||
// 也可以累积数据，最后 flush 写到 Writer 中
|
||
package main
|
||
|
||
import (
|
||
"bufio"
|
||
"bytes"
|
||
"fmt"
|
||
"os"
|
||
"strings"
|
||
)
|
||
|
||
func main() {
|
||
|
||
//
|
||
// 缓冲读
|
||
//
|
||
strR := strings.NewReader("cache and name things") // 21 字节
|
||
bufR := bufio.NewReader(strR) // 从现有 reader 中创建缓冲 Reader，会尽可能多的读取底层 reader 中的数据
|
||
b := make([]byte, 20) // 指定读取 20 字节
|
||
bufR.Read(b) // 将字节读如缓冲区，底层 reader 已读空
|
||
fmt.Println(string(b)) // cache and name thing
|
||
fmt.Println(bufR.Buffered()) // 1 // bufR 还剩 1 字节
|
||
|
||
strR.Reset("cache and name things")
|
||
bufR = bufio.NewReaderSize(strR, 10) // 缓冲区的最小长度是 10，bufR 还是会尽可能多的区读底层 reader 的数据
|
||
b = make([]byte, 12)
|
||
bufR.Read(b)
|
||
fmt.Println(string(b)) // cache and na
|
||
|
||
// 缓冲只读
|
||
peekB, _ := bufR.Peek(2) // Read() 在读取数据后会将该片缓冲区清除，而 Peek() 不会
|
||
fmt.Println(string(peekB)) // me
|
||
|
||
//
|
||
// 分割读
|
||
//
|
||
strR.Reset("name: 吴_胤")
|
||
bufR = bufio.NewReader(strR)
|
||
// r, _ := bufR.ReadRune() // 读 Unicode 字符
|
||
r, _ := bufR.ReadByte() // 读取下一个字节，没有则返回 EOF
|
||
fmt.Println(string(r)) // n
|
||
bufR.UnreadByte() // 还原已读取的最后一个字节
|
||
runes, _ := bufR.ReadBytes(':') // 读取数据直到指定分割字符出现，返回读取到的字节 slice
|
||
fmt.Println(string(runes)) // name:
|
||
str, _ := bufR.ReadString('_') // 分割后返回字符串
|
||
fmt.Printf("%q\n", str) // " 吴_"
|
||
|
||
strR.Reset("a_b")
|
||
bufR = bufio.NewReader(strR)
|
||
line, _ := bufR.ReadSlice('_') // ReadSlice 返回的 slice 值在下次 ReadSlice 后会被修改（引用），一般用得少
|
||
fmt.Println(string(line)) // a_
|
||
bufR.ReadSlice('_') // 返回的 slice 会按序填充到 line 中
|
||
fmt.Println(string(line)) // b_ // _ 被保留了下来
|
||
|
||
//
|
||
// 缓冲写
|
||
//
|
||
w := bytes.NewBuffer(nil)
|
||
bufW := bufio.NewWriter(w)
|
||
bufW.Write([]byte("PUBG")) // 将字节写入缓冲区
|
||
bufW.WriteByte('&') // 写一个字节
|
||
// bufW.WriteRune('&') // 写一个 Unicode 字符
|
||
bufW.WriteString("CSGO")
|
||
fmt.Println(bufW.Buffered()) // 9
|
||
fmt.Println(bufW.Available()) // 4087(4KB-9B) // 缓冲区中剩余缓冲字节数
|
||
bufW.Flush() // 将缓冲区的数据写到底层 writer，完成数据的真正写入
|
||
fmt.Println(string(w.Bytes()))
|
||
bufW = bufio.NewWriterSize(os.Stdout, 10) // 缓冲区的长度至少是 10 字节
|
||
bufW.Write([]byte("more than 10 runes\n")) // 19 nil
|
||
|
||
//
|
||
// 可读可写缓冲
|
||
//
|
||
bufRW := bufio.NewReadWriter(bufR, bufW)
|
||
fmt.Println(bufRW.ReadRune()) // m //可读
|
||
w = bytes.NewBuffer([]byte("magic"))
|
||
bufW = bufio.NewWriter(w)
|
||
bufRW.Flush()
|
||
fmt.Println(string(w.Bytes())) // magic // 可写
|
||
}
|