more timeout

weed/mq/kafka/protocol/handler.go

@@ -307,7 +307,8 @@ func (h *Handler) HandleConn(ctx context.Context, conn net.Conn) error {
 
 		// Start a timeout goroutine that will force completion after 1 second
 		go func() {
-			time.Sleep(1 * time.Second)
+			select {
+			case <-time.After(1 * time.Second):
 				select {
 				case done <- true:
 					fmt.Printf("DEBUG: [%s] Force timeout after 1 second, closing connection\n", connectionID)
@@ -315,6 +316,10 @@ func (h *Handler) HandleConn(ctx context.Context, conn net.Conn) error {
 				default:
 					// Already completed
 				}
+			case <-done:
+				// Operation completed, exit timeout goroutine
+				return
+			}
 		}()
 
 		// Main waiting logic