add tunnel feature for relay

This commit is contained in:
ginuerzh
2023-01-14 13:15:15 +08:00
parent 9b128534a0
commit 82cd924c86
20 changed files with 1000 additions and 45 deletions

37
registry/ingress.go Normal file
View File

@ -0,0 +1,37 @@
package registry
import (
"github.com/go-gost/core/ingress"
)
type ingressRegistry struct {
registry[ingress.Ingress]
}
func (r *ingressRegistry) Register(name string, v ingress.Ingress) error {
return r.registry.Register(name, v)
}
func (r *ingressRegistry) Get(name string) ingress.Ingress {
if name != "" {
return &ingressWrapper{name: name, r: r}
}
return nil
}
func (r *ingressRegistry) get(name string) ingress.Ingress {
return r.registry.Get(name)
}
type ingressWrapper struct {
name string
r *ingressRegistry
}
func (w *ingressWrapper) Get(host string) string {
v := w.r.get(w.name)
if v == nil {
return ""
}
return v.Get(host)
}

View File

@ -10,6 +10,7 @@ import (
"github.com/go-gost/core/bypass"
"github.com/go-gost/core/chain"
"github.com/go-gost/core/hosts"
"github.com/go-gost/core/ingress"
"github.com/go-gost/core/limiter/conn"
"github.com/go-gost/core/limiter/rate"
"github.com/go-gost/core/limiter/traffic"
@ -41,6 +42,8 @@ var (
trafficLimiterReg reg.Registry[traffic.TrafficLimiter] = new(trafficLimiterRegistry)
connLimiterReg reg.Registry[conn.ConnLimiter] = new(connLimiterRegistry)
rateLimiterReg reg.Registry[rate.RateLimiter] = new(rateLimiterRegistry)
ingressReg reg.Registry[ingress.Ingress] = new(ingressRegistry)
)
type registry[T any] struct {
@ -155,3 +158,7 @@ func ConnLimiterRegistry() reg.Registry[conn.ConnLimiter] {
func RateLimiterRegistry() reg.Registry[rate.RateLimiter] {
return rateLimiterReg
}
func IngressRegistry() reg.Registry[ingress.Ingress] {
return ingressReg
}