go-gost/x
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

add reload and plugin support for hop

Browse Source
This commit is contained in:
ginuerzh
2023-09-28 21:04:15 +08:00
parent ddc3c9392e
commit ea585fc25d
88 changed files with 2208 additions and 1538 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
registry/hop.go
View File

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