Merge pull request #201 from mengskysama/master

fix producer Stop() EOF
nsqio · Apr 15, 2017 · b2d0b44 · b2d0b44
2 parents b9762cd + a7c1e31
commit b2d0b44
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/conn.go b/conn.go
@@ -479,7 +479,11 @@ func (c *Conn) readLoop() {
 		}
 
 		frameType, data, err := ReadUnpackedResponse(c)
+
 		if err != nil {
+			if err == io.EOF && atomic.LoadInt32(&c.closeFlag) == 1 {
+				goto exit
+			}
 			if !strings.Contains(err.Error(), "use of closed network connection") {
 				c.log(LogLevelError, "IO error - %s", err)
 				c.delegate.OnIOError(c, err)