blob: 8fb3e202cf45427a0e76900482eda2a926ea23a7 [file] [log] [blame]
Girish Gowdra64503432020-01-07 10:59:10 +05301// Copyright (c) 2016 Uber Technologies, Inc.
2//
3// Permission is hereby granted, free of charge, to any person obtaining a copy
4// of this software and associated documentation files (the "Software"), to deal
5// in the Software without restriction, including without limitation the rights
6// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
7// copies of the Software, and to permit persons to whom the Software is
8// furnished to do so, subject to the following conditions:
9//
10// The above copyright notice and this permission notice shall be included in
11// all copies or substantial portions of the Software.
12//
13// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
14// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
15// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
16// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
17// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
18// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
19// THE SOFTWARE.
20
21package buffer
22
23import "sync"
24
25// A Pool is a type-safe wrapper around a sync.Pool.
26type Pool struct {
27 p *sync.Pool
28}
29
30// NewPool constructs a new Pool.
31func NewPool() Pool {
32 return Pool{p: &sync.Pool{
33 New: func() interface{} {
34 return &Buffer{bs: make([]byte, 0, _size)}
35 },
36 }}
37}
38
39// Get retrieves a Buffer from the pool, creating one if necessary.
40func (p Pool) Get() *Buffer {
41 buf := p.p.Get().(*Buffer)
42 buf.Reset()
43 buf.pool = p
44 return buf
45}
46
47func (p Pool) put(buf *Buffer) {
48 p.p.Put(buf)
49}