add chain group

registry/chain.go

@@ -2,26 +2,27 @@ package registry
 
 import (
 	"github.com/go-gost/core/chain"
+	"github.com/go-gost/core/metadata"
 )
 
 type chainRegistry struct {
 	registry
 }
 
-func (r *chainRegistry) Register(name string, v chain.Chainer) error {
+func (r *chainRegistry) Register(name string, v chain.SelectableChainer) error {
 	return r.registry.Register(name, v)
 }
 
-func (r *chainRegistry) Get(name string) chain.Chainer {
+func (r *chainRegistry) Get(name string) chain.SelectableChainer {
 	if name != "" {
 		return &chainWrapper{name: name, r: r}
 	}
 	return nil
 }
 
-func (r *chainRegistry) get(name string) chain.Chainer {
+func (r *chainRegistry) get(name string) chain.SelectableChainer {
 	if v := r.registry.Get(name); v != nil {
-		return v.(chain.Chainer)
+		return v.(chain.SelectableChainer)
 	}
 	return nil
 }
@@ -31,6 +32,22 @@ type chainWrapper struct {
 	r    *chainRegistry
 }
 
+func (w *chainWrapper) Marker() chain.Marker {
+	v := w.r.get(w.name)
+	if v == nil {
+		return nil
+	}
+	return v.Marker()
+}
+
+func (w *chainWrapper) Metadata() metadata.Metadata {
+	v := w.r.get(w.name)
+	if v == nil {
+		return nil
+	}
+	return v.Metadata()
+}
+
 func (w *chainWrapper) Route(network, address string) *chain.Route {
 	v := w.r.get(w.name)
 	if v == nil {

registry/registry.go

@@ -25,14 +25,14 @@ var (
 	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{}
+	serviceReg   Registry[service.Service]         = &serviceRegistry{}
+	chainReg     Registry[chain.SelectableChainer] = &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{}
 )
 
 type Registry[T any] interface {
@@ -99,7 +99,7 @@ func ServiceRegistry() Registry[service.Service] {
 	return serviceReg
 }
 
-func ChainRegistry() Registry[chain.Chainer] {
+func ChainRegistry() Registry[chain.SelectableChainer] {
 	return chainReg
 }