golang bytes buffer代码剖析

//上数据结构，bytes Buffer
type Buffer struct {
	buf       []byte            // byte切片
	off       int               // 从&buf[off]地址读数据, 从&buf[len(buf)]地址写数据
	runeBytes [utf8.UTFMax]byte // avoid allocation of slice on each WriteByte or Rune
	bootstrap [64]byte          // memory to hold first slice; helps small buffers (Printf) avoid allocation.
	lastRead  readOp            // last read operation, so that Unread* can work correctly.
}

再来看看我们bytes Buffer里面write是怎么实现的

func (b *Buffer) WriteString(s string) (n int, err error) {
	b.lastRead = opInvalid /// Non-read operation 不需要读的标志 等于0
	m := b.grow(len(s))     // 增长大小(准确来说是调整数据在buf中位置,也不一定增长)，m当然是老数据末尾
	return copy(b.buf[m:], s), nil //copy一下数据，从m开始
}

最最重要的看过来

func (b *Buffer) grow(n int) int {
	m := b.Len() //func (b *Buffer) Len() int { return len(b.buf) - b.off }
	// m就是buf的len 减去（-） b.off(读开始位置)
	if m == 0 && b.off != 0 {
		b.Truncate(0) //下面给予显示
	} 
	//调整大小和位置
	if len(b.buf)+n > cap(b.buf) {
		var buf []byte
		if b.buf == nil && n <= len(b.bootstrap) {
			buf = b.bootstrap[0:] //这个bootstrap缓存了buf的切片，说是防止重allocation
		} else if m+n <= cap(b.buf)/2 { // 二倍申请新的slice的原则
			copy(b.buf[:], b.buf[b.off:])
			buf = b.buf[:m]
		} else {
			// not enough space anywhere
			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
}

再看看它用到过的函数 Truncate函数，缩减切片

func (b *Buffer) Truncate(n int) {
	b.lastRead = opInvalid
	switch {
	case n < 0 || n > b.Len():
		panic("bytes.Buffer: truncation out of range")
	case n == 0:
		// Reuse buffer space.
		b.off = 0
	}
	b.buf = b.buf[0 : b.off+n]
}

makeSlice函数

func makeSlice(n int) []byte {
	// If the make fails, give a known error.
	defer func() {
		if recover() != nil {
			panic(ErrTooLarge)
		}
	}()
	return make([]byte, n) //其实还是调用这个make
}

最后看看一个使用示例咯

package main

import (
	"bytes"
	"fmt"
	"strconv"
	"time"
)

func main() {

	var buffer bytes.Buffer

	ttime := time.Now().UnixNano()

	for i := 0; i < 10000000; i++ {
		buffer.WriteString(strconv.Itoa(i))
	}

	ttime1 := time.Now().UnixNano()
	//取内容buffer.Bytes() 或者 buffer.String()
fmt.Printf("time cal %f %d\n", float64(ttime1-ttime)/float64(1*time.Second), len(buffer.String()))

}

总结：bytes buffer：就是写byte或者字符串string的一个容器