From 2a92354774496edf4a0dc9af8e1918a7bc639f37 Mon Sep 17 00:00:00 2001
From: tspiridonova <tspiridonova@ozon.ru>
Date: Mon, 15 Mar 2021 17:45:04 +0300
Subject: [PATCH] fix rpc_util

---
 rpc_util.go | 7 +++----
 1 file changed, 3 insertions(+), 4 deletions(-)

diff --git a/rpc_util.go b/rpc_util.go
index 366a0d7aa968..289877f6f300 100644
--- a/rpc_util.go
+++ b/rpc_util.go
@@ -758,11 +758,10 @@ func recv(p *parser, c baseCodec, s *transport.Stream, dc Decompressor, m interf
 		return err
 	}
 	if _, ok := m.(*io.PipeWriter); ok {
+		defer func() {
+			m.(*io.PipeWriter).Close()
+		}()
 		_, err = m.(*io.PipeWriter).Write(d)
-		if err != nil {
-			return err
-		}
-		err = m.(*io.PipeWriter).Close()
 		return err
 	}
 	if err = c.Unmarshal(d, m); err != nil {