micro/server/grpc/response.go

28 lines
439 B
Go
Raw Normal View History

2019-06-08 21:40:44 +03:00
package grpc
import (
"github.com/micro/go-micro/v3/codec"
2019-06-08 21:40:44 +03:00
)
type rpcResponse struct {
header map[string]string
codec codec.Codec
}
func (r *rpcResponse) Codec() codec.Writer {
return r.codec
}
func (r *rpcResponse) WriteHeader(hdr map[string]string) {
for k, v := range hdr {
r.header[k] = v
}
}
func (r *rpcResponse) Write(b []byte) error {
2019-06-18 20:51:52 +03:00
return r.codec.Write(&codec.Message{
2019-06-08 21:40:44 +03:00
Header: r.header,
Body: b,
2019-06-18 20:51:52 +03:00
}, nil)
2019-06-08 21:40:44 +03:00
}