From 6fa84624cb81ffa1d90bda1d8b5fbad0d9ef7707 Mon Sep 17 00:00:00 2001
From: v2ray <admin@v2ray.com>
Date: Fri, 29 Jul 2016 12:13:09 +0200
Subject: [PATCH] simplify logic
---
transport/ray/direct.go | 6 ++----
1 file changed, 2 insertions(+), 4 deletions(-)
diff --git a/transport/ray/direct.go b/transport/ray/direct.go
index 93f8271b..550aa28f 100644
--- a/transport/ray/direct.go
+++ b/transport/ray/direct.go
@@ -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 {