init

go-gost/x/registry/hop.go

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