42 lines
687 B
Go
42 lines
687 B
Go
package grpc
|
|
|
|
import (
|
|
"go.unistack.org/micro/v4/codec"
|
|
"go.unistack.org/micro/v4/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 nil
|
|
}
|
|
|
|
return metadata.Metadata(meta.Copy())
|
|
}
|
|
|
|
// Read the undecoded response
|
|
func (r *response) Read() ([]byte, error) {
|
|
f := &codec.Frame{}
|
|
wrap := &wrapStream{r.stream}
|
|
|
|
_, err := wrap.Read(f.Data)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return f.Data, nil
|
|
}
|