micro/server/rpc_response.go

36 lines
618 B
Go
Raw Normal View History

2019-01-09 22:11:47 +03:00
package server
import (
"net/http"
2019-01-09 22:28:13 +03:00
"github.com/micro/go-micro/codec"
2019-01-09 22:11:47 +03:00
"github.com/micro/go-micro/transport"
)
type rpcResponse struct {
header map[string]string
socket transport.Socket
2019-01-09 22:28:13 +03:00
codec codec.Codec
}
func (r *rpcResponse) Codec() codec.Writer {
return r.codec
2019-01-09 22:11:47 +03:00
}
func (r *rpcResponse) WriteHeader(hdr map[string]string) {
for k, v := range hdr {
r.header[k] = v
}
}
func (r *rpcResponse) Write(b []byte) error {
if _, ok := r.header["Content-Type"]; !ok {
r.header["Content-Type"] = http.DetectContentType(b)
}
return r.socket.Send(&transport.Message{
Header: r.header,
Body: b,
})
}