fix rpc_util

spirt-t · Mar 15, 2021 · 2a92354 · 2a92354
1 parent da06014
commit 2a92354
Showing 1 changed file with 3 additions and 4 deletions.
diff --git 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 {