mirror of
https://github.com/v2fly/v2ray-core.git
synced 2024-09-05 19:44:32 -04:00
46 lines
1.1 KiB
Go
46 lines
1.1 KiB
Go
package registry
|
|
|
|
import (
|
|
"github.com/golang/protobuf/proto"
|
|
"github.com/v2fly/v2ray-core/v4/common/protoext"
|
|
)
|
|
|
|
//go:generate go run github.com/v2fly/v2ray-core/v4/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")
|
|
}
|
|
|
|
func newImplementationSet() *implementationSet {
|
|
return &implementationSet{AliasLookup: map[string]*implementation{}}
|
|
}
|