ginuerzh
acce86c0e4
Log close errors in Unregister instead of silently discarding them. Add doc comments to all exported symbols across 20 registry files. Add comprehensive tests covering base registry, all wrapper types, hot-reload delegation, nil-fallback semantics, and global accessors.
71 lines
1.4 KiB
Go
71 lines
1.4 KiB
Go
package registry
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/go-gost/core/sd"
|
|
)
|
|
|
|
// sdRegistry implements a hot-reload-safe registry for sd.SD.
|
|
type sdRegistry struct {
|
|
registry[sd.SD]
|
|
}
|
|
|
|
// Register stores an SD under the given name.
|
|
func (r *sdRegistry) Register(name string, v sd.SD) error {
|
|
return r.registry.Register(name, v)
|
|
}
|
|
|
|
// Get returns a wrapper that delegates to the currently registered SD.
|
|
// Returns nil if name is empty.
|
|
func (r *sdRegistry) Get(name string) sd.SD {
|
|
if name != "" {
|
|
return &sdWrapper{name: name, r: r}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (r *sdRegistry) get(name string) sd.SD {
|
|
return r.registry.Get(name)
|
|
}
|
|
|
|
type sdWrapper struct {
|
|
name string
|
|
r *sdRegistry
|
|
}
|
|
|
|
func (w *sdWrapper) Register(ctx context.Context, service *sd.Service, opts ...sd.Option) error {
|
|
v := w.r.get(w.name)
|
|
if v == nil {
|
|
return nil
|
|
}
|
|
return v.Register(ctx, service, opts...)
|
|
}
|
|
|
|
func (w *sdWrapper) Deregister(ctx context.Context, service *sd.Service) error {
|
|
v := w.r.get(w.name)
|
|
if v == nil {
|
|
return nil
|
|
}
|
|
|
|
return v.Deregister(ctx, service)
|
|
}
|
|
|
|
func (w *sdWrapper) Renew(ctx context.Context, service *sd.Service) error {
|
|
v := w.r.get(w.name)
|
|
if v == nil {
|
|
return nil
|
|
}
|
|
|
|
return v.Renew(ctx, service)
|
|
}
|
|
|
|
func (w *sdWrapper) Get(ctx context.Context, name string) ([]*sd.Service, error) {
|
|
v := w.r.get(w.name)
|
|
if v == nil {
|
|
return nil, nil
|
|
}
|
|
|
|
return v.Get(ctx, name)
|
|
}
|