/
codec.go
67 lines (55 loc) · 1.54 KB
/
codec.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
/*
* Copyright (c) 2023 ivfzhou
* myrpc is licensed under Mulan PSL v2.
* You can use this software according to the terms and conditions of the Mulan PSL v2.
* You may obtain a copy of Mulan PSL v2 at:
* http://license.coscl.org.cn/MulanPSL2
* THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND,
* EITHER EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT,
* MERCHANTABILITY OR FIT FOR A PARTICULAR PURPOSE.
* See the Mulan PSL v2 for more details.
*/
package codec
import "sync"
var (
serializerMap sync.Map
compressorMap sync.Map
)
type Serializer interface {
Serialize(interface{}) ([]byte, error)
Deserialize([]byte, interface{}) error
}
type Compressor interface {
Compress([]byte) ([]byte, error)
Decompress([]byte) ([]byte, error)
}
type (
SerializerType string
CompressorType string
)
func GetSerializer(name SerializerType) Serializer {
value, ok := serializerMap.Load(name)
if ok && value != nil {
return value.(Serializer)
}
return nil
}
func GetCompressor(name CompressorType) Compressor {
value, ok := compressorMap.Load(name)
if ok && value != nil {
return value.(Compressor)
}
return nil
}
func RegisterSerializer(name SerializerType, serializer Serializer) {
serializerMap.Store(name, serializer)
}
func RegisterCompressor(name CompressorType, compressor Compressor) {
compressorMap.Store(name, compressor)
}
func DeregisterSerializer(name SerializerType) {
serializerMap.Delete(name)
}
func DeregisterCompressor(name CompressorType) {
compressorMap.Delete(name)
}