add hop

registry/hop.go

@@ -0,0 +1,51 @@
+package registry
+
+import (
+	"context"
+
+	"github.com/go-gost/core/chain"
+)
+
+type hopRegistry struct {
+	registry
+}
+
+func (r *hopRegistry) Register(name string, v chain.Hop) error {
+	return r.registry.Register(name, v)
+}
+
+func (r *hopRegistry) Get(name string) chain.Hop {
+	if name != "" {
+		return &hopWrapper{name: name, r: r}
+	}
+	return nil
+}
+
+func (r *hopRegistry) get(name string) chain.Hop {
+	if v := r.registry.Get(name); v != nil {
+		return v.(chain.Hop)
+	}
+	return nil
+}
+
+type hopWrapper struct {
+	name string
+	r    *hopRegistry
+}
+
+func (w *hopWrapper) Nodes() []*chain.Node {
+	v := w.r.get(w.name)
+	if v == nil {
+		return nil
+	}
+	return v.Nodes()
+}
+
+func (w *hopWrapper) Select(ctx context.Context, opts ...chain.SelectOption) *chain.Node {
+	v := w.r.get(w.name)
+	if v == nil {
+		return nil
+	}
+
+	return v.Select(ctx, opts...)
+}

registry/registry.go

@@ -30,6 +30,7 @@ var (
 
 	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{}
@@ -110,6 +111,10 @@ func ChainRegistry() Registry[chain.Chainer] {
 	return chainReg
 }
 
+func HopRegistry() Registry[chain.Hop] {
+	return hopReg
+}
+
 func AutherRegistry() Registry[auth.Authenticator] {
 	return autherReg
 }