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

5
registry/registry.go
View File

@ -9,6 +9,7 @@ import (
"github.com/go-gost/core/auth"
"github.com/go-gost/core/bypass"
"github.com/go-gost/core/chain"
"github.com/go-gost/core/hop"
"github.com/go-gost/core/hosts"
"github.com/go-gost/core/ingress"
"github.com/go-gost/core/limiter/conn"
@ -31,7 +32,7 @@ var (
connectorReg reg.Registry[NewConnector] = new(connectorRegistry)
serviceReg reg.Registry[service.Service] = new(serviceRegistry)
chainReg reg.Registry[chain.Chainer] = new(chainRegistry)
hopReg reg.Registry[chain.Hop] = new(hopRegistry)
hopReg reg.Registry[hop.Hop] = new(hopRegistry)
autherReg reg.Registry[auth.Authenticator] = new(autherRegistry)
admissionReg reg.Registry[admission.Admission] = new(admissionRegistry)
bypassReg reg.Registry[bypass.Bypass] = new(bypassRegistry)
@ -119,7 +120,7 @@ func ChainRegistry() reg.Registry[chain.Chainer] {
return chainReg
}
func HopRegistry() reg.Registry[chain.Hop] {
func HopRegistry() reg.Registry[hop.Hop] {
return hopReg
}