May 10, 2020 · May 10, 2020
diff --git a/ci/all.sh b/ci/all.sh

  ./ci/fmt.sh
  ./ci/lint.sh
  ./ci/test.sh
  ./ci/test.sh "$@"
 }

 main "$@"
diff --git a/write.go b/write.go
 if err != nil {
 return 0, err
 }
 defer func() {
 // We leave it locked when writing the close frame to avoid
 // any other goroutine writing any other frame.
 if opcode != opClose {
 c.writeFrameMu.unlock()
 defer c.writeFrameMu.unlock()

 // If the state says a close has already been written, we wait until
 // the connection is closed and return that error.
 //
 // However, if the frame being written is a close, that means its the close from
 // the state being set so we let it go through.
 c.closeMu.Lock()
 wroteClose := c.wroteClose
 c.closeMu.Unlock()
 if wroteClose && opcode != opClose {
 select {
 case <-ctx.Done():
 return 0, ctx.Err()
 case <-c.closed:
 return 0, c.closeErr
 }
 }()
 }

 select {
 case <-c.closed:
Original file line number	Diff line number	Diff line change
Expand Up		@@ -246,13 +246,24 @@ func (c *Conn) writeFrame(ctx context.Context, fin bool, flate bool, opcode opco
		if err != nil {
		return 0, err
		}
		defer func() {
		// We leave it locked when writing the close frame to avoid
		// any other goroutine writing any other frame.
		if opcode != opClose {
		c.writeFrameMu.unlock()
		defer c.writeFrameMu.unlock()

		// If the state says a close has already been written, we wait until
		// the connection is closed and return that error.
		//
		// However, if the frame being written is a close, that means its the close from
		// the state being set so we let it go through.
		c.closeMu.Lock()
		wroteClose := c.wroteClose
		c.closeMu.Unlock()
		if wroteClose && opcode != opClose {
		select {
		case <-ctx.Done():
		return 0, ctx.Err()
		case <-c.closed:
		return 0, c.closeErr
		}
		}()
		}

		select {
		case <-c.closed:
Expand Down