From d6fe25153cdefa9cd3f87f4bc7aa39af3764c4df Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E4=B8=96=E7=95=8C?= Date: Sun, 8 Oct 2023 12:00:00 +0800 Subject: [PATCH] Add `NetConn() net.Conn` support for cast --- common/upstream.go | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/common/upstream.go b/common/upstream.go index 30070cc..0bf7a5b 100644 --- a/common/upstream.go +++ b/common/upstream.go @@ -1,9 +1,15 @@ package common +import "net" + type WithUpstream interface { Upstream() any } +type stdWithUpstreamNetConn interface { + NetConn() net.Conn +} + func Cast[T any](obj any) (T, bool) { if c, ok := obj.(T); ok { return c, true @@ -11,6 +17,9 @@ func Cast[T any](obj any) (T, bool) { if u, ok := obj.(WithUpstream); ok { return Cast[T](u.Upstream()) } + if u, ok := obj.(stdWithUpstreamNetConn); ok { + return Cast[T](u.NetConn()) + } return DefaultValue[T](), false }