unistack-org/micro
4
0
Fork 3
Code Issues 14 Pull Requests 2 Actions Packages Projects Releases 254 Wiki Activity

Nitpick Headers to Header

Browse Source
This commit is contained in:
Asim 2016-01-28 18:11:13 +00:00
parent cd9801891e
commit d306559d8f
3 changed files with 21 additions and 21 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
client/rpc_codec.go
View File

@ -109,13 +109,13 @@ func (c *rpcPlusCodec) WriteRequest(req *request, body interface{}) error {
Target: req.Service, Target: req.Service,
Method: req.ServiceMethod, Method: req.ServiceMethod,
Type: codec.Request, Type: codec.Request,
Headers: map[string]string{}, Header: map[string]string{},
} }
if err := c.codec.Write(m, body); err != nil { if err := c.codec.Write(m, body); err != nil {
return err return err
} }
c.req.Body = c.buf.wbuf.Bytes() c.req.Body = c.buf.wbuf.Bytes()
for k, v := range m.Headers { for k, v := range m.Header {
c.req.Header[k] = v c.req.Header[k] = v
} }
return c.client.Send(c.req) return c.client.Send(c.req)

2
codec/codec.go
View File

@ -38,5 +38,5 @@ type Message struct {
Target string Target string
Method string Method string
Error string Error string
Headers map[string]string Header map[string]string
} }

10
server/rpc_codec.go
View File

@ -61,7 +61,7 @@ func newRpcPlusCodec(req *transport.Message, socket transport.Socket, c codec.Ne
} }
func (c *rpcPlusCodec) ReadRequestHeader(r *request, first bool) error { func (c *rpcPlusCodec) ReadRequestHeader(r *request, first bool) error {
m := codec.Message{Headers: c.req.Header} m := codec.Message{Header: c.req.Header}
if !first { if !first {
var tm transport.Message var tm transport.Message
@ -73,7 +73,7 @@ func (c *rpcPlusCodec) ReadRequestHeader(r *request, first bool) error {
return err return err
} }
m.Headers = tm.Header m.Header = tm.Header
} }
err := c.codec.ReadHeader(&m, codec.Request) err := c.codec.ReadHeader(&m, codec.Request)
@ -93,15 +93,15 @@ func (c *rpcPlusCodec) WriteResponse(r *response, body interface{}, last bool) e
Id: r.Seq, Id: r.Seq,
Error: r.Error, Error: r.Error,
Type: codec.Response, Type: codec.Response,
Headers: map[string]string{}, Header: map[string]string{},
} }
if err := c.codec.Write(m, body); err != nil { if err := c.codec.Write(m, body); err != nil {
return err return err
} }
m.Headers["Content-Type"] = c.req.Header["Content-Type"] m.Header["Content-Type"] = c.req.Header["Content-Type"]
return c.socket.Send(&transport.Message{ return c.socket.Send(&transport.Message{
Header: m.Headers, Header: m.Header,
Body: c.buf.wbuf.Bytes(), Body: c.buf.wbuf.Bytes(),
}) })
} }