simplify logic

2024-02-27 03:18:06 +00:00 · 2016-07-29 12:13:09 +02:00
parent 53cdcfdc2b
commit 6fa84624cb
1 changed files with 2 additions and 4 deletions
@@ -77,15 +77,13 @@ func (this *Stream) Read() (*alloc.Buffer, error) {
 }

 func (this *Stream) Write(data *alloc.Buffer) error {
-	if this.closed {
-		return io.EOF
-	}
-	for {
+	for !this.closed {
 		err := this.TryWriteOnce(data)
 		if err != ErrIOTimeout {
 			return err
 		}
 	}
+	return io.EOF
 }

 func (this *Stream) TryWriteOnce(data *alloc.Buffer) error {