Merge pull request #201 from Bytom/v0.1

[bytom/vapor.git] / vendor / github.com / bytom / encoding / bufpool / bufpool.go

diff --git a/vendor/github.com/bytom/encoding/bufpool/bufpool.go b/vendor/github.com/bytom/encoding/bufpool/bufpool.go
new file mode 100644 (file)
index 0000000..bd59317
--- /dev/null
+++ b/vendor/github.com/bytom/encoding/bufpool/bufpool.go
@@ -0,0 +1,35 @@
+// Package bufpool is a freelist for bytes.Buffer objects.
+package bufpool
+
+import (
+       "bytes"
+       "sync"
+)
+
+var pool = &sync.Pool{New: func() interface{} { return bytes.NewBuffer(nil) }}
+
+// Get returns an initialized bytes.Buffer object.
+// It is like new(bytes.Buffer) except it uses the free list.
+// The caller should call Put when finished with the returned object.
+// Since Buffer.Bytes() returns the buffer's underlying slice,
+// it is not safe for that slice to escape the caller.
+// If the bytes need to escape, CopyBytes should be used.
+func Get() *bytes.Buffer {
+       return pool.Get().(*bytes.Buffer)
+}
+
+// Put resets the buffer and adds it to the freelist.
+func Put(b *bytes.Buffer) {
+       b.Reset()
+       pool.Put(b)
+}
+
+// CopyBytes returns a copy of the bytes contained in the buffer.
+// This slice is safe from updates in the underlying buffer,
+// allowing the buffer to be placed back in the free list.
+func CopyBytes(buf *bytes.Buffer) []byte {
+       b := buf.Bytes()
+       b2 := make([]byte, len(b))
+       copy(b2, b)
+       return b2
+}