with *[]byte interface

spirt-t · Mar 18, 2021 · 964f1a9 · 964f1a9
1 parent 2a92354
commit 964f1a9
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/rpc_util.go b/rpc_util.go
@@ -757,12 +757,12 @@ func recv(p *parser, c baseCodec, s *transport.Stream, dc Decompressor, m interf
 	if err != nil {
 		return err
 	}
-	if _, ok := m.(*io.PipeWriter); ok {
-		defer func() {
-			m.(*io.PipeWriter).Close()
-		}()
-		_, err = m.(*io.PipeWriter).Write(d)
-		return err
+	if _, ok := m.(*[]byte); ok {
+		*m.(*[]byte) = d
+		if payInfo != nil {
+			payInfo.uncompressedBytes = d
+		}
+		return nil
 	}
 	if err = c.Unmarshal(d, m); err != nil {
 		return status.Errorf(codes.Internal, "grpc: failed to unmarshal the received message %v", err)