From 08e34ffb83d7cac434181142d51d77c0e96007fb Mon Sep 17 00:00:00 2001
From: kiranrg <kiranrg@uber.com>
Date: Thu, 27 Oct 2016 14:17:55 -0700
Subject: [PATCH] Don't close FD on write errors

---
 conn.go | 12 ++++++------
 1 file changed, 6 insertions(+), 6 deletions(-)

diff --git a/conn.go b/conn.go
index 5b26b53..fbc5c4b 100644
--- a/conn.go
+++ b/conn.go
@@ -177,6 +177,7 @@ var (
 	errBadWriteOpCode      = errors.New("websocket: bad write message type")
 	errWriteClosed         = errors.New("websocket: write closed")
 	errInvalidControlFrame = errors.New("websocket: invalid control frame")
+	errPartialWrite        = errors.New("websocket: partial write")
 )
 
 func hideTempErr(err error) error {
@@ -322,13 +323,12 @@ func (c *Conn) write(frameType int, deadline time.Time, bufs ...[]byte) error {
 	for _, buf := range bufs {
 		if len(buf) > 0 {
 			n, err := c.conn.Write(buf)
-			if n != len(buf) {
-				// Close on partial write.
-				c.conn.Close()
-			}
 			if err != nil {
 				return err
 			}
+			if n != len(buf) {
+				return errPartialWrite
+			}
 		}
 	}
 	return nil
@@ -387,8 +387,8 @@ func (c *Conn) WriteControl(messageType int, data []byte, deadline time.Time) er
 
 	c.conn.SetWriteDeadline(deadline)
 	n, err := c.conn.Write(buf)
-	if n != 0 && n != len(buf) {
-		c.conn.Close()
+	if err == nil && n != 0 && n != len(buf) {
+		err = errPartialWrite
 	}
 	return hideTempErr(err)
 }