mirror of
https://github.com/SagerNet/sing-box.git
synced 2024-12-05 04:46:40 +00:00
69 lines
2.2 KiB
Go
69 lines
2.2 KiB
Go
|
package inbound
|
||
|
|
||
|
import (
|
||
|
"context"
|
||
|
"sync"
|
||
|
|
||
|
"github.com/sagernet/sing-box/adapter"
|
||
|
"github.com/sagernet/sing-box/log"
|
||
|
"github.com/sagernet/sing/common"
|
||
|
E "github.com/sagernet/sing/common/exceptions"
|
||
|
)
|
||
|
|
||
|
type ConstructorFunc[T any] func(ctx context.Context, router adapter.Router, logger log.ContextLogger, tag string, options T) (adapter.Inbound, error)
|
||
|
|
||
|
func Register[Options any](registry *Registry, outboundType string, constructor ConstructorFunc[Options]) {
|
||
|
registry.register(outboundType, func() any {
|
||
|
return new(Options)
|
||
|
}, func(ctx context.Context, router adapter.Router, logger log.ContextLogger, tag string, options any) (adapter.Inbound, error) {
|
||
|
return constructor(ctx, router, logger, tag, common.PtrValueOrDefault(options.(*Options)))
|
||
|
})
|
||
|
}
|
||
|
|
||
|
var _ adapter.InboundRegistry = (*Registry)(nil)
|
||
|
|
||
|
type (
|
||
|
optionsConstructorFunc func() any
|
||
|
constructorFunc func(ctx context.Context, router adapter.Router, logger log.ContextLogger, tag string, options any) (adapter.Inbound, error)
|
||
|
)
|
||
|
|
||
|
type Registry struct {
|
||
|
access sync.Mutex
|
||
|
optionsType map[string]optionsConstructorFunc
|
||
|
constructors map[string]constructorFunc
|
||
|
}
|
||
|
|
||
|
func NewRegistry() *Registry {
|
||
|
return &Registry{
|
||
|
optionsType: make(map[string]optionsConstructorFunc),
|
||
|
constructors: make(map[string]constructorFunc),
|
||
|
}
|
||
|
}
|
||
|
|
||
|
func (r *Registry) CreateOptions(outboundType string) (any, bool) {
|
||
|
r.access.Lock()
|
||
|
defer r.access.Unlock()
|
||
|
optionsConstructor, loaded := r.optionsType[outboundType]
|
||
|
if !loaded {
|
||
|
return nil, false
|
||
|
}
|
||
|
return optionsConstructor(), true
|
||
|
}
|
||
|
|
||
|
func (r *Registry) CreateInbound(ctx context.Context, router adapter.Router, logger log.ContextLogger, tag string, outboundType string, options any) (adapter.Inbound, error) {
|
||
|
r.access.Lock()
|
||
|
defer r.access.Unlock()
|
||
|
constructor, loaded := r.constructors[outboundType]
|
||
|
if !loaded {
|
||
|
return nil, E.New("outbound type not found: " + outboundType)
|
||
|
}
|
||
|
return constructor(ctx, router, logger, tag, options)
|
||
|
}
|
||
|
|
||
|
func (r *Registry) register(outboundType string, optionsConstructor optionsConstructorFunc, constructor constructorFunc) {
|
||
|
r.access.Lock()
|
||
|
defer r.access.Unlock()
|
||
|
r.optionsType[outboundType] = optionsConstructor
|
||
|
r.constructors[outboundType] = constructor
|
||
|
}
|