Merge pull request #5254 from turuslan/fix/chainxchg-buffered

Chainxchg write response timeout
filecoin-project · Jan 5, 2021 · a15080b · a15080b
2 parents ab529f4 + f9aca39
commit a15080b
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/chain/exchange/server.go b/chain/exchange/server.go
@@ -56,7 +56,11 @@ func (s *server) HandleStream(stream inet.Stream) {
 	}
 
 	_ = stream.SetDeadline(time.Now().Add(WriteResDeadline))
-	if err := cborutil.WriteCborRPC(stream, resp); err != nil {
+	buffered := bufio.NewWriter(stream)
+	if err = cborutil.WriteCborRPC(buffered, resp); err == nil {
+		err = buffered.Flush()
+	}
+	if err != nil {
 		_ = stream.SetDeadline(time.Time{})
 		log.Warnw("failed to write back response for handle stream",
 			"err", err, "peer", stream.Conn().RemotePeer())