From b70a3af443e7ea9968a9ec3f6430fc480d8602d0 Mon Sep 17 00:00:00 2001 From: pyihe <785131182@qq.com> Date: Fri, 5 Aug 2022 17:56:18 +0800 Subject: [PATCH] feature(buffers): add buffers --- buffers/buffer.go | 35 +++++++++++++++++++++++++++++++++++ bytes/bytes.go | 12 ------------ 2 files changed, 35 insertions(+), 12 deletions(-) create mode 100644 buffers/buffer.go diff --git a/buffers/buffer.go b/buffers/buffer.go new file mode 100644 index 0000000..8a88eed --- /dev/null +++ b/buffers/buffer.go @@ -0,0 +1,35 @@ +package buffers + +import ( + "bytes" + "sync" +) + +//type ByteBuffer = bytebufferpool.ByteBuffer +// +//var ( +// Get = bytebufferpool.Get +// Put = func(b *ByteBuffer) { +// if b != nil { +// bytebufferpool.Put(b) +// } +// } +//) + +var bp sync.Pool + +func Get() *bytes.Buffer { + buffer, ok := bp.Get().(*bytes.Buffer) + if ok { + return buffer + } + return &bytes.Buffer{} +} + +func Put(b *bytes.Buffer) { + if b == nil { + return + } + b.Reset() + bp.Put(b) +} diff --git a/bytes/bytes.go b/bytes/bytes.go index ffd2572..0fd4f42 100644 --- a/bytes/bytes.go +++ b/bytes/bytes.go @@ -5,18 +5,6 @@ import ( "unsafe" "github.com/pyihe/go-pkg/errors" - "github.com/valyala/bytebufferpool" -) - -type ByteBuffer = bytebufferpool.ByteBuffer - -var ( - Get = bytebufferpool.Get - Put = func(b *ByteBuffer) { - if b != nil { - bytebufferpool.Put(b) - } - } ) func Int64(b []byte) (v int64, err error) {