hop.go

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...)
}