Manually close conn when mux read fails

mohanson · Aug 23, 2024 · 85f4794 · 85f4794
1 parent f7b2db8
commit 85f4794
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/protocol/czar/engine.go b/protocol/czar/engine.go
@@ -179,13 +179,14 @@ func (c *Client) Run() {
 			case c.Mux <- mux:
 			case <-mux.rer.Sig():
 				log.Println("czar: mux done")
+				mux.Close()
 				sid = 0
 			case <-c.Cancel:
 				log.Println("czar: mux done")
+				mux.Close()
 				sid = 2
 			}
 		case 2:
-			mux.Close()
 			return
 		}
 	}