forked from lunny/tango
/
pool.go
42 lines (37 loc) · 799 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
// Copyright 2015 The Tango Authors. All rights reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.
package tango
import (
"reflect"
"sync"
)
type pool struct {
size int
tp reflect.Type
stp reflect.Type
pool reflect.Value
cur int
lock sync.Mutex
}
func newPool(size int, tp reflect.Type) *pool {
return &pool{
size: size,
cur: 0,
pool: reflect.MakeSlice(reflect.SliceOf(tp), 0, 0), // init for don't allocate memory
tp: reflect.SliceOf(tp),
stp: tp,
}
}
func (p *pool) New() reflect.Value {
//return reflect.New(p.stp)
p.lock.Lock()
if p.cur == p.pool.Len() {
p.pool = reflect.MakeSlice(p.tp, p.size, p.size)
p.cur = 0
}
res := p.pool.Index(p.cur).Addr()
p.cur++
p.lock.Unlock()
return res
}