Buffered stream for chainxchg response

2020-12-25 16:03:46 +03:00
parent bb5a92e2f4
commit f9aca39caf
1 changed files with 5 additions and 1 deletions
@@ -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())