1
0
mirror of https://github.com/go-micro/go-micro.git synced 2024-12-06 08:16:03 +02:00
go-micro/server/grpc/response.go

28 lines
436 B
Go
Raw Normal View History

2019-06-08 20:40:44 +02:00
package grpc
import (
"github.com/micro/go-micro/codec"
)
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 19:51:52 +02:00
return r.codec.Write(&codec.Message{
2019-06-08 20:40:44 +02:00
Header: r.header,
Body: b,
2019-06-18 19:51:52 +02:00
}, nil)
2019-06-08 20:40:44 +02:00
}