go原生庫的中bytes.Buffer用法
1 bytes.Buffer定義
bytes.Buffer提供可擴容的字節(jié)緩沖區(qū),實質(zhì)是對切片的封裝;結(jié)構(gòu)中包含一個64字節(jié)的小切片,避免小內(nèi)存分配:
// A Buffer is a variable-sized buffer of bytes with Read and Write methods.
// The zero value for Buffer is an empty buffer ready to use.
type Buffer struct {
buf []byte // contents are the bytes buf[off : len(buf)]
off int // read at &buf[off], write at &buf[len(buf)]--->指示讀指針
bootstrap [64]byte // memory to hold first slice; helps small buffers avoid allocation.
lastRead readOp // last read operation, so that Unread* can work correctly.
}
2 初始化bytes.Buffer的方法
1) var buf bytes.Buffer ->定義一個空的字節(jié)緩沖區(qū)
2) func NewBuffer(buf []byte) *Buffer { return &Buffer{buf: buf} } -->將字節(jié)切片初始化為緩沖區(qū)
3) func NewBufferString(s string) *Buffer {return &Buffer{buf: []byte(s)}} -->將字符串初始化為緩沖區(qū)
3 提供的主要API函數(shù)
1)寫字節(jié)流數(shù)據(jù)到緩沖區(qū)
// Write appends the contents of p to the buffer, growing the buffer as
// needed. The return value n is the length of p; err is always nil. If the
// buffer becomes too large, Write will panic with ErrTooLarge.
func (b *Buffer) Write(p []byte) (n int, err error) {
b.lastRead = opInvalid
m := b.grow(len(p))
return copy(b.buf[m:], p), nil
}
2)寫字符串到緩沖區(qū)
// WriteString appends the contents of s to the buffer, growing the buffer as
// needed. The return value n is the length of s; err is always nil. If the
// buffer becomes too large, WriteString will panic with ErrTooLarge.
func (b *Buffer) WriteString(s string) (n int, err error) {
b.lastRead = opInvalid
//返回寫入的index
m := b.grow(len(s))
return copy(b.buf[m:], s), nil
}
3)從緩沖區(qū)中讀取數(shù)據(jù)
// Read reads the next len(p) bytes from the buffer or until the buffer
// is drained. The return value n is the number of bytes read. If the
// buffer has no data to return, err is io.EOF (unless len(p) is zero);
// otherwise it is nil.
func (b *Buffer) Read(p []byte) (n int, err error) {
b.lastRead = opInvalid
if b.off >= len(b.buf) {
// Buffer is empty, reset to recover space.
b.Truncate(0)
if len(p) == 0 {
return
}
return 0, io.EOF
}
n = copy(p, b.buf[b.off:])
b.off += n
if n > 0 {
b.lastRead = opRead
}
return
}
4)從緩沖區(qū)中讀取字符串,直到分隔符delim 位置
// ReadString reads until the first occurrence of delim in the input,
// returning a string containing the data up to and including the delimiter.
// If ReadString encounters an error before finding a delimiter,
// it returns the data read before the error and the error itself (often io.EOF).
// ReadString returns err != nil if and only if the returned data does not end
// in delim.
func (b *Buffer) ReadString(delim byte) (line string, err error) {
slice, err := b.readSlice(delim)
return string(slice), err
}
5)將未被讀取的字節(jié)數(shù)據(jù)返回
// Bytes returns a slice of length b.Len() holding the unread portion of the buffer.
// The slice is valid for use only until the next buffer modification (that is,
// only until the next call to a method like Read, Write, Reset, or Truncate).
// The slice aliases the buffer content at least until the next buffer modification,
// so immediate changes to the slice will affect the result of future reads.
func (b *Buffer) Bytes() []byte { return b.buf[b.off:] }
6)將未被讀取的字節(jié)數(shù)據(jù)以字符串形式返回
// String returns the contents of the unread portion of the buffer
// as a string. If the Buffer is a nil pointer, it returns "<nil>".
func (b *Buffer) String() string {
if b == nil {
// Special case, useful in debugging.
return "<nil>"
}
return string(b.buf[b.off:])
}
7)返回緩沖區(qū)當(dāng)前容量
// Cap returns the capacity of the buffer's underlying byte slice, that is, the
// total space allocated for the buffer's data.
func (b *Buffer) Cap() int { return cap(b.buf) }
8)返回未被讀取的字節(jié)數(shù)據(jù)大小
// Len returns the number of bytes of the unread portion of the buffer;
// b.Len() == len(b.Bytes()).
func (b *Buffer) Len() int { return len(b.buf) - b.off }
4 bytes.Buffer自動擴容機制
當(dāng)向緩沖區(qū)寫入數(shù)據(jù)時,首先會檢查當(dāng)前容量是否滿足需求,如果不滿足分三種情況處理:
1)當(dāng)前內(nèi)置緩沖區(qū)切片buf為空,且寫入數(shù)據(jù)量小于bootstrap的大小(64字節(jié)),則bootstrap作為buf
2)當(dāng)前未讀數(shù)據(jù)長度+新寫入數(shù)據(jù)長度小于等于緩沖區(qū)容量的1/2,則挪動數(shù)據(jù)(將未讀的數(shù)據(jù)放到已讀數(shù)據(jù)位置)
3)以上條件不滿足,只能重新分配切片,容量設(shè)定為2*cap(b.buf) + n,即兩倍原來的緩沖區(qū)容量+寫入數(shù)據(jù)量大小
// grow grows the buffer to guarantee space for n more bytes.
// It returns the index where bytes should be written.
// If the buffer can't grow it will panic with ErrTooLarge.
func (b *Buffer) grow(n int) int {
m := b.Len()
// If buffer is empty, reset to recover space.
if m == 0 && b.off != 0 {
b.Truncate(0)
}
//如果需要的容量大于現(xiàn)在的容量--->
if len(b.buf)+n > cap(b.buf) {
var buf []byte
//現(xiàn)有的預(yù)備64byte可以滿足
if b.buf == nil && n <= len(b.bootstrap) {
buf = b.bootstrap[0:]
//實際需要的小于本身切片容量
} else if m+n <= cap(b.buf)/2 {
// We can slide things down instead of allocating a new
// slice. We only need m+n <= cap(b.buf) to slide, but
// we instead let capacity get twice as large so we
// don't spend all our time copying.
copy(b.buf[:], b.buf[b.off:])
buf = b.buf[:m]
} else {
// not enough space anywhere
//不夠,那就分配2倍加n的容量
buf = makeSlice(2*cap(b.buf) + n)
copy(buf, b.buf[b.off:])
}
b.buf = buf
b.off = 0
}
b.buf = b.buf[0 : b.off+m+n]
return b.off + m
}
5 bytes.Buffer的局限
bytes.Buffer提供了對切片的初步封裝,但也沒做太多的事;對于已讀的數(shù)據(jù)無法操作。
補充:Golang bytes.Buffer 用法精述
1.簡介
bytes.Buffer 是 Golang 標(biāo)準(zhǔn)庫中的緩沖區(qū),具有讀寫方法和可變大小的字節(jié)存儲功能。緩沖區(qū)的零值是一個待使用的空緩沖區(qū)。
定義如下:
type Buffer struct {
buf []byte // contents are the bytes buf[off : len(buf)]
off int // read at &buf[off], write at &buf[len(buf)]
lastRead readOp // last read operation, so that Unread* can work correctly.
}
注意要點:
(1)從 bytes.Buffer 讀取數(shù)據(jù)后,被成功讀取的數(shù)據(jù)仍保留在原緩沖區(qū),只是無法被使用,因為緩沖區(qū)的可見數(shù)據(jù)從偏移 off 開始,即buf[off : len(buf)]。
2.常用方法
(1)聲明一個 Buffer
var b bytes.Buffer //直接定義一個Buffer變量,不用初始化,可以直接使用 b := new(bytes.Buffer) //使用New返回Buffer變量 b := bytes.NewBuffer(s []byte) //從一個[]byte切片,構(gòu)造一個Buffer b := bytes.NewBufferString(s string) //從一個string變量,構(gòu)造一個Buffer
(2)往 Buffer 中寫入數(shù)據(jù)
b.Write(d []byte) (n int, err error) //將切片d寫入Buffer尾部 b.WriteString(s string) (n int, err error) //將字符串s寫入Buffer尾部 b.WriteByte(c byte) error //將字符c寫入Buffer尾部 b.WriteRune(r rune) (n int, err error) //將一個rune類型的數(shù)據(jù)放到緩沖區(qū)的尾部 b.ReadFrom(r io.Reader) (n int64, err error) //從實現(xiàn)了io.Reader接口的可讀取對象寫入Buffer尾部
(3)從 Buffer 中讀取數(shù)據(jù)
//讀取 n 個字節(jié)數(shù)據(jù)并返回,如果 buffer 不足 n 字節(jié),則讀取全部 b.Next(n int) []byte //一次讀取 len(p) 個 byte 到 p 中,每次讀取新的內(nèi)容將覆蓋p中原來的內(nèi)容。成功返回實際讀取的字節(jié)數(shù),off 向后偏移 n,buffer 沒有數(shù)據(jù)返回錯誤 io.EOF b.Read(p []byte) (n int, err error) //讀取第一個byte并返回,off 向后偏移 n b.ReadByte() (byte, error) //讀取第一個 UTF8 編碼的字符并返回該字符和該字符的字節(jié)數(shù),b的第1個rune被拿掉。如果buffer為空,返回錯誤 io.EOF,如果不是UTF8編碼的字符,則消費一個字節(jié),返回 (U+FFFD,1,nil) b.ReadRune() (r rune, size int, err error) //讀取緩沖區(qū)第一個分隔符前面的內(nèi)容以及分隔符并返回,緩沖區(qū)會清空讀取的內(nèi)容。如果沒有發(fā)現(xiàn)分隔符,則返回讀取的內(nèi)容并返回錯誤io.EOF b.ReadBytes(delimiter byte) (line []byte, err error) //讀取緩沖區(qū)第一個分隔符前面的內(nèi)容以及分隔符并作為字符串返回,緩沖區(qū)會清空讀取的內(nèi)容。如果沒有發(fā)現(xiàn)分隔符,則返回讀取的內(nèi)容并返回錯誤 io.EOF b.ReadString(delimiter byte) (line string, err error) //將 Buffer 中的內(nèi)容輸出到實現(xiàn)了 io.Writer 接口的可寫入對象中,成功返回寫入的字節(jié)數(shù),失敗返回錯誤 b.WriteTo(w io.Writer) (n int64, err error)
(4)其它操作
b.Bytes() []byte //返回字節(jié)切片 b.Cap() int //返回 buffer 內(nèi)部字節(jié)切片的容量 b.Grow(n int) //為 buffer 內(nèi)部字節(jié)切片的容量增加 n 字節(jié) b.Len() int //返回緩沖區(qū)數(shù)據(jù)長度,等于 len(b.Bytes()) b.Reset() //清空數(shù)據(jù) b.String() string //字符串化 b.Truncate(n int) //丟棄緩沖區(qū)中除前n個未讀字節(jié)以外的所有字節(jié)。如果 n 為負(fù)數(shù)或大于緩沖區(qū)長度,則引發(fā) panic b.UnreadByte() error //將最后一次讀取操作中被成功讀取的字節(jié)設(shè)為未被讀取的狀態(tài),即將已讀取的偏移 off 減 1 b.UnreadRune() error //將最后一次 ReadRune() 讀取操作返回的 UTF8 字符 rune設(shè)為未被讀取的狀態(tài),即將已讀取的偏移 off 減去 字符 rune 的字節(jié)數(shù)
3.使用示例
(1)從文件 test.txt 中讀取全部內(nèi)容追加到 buffer 尾部
test.txt 的內(nèi)容為:
My name is dablelv
具體實現(xiàn):
package main
import (
"os"
"fmt"
"bytes"
)
func main() {
file, _ := os.Open("./test.txt")
buf := bytes.NewBufferString("Hello world ")
buf.ReadFrom(file) //將text.txt內(nèi)容追加到緩沖器的尾部
fmt.Println(buf.String())
}
編譯運行輸出:
Hello world My name is dablelv
以上為個人經(jīng)驗,希望能給大家一個參考,也希望大家多多支持腳本之家。如有錯誤或未考慮完全的地方,望不吝賜教。
相關(guān)文章
Go中函數(shù)的使用細(xì)節(jié)與注意事項詳解
在Go語言中函數(shù)可是一等的(first-class)公民,函數(shù)類型也是一等的數(shù)據(jù)類型,下面這篇文章主要給大家介紹了關(guān)于Go中函數(shù)的使用細(xì)節(jié)與注意事項的相關(guān)資料,文中通過實例代碼介紹的非常詳細(xì),需要的朋友可以參考下2022-11-11
詳解如何使用unsafe標(biāo)準(zhǔn)庫突破Golang中的類型限制
在使用c語言編程時,常常因為類型的問題大傷腦筋,而,golang提供了一些方式用于喜歡hack的用戶,下面我們就來講講如何使用unsafe標(biāo)準(zhǔn)庫突破Golang中的類型限制吧2024-03-03
Golang實現(xiàn)Redis網(wǎng)絡(luò)協(xié)議實例探究
這篇文章主要為大家介紹了Golang實現(xiàn)Redis網(wǎng)絡(luò)協(xié)議實例探究,有需要的朋友可以借鑒參考下,希望能夠有所幫助,祝大家多多進步,早日升職加薪2024-01-01
Go調(diào)度器學(xué)習(xí)之協(xié)作與搶占詳解
如果某個G執(zhí)行時間過長,其他的G如何才能被正常調(diào)度,這就引出了接下來的話題:協(xié)作與搶占。本文將通過一些示例為大家詳細(xì)講講調(diào)度器中協(xié)作與搶占的相關(guān)知識,需要的可以參考一下2023-04-04

