if socket is closed, unblock the writer until done

dimkr · Oct 14, 2024 · cffff5f · cffff5f
1 parent 138e4ce
commit cffff5f
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/front/text/line.go b/front/text/line.go
@@ -73,6 +73,14 @@ func LineBuffered(inner io.Writer) *LineWriter {
 					_, err := w.inner.Write(w.buffer.Bytes())
 					if err != nil {
 						w.done <- err
+
+						// continue reading until closed, to unblock the writing routine
+						for {
+							if _, ok := <-w.c; !ok {
+								break
+							}
+						}
+
 						return
 					}