-
Notifications
You must be signed in to change notification settings - Fork 9
/
type.go
115 lines (101 loc) · 2.51 KB
/
type.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
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
package protocol
import "sync"
// Type determines connection protocol type.
type Type string
const (
// TypeJSON means JSON protocol.
TypeJSON Type = "json"
// TypeProtobuf means Protobuf protocol.
TypeProtobuf Type = "protobuf"
)
var (
DefaultJsonPushEncoder = NewJSONPushEncoder()
DefaultProtobufPushEncoder = NewProtobufPushEncoder()
)
// GetPushEncoder ...
func GetPushEncoder(protoType Type) PushEncoder {
if protoType == TypeJSON {
return DefaultJsonPushEncoder
}
return DefaultProtobufPushEncoder
}
var (
DefaultJsonReplyEncoder = NewJSONReplyEncoder()
DefaultProtobufReplyEncoder = NewProtobufReplyEncoder()
)
// GetReplyEncoder ...
func GetReplyEncoder(protoType Type) ReplyEncoder {
if protoType == TypeJSON {
return DefaultJsonReplyEncoder
}
return DefaultProtobufReplyEncoder
}
var (
jsonDataEncoderPool sync.Pool
protobufDataEncoderPool sync.Pool
jsonCommandDecoderPool sync.Pool
protobufCommandDecoderPool sync.Pool
)
// GetDataEncoder ...
func GetDataEncoder(protoType Type) DataEncoder {
if protoType == TypeJSON {
e := jsonDataEncoderPool.Get()
if e == nil {
return NewJSONDataEncoder()
}
protoEncoder := e.(DataEncoder)
protoEncoder.Reset()
return protoEncoder
}
e := protobufDataEncoderPool.Get()
if e == nil {
return NewProtobufDataEncoder()
}
protoEncoder := e.(DataEncoder)
protoEncoder.Reset()
return protoEncoder
}
// PutDataEncoder ...
func PutDataEncoder(protoType Type, e DataEncoder) {
if protoType == TypeJSON {
jsonDataEncoderPool.Put(e)
return
}
protobufDataEncoderPool.Put(e)
}
// GetCommandDecoder ...
func GetCommandDecoder(protoType Type, data []byte) CommandDecoder {
if protoType == TypeJSON {
e := jsonCommandDecoderPool.Get()
if e == nil {
return NewJSONCommandDecoder(data)
}
commandDecoder := e.(*JSONCommandDecoder)
_ = commandDecoder.Reset(data)
return commandDecoder
}
e := protobufCommandDecoderPool.Get()
if e == nil {
return NewProtobufCommandDecoder(data)
}
commandDecoder := e.(*ProtobufCommandDecoder)
_ = commandDecoder.Reset(data)
return commandDecoder
}
// PutCommandDecoder ...
func PutCommandDecoder(protoType Type, e CommandDecoder) {
if protoType == TypeJSON {
jsonCommandDecoderPool.Put(e)
return
}
protobufCommandDecoderPool.Put(e)
}
// GetResultEncoder ...
func GetResultEncoder(protoType Type) ResultEncoder {
if protoType == TypeJSON {
return NewJSONResultEncoder()
}
return NewProtobufResultEncoder()
}
// PutResultEncoder ...
func PutResultEncoder(_ Type, _ ReplyEncoder) {}