batchwriter.go RawHistoryFolder

1package term
2
3import (
4	"bufio"
5	"io"
6	"sync"
7)
8
9// batchWriter is a thread-safe buffered writer. Writes accumulate in an
10// internal buffer until Flush is called; a single write larger than the
11// buffer's capacity passes straight through to the underlying sink
12// (bufio.Writer's standard behavior).
13type batchWriter struct {
14	mu sync.Mutex
15	bw *bufio.Writer
16}
17
18// newBatchWriter wraps out with a buffer of the given size.
19func newBatchWriter(out io.Writer, size int) *batchWriter {
20	return &batchWriter{bw: bufio.NewWriterSize(out, size)}
21}
22
23// Write buffers p, flushing to the underlying sink only as bufio.Writer
24// requires (e.g. when p would overflow the buffer).
25func (w *batchWriter) Write(p []byte) (int, error) {
26	w.mu.Lock()
27	defer w.mu.Unlock()
28	return w.bw.Write(p)
29}
30
31// WriteString is the string analogue of Write.
32func (w *batchWriter) WriteString(s string) (int, error) {
33	w.mu.Lock()
34	defer w.mu.Unlock()
35	return w.bw.WriteString(s)
36}
37
38// Flush writes any buffered data to the underlying sink.
39func (w *batchWriter) Flush() error {
40	w.mu.Lock()
41	defer w.mu.Unlock()
42	return w.bw.Flush()
43}
44
45// Buffered reports the number of bytes currently buffered and not yet
46// flushed to the underlying sink.
47func (w *batchWriter) Buffered() int {
48	w.mu.Lock()
49	defer w.mu.Unlock()
50	return w.bw.Buffered()
51}