package grpc import ( "strings" "github.com/micro/go-micro/codec" "github.com/micro/go-micro/codec/bytes" "google.golang.org/grpc" "google.golang.org/grpc/encoding" ) type response struct { conn *grpc.ClientConn stream grpc.ClientStream codec encoding.Codec gcodec codec.Codec } // Read the response func (r *response) Codec() codec.Reader { return r.gcodec } // read the header func (r *response) Header() map[string]string { md, err := r.stream.Header() if err != nil { return map[string]string{} } hdr := make(map[string]string) for k, v := range md { hdr[k] = strings.Join(v, ",") } return hdr } // Read the undecoded response func (r *response) Read() ([]byte, error) { f := &bytes.Frame{} if err := r.gcodec.ReadBody(f); err != nil { return nil, err } return f.Data, nil }