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

13 years ago · 311fb2064d
parent a6299fc49a
commit 311fb2064d
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)
 	}