package serialer

import (
	"bytes"
	"sync"
)

// SafeBuffer Tread safe
type SafeBuffer struct {
	b  bytes.Buffer
	rw sync.Mutex
}

// Read read
func (b *SafeBuffer) Read(p []byte) (n int, err error) {
	b.rw.Lock()
	defer b.rw.Unlock()
	return b.b.Read(p)
}

// Write write
func (b *SafeBuffer) Write(p []byte) (n int, err error) {
	b.rw.Lock()
	defer b.rw.Unlock()
	return b.b.Write(p)
}

// Reset reset
func (b *SafeBuffer) Reset() {
	b.rw.Lock()
	defer b.rw.Unlock()
	b.b.Reset()
}

// RemoveHead4K 删除头部4K 字节
func (b *SafeBuffer) RemoveHead4K() {
	p := make([]byte, 4096)
	b.rw.Lock()
	b.b.Read(p)
	b.rw.Unlock()
}

// Len len
func (b *SafeBuffer) Len() int {
	b.rw.Lock()
	defer b.rw.Unlock()
	return b.b.Len()
}