fix

Signed-off-by: kpango <[email protected]>
vdaas · Oct 27, 2024 · 3357c09 · 3357c09
1 parent f7d704f
commit 3357c09
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/internal/net/dialer.go b/internal/net/dialer.go
@@ -119,7 +119,6 @@ func NewDialer(opts ...DialerOption) (der Dialer, err error) {
 
 	netpoll.SetLoadBalance(netpoll.RoundRobin)
 	d.npDialer = netpoll.NewDialer()
-
 	d.dialer = d.dial
 	if d.enableDNSCache {
 		if d.dnsRefreshDuration > d.dnsCacheExpiration {
@@ -330,7 +329,14 @@ func (d *dialer) dial(ctx context.Context, network, addr string) (conn Conn, err
 		} else {
 			conn, err = d.npDialer.DialConnection(network, addr, d.der.Timeout)
 			if err != nil {
-				conn, err = d.der.DialContext(ctx, network, addr)
+				if conn != nil {
+					err = errors.Join(err, conn.Close())
+				}
+				var ierr error
+				conn, ierr = d.der.DialContext(ctx, network, addr)
+				if ierr != nil {
+					err = errors.Join(err, ierr)
+				}
 			}
 		}
 		return err