-
Notifications
You must be signed in to change notification settings - Fork 8
/
type.go
96 lines (83 loc) · 2.02 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
package protocol
import "sync"
// Type determines connection protocol type.
type Type string
const (
// TypeJSON means JSON protocol - i.e. data encoded in
// JSON-streaming format.
TypeJSON Type = "json"
// TypeProtobuf means protobuf protocol - i.e. data encoded
// as length-delimited protobuf messages.
TypeProtobuf Type = "protobuf"
)
// GetPushEncoder ...
func GetPushEncoder(protoType Type) PushEncoder {
if protoType == TypeJSON {
return NewJSONPushEncoder()
}
return NewProtobufPushEncoder()
}
var (
jsonReplyEncoderPool sync.Pool
protobufReplyEncoderPool sync.Pool
)
// GetReplyEncoder ...
func GetReplyEncoder(protoType Type) ReplyEncoder {
if protoType == TypeJSON {
e := jsonReplyEncoderPool.Get()
if e == nil {
return NewJSONReplyEncoder()
}
protoTypeoder := e.(ReplyEncoder)
protoTypeoder.Reset()
return protoTypeoder
}
e := protobufReplyEncoderPool.Get()
if e == nil {
return NewProtobufReplyEncoder()
}
protoTypeoder := e.(ReplyEncoder)
protoTypeoder.Reset()
return protoTypeoder
}
// PutReplyEncoder ...
func PutReplyEncoder(protoType Type, e ReplyEncoder) {
if protoType == TypeJSON {
jsonReplyEncoderPool.Put(e)
return
}
protobufReplyEncoderPool.Put(e)
}
// GetCommandDecoder ...
func GetCommandDecoder(protoType Type, data []byte) CommandDecoder {
if protoType == TypeJSON {
return NewJSONCommandDecoder(data)
}
return NewProtobufCommandDecoder(data)
}
// PutCommandDecoder ...
func PutCommandDecoder(protoType Type, e CommandDecoder) {
return
}
// GetResultEncoder ...
func GetResultEncoder(protoType Type) ResultEncoder {
if protoType == TypeJSON {
return NewJSONResultEncoder()
}
return NewProtobufResultEncoder()
}
// PutResultEncoder ...
func PutResultEncoder(protoType Type, e ReplyEncoder) {
return
}
// GetParamsDecoder ...
func GetParamsDecoder(protoType Type) ParamsDecoder {
if protoType == TypeJSON {
return NewJSONParamsDecoder()
}
return NewProtobufParamsDecoder()
}
// PutParamsDecoder ...
func PutParamsDecoder(protoType Type, e ParamsDecoder) {
return
}