141 lines
3.2 KiB
Go
141 lines
3.2 KiB
Go
package registry
|
|
|
|
import (
|
|
"errors"
|
|
"io"
|
|
"sync"
|
|
|
|
"github.com/go-gost/core/admission"
|
|
"github.com/go-gost/core/auth"
|
|
"github.com/go-gost/core/bypass"
|
|
"github.com/go-gost/core/chain"
|
|
"github.com/go-gost/core/hosts"
|
|
"github.com/go-gost/core/limiter/conn"
|
|
"github.com/go-gost/core/limiter/traffic"
|
|
"github.com/go-gost/core/recorder"
|
|
"github.com/go-gost/core/resolver"
|
|
"github.com/go-gost/core/service"
|
|
)
|
|
|
|
var (
|
|
ErrDup = errors.New("registry: duplicate object")
|
|
)
|
|
|
|
var (
|
|
listenerReg Registry[NewListener] = &listenerRegistry{}
|
|
handlerReg Registry[NewHandler] = &handlerRegistry{}
|
|
dialerReg Registry[NewDialer] = &dialerRegistry{}
|
|
connectorReg Registry[NewConnector] = &connectorRegistry{}
|
|
|
|
serviceReg Registry[service.Service] = &serviceRegistry{}
|
|
chainReg Registry[chain.Chainer] = &chainRegistry{}
|
|
autherReg Registry[auth.Authenticator] = &autherRegistry{}
|
|
admissionReg Registry[admission.Admission] = &admissionRegistry{}
|
|
bypassReg Registry[bypass.Bypass] = &bypassRegistry{}
|
|
resolverReg Registry[resolver.Resolver] = &resolverRegistry{}
|
|
hostsReg Registry[hosts.HostMapper] = &hostsRegistry{}
|
|
recorderReg Registry[recorder.Recorder] = &recorderRegistry{}
|
|
trafficLimiterReg Registry[traffic.TrafficLimiter] = &trafficLimiterRegistry{}
|
|
connLimiterReg Registry[conn.ConnLimiter] = &connLimiterRegistry{}
|
|
)
|
|
|
|
type Registry[T any] interface {
|
|
Register(name string, v T) error
|
|
Unregister(name string)
|
|
IsRegistered(name string) bool
|
|
Get(name string) T
|
|
}
|
|
|
|
type registry struct {
|
|
m sync.Map
|
|
}
|
|
|
|
func (r *registry) Register(name string, v any) error {
|
|
if name == "" || v == nil {
|
|
return nil
|
|
}
|
|
if _, loaded := r.m.LoadOrStore(name, v); loaded {
|
|
return ErrDup
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (r *registry) Unregister(name string) {
|
|
if v, ok := r.m.Load(name); ok {
|
|
if closer, ok := v.(io.Closer); ok {
|
|
closer.Close()
|
|
}
|
|
r.m.Delete(name)
|
|
}
|
|
}
|
|
|
|
func (r *registry) IsRegistered(name string) bool {
|
|
_, ok := r.m.Load(name)
|
|
return ok
|
|
}
|
|
|
|
func (r *registry) Get(name string) any {
|
|
if name == "" {
|
|
return nil
|
|
}
|
|
v, _ := r.m.Load(name)
|
|
return v
|
|
}
|
|
|
|
func ListenerRegistry() Registry[NewListener] {
|
|
return listenerReg
|
|
}
|
|
|
|
func HandlerRegistry() Registry[NewHandler] {
|
|
return handlerReg
|
|
}
|
|
|
|
func DialerRegistry() Registry[NewDialer] {
|
|
return dialerReg
|
|
}
|
|
|
|
func ConnectorRegistry() Registry[NewConnector] {
|
|
return connectorReg
|
|
}
|
|
|
|
func ServiceRegistry() Registry[service.Service] {
|
|
return serviceReg
|
|
}
|
|
|
|
func ChainRegistry() Registry[chain.Chainer] {
|
|
return chainReg
|
|
}
|
|
|
|
func AutherRegistry() Registry[auth.Authenticator] {
|
|
return autherReg
|
|
}
|
|
|
|
func AdmissionRegistry() Registry[admission.Admission] {
|
|
return admissionReg
|
|
}
|
|
|
|
func BypassRegistry() Registry[bypass.Bypass] {
|
|
return bypassReg
|
|
}
|
|
|
|
func ResolverRegistry() Registry[resolver.Resolver] {
|
|
return resolverReg
|
|
}
|
|
|
|
func HostsRegistry() Registry[hosts.HostMapper] {
|
|
return hostsReg
|
|
}
|
|
|
|
func RecorderRegistry() Registry[recorder.Recorder] {
|
|
return recorderReg
|
|
}
|
|
|
|
func TrafficLimiterRegistry() Registry[traffic.TrafficLimiter] {
|
|
return trafficLimiterReg
|
|
}
|
|
|
|
func ConnLimiterRegistry() Registry[conn.ConnLimiter] {
|
|
return connLimiterReg
|
|
}
|