SagerNet/sing
1
0
Fork 0
mirror of https://github.com/SagerNet/sing.git synced 2025-04-04 20:37:40 +03:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add NetConn() net.Conn support for cast

Browse source
This commit is contained in:
世界 2023-10-08 12:00:00 +08:00
parent 875ad36cc6
commit 4fbbd19320
No known key found for this signature in database
GPG key ID: CD109927C34A63C4
1 changed files with 12 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

12
common/upstream.go
View file

@ -1,9 +1,15 @@
package common package common
import "net"
type WithUpstream interface { type WithUpstream interface {
Upstream() any Upstream() any
} }
type stdWithUpstreamNetConn interface {
NetConn() net.Conn
}
func Cast[T any](obj any) (T, bool) { func Cast[T any](obj any) (T, bool) {
if c, ok := obj.(T); ok { if c, ok := obj.(T); ok {
return c, true return c, true
@ -11,6 +17,9 @@ func Cast[T any](obj any) (T, bool) {
if u, ok := obj.(WithUpstream); ok { if u, ok := obj.(WithUpstream); ok {
return Cast[T](u.Upstream()) return Cast[T](u.Upstream())
} }
if u, ok := obj.(stdWithUpstreamNetConn); ok {
return Cast[T](u.NetConn())
}
return DefaultValue[T](), false return DefaultValue[T](), false
} }
@ -27,5 +36,8 @@ func Top(obj any) any {
if u, ok := obj.(WithUpstream); ok { if u, ok := obj.(WithUpstream); ok {
return Top(u.Upstream()) return Top(u.Upstream())
} }
if u, ok := obj.(stdWithUpstreamNetConn); ok {
return Top(u.NetConn())
}
return obj return obj
} }