fix quic config
This commit is contained in:
@ -5,7 +5,7 @@ import (
|
||||
)
|
||||
|
||||
type admissionRegistry struct {
|
||||
registry
|
||||
registry[admission.Admission]
|
||||
}
|
||||
|
||||
func (r *admissionRegistry) Register(name string, v admission.Admission) error {
|
||||
|
@ -5,7 +5,7 @@ import (
|
||||
)
|
||||
|
||||
type autherRegistry struct {
|
||||
registry
|
||||
registry[auth.Authenticator]
|
||||
}
|
||||
|
||||
func (r *autherRegistry) Register(name string, v auth.Authenticator) error {
|
||||
|
@ -5,7 +5,7 @@ import (
|
||||
)
|
||||
|
||||
type bypassRegistry struct {
|
||||
registry
|
||||
registry[bypass.Bypass]
|
||||
}
|
||||
|
||||
func (r *bypassRegistry) Register(name string, v bypass.Bypass) error {
|
||||
|
@ -9,7 +9,7 @@ import (
|
||||
)
|
||||
|
||||
type chainRegistry struct {
|
||||
registry
|
||||
registry[chain.Chainer]
|
||||
}
|
||||
|
||||
func (r *chainRegistry) Register(name string, v chain.Chainer) error {
|
||||
|
@ -8,7 +8,7 @@ import (
|
||||
type NewConnector func(opts ...connector.Option) connector.Connector
|
||||
|
||||
type connectorRegistry struct {
|
||||
registry
|
||||
registry[NewConnector]
|
||||
}
|
||||
|
||||
func (r *connectorRegistry) Register(name string, v NewConnector) error {
|
||||
@ -17,10 +17,3 @@ func (r *connectorRegistry) Register(name string, v NewConnector) error {
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func (r *connectorRegistry) Get(name string) NewConnector {
|
||||
if v := r.registry.Get(name); v != nil {
|
||||
return v.(NewConnector)
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
@ -8,7 +8,7 @@ import (
|
||||
type NewDialer func(opts ...dialer.Option) dialer.Dialer
|
||||
|
||||
type dialerRegistry struct {
|
||||
registry
|
||||
registry[NewDialer]
|
||||
}
|
||||
|
||||
func (r *dialerRegistry) Register(name string, v NewDialer) error {
|
||||
@ -17,10 +17,3 @@ func (r *dialerRegistry) Register(name string, v NewDialer) error {
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func (r *dialerRegistry) Get(name string) NewDialer {
|
||||
if v := r.registry.Get(name); v != nil {
|
||||
return v.(NewDialer)
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
@ -8,7 +8,7 @@ import (
|
||||
type NewHandler func(opts ...handler.Option) handler.Handler
|
||||
|
||||
type handlerRegistry struct {
|
||||
registry
|
||||
registry[NewHandler]
|
||||
}
|
||||
|
||||
func (r *handlerRegistry) Register(name string, v NewHandler) error {
|
||||
@ -17,10 +17,3 @@ func (r *handlerRegistry) Register(name string, v NewHandler) error {
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func (r *handlerRegistry) Get(name string) NewHandler {
|
||||
if v := r.registry.Get(name); v != nil {
|
||||
return v.(NewHandler)
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
@ -7,7 +7,7 @@ import (
|
||||
)
|
||||
|
||||
type hopRegistry struct {
|
||||
registry
|
||||
registry[chain.Hop]
|
||||
}
|
||||
|
||||
func (r *hopRegistry) Register(name string, v chain.Hop) error {
|
||||
|
@ -7,7 +7,7 @@ import (
|
||||
)
|
||||
|
||||
type hostsRegistry struct {
|
||||
registry
|
||||
registry[hosts.HostMapper]
|
||||
}
|
||||
|
||||
func (r *hostsRegistry) Register(name string, v hosts.HostMapper) error {
|
||||
|
@ -7,7 +7,7 @@ import (
|
||||
)
|
||||
|
||||
type trafficLimiterRegistry struct {
|
||||
registry
|
||||
registry[traffic.TrafficLimiter]
|
||||
}
|
||||
|
||||
func (r *trafficLimiterRegistry) Register(name string, v traffic.TrafficLimiter) error {
|
||||
@ -50,7 +50,7 @@ func (w *trafficLimiterWrapper) Out(key string) traffic.Limiter {
|
||||
}
|
||||
|
||||
type connLimiterRegistry struct {
|
||||
registry
|
||||
registry[conn.ConnLimiter]
|
||||
}
|
||||
|
||||
func (r *connLimiterRegistry) Register(name string, v conn.ConnLimiter) error {
|
||||
@ -85,7 +85,7 @@ func (w *connLimiterWrapper) Limiter(key string) conn.Limiter {
|
||||
}
|
||||
|
||||
type rateLimiterRegistry struct {
|
||||
registry
|
||||
registry[rate.RateLimiter]
|
||||
}
|
||||
|
||||
func (r *rateLimiterRegistry) Register(name string, v rate.RateLimiter) error {
|
||||
|
@ -8,7 +8,7 @@ import (
|
||||
type NewListener func(opts ...listener.Option) listener.Listener
|
||||
|
||||
type listenerRegistry struct {
|
||||
registry
|
||||
registry[NewListener]
|
||||
}
|
||||
|
||||
func (r *listenerRegistry) Register(name string, v NewListener) error {
|
||||
@ -17,10 +17,3 @@ func (r *listenerRegistry) Register(name string, v NewListener) error {
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func (r *listenerRegistry) Get(name string) NewListener {
|
||||
if v := r.registry.Get(name); v != nil {
|
||||
return v.(NewListener)
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
@ -7,7 +7,7 @@ import (
|
||||
)
|
||||
|
||||
type recorderRegistry struct {
|
||||
registry
|
||||
registry[recorder.Recorder]
|
||||
}
|
||||
|
||||
func (r *recorderRegistry) Register(name string, v recorder.Recorder) error {
|
||||
|
@ -23,24 +23,24 @@ var (
|
||||
)
|
||||
|
||||
var (
|
||||
listenerReg Registry[NewListener] = &listenerRegistry{}
|
||||
handlerReg Registry[NewHandler] = &handlerRegistry{}
|
||||
dialerReg Registry[NewDialer] = &dialerRegistry{}
|
||||
connectorReg Registry[NewConnector] = &connectorRegistry{}
|
||||
listenerReg Registry[NewListener] = new(listenerRegistry)
|
||||
handlerReg Registry[NewHandler] = new(handlerRegistry)
|
||||
dialerReg Registry[NewDialer] = new(dialerRegistry)
|
||||
connectorReg Registry[NewConnector] = new(connectorRegistry)
|
||||
|
||||
serviceReg Registry[service.Service] = &serviceRegistry{}
|
||||
chainReg Registry[chain.Chainer] = &chainRegistry{}
|
||||
hopReg Registry[chain.Hop] = &hopRegistry{}
|
||||
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{}
|
||||
serviceReg Registry[service.Service] = new(serviceRegistry)
|
||||
chainReg Registry[chain.Chainer] = new(chainRegistry)
|
||||
hopReg Registry[chain.Hop] = new(hopRegistry)
|
||||
autherReg Registry[auth.Authenticator] = new(autherRegistry)
|
||||
admissionReg Registry[admission.Admission] = new(admissionRegistry)
|
||||
bypassReg Registry[bypass.Bypass] = new(bypassRegistry)
|
||||
resolverReg Registry[resolver.Resolver] = new(resolverRegistry)
|
||||
hostsReg Registry[hosts.HostMapper] = new(hostsRegistry)
|
||||
recorderReg Registry[recorder.Recorder] = new(recorderRegistry)
|
||||
|
||||
trafficLimiterReg Registry[traffic.TrafficLimiter] = &trafficLimiterRegistry{}
|
||||
connLimiterReg Registry[conn.ConnLimiter] = &connLimiterRegistry{}
|
||||
rateLimiterReg Registry[rate.RateLimiter] = &rateLimiterRegistry{}
|
||||
trafficLimiterReg Registry[traffic.TrafficLimiter] = new(trafficLimiterRegistry)
|
||||
connLimiterReg Registry[conn.ConnLimiter] = new(connLimiterRegistry)
|
||||
rateLimiterReg Registry[rate.RateLimiter] = new(rateLimiterRegistry)
|
||||
)
|
||||
|
||||
type Registry[T any] interface {
|
||||
@ -48,14 +48,15 @@ type Registry[T any] interface {
|
||||
Unregister(name string)
|
||||
IsRegistered(name string) bool
|
||||
Get(name string) T
|
||||
GetAll() map[string]T
|
||||
}
|
||||
|
||||
type registry struct {
|
||||
type registry[T any] struct {
|
||||
m sync.Map
|
||||
}
|
||||
|
||||
func (r *registry) Register(name string, v any) error {
|
||||
if name == "" || v == nil {
|
||||
func (r *registry[T]) Register(name string, v T) error {
|
||||
if name == "" {
|
||||
return nil
|
||||
}
|
||||
if _, loaded := r.m.LoadOrStore(name, v); loaded {
|
||||
@ -65,7 +66,7 @@ func (r *registry) Register(name string, v any) error {
|
||||
return nil
|
||||
}
|
||||
|
||||
func (r *registry) Unregister(name string) {
|
||||
func (r *registry[T]) Unregister(name string) {
|
||||
if v, ok := r.m.Load(name); ok {
|
||||
if closer, ok := v.(io.Closer); ok {
|
||||
closer.Close()
|
||||
@ -74,17 +75,29 @@ func (r *registry) Unregister(name string) {
|
||||
}
|
||||
}
|
||||
|
||||
func (r *registry) IsRegistered(name string) bool {
|
||||
func (r *registry[T]) IsRegistered(name string) bool {
|
||||
_, ok := r.m.Load(name)
|
||||
return ok
|
||||
}
|
||||
|
||||
func (r *registry) Get(name string) any {
|
||||
func (r *registry[T]) Get(name string) (t T) {
|
||||
if name == "" {
|
||||
return nil
|
||||
return
|
||||
}
|
||||
v, _ := r.m.Load(name)
|
||||
return v
|
||||
t, _ = v.(T)
|
||||
return
|
||||
}
|
||||
|
||||
func (r *registry[T]) GetAll() (m map[string]T) {
|
||||
m = make(map[string]T)
|
||||
r.m.Range(func(key, value any) bool {
|
||||
k, _ := key.(string)
|
||||
v, _ := value.(T)
|
||||
m[k] = v
|
||||
return true
|
||||
})
|
||||
return
|
||||
}
|
||||
|
||||
func ListenerRegistry() Registry[NewListener] {
|
||||
|
@ -8,7 +8,7 @@ import (
|
||||
)
|
||||
|
||||
type resolverRegistry struct {
|
||||
registry
|
||||
registry[resolver.Resolver]
|
||||
}
|
||||
|
||||
func (r *resolverRegistry) Register(name string, v resolver.Resolver) error {
|
||||
|
@ -5,16 +5,5 @@ import (
|
||||
)
|
||||
|
||||
type serviceRegistry struct {
|
||||
registry
|
||||
}
|
||||
|
||||
func (r *serviceRegistry) Register(name string, v service.Service) error {
|
||||
return r.registry.Register(name, v)
|
||||
}
|
||||
|
||||
func (r *serviceRegistry) Get(name string) service.Service {
|
||||
if v := r.registry.Get(name); v != nil {
|
||||
return v.(service.Service)
|
||||
}
|
||||
return nil
|
||||
registry[service.Service]
|
||||
}
|
||||
|
Reference in New Issue
Block a user