-
-
Notifications
You must be signed in to change notification settings - Fork 577
/
pool.go
45 lines (38 loc) · 818 Bytes
/
pool.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
package sockjs
import (
"bytes"
"strings"
"sync"
)
var stringReaderPool sync.Pool
// GetStringReader from pool.
func GetStringReader(data string) *strings.Reader {
r := bytesReaderPool.Get()
if r == nil {
return strings.NewReader(data)
}
reader := r.(*strings.Reader)
reader.Reset(data)
return reader
}
// PutStringReader to pool.
func PutStringReader(reader *strings.Reader) {
reader.Reset("")
stringReaderPool.Put(reader)
}
var bytesReaderPool sync.Pool
// GetBytesReader from pool.
func GetBytesReader(data []byte) *bytes.Reader {
r := bytesReaderPool.Get()
if r == nil {
return bytes.NewReader(data)
}
reader := r.(*bytes.Reader)
reader.Reset(data)
return reader
}
// PutBytesReader to pool.
func PutBytesReader(reader *bytes.Reader) {
reader.Reset(nil)
bytesReaderPool.Put(reader)
}