micro-server-grpc/response.go

35 lines
605 B
Go
Raw Normal View History

2019-06-08 19:40:44 +01:00
package grpc
import (
"io"
"github.com/unistack-org/micro/v3/codec"
"github.com/unistack-org/micro/v3/metadata"
"github.com/unistack-org/micro/v3/server"
2019-06-08 19:40:44 +01:00
)
var _ server.Response = &rpcResponse{}
2019-06-08 19:40:44 +01:00
type rpcResponse struct {
rw io.ReadWriter
header metadata.Metadata
codec codec.Codec
2019-06-08 19:40:44 +01:00
}
func (r *rpcResponse) Codec() codec.Codec {
2019-06-08 19:40:44 +01:00
return r.codec
}
func (r *rpcResponse) WriteHeader(hdr metadata.Metadata) {
2019-06-08 19:40:44 +01:00
for k, v := range hdr {
r.header[k] = v
}
}
func (r *rpcResponse) Write(b []byte) error {
return r.codec.Write(r.rw, &codec.Message{
2019-06-08 19:40:44 +01:00
Header: r.header,
Body: b,
2019-06-18 18:51:52 +01:00
}, nil)
2019-06-08 19:40:44 +01:00
}