Use client.Call for non streaming requests
This commit is contained in:
parent
0b0eee41d0
commit
4495ca3839
@ -59,9 +59,6 @@ type CallOptions struct {
|
|||||||
// Middleware for low level call func
|
// Middleware for low level call func
|
||||||
CallWrappers []CallWrapper
|
CallWrappers []CallWrapper
|
||||||
|
|
||||||
// SendEOS specifies whether to send EOS
|
|
||||||
SendEOS bool
|
|
||||||
|
|
||||||
// Other options for implementations of the interface
|
// Other options for implementations of the interface
|
||||||
// can be stored in a context
|
// can be stored in a context
|
||||||
Context context.Context
|
Context context.Context
|
||||||
@ -308,13 +305,6 @@ func WithDialTimeout(d time.Duration) CallOption {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// SendEOS specifies whether to send the end of stream message
|
|
||||||
func SendEOS(b bool) CallOption {
|
|
||||||
return func(o *CallOptions) {
|
|
||||||
o.SendEOS = b
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// Request Options
|
// Request Options
|
||||||
|
|
||||||
func WithContentType(ct string) RequestOption {
|
func WithContentType(ct string) RequestOption {
|
||||||
|
@ -118,7 +118,7 @@ func (r *rpcClient) call(ctx context.Context, node *registry.Node, req Request,
|
|||||||
codec: codec,
|
codec: codec,
|
||||||
closed: make(chan bool),
|
closed: make(chan bool),
|
||||||
release: func(err error) { r.pool.Release(c, err) },
|
release: func(err error) { r.pool.Release(c, err) },
|
||||||
sendEOS: opts.SendEOS,
|
sendEOS: false,
|
||||||
}
|
}
|
||||||
// close the stream on exiting this function
|
// close the stream on exiting this function
|
||||||
defer stream.Close()
|
defer stream.Close()
|
||||||
@ -244,7 +244,7 @@ func (r *rpcClient) stream(ctx context.Context, node *registry.Node, req Request
|
|||||||
// used to close the stream
|
// used to close the stream
|
||||||
closed: make(chan bool),
|
closed: make(chan bool),
|
||||||
// signal the end of stream,
|
// signal the end of stream,
|
||||||
sendEOS: opts.SendEOS,
|
sendEOS: true,
|
||||||
// release func
|
// release func
|
||||||
release: func(err error) { r.pool.Release(c, err) },
|
release: func(err error) { r.pool.Release(c, err) },
|
||||||
}
|
}
|
||||||
|
@ -249,6 +249,12 @@ func (c *rpcCodec) ReadHeader(m *codec.Message, r codec.MessageType) error {
|
|||||||
|
|
||||||
func (c *rpcCodec) ReadBody(b interface{}) error {
|
func (c *rpcCodec) ReadBody(b interface{}) error {
|
||||||
// read body
|
// read body
|
||||||
|
// read raw data
|
||||||
|
if v, ok := b.(*raw.Frame); ok {
|
||||||
|
v.Data = c.buf.rbuf.Bytes()
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
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())
|
||||||
}
|
}
|
||||||
|
@ -220,9 +220,21 @@ func (p *Proxy) ServeRequest(ctx context.Context, req server.Request, rsp server
|
|||||||
// create new request with raw bytes body
|
// create new request with raw bytes body
|
||||||
creq := p.Client.NewRequest(service, endpoint, &bytes.Frame{body}, client.WithContentType(req.ContentType()))
|
creq := p.Client.NewRequest(service, endpoint, &bytes.Frame{body}, client.WithContentType(req.ContentType()))
|
||||||
|
|
||||||
|
// not a stream so make a client.Call request
|
||||||
if !req.Stream() {
|
if !req.Stream() {
|
||||||
// specify not to send eos
|
crsp := new(bytes.Frame)
|
||||||
opts = append(opts, client.SendEOS(false))
|
|
||||||
|
// make a call to the backend
|
||||||
|
if err := p.Client.Call(ctx, creq, crsp, opts...); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
// write the response
|
||||||
|
if err := rsp.Write(crsp.Data); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
// create new stream
|
// create new stream
|
||||||
@ -233,9 +245,7 @@ func (p *Proxy) ServeRequest(ctx context.Context, req server.Request, rsp server
|
|||||||
defer stream.Close()
|
defer stream.Close()
|
||||||
|
|
||||||
// create client request read loop if streaming
|
// create client request read loop if streaming
|
||||||
if req.Stream() {
|
|
||||||
go readLoop(req, stream)
|
go readLoop(req, stream)
|
||||||
}
|
|
||||||
|
|
||||||
// get raw response
|
// get raw response
|
||||||
resp := stream.Response()
|
resp := stream.Response()
|
||||||
@ -273,11 +283,6 @@ func (p *Proxy) ServeRequest(ctx context.Context, req server.Request, rsp server
|
|||||||
} else if err != nil {
|
} else if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
// we don't continue unless its a stream
|
|
||||||
if !req.Stream() {
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
Loading…
Reference in New Issue
Block a user