add sd for tunnel

registry/registry.go

@@ -18,6 +18,7 @@ import (
 	"github.com/go-gost/core/recorder"
 	reg "github.com/go-gost/core/registry"
 	"github.com/go-gost/core/resolver"
+	"github.com/go-gost/core/sd"
 	"github.com/go-gost/core/service"
 )
 
@@ -32,7 +33,7 @@ var (
 	connectorReg reg.Registry[NewConnector]        = new(connectorRegistry)
 	serviceReg   reg.Registry[service.Service]     = new(serviceRegistry)
 	chainReg     reg.Registry[chain.Chainer]       = new(chainRegistry)
-	hopReg       reg.Registry[hop.Hop]           = new(hopRegistry)
+	hopReg       reg.Registry[hop.Hop]             = new(hopRegistry)
 	autherReg    reg.Registry[auth.Authenticator]  = new(autherRegistry)
 	admissionReg reg.Registry[admission.Admission] = new(admissionRegistry)
 	bypassReg    reg.Registry[bypass.Bypass]       = new(bypassRegistry)
@@ -45,6 +46,7 @@ var (
 	rateLimiterReg    reg.Registry[rate.RateLimiter]       = new(rateLimiterRegistry)
 
 	ingressReg reg.Registry[ingress.Ingress] = new(ingressRegistry)
+	sdReg      reg.Registry[sd.SD]           = new(sdRegistry)
 )
 
 type registry[T any] struct {
@@ -163,3 +165,7 @@ func RateLimiterRegistry() reg.Registry[rate.RateLimiter] {
 func IngressRegistry() reg.Registry[ingress.Ingress] {
 	return ingressReg
 }
+
+func SDRegistry() reg.Registry[sd.SD] {
+	return sdReg
+}

registry/sd.go

@@ -0,0 +1,66 @@
+package registry
+
+import (
+	"context"
+
+	"github.com/go-gost/core/sd"
+)
+
+type sdRegistry struct {
+	registry[sd.SD]
+}
+
+func (r *sdRegistry) Register(name string, v sd.SD) error {
+	return r.registry.Register(name, v)
+}
+
+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, name string, network, address string, opts ...sd.Option) error {
+	v := w.r.get(w.name)
+	if v == nil {
+		return nil
+	}
+	return v.Register(ctx, name, network, address, opts...)
+}
+
+func (w *sdWrapper) Deregister(ctx context.Context, name string) error {
+	v := w.r.get(w.name)
+	if v == nil {
+		return nil
+	}
+
+	return v.Deregister(ctx, name)
+}
+
+func (w *sdWrapper) Renew(ctx context.Context, name string) error {
+	v := w.r.get(w.name)
+	if v == nil {
+		return nil
+	}
+
+	return v.Renew(ctx, name)
+}
+
+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)
+}