forked from TrueCloudLab/neoneo-go
168ba7960c
Running time becomes faster under high load while staying the same in the average case. Memory allocation done in `Trie` goes down by about ~10% (even more, actually).
50 lines
1.4 KiB
Go
50 lines
1.4 KiB
Go
package io
|
|
|
|
import (
|
|
"bytes"
|
|
"errors"
|
|
)
|
|
|
|
// BufBinWriter is an additional layer on top of BinWriter that
|
|
// automatically creates buffer to write into that you can get after all
|
|
// writes via Bytes().
|
|
type BufBinWriter struct {
|
|
*BinWriter
|
|
buf *bytes.Buffer
|
|
}
|
|
|
|
// NewBufBinWriter makes a BufBinWriter with an empty byte buffer.
|
|
func NewBufBinWriter() *BufBinWriter {
|
|
b := new(bytes.Buffer)
|
|
return &BufBinWriter{BinWriter: NewBinWriterFromIO(b), buf: b}
|
|
}
|
|
|
|
// NewBufBinWriterPreAlloc makes a BufBinWriter using preallocated buffer.
|
|
func NewBufBinWriterPreAlloc(buf []byte) *BufBinWriter {
|
|
b := bytes.NewBuffer(buf[:0])
|
|
return &BufBinWriter{BinWriter: NewBinWriterFromIO(b), buf: b}
|
|
}
|
|
|
|
// Len returns the number of bytes of the unread portion of the buffer.
|
|
func (bw *BufBinWriter) Len() int {
|
|
return bw.buf.Len()
|
|
}
|
|
|
|
// Bytes returns resulting buffer and makes future writes return an error.
|
|
func (bw *BufBinWriter) Bytes() []byte {
|
|
if bw.Err != nil {
|
|
return nil
|
|
}
|
|
bw.Err = errors.New("buffer already drained")
|
|
return bw.buf.Bytes()
|
|
}
|
|
|
|
// Reset resets the state of the buffer, making it usable again. It can
|
|
// make buffer usage somewhat more efficient, because you don't need to
|
|
// create it again, but beware that the buffer is gonna be the same as the one
|
|
// returned by Bytes(), so if you need that data after Reset() you have to copy
|
|
// it yourself.
|
|
func (bw *BufBinWriter) Reset() {
|
|
bw.Err = nil
|
|
bw.buf.Reset()
|
|
}
|