cannot close twice for grpc

isayme · May 14, 2024 · 39882e8 · 39882e8
1 parent aa50c8b
commit 39882e8
Showing 1 changed file with 12 additions and 2 deletions.
diff --git a/cmd/local.go b/cmd/local.go
@@ -2,6 +2,7 @@ package cmd
 
 import (
 	"context"
+	"io"
 	"net"
 	"sync"
 	"time"
@@ -66,7 +67,12 @@ func handleConnection(conn net.Conn, tc tunnel.Client) {
 		logger.Errorw("connect tunnel server fail", "err", err)
 		return
 	}
-	defer remote.Close()
+	var once sync.Once
+	defer func() {
+		once.Do(func() {
+			remote.Close()
+		})
+	}()
 
 	logger.Debug("connect tunnel server ok")
 
@@ -80,7 +86,11 @@ func handleConnection(conn net.Conn, tc tunnel.Client) {
 		var n int64
 		n, err = util.CopyBuffer(remote, conn)
 		logger.Debugw("copy from client end", "n", n, "err", err)
-		remote.Close()
+		if err != nil && err != io.EOF {
+			once.Do(func() {
+				remote.Close()
+			})
+		}
 	}()
 
 	go func() {