concurrent_buffer.go

package core

import (
	"bytes"
	"sync"
)

// Implement something very much like bytes.Buffer, but with
// thread-safe methods.
type concurrentbuffer struct {
	buf bytes.Buffer
	mu  sync.RWMutex
}

// Implement io.Reader.
func (b *concurrentbuffer) Read(p []byte) (n int, err error) {
	b.mu.Lock()
	defer b.mu.Unlock()

	return b.buf.Read(p)
}

// Implement io.Writer.
func (b *concurrentbuffer) Write(p []byte) (n int, err error) {
	b.mu.Lock()
	defer b.mu.Unlock()

	return b.buf.Write(p)
}

// Truncate the buffer to 0 elements.
func (b *concurrentbuffer) Reset() {
	b.mu.Lock()
	defer b.mu.Unlock()

	b.buf.Reset()
}

// Return a deep copy of the underlying bytes.
func (b *concurrentbuffer) Bytes() []byte {
	b.mu.RLock()
	defer b.mu.RUnlock()

	src := b.buf.Bytes()
	return append([]byte(nil), src...)
}

// Return a printable version of the buffer.
func (b *concurrentbuffer) String() string {
	b.mu.RLock()
	defer b.mu.RUnlock()

	return b.buf.String()
}