/
register.go
executable file
·46 lines (37 loc) · 957 Bytes
/
register.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
// Copyright (c) 2024 RoseLoverX
package tl
import (
"fmt"
"reflect"
)
var (
// used by decoder, guaranteed that types are convertible to tl.Object
objectByCrc = make(map[uint32]reflect.Type) // this value setting by registerObject(), DO NOT CALL IT BY HANDS
enumCrcs = make(map[uint32]struct{})
)
func registerObject(o Object) {
if o == nil {
panic("object is nil")
}
objectByCrc[o.CRC()] = reflect.TypeOf(o)
}
func registerEnum(o Object) {
registerObject(o)
enumCrcs[o.CRC()] = struct{}{}
}
func RegisterObjects(obs ...Object) {
for _, o := range obs {
if val, found := objectByCrc[o.CRC()]; found {
panic(fmt.Errorf("object with that crc already registered as %v: 0x%08x", val.String(), o.CRC()))
}
registerObject(o)
}
}
func RegisterEnums(enums ...Object) {
for _, e := range enums {
if _, found := enumCrcs[e.CRC()]; found {
panic(fmt.Errorf("enum with that crc already registered"))
}
registerEnum(e)
}
}