micro-client-grpc/response.go

package grpc

import (
	"strings"

	"github.com/unistack-org/micro/v3/codec"
	"github.com/unistack-org/micro/v3/metadata"
	"google.golang.org/grpc"
)

type response struct {
	conn   *poolConn
	stream grpc.ClientStream
	codec  codec.Codec
}

// Read the response
func (r *response) Codec() codec.Codec {
	return r.codec
}

// read the header
func (r *response) Header() metadata.Metadata {
	meta, err := r.stream.Header()
	if err != nil {
		return metadata.New(0)
	}
	md := metadata.New(len(meta))
	for k, v := range meta {
		md.Set(k, strings.Join(v, ","))
	}
	return md
}

// Read the undecoded response
func (r *response) Read() ([]byte, error) {
	f := &codec.Frame{}
	if err := r.codec.ReadBody(&wrapStream{r.stream}, f); err != nil {
		return nil, err
	}
	return f.Data, nil
}