unistack-org/micro
4
0
Fork 3
Code Issues 17 Pull Requests 1 Actions Packages Projects Releases 258 Wiki Activity

rename codec

Browse Source
This commit is contained in:
Asim Aslam 2018-12-31 17:28:19 +00:00
parent 460fb3e70c
commit dcf7a56f9b
1 changed files with 7 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
client/rpc_codec.go
View File

@ -28,7 +28,7 @@ var (
errShutdown = errs.New("connection is shut down") errShutdown = errs.New("connection is shut down")
) )
type rpcPlusCodec struct { type rpcCodec struct {
client transport.Client client transport.Client
codec codec.Codec codec codec.Codec
@ -89,12 +89,12 @@ func (rwc *readWriteCloser) Close() error {
return nil return nil
} }
func newRpcPlusCodec(req *transport.Message, client transport.Client, c codec.NewCodec) *rpcPlusCodec { func newRpcPlusCodec(req *transport.Message, client transport.Client, c codec.NewCodec) *rpcCodec {
rwc := &readWriteCloser{ rwc := &readWriteCloser{
wbuf: bytes.NewBuffer(nil), wbuf: bytes.NewBuffer(nil),
rbuf: bytes.NewBuffer(nil), rbuf: bytes.NewBuffer(nil),
} }
r := &rpcPlusCodec{ r := &rpcCodec{
buf: rwc, buf: rwc,
client: client, client: client,
codec: c(rwc), codec: c(rwc),
@ -103,7 +103,7 @@ func newRpcPlusCodec(req *transport.Message, client transport.Client, c codec.Ne
return r return r
} }
func (c *rpcPlusCodec) WriteRequest(req *request, body interface{}) error { func (c *rpcCodec) WriteRequest(req *request, body interface{}) error {
c.buf.wbuf.Reset() c.buf.wbuf.Reset()
m := &codec.Message{ m := &codec.Message{
@ -129,7 +129,7 @@ func (c *rpcPlusCodec) WriteRequest(req *request, body interface{}) error {
return nil return nil
} }
func (c *rpcPlusCodec) ReadResponseHeader(r *response) error { func (c *rpcCodec) ReadResponseHeader(r *response) error {
var m transport.Message var m transport.Message
if err := c.client.Recv(&m); err != nil { if err := c.client.Recv(&m); err != nil {
return errors.InternalServerError("go.micro.client.transport", err.Error()) return errors.InternalServerError("go.micro.client.transport", err.Error())
@ -147,14 +147,14 @@ func (c *rpcPlusCodec) ReadResponseHeader(r *response) error {
return nil return nil
} }
func (c *rpcPlusCodec) ReadResponseBody(b interface{}) error { func (c *rpcCodec) ReadResponseBody(b interface{}) error {
if err := c.codec.ReadBody(b); err != nil { if err := c.codec.ReadBody(b); err != nil {
return errors.InternalServerError("go.micro.client.codec", err.Error()) return errors.InternalServerError("go.micro.client.codec", err.Error())
} }
return nil return nil
} }
func (c *rpcPlusCodec) Close() error { func (c *rpcCodec) Close() error {
c.buf.Close() c.buf.Close()
c.codec.Close() c.codec.Close()
if err := c.client.Close(); err != nil { if err := c.client.Close(); err != nil {