aboutsummaryrefslogtreecommitdiff
path: root/vendor/github.com/json-iterator/go/feature_pool.go
blob: 73962bc6f6ccdc66dfa62cd32dc4ddbd1108dec9 (plain)
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
46
47
48
49
50
51
52
53
54
55
56
57
package jsoniter

import (
	"io"
)

// IteratorPool a thread safe pool of iterators with same configuration
type IteratorPool interface {
	BorrowIterator(data []byte) *Iterator
	ReturnIterator(iter *Iterator)
}

// StreamPool a thread safe pool of streams with same configuration
type StreamPool interface {
	BorrowStream(writer io.Writer) *Stream
	ReturnStream(stream *Stream)
}

func (cfg *frozenConfig) BorrowStream(writer io.Writer) *Stream {
	select {
	case stream := <-cfg.streamPool:
		stream.Reset(writer)
		return stream
	default:
		return NewStream(cfg, writer, 512)
	}
}

func (cfg *frozenConfig) ReturnStream(stream *Stream) {
	stream.Error = nil
	select {
	case cfg.streamPool <- stream:
		return
	default:
		return
	}
}

func (cfg *frozenConfig) BorrowIterator(data []byte) *Iterator {
	select {
	case iter := <-cfg.iteratorPool:
		iter.ResetBytes(data)
		return iter
	default:
		return ParseBytes(cfg, data)
	}
}

func (cfg *frozenConfig) ReturnIterator(iter *Iterator) {
	iter.Error = nil
	select {
	case cfg.iteratorPool <- iter:
		return
	default:
		return
	}
}