fix panic

Snawoot · Oct 14, 2024 · d2fb7a8 · d2fb7a8
1 parent fb732f1
commit d2fb7a8
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/pool/connpool.go b/pool/connpool.go
@@ -101,7 +101,7 @@ func (p *ConnPool) worker() {
 		queue_id := p.prepared.Push(output_ch)
 		p.qmux.Unlock()
 		readctx, readcancel := context.WithCancel(p.ctx)
-		readdone := make(chan struct{}, 1)
+		readdone := make(chan struct{})
 		go func() {
 			connReadContext(readctx, conn, dummybuf)
 			close(readdone)

diff --git a/server/handler.go b/server/handler.go
@@ -55,6 +55,7 @@ func (h *ConnHandler) Handle(ctx context.Context, c net.Conn) {
 	if err != nil {
 		h.logger.Error("Error on connection retrieve from pool: %v", err)
 		c.Close()
+		return
 	}
 	h.proxy(ctx, c, tlsconn)
 }