update

Hunter0x7c7

2022-08-11 b8230139fb40edea387617b6accd8371e37eda58

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
package registry 
 
import ( 
    "github.com/golang/protobuf/proto" 
 
    "github.com/v2fly/v2ray-core/v5/common/protoext" 
) 
 
//go:generate go run github.com/v2fly/v2ray-core/v5/common/errors/errorgen 
 
type implementationSet struct { 
    AliasLookup map[string]*implementation 
} 
 
type CustomLoader func(data []byte, loader LoadByAlias) (proto.Message, error) 
 
type implementation struct { 
    FullName string 
    Alias    []string 
    Loader   CustomLoader 
} 
 
func (i *implementationSet) RegisterImplementation(name string, opt *protoext.MessageOpt, loader CustomLoader) { 
    alias := opt.GetShortName() 
 
    impl := &implementation{ 
        FullName: name, 
        Alias:    alias, 
    } 
 
    for _, aliasName := range alias { 
        i.AliasLookup[aliasName] = impl 
    } 
} 
 
func (i *implementationSet) findImplementationByAlias(alias string) (string, CustomLoader, error) { 
    impl, found := i.AliasLookup[alias] 
    if found { 
        return impl.FullName, impl.Loader, nil 
    } 
    return "", nil, newError("cannot find implementation by alias: ", alias) 
} 
 
func newImplementationSet() *implementationSet { 
    return &implementationSet{AliasLookup: map[string]*implementation{}} 
}