fix quic config

This commit is contained in:
ginuerzh
2022-11-03 21:30:01 +08:00
parent 76fbcb2046
commit 30d44c7376
31 changed files with 218 additions and 138 deletions

View File

@ -5,7 +5,7 @@ import (
)
type admissionRegistry struct {
registry
registry[admission.Admission]
}
func (r *admissionRegistry) Register(name string, v admission.Admission) error {

View File

@ -5,7 +5,7 @@ import (
)
type autherRegistry struct {
registry
registry[auth.Authenticator]
}
func (r *autherRegistry) Register(name string, v auth.Authenticator) error {

View File

@ -5,7 +5,7 @@ import (
)
type bypassRegistry struct {
registry
registry[bypass.Bypass]
}
func (r *bypassRegistry) Register(name string, v bypass.Bypass) error {

View File

@ -9,7 +9,7 @@ import (
)
type chainRegistry struct {
registry
registry[chain.Chainer]
}
func (r *chainRegistry) Register(name string, v chain.Chainer) error {

View File

@ -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
}

View File

@ -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
}

View File

@ -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
}

View File

@ -7,7 +7,7 @@ import (
)
type hopRegistry struct {
registry
registry[chain.Hop]
}
func (r *hopRegistry) Register(name string, v chain.Hop) error {

View File

@ -7,7 +7,7 @@ import (
)
type hostsRegistry struct {
registry
registry[hosts.HostMapper]
}
func (r *hostsRegistry) Register(name string, v hosts.HostMapper) error {

View File

@ -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 {

View File

@ -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
}

View File

@ -7,7 +7,7 @@ import (
)
type recorderRegistry struct {
registry
registry[recorder.Recorder]
}
func (r *recorderRegistry) Register(name string, v recorder.Recorder) error {

View File

@ -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] {

View File

@ -8,7 +8,7 @@ import (
)
type resolverRegistry struct {
registry
registry[resolver.Resolver]
}
func (r *resolverRegistry) Register(name string, v resolver.Resolver) error {

View File

@ -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]
}