packer/rpc: can write while MuxConn is in CloseWait state

13 years ago · 6f31b9a151
parent a957a427c4
commit 6f31b9a151
1 changed files with 1 additions and 1 deletions
--- a/packer/rpc/muxconn.go
+++ b/packer/rpc/muxconn.go
@ -430,7 +430,7 @@ func (s *Stream) Write(p []byte) (int, error) {
 	state := s.state
 	s.mu.Unlock()

-	if state != streamStateEstablished {
+	if state != streamStateEstablished && state != streamStateCloseWait {
 		return 0, fmt.Errorf("Stream %d in bad state to send: %d", s.id, state)
 	}